← Back
Editing: reflection_test.cpython-36.pyc
3 ���h�� � @ s� d Z ddlZddlZddlZddlZddlZy ddlZW n ek rT ddlZY nX ddl m Z ddl m Z ddl m Z ddl m Z ddl mZ ddl mZ dd l mZ dd l mZ dd lmZ dd lmZ dd lmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ejZG dd� de �Z G dd� de�Z G dd� de�Z G dd� de�Z G dd � d e�Z!G d d � d e�Z"G d d � d e�Z#G d!d"� d"e�Z$G d#d$� d$e�Z%G d%d&� d&e�Z&e'd'k�r�ej(� dS )(ziUnittest for reflection.py, which also indirectly tests the output of the pure-Python protocol compiler. � N)�unittest_import_pb2)�unittest_mset_pb2)� unittest_pb2)�descriptor_pb2)� descriptor)�message)� reflection)� text_format)�api_implementation)�more_extensions_pb2)�more_messages_pb2)�message_set_extensions_pb2)� wire_format)� test_util)�testing_refleaks)�decoderc @ s\ e Zd ZdZdd� Zdd� ZeZeZeZeZ dd� Z e Z dd � Z d d � Z d d � Zdd� ZdS )� _MiniDecodera� Decodes a stream of values from a string. Once upon a time we actually had a class called decoder.Decoder. Then we got rid of it during a redesign that made decoding much, much faster overall. But a couple tests in this file used it to check that the serialized form of a message was correct. So, this class implements just the methods that were used by said tests, so that we don't have to rewrite the tests. c C s || _ d| _d S )Nr )�_bytes�_pos)�self�bytes� r �%/usr/lib/python3.6/reflection_test.py�__init__O s z_MiniDecoder.__init__c C s t j| j| j�\}| _|S )N)r Z _DecodeVarintr r )r �resultr r r � ReadVarintS s z_MiniDecoder.ReadVarintc C s t j| j� �S )N)r Z ZigZagDecoder )r r r r � ReadSInt64\ s z_MiniDecoder.ReadSInt64c C s t j| j� �S )N)r Z UnpackTagr )r r r r �ReadFieldNumberAndWireTypea s z'_MiniDecoder.ReadFieldNumberAndWireTypec C s4 t jd| j| j| jd � �d }| jd7 _|S )Nz<f� r )�struct�unpackr r )r r r r r � ReadFloatd s "z_MiniDecoder.ReadFloatc C s4 t jd| j| j| jd � �d }| jd7 _|S )Nz<d� r )r r r r )r r r r r � ReadDoublei s "z_MiniDecoder.ReadDoublec C s | j t| j�kS )N)r �lenr )r r r r � EndOfStreamn s z_MiniDecoder.EndOfStreamN)�__name__� __module__� __qualname__�__doc__r r � ReadInt32� ReadInt64� ReadUInt32� ReadUInt64r � ReadSInt32r r! r# r% r r r r r E s r c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Zd3d4� Z d5d6� Z d7d8� Z d9d:� Z d;d<� Z d=d>� Z!d?d@� Z"dAdB� Z#dCdD� Z$dEdF� Z%dGdH� Z&dIdJ� Z'dKdL� Z(dMdN� Z)dOdP� Z*dQdR� Z+dSdT� Z,dUdV� Z-dWdX� Z.dYdZ� Z/d[d\� Z0d]d^� Z1d_d`� Z2dadb� Z3dcdd� Z4dedf� Z5dgdh� Z6didj� Z7dkdl� Z8dmdn� Z9dodp� Z:dqdr� Z;dsdt� Z<dudv� Z=dwdx� Z>dydz� Z?d{d|� Z@d}d~� ZAdd�� ZBd�d�� ZCd�d�� ZDd�d�� ZEd�d�� ZFd�d�� ZGd�d�� ZHd�d�� ZId�d�� ZJd�d�� ZKd�d�� ZLd�d�� ZMd�d�� ZNd�d�� ZOd�d�� ZPd�d�� ZQd�d�d��ZRd�d�� ZSeTjUeVjW� d�k�p�eVjX� d�kd��d�d�� �ZYd�d�� ZZd�d�� Z[d�d�� Z\d�d�� Z]d�d�� Z^d�d�� Z_d�S )��ReflectionTestc C sB | j t|�t|�� x(tt|��D ]}| j || || � q"W d S )N)� assertEqualr$ �range)r �valuesZothers�ir r r �assertListsEqualt s z ReflectionTest.assertListsEqualc C sP t jdddd d�}| jd|j� | jd|j� | jd|j� | j|jd�� d S )N� g +�)K@�optional_string)�optional_int32�optional_doubler6 �optional_floatr9 )r � TestAllTypesr0 r7 r8 r6 � assertFalse�HasField)r �protor r r �testScalarConstructory s z$ReflectionTest.testScalarConstructorc C s� t jddddgddgdddgd gd d �}| jddddgt|j�� | jddgt|j�� | jdddgt|j�� | jd gt|j�� | jg t|j�� d S ) N� � � r g�G�z��?g +�)K@TFr6 )�repeated_int32�repeated_double� repeated_bool�repeated_string�repeated_float) r r: r0 �listrB rC rD rE rF )r r= r r r � testRepeatedScalarConstructor� s z,ReflectionTest.testRepeatedScalarConstructorc C s t jt jjt jjd�t jjt jjd�gt jd d�t jdd�t jdd�gt jj� t jjdd�t jjdd�gd �}| jt jjt jjd�t jjt jjd�gt|j �� | jt jd d�t jdd�t jdd�gt|j �� | jt jj� t jjdd�t jjdd�gt|j �� d S ) N)�bb�+ )�ci � � r? )�ar@ )�repeated_nested_message�repeated_foreign_message� repeatedgroupi���i���) r r: � NestedMessage�FOO�BAR�ForeignMessageZ RepeatedGroupr0 rG rN rO rP )r r= r r r � testRepeatedCompositeConstructor� s6 z/ReflectionTest.testRepeatedCompositeConstructorc C s t jddddgdddgt jjt jjd�t jjt jjd�gt jdd �t jd d �t jd d �gd d �}| jd|j� | jd|j� | jddgt |j �� | jdddgt |j �� | jt jjt jjd�t jjt jjd�gt |j �� | jt jdd �t jd d �t jd d �gt |j �� | j|jd �� d S )Nr5 r6 g�G�z��?g +�)K@TF)rI rJ )rK i � rL )r7 r6 rC rD rN rO �optional_nested_messagerV i���i���)r r: rQ rR rS rT r0 r7 r6 rG rC rD rN rO r; r< )r r= r r r �testMixedConstructor� s: z#ReflectionTest.testMixedConstructorc C s� | j ttjdd� | j ttjdd� | j ttjdd� | j ttjdd� | j ttjdgd� | j ttjdd� | j ttjdgd� | j ttjdd� | j ttjdgd� d S ) N�foo)r7 i� )r6 )rV )rB )rE )rN )� assertRaises� TypeErrorr r: )r r r r �testConstructorTypeError� s$ z'ReflectionTest.testConstructorTypeErrorc C s� t jdd�}| jd|j� � t jt jj� d�}| jd|j� � t jdgd�}| jd|j� � t jt jj� gd�}| jd|j� � d S )NrL )r7 r@ )rV rA )rB )rN )r r: r0 �ByteSizerQ )r r r r r �(testConstructorInvalidatesCachedByteSize� s z7ReflectionTest.testConstructorInvalidatesCachedByteSizec C sp t j� }| j|jd� � | jd|j� | j|jd� � d|_| j|jd�� |jd� | j|jd� � d S )Nr7 r r? )r r: � assertTruer< r0 r7 � ClearField)r r= r r r �testSimpleHasBits� s z ReflectionTest.testSimpleHasBitsc s8 � fdd�}|dd� |dd� |dd� |d d � d S ) Nc s0 t j� }t|| �}t||�}� jd|� � j|j|� � � j|j| � � d}t|||� � j|t||�� |}� j|j|�� � j|j| �� |j| � t|| �}� j|j|� � � j|j| � � � jdt||�� � j||k � t|||� � j|j|� � � j|j| � � � jdt||�� d S )Nr � )r r: �getattrr0 r^ r< �setattrr_ )Zcomposite_field_nameZscalar_field_namer= Zcomposite_fieldZoriginal_scalar_valueZnew_valZold_composite_field)r r r �TestCompositeHasBits. s, zNReflectionTest.testHasBitsWithSinglyNestedScalar.<locals>.TestCompositeHasBits� optionalgrouprM rV rI �optional_foreign_messagerK Zoptional_import_message�dr )r rd r )r r �!testHasBitsWithSinglyNestedScalar s " * z0ReflectionTest.testHasBitsWithSinglyNestedScalarc C s t j� }|j}~d|_d S )N� )r r: rV rI )r r= �nestedr r r � testReferencesToNestedMessage] s z,ReflectionTest.testReferencesToNestedMessagec C sT t j� }|j}|jd� | j||jk � d|_| j|jd� � | jd|jj� d S )NrV ri r )r r: rV r_ r^ rI r< r0 )r r= rj r r r �0testDisconnectingNestedMessageBeforeSettingFielde s z?ReflectionTest.testDisconnectingNestedMessageBeforeSettingFieldc C s6 t j� }|j}|jd� ~~tj� t j� }|j}d S )NrV )r r: rV r_ �gc�collect)r r= rj r r r �5testGetDefaultMessageAfterDisconnectingDefaultMessagen s zDReflectionTest.testGetDefaultMessageAfterDisconnectingDefaultMessagec C s� t j� }|j}d|_| j|jd�� |jd� | jd|j� | jd|jj� | j||jk � d|_| j|jd� � | jd|jj� d S )N� rV r ri )r r: rV rI r^ r< r_ r0 )r r= rj r r r �/testDisconnectingNestedMessageAfterSettingField{ s z>ReflectionTest.testDisconnectingNestedMessageAfterSettingFieldc C s: t j� }| j|jd� � |jd� | j|jd� � d S )NrV )r r: r^ r< r_ )r r= r r r �0testDisconnectingNestedMessageBeforeGettingField� s z?ReflectionTest.testDisconnectingNestedMessageBeforeGettingFieldc C sR t j� }t j� }d|j_|j|� | j|jd�� |jd� | j|jd� � d S )Nrp rV )r r: rV rI � MergeFromr^ r< r_ )r �proto1�proto2r r r �(testDisconnectingNestedMessageAfterMerge� s z7ReflectionTest.testDisconnectingNestedMessageAfterMergec C s8 t j� dkrd S tj� }d|j_|jd� ~tj� d S )N�pythonrp �optional_lazy_message) r �Typer r: rx rI r_ rm rn )r r= r r r �"testDisconnectingLazyNestedMessage� s z1ReflectionTest.testDisconnectingLazyNestedMessagec C st t j� }|jjjd� | jdg|jj� | j|jd�� |jd� | j|jd� � |jj j � | j|jd�� d S )Nrp rV ) r ZTestNestedMessageHasBitsrV Z nestedmessage_repeated_int32�appendr0 r^ r< r_ Z%nestedmessage_repeated_foreignmessage�add)r r= r r r �&testHasBitsWhenModifyingRepeatedFields� s z5ReflectionTest.testHasBitsWhenModifyingRepeatedFieldsc C s t j� }| j|jd� � | jd|jjjjjj� | j|jd� � d|jjjjj_| jd|jjjjjj� | j|jd�� | j|jjd�� | j|jjjd�� | j|jjjjd�� | j|jjjjjd�� | j|jjjjjjd� � | j|jjjjjjd�� d S )NrI r rp rM r7 )r �TestMutualRecursionAr^ r< r0 rI rM r7 )r Zrecursive_protor r r �!testHasBitsForManyLevelsOfNesting� s z0ReflectionTest.testHasBitsForManyLevelsOfNestingc C s� t j� }d|_d|_d|_|j}| j|jjd df|jjd df|jjd dfg|j � � d|j_ | j|jjd df|jjd df|jjd df|jjd |fg|j � � d S ) Nr? rp rX r7 �optional_fixed32r6 �{ rV ) r r: r� r7 r6 rV r0 � DESCRIPTOR�fields_by_name� ListFieldsrI )r r= Znested_messager r r �testSingularListFields� s$ z%ReflectionTest.testSingularListFieldsc C s� t j� }|jjd� |jjd� |jjd� |jjddg� |jjg � |jjd� |jjdd� td �D �� d |_|j | j |j j d d f|j j d ddgf|j j d dgf|j j d dddddgfg|j � � d S )Nr? rp � rX �bar�bazc s s | ]}t |�V qd S )N)�str)�.0�xr r r � <genexpr>� s z8ReflectionTest.testRepeatedListFields.<locals>.<genexpr>r@ � r7 rB �repeated_fixed32rE �0�1)r r: r� r{ rB rE �extendr1 r7 rD r0 r� r� r� )r r= r r r �testRepeatedListFields� s" z%ReflectionTest.testRepeatedListFieldsc C sX t j� }d|jt j< d|jt j< d|jt j< | jt jdft jdft jdfg|j� � d S )Nr? rp rX )r �TestAllExtensions� ExtensionsZoptional_fixed32_extension�optional_int32_extensionZoptional_string_extensionr0 r� )r r= r r r �testSingularListExtensions� s z)ReflectionTest.testSingularListExtensionsc C s� t j� }|jt j jd� |jt j jd� |jt j jd� |jt j jd� |jt j jd� |jt j jd� d|jt j< | jt jdft jddgft jdgft jdddgfg|j � � d S )Nr? rp r� rX r� r� r� ) r r� r� Zrepeated_fixed32_extensionr{ �repeated_int32_extension�repeated_string_extensionr� r0 r� )r r= r r r �testRepeatedListExtensions� s z)ReflectionTest.testRepeatedListExtensionsc C sf t j� }tj|� t j | j|jjd dft jdf|jjd dft jdf|jjd dfg|j � � d S ) NZmy_intr? ri Z my_stringrX r� Zmy_floatg �?) r �TestFieldOrderingsr �SetAllFieldsAndExtensionsZmy_extension_intr0 r� r� Zmy_extension_stringr� )r r= r r r �testListFieldsAndExtensions s z*ReflectionTest.testListFieldsAndExtensionsc C s� t j� }| jd|j� | jd|j� | jd|j� | jd|j� | jd|j� | jd|j� | jd|j � | jd|j � | jd|j � | jd|j � | jd|j � | jd|j� | jd|j� | jd|j� | jd|j� | jd|j� | jd|j� | jd|j� | jd |j� | jd|j� | jd |j� | jd |j� | jd |j� | jd|j� | jd|j� | jd|j � | jd|j � | jd|j � | jd|j � | jd|j � | jt jj!|j"� | jt j#|j$� | jt%j&|j'� t j(� }| jd|j)� d S )Nr g F� � �) �* rJ �, �- �. �/ �0 �1 �2 g �I@g d�@TZhellos worldu ሴi���i���)*r r: r0 r7 �optional_int64�optional_uint32�optional_uint64�optional_sint32�optional_sint64r� �optional_fixed64�optional_sfixed32�optional_sfixed64r9 r8 � optional_boolr6 �optional_bytesZ default_int32Z default_int64Zdefault_uint32Zdefault_uint64Zdefault_sint32Zdefault_sint64Zdefault_fixed32Zdefault_fixed64Zdefault_sfixed32Zdefault_sfixed64Z default_floatZdefault_doubleZ default_boolZdefault_stringZ default_bytesrS Zdefault_nested_enum� FOREIGN_BARZdefault_foreign_enumr Z IMPORT_BARZdefault_import_enumZTestExtremeDefaultValuesZ utf8_string)r r= r r r �testDefaultValues sJ z ReflectionTest.testDefaultValuesc C s t j� }| jt|jd� d S )N�nonexistent_field)r r: rY � ValueErrorr< )r r= r r r � testHasFieldWithUnknownFieldName@ s z/ReflectionTest.testHasFieldWithUnknownFieldNamec C s t j� }| jt|jd� d S )Nr� )r r: rY r� r_ )r r= r r r �"testClearFieldWithUnknownFieldNameD s z1ReflectionTest.testClearFieldWithUnknownFieldNamec sN t j� � xtd�D ]}� jj� qW t j� }� j|� | jt� fdd�� d S )N� c s � j d S )Nrp )�repeated_messager )r= r r �<lambda>Q s z9ReflectionTest.testClearRemovesChildren.<locals>.<lambda>)r �TestRequiredForeignr1 r� r| �CopyFromrY � IndexError)r r3 ru r )r= r �testClearRemovesChildrenH s z'ReflectionTest.testClearRemovesChildrenc C s~ t j� }| jtt|dd� | jtt|ddg� | jtt|dd� | jtt|jdd� | jtt|jdd� | jtt|dd� d S ) NrB r� rV ri rI �" Zsome_attributer� )r r: rY �AttributeErrorrc rN rF )r r= r r r �testDisallowedAssignmentsS s z(ReflectionTest.testDisallowedAssignmentsc C s� t j� }| jtt|dd� | jtt|dd� | jtt|dd� | jtt|dd� | jtt|dd� | jtt|dd� | jtt|d d� tj� d kr�| jtt|dd� nd|_d S ) Nr7 g�������?rX r6 r� r� r� r9 r8 rw )r r: rY rZ rc r ry r� )r r= r r r �testSingleScalarTypeSafetyh s z)ReflectionTest.testSingleScalarTypeSafetyc s� � �fdd�}|ddt � |ddt � |ddt � yt }W n tk rR t }Y nX tjd�dkrp|dd|� n |ddt � |d d|� |d d|� |d d|� |d d|� d S )z}Verifies setting of scalar integers. Args: integer_fn: A function to wrap the integers that will be assigned. c sZ t j� }� |�}t|| |� �jt|| �|� t j� }|j|j� � �jt|| �|� d S )N)r r: rc �assertIsInstancerb �ParseFromString�SerializeToString)� field_name�valueZ expected_typer= ru )� integer_fnr r r �TestGetAndDeserialize} s z@ReflectionTest.assertIntegerTypes.<locals>.TestGetAndDeserializer7 r? � r� �Lr � r� �< r� Ni @i @l l i @l i @l )�int� NameErrorr �calcsize)r r� r� Z integer_64r )r� r r �assertIntegerTypesw s z!ReflectionTest.assertIntegerTypesc C s | j dd� � d S )Nc S s | S )Nr )r� r r r r� � s z1ReflectionTest.testIntegerTypes.<locals>.<lambda>)r� )r r r r �testIntegerTypes� s z ReflectionTest.testIntegerTypesc C s | j tj� d S )N)r� r �NonStandardInteger)r r r r �testNonStandardIntegerTypes� s z*ReflectionTest.testNonStandardIntegerTypesc C sN t j� }| jt�� d|_W d Q R X | jtd�� tjdd�|_W d Q R X d S )N�2Zmy_errorrp ) r r: rY rZ r� ZassertRaisesRegex� RuntimeErrorr r� )r �pbr r r � testIllegalValuesForIntegers� s z+ReflectionTest.testIllegalValuesForIntegersc s� � �fdd�}|ddd� |ddd� |d dd� |d dd � t j� }�jttf�� � d�|_W d Q R X t j� }� d�|_�jd|j� d S )z�Verifies bounds checking for scalar integer fields. Args: integer_fn: A function to wrap the integers that will be assigned. c s� t j� }� |�}� |�}t|| |� �j|t|| �� t|| |� �j|t|| �� �jttft|| |d � �jttft|| |d � d S )Nr? )r r: rc r0 rb rY r� rZ )r� Z expected_minZ expected_maxr� )r� r r r �TestMinAndMaxIntegers� s zHReflectionTest.assetIntegerBoundsChecking.<locals>.TestMinAndMaxIntegersr7 r? r� r� r l �� r� �? r� l ���� Nl i �l i���l l���� l l ���� l l���� )r r: rY r� rZ r� �optional_nested_enumr0 )r r� r� r� r )r� r r �assetIntegerBoundsChecking� s z)ReflectionTest.assetIntegerBoundsCheckingc C s | j dd� � d S )Nc S s | S )Nr )r� r r r r� � s z?ReflectionTest.testSingleScalarBoundsChecking.<locals>.<lambda>)r� )r r r r � testSingleScalarBoundsChecking� s z-ReflectionTest.testSingleScalarBoundsCheckingc C s | j tj� d S )N)r� r r� )r r r r �)testNonStandardSingleScalarBoundsChecking� s z8ReflectionTest.testNonStandardSingleScalarBoundsCheckingc C s� t j� }| jt|jjd� | jt|jjd� | jt|jd� | jt|jd� |jjd� d|jd< | jt|jj dd� | jt|jj dd� | jt|jj dg � | jt|jj dd� |jjd � | jt|jj dd� d S ) Ng�������?rX r� ri r i� �abc�indexr� ) r r: rY rZ rB r{ rE Zrepeated_bytesr� � __setitem__)r r= r r r � testRepeatedScalarTypeSafety� s z+ReflectionTest.testRepeatedScalarTypeSafetyc C sV t j� }| jd|j� d|_| jd|j� d|_| jd|j� d|_| jd|j� d S )Nr r? l ��� l ���� )r r: r0 r7 r� )r r= r r r �!testSingleScalarGettersAndSetters� s z0ReflectionTest.testSingleScalarGettersAndSettersc C sV t j� }|jd� d|_| j|jd�� |jd� | jd|j� | j|jd� � d S )Nr7 r? r )r r: r_ r7 r^ r< r0 )r r= r r r �testSingleScalarClearField� s z)ReflectionTest.testSingleScalarClearFieldc C sf t j� }| jd|j� | jdt jj� | jd|j� | jdt jj� | jd|j� | jdt jj� d S )Nr? r@ rA )r r: r0 rR rS �BAZ)r r= r r r � testEnums� s zReflectionTest.testEnumsc C s | j dtjjtj�� | j dtjjtj�� | j dtjjtj�� | jttjjd� tj � }| j d|j j|j �� | j dtj j j|j �� | j d|j j|j �� | j dtj j j|j �� | j d|j j|j �� | j dtj j j|j �� | jt|j jd� | jttj j jd� d S )N� FOREIGN_FOOr� � FOREIGN_BAZi0, rR rS r� )r0 r � ForeignEnum�Namer� r� r� rY r� r: � NestedEnumrR rS r� )r r= r r r � testEnum_Name s2 z ReflectionTest.testEnum_Namec C s | j tjtjjd�� | j tjtjjd�� | j tjtjjd�� | jttjjd� tj � }| j |j |j jd�� | j |j tj j jd�� | j |j |j jd�� | j |j tj j jd�� | j |j |j jd�� | j |j tj j jd�� | jt|j jd� | jttj j jd� d S ) Nr� r� r� ZFOrR rS r� ZFoo)r0 r r� r� ZValuer� r� rY r� r: rR r� rS r� )r r= r r r �testEnum_Value" s2 z ReflectionTest.testEnum_Valuec C s� | j dddgttjj� �� | j dddgttjj� �� | j dddgttjj� �� tj� }| j ddd d gt|jj� �� | j d d d dgt|jj� �� | j ddddgt|jj� �� d S )Nr� r� r� r rp � rR rS r� �NEGr? r@ rA )r� r )r� rp )r� r� ���)rR r? )rS r@ )r� rA r� )r� r� ) r0 rG r r� �keysr2 �itemsr: r� )r r= r r r �testEnum_KeysAndValues> s z%ReflectionTest.testEnum_KeysAndValuesc C s| t j� }| j|j � | jdt|j�� |jjd� |jjd� |jjd� | j|j� | jdt|j�� | jdddg|j� | jd|jd � | jd|jd � | jt|jj d� | jt|jj d� | jt |jj d� | jt |jj d � d |jd< | jdd dg|j� |jj dd � | jdd d dg|j� |jjd � | jd d dg|jdd � � | jdd d dd g|jd d � � d d� t d�D �|jdd �< | jddddd g|j� dddg|jdd �< | jddddd g|j� g }x|jD ]}|j|� �q�W | jddddd g|� |jd= | jdddd g|j� |jdd �= | jddg|j� |jj ddg� | jddddg|j� |jd� | j|j � | jdt|j�� |jjd� | jd|jd � d|jd< | jd|jd � ddddg|jd d �< |jd = | jdddg|j� |jd = | jddg|j� | jt|jjd � | jt|jjd� |jd d!�= | jdg|j� |jdd�= | jdg|j� d S )"Nr rp r� � rA r? i� rX ra � r� r c s s | ] }|V qd S )Nr )r� r3 r r r r� r s z5ReflectionTest.testRepeatedScalars.<locals>.<genexpr>r@ �# �( r� � rB i, �d i' r� i.���r� r� r� r� ������r� r� )r r: r^ rB r0 r$ r{ rY r� � __getitem__rZ �insertr1 r� r_ � __delitem__)r r= r r3 r r r �testRepeatedScalarsM sl z"ReflectionTest.testRepeatedScalarsc C sF t j� }| j|j � | jdt|j�� |jjd� |jjd� |jjd� |jjd� | jdt|j�� |jjd� | jdt|j�� | jd|jd � | jd|jd � | jd|jd � |jjd� | jdt|j�� | jd|jd � | jd|jd � |jjd� | jdt|j�� | jd|jd � | jt |jjd� d S ) Nr rp r� r rA r? r@ r� ) r r: r^ rB r0 r$ r{ �removerY r� )r r= r r r �testRepeatedScalarsRemove� s* z(ReflectionTest.testRepeatedScalarsRemovec C s( t j� }| j|j � | jdt|j�� |jj� }|jj� }| j|j� | jdt|j�� | j||g|j� | j|t jj � | j t |jj d� | j t |jj d � | j t |jj d� | j t |jj d � |jj� }|jj� }|jj� }| j|||g|jdd� � | j|||||g|jd d � � | j||g|jd d� � | j|||g|jdd � � | j||jd � | j|g|jd d� � g }x|jD ]}|j|� �q�W | j|||||g|� |jd= | j||||g|j� |jdd �= | j||g|j� t jj dd�} t jj dd�} |jj| | g� | jdt|j�� | j| |jd � | j| |jd � | j t |jj| � | j t |jjdg� t j� } | j t |jj| g� |jd � | j|j � | jdt|j�� |jjd d� | jdt|j�� | jd |jd j� | j t |jjd � | j t�� d |jd< W d Q R X d S ) Nr r@ i� rX r? r )rI rA rN ri i.���)r r: r^ rN r0 r$ r| r4 r� rQ rY r� r� rZ r{ r� r_ rI � Exception) r r= �m0�m1�m2Zm3Zm4r r3 Zn1Zn2Zwrong_message_typer r r �testRepeatedComposites� s~ z%ReflectionTest.testRepeatedCompositesc C s< t j� }| jdt|j�� |jj� }t|j�|_|jj� }t|j�|_| j||k� |jj� }t|j�|_| j|||g|j� | jdt|j�� |jj |� | jdt|j�� | j||jd � | j||jd � | j t |jj |� | j t |jj d � | jdt|j�� |jj |� | jdt|j�� | j||jd � d S )Nr rA r@ r? ) r r: r0 r$ rN r| rI r^ r4 r rY r� )r r= r r r r r r �testRepeatedCompositeRemove s* z*ReflectionTest.testRepeatedCompositeRemovec s� t j� dkrd S tj}|dddd|j|j|jdd d d dd tj� d�}tj ddd d g g |gg tj � d � � G � fd d �d t j t jtj��}|� }| jd|j� | j|jd� � d |_| jd |j� | j|jd�� d S )Nrw � foo_fieldzMyProto.foo_fieldr r? F)�name� full_namer� �number�typeZcpp_type�labelZ default_value�containing_type� message_typeZ enum_typeZ is_extensionZextension_scope�optionsZMyProtoZignored) r r �filenamer � nested_types� enum_types�fields� extensionsr c s e Zd Z� ZdS )z>ReflectionTest.testHandWrittenReflection.<locals>.MyProtoClassN)r& r' r( r� r )� mydescriptorr r � MyProtoClassH s r ri )r ry r �FieldDescriptor� TYPE_INT64Z CPPTYPE_INT64�LABEL_OPTIONALr Z FieldOptions� DescriptorZMessageOptions�six�with_metaclassr � GeneratedProtocolMessageTyper �Messager0 r r^ r< )r r Zfoo_field_descriptorr Zmyproto_instancer )r r �testHandWrittenReflection4 s. z(ReflectionTest.testHandWrittenReflectionc s� t j� dkrd S � fdd�� d� _tj� }d|_tj}� |d|j� � |d|j� � |d|j � � |d |j � � jd 7 _|j j � }d |_|j|_ � j|_tjj|_tj|��| jd�jk� | jd�jk� | jd�jk� | jd �jk� | jd �jk� G �fd d �d tjtjtj��}|� }d|_d|_d|_ d|_ |j j ddg� |j � }tj!�|�}| j||k � | j"||� | j"|j|j� | j"|j|j� | j"|j |j � | j"|j |j � | j"|j |j � d S )Nrw c s: � j d7 _ | jj� }||_||_� j |_tjj|_ d S )Nr? ) � field_index�fieldr| r r r r �FieldDescriptorProtor r )r= r� Z field_type� new_field)�AddDescriptorFieldr r r% W s zEReflectionTest.testDescriptorProtoSupport.<locals>.AddDescriptorFieldr ZCarr �year� automatic�pricer? �ownersc s e Zd Z� ZdS )z=ReflectionTest.testDescriptorProtoSupport.<locals>.CarMessageN)r& r' r( r� r )�descr r � CarMessagew s r+ �priusi� Tg ���@ZbobZsusan)#r ry r! r ZDescriptorProtor r# Z TYPE_STRINGr Z TYPE_BOOLZ TYPE_DOUBLEr" r| r r �LABEL_REPEATEDr r �MakeDescriptorr^ r� r r r r r r r& r' r( r) r� r� Z ParseMessager0 )r Z desc_protoZfdpr$ r+ r, Zserialized_priusZ new_priusr )r% r* r �testDescriptorProtoSupportQ sN z)ReflectionTest.testDescriptorProtoSupportc C s� t j� }t j}| j|j|� � | jd|j| � | j|j|� � d|j|< | jd|j| � | j|j|�� |j|� | jd|j| � | j|j|� � d S )Nr ri )r r� r� r^ � HasExtensionr0 r� �ClearExtension)r �extendee_proto� extensionr r r �'testTopLevelExtensionsForOptionalScalar� s z6ReflectionTest.testTopLevelExtensionsForOptionalScalarc C s� t j� }t j}| jdt|j| �� |j| jd� | jdg|j| � |j| }|j|� | jdt|j| �� | j||j| k � | j t t j |j|d� d S )Nr rX rM ) r r� r� r0 r$ r� r{ r1 r^ rY rZ �operator�setitem)r r2 r3 Z string_listr r r �'testTopLevelExtensionsForRepeatedScalar� s z6ReflectionTest.testTopLevelExtensionsForRepeatedScalarc C s� t j� }t j}| j|j|� � | jd|j| j� | j|j|� � d|j| _| jd|j| j� | j|j|�� |j| }|j|� | j||j| k � d|_| jd|j� | j|j d�� | j|j|� � | j t t j |j|d� d S )Nr ri r� rK rM )r r� Z"optional_foreign_message_extensionr^ r0 r0 r� rK r1 r< rY rZ r5 r6 )r r2 r3 Zforeign_messager r r �(testTopLevelExtensionsForOptionalMessage� s" z7ReflectionTest.testTopLevelExtensionsForOptionalMessagec C s� t j� }t j}| jdt|j| �� |j| j� }d|_| jd|j| d j� d|_| jd|j| d j� |j| }|j|� | jdt|j| �� | j ||j| k � | j t t j |j|d� d S )Nr ri r� rM )r r� �repeatedgroup_extensionr0 r$ r� r| rM r1 r^ rY rZ r5 r6 )r r2 r3 �groupZ group_listr r r �(testTopLevelExtensionsForRepeatedMessage� s z7ReflectionTest.testTopLevelExtensionsForRepeatedMessagec C s� t j� }t jj}| j|j|� � |j| }| jd|j� | j|j|� � d|_| jd|j| j� | j|j|�� |j |� | j||j| k � | j|j|� � d S )Nr ri ) r r� � TestRequired�singler^ r0 r� r0 rM r1 )r r2 r3 Zrequiredr r r �testNestedExtensions� s z#ReflectionTest.testNestedExtensionsc C sT t jj}| j|jt jjd�� | j|jd�jt jj� | j dt |j t j j��� d S )Nr? z*protobuf_unittest.optional_int32_extensionr ) r r� �poolr^ ZFindExtensionByNumberr� ZassertIsZFindExtensionByNamer r0 r$ ZFindAllExtensionsr: )r r? r r r �testRegisteredExtensions� s z'ReflectionTest.testRegisteredExtensionsc C s� t j� }| j|jd� � | jd|jjt j � | j|jd� � d|jjt j< | jd|jjt j � | j|jd�� t j� }| j|jd� � | jg |jjt j � | j|jd� � |jjt j j d� | jdg|jjt j � | j|jd�� t j� }| j|jd� � | jd|jjt j j � | j|jd� � d|jjt j _ | jd|jjt j j � | j|jd�� t j� }| j|jd� � | jdt |jjt j �� | j|jd� � |jjt j j� }| j||jjt j d � | j|jd�� d S )N� submessager ri )r �TopLevelMessager^ r< r0 rA r� �optional_int_extension�repeated_int_extensionr{ �optional_message_extension�foreign_message_intr$ �repeated_message_extensionr| )r �toplevel�foreignr r r �(testHasBitsForAncestorsOfExtendedMessage� sL z7ReflectionTest.testHasBitsForAncestorsOfExtendedMessagec C sF t j� }|j}t j}|j| }|j|� d|_| j||j| k � d S )Nri )r rB rA rE r� r1 rF r^ )r rH r2 r3 Zextension_protor r r �*testDisconnectionAfterClearingEmptyMessage* s z9ReflectionTest.testDisconnectionAfterClearingEmptyMessagec C s� t j� }| jt|jd� | jt|jd� | jt|jjd� | jt|jjdd� xbt j t j t j t j fD ]J}| jt|j|� | jt|j|� | jt|jj|� | jt|jj|d� qdW | jt|jt j� d S )Ni� rp )r r� rY �KeyErrorr0 r1 r� r� r� r rC rE rD rG r� )r r2 Zunknown_handler r r �testExtensionFailureModes4 s& z(ReflectionTest.testExtensionFailureModesc C s6 t j� }tj|� |j� }t jj|�}| j||� d S )N)r r: r � SetAllFieldsr� � FromStringr0 )r rt �string1ru r r r �testStaticParseFromS s z"ReflectionTest.testStaticParseFromc C sF t j� }d|_t j� }d|_|j|� | jd|j� | jd|j� d S )Nr? r� )r r: r7 r6 rs r0 )r rt ru r r r �testMergeFromSingularField] s z)ReflectionTest.testMergeFromSingularFieldc C sx t j� }|jjd� |jjd� t j� }|jjd� |j|� | jd|jd � | jd|jd � | jd|jd � d S )Nr? r@ r )r r: rB r{ rs r0 )r rt ru r r r �testMergeFromRepeatedFieldj s z)ReflectionTest.testMergeFromRepeatedFieldc C s6 t j� }d|j_t j� }|j|� | jd|jj� d S )NrL )r r: re rM rs r0 )r rt ru r r r �testMergeFromOptionalGroupx s z)ReflectionTest.testMergeFromOptionalGroupc C s� t j� }|jj� }d|_|jj� }d|_t j� }|jj� }d|_|j|� | jd|jd j� | jd|jd j� | jd|jd j� t j� }|jj|j� | jd|jd j� | jd|jd j� | jd|jd j� d S )Nr� iA i� r r? r@ )r r: rN r| rI rs r0 )r rt �mru Zproto3r r r �"testMergeFromRepeatedNestedMessage� s" z1ReflectionTest.testMergeFromRepeatedNestedMessagec C sP t j� }tj|� t j� }|j|� | j||� |j� }|j� }| j||� d S )N)r r: r rN rs r0 r� )r rt ru rP �string2r r r �testMergeFromAllFields� s z%ReflectionTest.testMergeFromAllFieldsc C s> t j� }d|jt j< t j� }|j|� | jd|jt j � d S )Nr? )r r� r� r� rs r0 )r rt ru r r r � testMergeFromExtensionsSingular� s z.ReflectionTest.testMergeFromExtensionsSingularc C s� t j� }|jt j jd� |jt j jd� t j� }|jt j jd� |j|� | jdt|jt j �� | jd|jt j d � | jd|jt j d � | jd|jt j d � d S )Nr? r@ r rA )r r� r� r� r{ rs r0 r$ )r rt ru r r r � testMergeFromExtensionsRepeated� s z.ReflectionTest.testMergeFromExtensionsRepeatedc C s� t j� }|jt j }|j� }d|_|j� }d|_t j� }|jt j }|j� }d|_|j|� |jt j }| jdt|�� | jd|d j� | jd|d j� | jd|d j� d S )N�� iM �o rA r r? r@ ) r r� r� �!repeated_nested_message_extensionr| rI rs r0 r$ )r rt Zext1rU ru Zext2r r r �$testMergeFromExtensionsNestedMessage� s& z3ReflectionTest.testMergeFromExtensionsNestedMessagec C sD t j� }t j� }|j | j|jd�� |j|� | j|jd�� d S )NrV )r r: rV r; r< rs )r �message1�message2r r r �testMergeFromBug� s z ReflectionTest.testMergeFromBugc C sL t j� }d|_d|_t j� }d|_|j|� | jd|j� | jd|j� d S )Nr? zimportant-textr� )r r: r7 r6 r� r0 )r rt ru r r r �testCopyFromSingularField� s z(ReflectionTest.testCopyFromSingularFieldc C sf t j� }|jjd� |jjd� t j� }|jjd� |j|� | jd|jd � | jd|jd � d S )Nr? r@ r )r r: rB r{ r� r0 )r rt ru r r r �testCopyFromRepeatedField� s z(ReflectionTest.testCopyFromRepeatedFieldc C sP t j� }tj|� t j� }|j|� | j||� |j� }|j� }| j||� d S )N)r r: r rN r� r0 r� )r rt ru rP rW r r r �testCopyFromAllFields� s z$ReflectionTest.testCopyFromAllFieldsc C s\ t j� }|jjd� d|_d|_|j|� | jd|jd � | jd|j� | jd|j� d S )Nr? r@ zimportant-textr )r r: rB r{ r7 r6 r� r0 )r rt r r r �testCopyFromSelf s z ReflectionTest.testCopyFromSelfc C s4 t j� dkrd S tj� }tj� }| jt|j|� d S )Nrw )r ry r r: r� rY rZ r� )r rt ru r r r �testCopyFromBadType s z"ReflectionTest.testCopyFromBadTypec C s� t j� }d|_tj|�}| jd|j� |jjd� |jjd� tj|j�}| jddg|� |jj � }d|_ tj|j�}| j|j|� d|_ | j |j|� d S )Nr? r@ rA ) r r: r7 �copy�deepcopyr0 rB r{ rN r| rI �assertNotEqual)r rt ru Z containerr_ Zmessagesr r r � testDeepCopy s zReflectionTest.testDeepCopyc C s� t j� }tj� dkr tj|� n tj|� |j� | j|j � d� t j� }| j||� t j � }tj |� |j� | j|j � d� t j � }| j||� d S )Nrw r ) r r: r ry r rN ZSetAllNonLazyFields�Clearr0 r\ r� �SetAllExtensions)r r= Z empty_protor r r � testClear4 s zReflectionTest.testClearc C s t j� }|j}|j� | j||jk � d|_| j|jd� � | jd|jj� t j� }|j}d|_|j}d|_ |j� | j||jk � | j||jk � | jd|j� | jd|j � d|_d|_ | j |jd�� | jd|jj� | j |jd�� | jd|jj � d S ) Nri rV r rp r� r� � rf ) r r: rV rk r^ rI r< r0 rf rK r; )r r= rj rI r r r � testDisconnectingBeforeClearK s. z+ReflectionTest.testDisconnectingBeforeClearc C sf t j� }d|j_|j}| j|jjd� | j|jjd� d|j_| j|jjd� | j|jd� d|_d S )Nrp r r� � )r Z TestOneof2Z foo_messageZqux_intr0 Zfoo_lazy_message)r rU Z sub_messager r r �testDisconnectingInOneoff s z'ReflectionTest.testDisconnectingInOneofc C sr t j� }d|_d|j_| jd|jj� | j|jd�� |j}d|_| jd|j� | jd|j� | j|jd�� d S )Nr� r� � oneof_uint32r� �oneof_nested_message) r r: rr rs rI r0 r; r< Z oneof_string)r r= rj r r r � testOneOft s zReflectionTest.testOneOfc C s" | j |j� � |j� |j� d S )N)r^ � IsInitializedr� �SerializePartialToString)r r= r r r �assertInitialized� s z ReflectionTest.assertInitializedNc C sN g }| j |j� � | j |j|�� | j|t|�� | jtj|j� |j� d S )N) r; ru r0 r$ rY r � EncodeErrorr� rv )r r= Z error_size�errorsr r r �assertNotInitialized� s z#ReflectionTest.assertNotInitializedc C s� t j� }| j|� t j� }| j|� t j� }| j|d� d |_ |_|_| j|� t j � }| j|� d|j _| j|d� d|j _d|j _| j|� |j j � }| j|d� d |_ |_|_| j|� t j� }t jj }|j| j � }|j| j � }| j|d� d|_d|_d|_| j|d� d|_d|_d|_| j|� t j� }t jj}d|j| _| j|d� d|j| _d|j| _| j|� g }t j� }| j|j|�� | j|dddg� d S ) NrA r@ r? r r� rM �brK )r r: rw r� r< rz rM r{ rK r� �optional_messager� r| �multir� r= r; ru r0 )r r= r_ r3 r` ry r r r �testIsInitialized� sV z ReflectionTest.testIsInitialized�cppr@ zBErrors are only available from the most recent C++ implementation.c C s� d}d}t j� }||_||_|jj� }d|_tj|||j� d� d}||_|jj� }d|_| j t ��`}tj|||j� d� | j t |d�dt |jd|j� � | jdt|j�� | jd t|j�� W d Q R X d S ) Nz!test_file_descriptor_errors.protoZmsg1)Z serialized_pbz)another_test_file_descriptor_errors.protoZmsg2� exceptionz %s not raisedr& z test_file_descriptor_errors.msg1)r �FileDescriptorProtor �packager r| r ZFileDescriptorr� rY rZ r^ �hasattrrb �expected�assertInr� r� )r � file_nameZ package_nameZfile_descriptor_protor Zanother_file_namer �cmr r r �testFileDescriptorErrors� s0 z'ReflectionTest.testFileDescriptorErrorsc C s� t j� }| jtt|dd� | jt|j�tj � tj d�|_| j|jt d�� t d�|_| j|jtj d�� | jt t|dd� dj d�}||_d|_d|_d S ) Nr� zunicode objectZTestingr6 s a�au Тестzutf-8r� ) r r: rY rZ rc r0 r r6 r � text_typer� r� �encode)r r= Z utf8_bytesr r r �testStringUTF8Encoding� s z%ReflectionTest.testStringUTF8Encodingc C sj t j� }t j}|j}d}|jd�}||j| _|j� }| j|j � t |�� t j � }|j |�}| jt |�|� t j� } | jdt |j�� | j|jd jd� | j|jd jj|�� | j |jd j�}| jt |jd j�|� | jt| j�tj� | j| j|� |jd jj|t |�d �} d} y| j | � W n tk �rF d} Y nX | j} | j| �pbt| �tk� d S ) Nu Тестzutf-8r? r iʿ�� �FT)r �TestMessageSet�TestMessageSetExtension2�message_set_extensionr� r� r� r� r0 r\ r$ r � RawMessageSet�MergeFromString�item�type_idr^ r �endswithr r r� �replace�UnicodeDecodeErrorr ) r r= Zextension_messager3 Z test_utf8Ztest_utf8_bytes� serialized�raw� bytes_readr` ZbadbytesZunicode_decode_failedZ string_fieldr r r �testStringUTF8Serialization s8 z*ReflectionTest.testStringUTF8Serializationc C s" t jdd�}| jdtj|�� d S )Ns ��)r� z#optional_bytes: "\000\177\200\377" )r r: r0 r r� )r r= r r r �testBytesInTextFormat= s z$ReflectionTest.testBytesInTextFormatc C s� t j� }|jjt jj� � | j|jd�� t j� }|jjt jj� � | j|jd�� t j� }|jjd�}| j d|� | j|jd�� t j� }|jj d� | j|jd�� |j � }t j� }| j t |�|j|�� | j|jd�� d S )NrV r� r ) r r: rV rs rQ r^ r< r� r� r0 r� r� r$ )r r= r� r� ru r r r �testEmptyNestedMessageB s* z%ReflectionTest.testEmptyNestedMessagec C s6 t j� }| j|jd�� |jj� | j|jd�� d S )Nre )r r: r; r< re Z SetInParentr^ )r r= r r r �testSetInParent] s z ReflectionTest.testSetInParentc C s* ddl m} |jj� }| jd|jj� dS )a� Test that we can import nested messages from their __init__.py. Such setup is not trivial since at the time of processing of __init__.py one can't refer to its submodules by name in code, so expressions like google.protobuf.internal.import_test_package.inner_pb2 don't work. They do work in imports, so we have assign an alias at import and then use that alias in generated code. r )�import_test_package�9 N)�google.protobuf.internalr� ZmyprotoZOuterr0 �innerr� )r r� �msgr r r � testPackageInitializationImportc s z.ReflectionTest.testPackageInitializationImport)N)`r& r' r( r4 r> rH rU rW r[ r] r` rh rk rl ro rq rr rv rz r} r r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r/ r4 r7 r8 r; r> r@ rJ rK rM rQ rR rS rT rV rX rY rZ r^ ra rb rc rd re rf rj rm ro rq rt rw rz r~ �unittest�skipIfr ry �Versionr� r� r� r� r� r� r� r r r r r/ r s� "#Q ) " ! Z S =1 : 6r/ c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) �TestAllTypesEqualityTestc C s t j� | _t j� | _d S )N)r r: � first_proto� second_proto)r r r r �setUp{ s z TestAllTypesEqualityTest.setUpc C s | j tt| j� d S )N)rY rZ �hashr� )r r r r �testNotHashable s z(TestAllTypesEqualityTest.testNotHashablec C s | j | j| j� d S )N)r0 r� )r r r r �testSelfEquality� s z)TestAllTypesEqualityTest.testSelfEqualityc C s | j | j| j� d S )N)r0 r� r� )r r r r �testEmptyProtosEqual� s z-TestAllTypesEqualityTest.testEmptyProtosEqualN)r& r' r( r� r� r� r� r r r r r� y s r� c @ sh e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dd� Z dd� ZdS )�FullProtosEqualityTestz@Equality tests using completely-full protos as a starting point.c C s0 t j� | _t j� | _tj| j� tj| j� d S )N)r r: r� r� r rN )r r r r r� � s z FullProtosEqualityTest.setUpc C s | j tt| j� d S )N)rY rZ r� r� )r r r r r� � s z&FullProtosEqualityTest.testNotHashablec C s | j | jd � | j d | j� d S )N)ri r� r� )r r r r �testNoneNotEqual� s z'FullProtosEqualityTest.testNoneNotEqualc C s( t j� }| j| j|� | j|| j� d S )N)r r< ri r� r� )r Z third_protor r r �testNotEqualToOtherMessage� s z1FullProtosEqualityTest.testNotEqualToOtherMessagec C s | j | j| j� d S )N)r0 r� r� )r r r r �testAllFieldsFilledEquality� s z2FullProtosEqualityTest.testAllFieldsFilledEqualityc C s@ | j jd7 _| j| j | j� | j jd� | j| j | j� d S )Nr? r7 )r� r7 ri r� r_ )r r r r �testNonRepeatedScalar� s z,FullProtosEqualityTest.testNonRepeatedScalarc C s� | j j jd7 _| j| j | j� | j j jd8 _| j| j | j� | j jjd� | j| j | j� | jjj| j j_| j| j | j� | j jd� | j| j | j� d S )Nr? rI rV )r� rV rI ri r� r0 r_ )r r r r �testNonRepeatedComposite� s z/FullProtosEqualityTest.testNonRepeatedCompositec C s> | j jjd� | j| j | j� | j jd� | j| j | j� d S )Nrp rB )r� rB r{ ri r� r_ )r r r r �testRepeatedScalar� s z)FullProtosEqualityTest.testRepeatedScalarc C s� | j jd jd7 _| j| j | j� | j jd jd8 _| j| j | j� | j jj� | j| j | j� | jjj� | j| j | j� d S )Nr r? )r� rN rI ri r� r0 r| )r r r r �testRepeatedComposite� s z,FullProtosEqualityTest.testRepeatedCompositec C s( | j jd� d| j_| j| j | j� d S )Nr7 r )r� r_ r� r7 ri )r r r r � testNonRepeatedScalarHasBits� s z3FullProtosEqualityTest.testNonRepeatedScalarHasBitsc C sV | j jd� | jjjd� | j| j | j� d| j j_| j jjd� | j| j | j� d S )NrV rI r )r� r_ r� rV ri rI r0 )r r r r � testNonRepeatedCompositeHasBits� s z6FullProtosEqualityTest.testNonRepeatedCompositeHasBitsN)r& r' r( r) r� r� r� r� r� r� r� r� r� r� r� r r r r r� � s r� c @ s e Zd Zdd� ZdS )�ExtensionEqualityTestc C s� t j� }t j� }| j||� tj|� | j||� tj|� | j||� |jt j d7 < | j||� |jt j d8 < | j||� |jt j� d|jt j< | j||� d|jt j< | j||� t j� }t j� }| jd|jt j � | j||� d S )Nr? r ) r r� r0 r rl ri r� r� r1 )r r� r� r r r �testExtensionEquality� s* z+ExtensionEqualityTest.testExtensionEqualityN)r& r' r( r� r r r r r� � s r� c @ s e Zd Zdd� ZdS )�MutualRecursionEqualityTestc C sP t j� }t j� }| j||� d|jjj_| j||� d|jjj_| j||� d S )Nri )r r~ r0 rI rM r7 ri )r r� r� r r r � testEqualityWithMutualRecursion s z;MutualRecursionEqualityTest.testEqualityWithMutualRecursionN)r& r' r( r� r r r r r� s r� c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-S ).� ByteSizeTestc C s, t j� | _tj� | _t j� | _t j� | _ d S )N) r r: r= r ZExtendedMessage�extended_proto�TestPackedTypes� packed_proto�TestPackedExtensions�packed_extended_proto)r r r r r� s zByteSizeTest.setUpc C s | j j� S )N)r= r\ )r r r r �Size s zByteSizeTest.Sizec C s | j d| jj� � d S )Nr )r0 r= r\ )r r r r �testEmptyMessage s z ByteSizeTest.testEmptyMessagec C s8 t j� }| jd|j� � t jdd�}| jd|j� � d S )Nr r? )r� r@ )r r: r0 r\ )r r= Z proto_kwargsr r r �testSizedOnKwargs s z ByteSizeTest.testSizedOnKwargsc s� � fdd�}|dd� |dd� x>t ttddd��ttdd���D ]\}}|d|> d |� qBW |d d� |d d� |d d� d S )Nc s* � j j� | � j _� j|d � j� � d S )Nr? )r= rk r� r0 r� )r3 Zexpected_varint_size)r r r �Test# s z&ByteSizeTest.testVarints.<locals>.Testr r? rp r� i' r� r@ r� r� l l���� )�ziprG r1 )r r� r3 Z num_bytesr )r r � testVarints" s * zByteSizeTest.testVarintsc C sh d| j _| jd| j� � d| j _| jdt| j j� | j� � dd | j _| jdt| j j� | j� � d S )Nr� r@ r� r� � rA )r= r6 r0 r� r$ )r r r r � testStrings1 s zByteSizeTest.testStringsc C s� d| j _| jd| j� � tj� | _ d| j _| jd| j� � tj� | _ d| j _| jd| j� � tj� | _ d| j _| jd| j� � tj� | _ d| j _ | jd| j� � tj� | _ d S )Ni� rp � gX9��v��?�@ rA ) r= r� r0 r� r r: r� r9 r8 r� )r r r r �testOtherNumerics> s z ByteSizeTest.testOtherNumericsc C s d| j j_| jd| j� � d S ) Nr? � rA r@ i @ r rp rp )r= rV rI r0 r� )r r r r �testCompositesX s zByteSizeTest.testCompositesc C s d| j j_| jd| j� � d S ) Nr? r� r r@ i r� r r� )r= re rM r0 r� )r r r r � testGroups` s zByteSizeTest.testGroupsc C s0 | j jjd� | j jjd� | jd| j� � d S )Nr� r� r? r@ rA r rp )r= rB r{ r0 r� )r r r r r� g s z ByteSizeTest.testRepeatedScalarsc C s& | j jjddg� | jd| j� � d S )Nr� r� r? r@ rA r rp )r= rB r� r0 r� )r r r r �testRepeatedScalarsExtendm s z&ByteSizeTest.testRepeatedScalarsExtendc C sN | j jjd� | j jjd� | jd| j� � | j jjd� | jd| j� � d S ) Nr� r� r? r@ rA r rp rA )r= rB r{ r0 r� r )r r r r r r s z&ByteSizeTest.testRepeatedScalarsRemovec C s2 | j jj� }| j jj� }d|_| jd| j� � d S ) Nrp r@ r? rA rp r� rp r" )r= rN r| rI r0 r� )r �foreign_message_0�foreign_message_1r r r r z s z#ByteSizeTest.testRepeatedCompositesc C s | j jj� }| j jj� }d|_| jd | j� � tj| j j�}| j jd= | jd| j� � | j jj� }d|_| jd| j� � | j jd= | jd| j� � | j jd= | jd| j� � | jdt|�� |dd�= t j � dk� r�| jdt|�� |d= t j � dk�r| jdt|�� d S )Nr� r@ r? r rL rw rA rp r� rp r" rA r rp rA r rp rp r" r� r� rA r rp r� ) r= rN r| rI r0 r� rg rh r$ r ry )r r� r� rN Zforeign_message_2r r r � testRepeatedCompositesDelete� s, z)ByteSizeTest.testRepeatedCompositesDeletec C s2 | j jj� }| j jj� }d|_| jd| j� � d S ) Nrp r@ r? r r� r" r� r� )r= rP r| rM r0 r� )r Zgroup_0Zgroup_1r r r �testRepeatedGroups� s z ByteSizeTest.testRepeatedGroupsc C sj t j� }| jd|j� � t j}d|j|< | jd|j� � t jjjd }| j t �� d|j|< W d Q R X d S )Nr ri r@ r7 ) r r� r0 r\ r� r� r: r� r� rY rL )r r= r3 r" r r r �testExtensions� s zByteSizeTest.testExtensionsc C s� d| j _| jd| j j� � d| j _| jd| j j� � | j jd� | jd| j j� � tj}d| jj|< | jd| jj� � d| jj|< | jd| jj� � | jj |� | jd| jj� � d S )Nr? r@ r� rA r7 r ) r= r7 r0 r\ r_ r rC r� r� r1 )r r3 r r r �)testCacheInvalidationForNonrepeatedScalar� s z6ByteSizeTest.testCacheInvalidationForNonrepeatedScalarc C s | j jjd� | jd| j j� � | j jjd� | jd| j j� � d| j jd< | jd| j j� � | j jd� | jd| j j� � tj}| jj | }|jd� | jd| jj� � |jd� | jd | jj� � d|d< | jd | jj� � | jj |� | jd| jj� � d S ) Nr? rA r� r� rp rB r r@ r rp ) r= rB r{ r0 r\ r_ r rD r� r� r1 )r r3 Zrepeatedr r r �&testCacheInvalidationForRepeatedScalar� s$ z3ByteSizeTest.testCacheInvalidationForRepeatedScalarc C s$ d| j j_| jd| j j� � d| j j_| jd| j j� � | j jjd� | jd| j j� � | j jd� | jd| j j� � tj� d kr�| j j}| j jd� d|_| jd| j j� � tj }| j j | }| jd| j j� � d|_ | jd | j j� � d|_ | jd| j j� � | j j |� | jd| j j� � d S ) Nr? rp r� r� rK rA rf r rw r )r= rf rK r0 r\ r_ r ry r rE r� r� rF r1 )r �childr3 r r r �*testCacheInvalidationForNonrepeatedMessage� s, z7ByteSizeTest.testCacheInvalidationForNonrepeatedMessagec C s | j jj� }| jd| j j� � | j jj� | jd| j j� � d|_| jd| j j� � | j jd� | jd| j j� � tj}| j j | }|j� }| jd| j j� � |j� | jd| j j� � d|_ | jd| j j� � |jd � | jd| j j� � | j j |� | jd| j j� � d S ) NrA r� r? r" rO r r@ r rF ) r= rO r| r0 r\ rK r_ r rG r� r� rF r1 )r Zchild0r3 Z child_listr r r �'testCacheInvalidationForRepeatedMessage s( z4ByteSizeTest.testCacheInvalidationForRepeatedMessagec C s� | j d| jj� � | jjjd� | jjjd� d }| j || jj� � | jjjd� | jjjd� d}| j || | jj� � | jjd � | j || jj� � d S )Nr r� r� r? r@ rA g������@g @r" � packed_int32rA r� rn � )r0 r� r\ r� r{ � packed_doubler_ )r Zint_sizeZ double_sizer r r �testPackedRepeatedScalars s z&ByteSizeTest.testPackedRepeatedScalarsc C sH | j d| jj� � | jjtj }|jddddg� | j d| jj� � d S )Nr r? r@ rA r r� )r0 r� r\ r� r Zpacked_fixed32_extensionr� )r r3 r r r �testPackedExtensions0 s z!ByteSizeTest.testPackedExtensionsN)r& r' r( r� r� r� r� r� r� r� r� r� r� r� r r r� r� r� r� r� r� r� r� r� r r r r r� s, ' r� c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Zd3d4� Z d5d6� Z d7d8� Z d9S ):�SerializationTestc C sR t j� }t j� }|j� }| j|j� t|�� | jt|�|j|�� | j||� d S )N)r r: r� r0 r\ r$ r� )r r� r� r� r r r �testSerializeEmtpyMessageF s z+SerializationTest.testSerializeEmtpyMessagec C s\ t j� }t j� }tj|� |j� }| j|j� t|�� | jt|�|j|�� | j||� d S )N) r r: r rN r� r0 r\ r$ r� )r r� r� r� r r r �testSerializeAllFieldsP s z(SerializationTest.testSerializeAllFieldsc C sH t j� }t j� }tj|� |j� }| jt|�|j|�� | j||� d S )N)r r� r rl r� r0 r$ r� )r r� r� r� r r r �testSerializeAllExtensions[ s z,SerializationTest.testSerializeAllExtensionsc C sF t j� }t j� }d|j_|j� }| jt|�|j|�� | j||� d S )N�� )r r: re rM r� r0 r$ r� )r r� r� r� r r r � testSerializeWithOptionalGroupe s z0SerializationTest.testSerializeWithOptionalGroupc C sL t j� }d|_d |_d |_d |_d|_d|_t jj|j � �}| j ||� d S )Nr? r@ r� rA r rp r� r� l l���� r� l l���� ���l l���� ) r r: r7 r� r� r� r� r� rO r� r0 )r r� r� r r r �testSerializeNegativeValueso s z-SerializationTest.testSerializeNegativeValuesc C s� t j� dkrd S tj� }tj|� |j� }x�tt|�d �D ]�}yntj� }tj � }|j |d|�}| j ||� y |j |d|�}| j ||� W n t j k r� | jd� Y nX W q< t j k r� | jt j |j |d|� Y q<X q<W d S )Nrw r? r z@Parsing unknown fields failed when parsing known fields did not.)r ry r r: r rN r� r1 r$ �TestEmptyMessageZ_InternalParser0 r Z DecodeErrorZfailrY )r r� r� Ztruncation_pointr� Zunknown_fields�posZpos2r r r �testParseTruncated~ s& z$SerializationTest.testParseTruncatedc C s t j� }d|_d|jt j< d|_d|jt j< d|_|j� }| j |j � t |�� t |�}|j }| j dtjf|� � | j d|j� � | j dtjf|� � | j d|j� � | j dtjf|� � | j d|j� � | j dtjf|� � | j d|j� � | j dtjf|� � | j d|j� � d S )Nrp r rA r@ r? )r ZOutOfOrderFieldsr� r� r� r� r� r7 r� r0 r\ r$ r r r ZWIRETYPE_VARINTr* r+ r, r- r. )r r= r� rg �ReadTagr r r � testCanonicalSerializationOrder� s( z1SerializationTest.testCanonicalSerializationOrderc C s( t j� }tj|� |j� }tj|� d S )N)r r� r r� r� Z#ExpectAllFieldsAndExtensionsInOrder)r r= r� r r r �(testCanonicalSerializationOrderSameAsCpp� s z:SerializationTest.testCanonicalSerializationOrderSameAsCppc C s� t j� }|jjd� d|_d|j_|j� }t j� }|jjd� d|_d|j_|j|�}| j t |�|� | j ddgt |j�� | j d|j� | j d|jj� d S )NZfoobarri r� r� r� i� ) r r: rE r{ r7 rV rI r� r� r0 r$ rG )r r� r� r� Z bytes_parsedr r r �'testMergeFromStringWhenFieldsAlreadySet� s z9SerializationTest.testMergeFromStringWhenFieldsAlreadySetc C s� t j� }t j}t j}|j}|j}t j}d|j| _d|j| _d|j| _ |j � }t j � }| j d|jj� j� | j t|�|j|�� | j dt|j�� t j� } | j t|jd j�| j|jd j�� | j d| j� t j� } | j t|jd j�| j|jd j�� | j d| j� t j� } | j t|jd j�| j|jd j�� | j d| j � t j� } | j t|�| j|�� | j d| j| j� | j d| j| j� | j d| j| j � | j | j� t|�� | j |j� t|�� d S ) Nr� rX r� FrA r r? r@ )r r� �TestMessageSetExtension1r� r� Zmessage_set_extension3r� r3 r� �textr� r r� r0 r� � GetOptions�message_set_wire_formatr$ r� r� r ZTestMessageSetExtension3r\ ) r r= �extension_message1Zextension_message2� extension1Z extension2Z extension3r� r� r_ r` Zmessage3ru r r r �testMessageSetWireFormat� sR z*SerializationTest.testMessageSetWireFormatc C s� t j� }|jj� }d|_tj}tj� }d|_|j� |_ |jj� }d|_tj}tj� }d|_|j� |_ |jj� }d|_tj � }d|_ |j� |_ |j� }tj � }| j t|�|j|�� tj}|j}| j d|j| j� d S )Ni���i90 i���i:0 i���rX )r r� r� r| r� r r� r3 r� r r� r� r� r0 r$ r� r� r� )r r� r� r� r_ r� r= r� r r r �(testMessageSetWireFormatUnknownExtension s4 z:SerializationTest.testMessageSetWireFormatUnknownExtensionc C sp t j� }tj|� |j� }t j� }| jt|�|j|�� t j� }d|_ |j� }t j� }| jt|�|j|�� d S )Nl ����) r r: r rN r� r� r0 r$ r� r� )r r= r� ru r r r �testUnknownFields2 s z#SerializationTest.testUnknownFieldsc C sR y |� W n0 |k r: } z| j |t|�� dS d}~X nX | jdt|� ��dS )zEThis method checks if the excpetion type and message are as expected.Nz %s not raised)r0 r� ZfailureException)r Z exc_classZ callable_objr� Zexr r r � _CheckRaisesM s z SerializationTest._CheckRaisesc C s6 t j� }| jtj|jd� |j� }t j� }| j|jd�� |j |� | j|jd�� d|_ | jtj|jd� |j� }d|_ | jtj|jd� |j� }d|_ |j� }|j� }t j� }| j t|�|j|�� | j d|j � | j d|j � | j d|j � | j t|�|j|�� | j d|j � | j d|j � | j d|j � d S )NzHMessage protobuf_unittest.TestRequired is missing required fields: a,b,crM r? zFMessage protobuf_unittest.TestRequired is missing required fields: b,cr@ zDMessage protobuf_unittest.TestRequired is missing required fields: crA )r r< r� r rx r� rv r; r< r� rM r{ rK r0 r$ r� )r r= �partialru r� r r r �testSerializeUninitializedX sL z,SerializationTest.testSerializeUninitializedc C s� t j� }|j� d|j_| jtj|jd� d|j_d|j_ |j� d|j j � _d|j j � _| jtj|jd� d|j d _d|j d _ d|j d _d|j d _ |j� d S )Nr? zoMessage protobuf_unittest.TestRequiredForeign is missing required fields: optional_message.b,optional_message.cr@ rA z�Message protobuf_unittest.TestRequiredForeign is missing required fields: repeated_message[0].b,repeated_message[0].c,repeated_message[1].a,repeated_message[1].cr ) r r� r� r| rM r� r rx r{ rK r� r| )r r= r r r �$testSerializeUninitializedSubMessage� s* z6SerializationTest.testSerializeUninitializedSubMessagec C s` t j� }t j� }tj|� |j� }| j|j� t|�� |j|�}| j|j� |� | j||� d S )N) r r� r �SetAllPackedFieldsr� r0 r\ r$ r� )r r� r� r� r� r r r � testSerializeAllPackedFields� s z.SerializationTest.testSerializeAllPackedFieldsc C sL t j� }t j� }tj|� |j� }|j|�}| j|j� |� | j||� d S )N)r r� r ZSetAllPackedExtensionsr� r� r0 r\ )r r� r� r� r� r r r � testSerializeAllPackedExtensions� s z2SerializationTest.testSerializeAllPackedExtensionsc C s� t j� }|jjddg� |jjd� |j� }t j� }|jjd� |jjddg� |jjd� | jt |�|j |�� | jdddg|j� | jdddg|j� | jdg|j� d S )Nr? r@ g @rA g �?g @r ) r r� r� r� r� r{ r� Z packed_sint32r0 r$ r� )r r� r� r� r r r �1testMergePackedFromStringWhenSomeFieldsAlreadySet� s zCSerializationTest.testMergePackedFromStringWhenSomeFieldsAlreadySetc C sP t j� }|jjddddg� |jjddg� |jjd� |j� }| j|j � t |�� t |�}|j }| jdt jf|� � | jd|j� � | jd|j� � | jd|j� � | jd|j� � | jd|j� � | jd t jf|� � | jd |j� � | jd|j� � | jd t jf|� � | jd|j� � | jd|j� � | jd|j� � | j|j� � d S )Nr? r@ � rA g �?g @�@g @�Z r� r �e r" r@ rA rp rn )r r� r� r� r� Z packed_floatr{ r� r0 r\ r$ r r r ZWIRETYPE_LENGTH_DELIMITEDr* r! r# r^ r% )r r= r� rg r� r r r �testPackedFieldsWireFormat� s, z,SerializationTest.testPackedFieldsWireFormatc C sZ t j� }tj|� t j� }|j� }| jt|�|j|�� t j� }tj |� | j||� d S )N) r �TestUnpackedTypesr �SetAllUnpackedFieldsr� r� r0 r$ r� r� )r �unpacked�packedr� r� r r r �testParsePackedFromUnpacked� s z-SerializationTest.testParsePackedFromUnpackedc C sZ t j� }tj|� t j� }|j� }| jt|�|j|�� t j� }tj |� | j||� d S )N) r r� r r� r r� r0 r$ r� r )r r r r� r� r r r �testParseUnpackedFromPacked� s z-SerializationTest.testParseUnpackedFromPackedc C s� t j� }| jt jjjd� | jt jjd� | jt jjd� | jt jjd� | jt jjd� | jt jj d� | jt jj d� | jt jj d� | jt jj d� d S ) Nr? rn � r� r� r� r� �3 ) r r: r0 rQ ZBB_FIELD_NUMBERZOPTIONAL_INT32_FIELD_NUMBERZOPTIONALGROUP_FIELD_NUMBERZ$OPTIONAL_NESTED_MESSAGE_FIELD_NUMBERZ!OPTIONAL_NESTED_ENUM_FIELD_NUMBERZREPEATED_INT32_FIELD_NUMBERZREPEATEDGROUP_FIELD_NUMBERZ$REPEATED_NESTED_MESSAGE_FIELD_NUMBERZ!REPEATED_NESTED_ENUM_FIELD_NUMBER)r r= r r r �testFieldNumbers s z"SerializationTest.testFieldNumbersc C s8 | j tjjjd� | j tjjd� | j tjjjd� | j tjjd� | j tjjd� | j tj d� | j tj jd� | j tj d� | j tj jd� | j tj d� | j tjjd� | j tjd� | j tjjd� | j tjd� | j tjjd� | j tjd� | j tjjd � | j tjd � | j tjjd � | j tjd � d S ) Ni� i� r? rn r r� r� r� r� r )r0 r r< r= r ZSINGLE_FIELD_NUMBERr} ZMULTI_FIELD_NUMBERr� Z%OPTIONAL_INT32_EXTENSION_FIELD_NUMBERZoptionalgroup_extensionZ$OPTIONALGROUP_EXTENSION_FIELD_NUMBERZ!optional_nested_message_extensionZ.OPTIONAL_NESTED_MESSAGE_EXTENSION_FIELD_NUMBERZ optional_nested_enum_extensionZ+OPTIONAL_NESTED_ENUM_EXTENSION_FIELD_NUMBERr� Z%REPEATED_INT32_EXTENSION_FIELD_NUMBERr9 Z$REPEATEDGROUP_EXTENSION_FIELD_NUMBERr] Z.REPEATED_NESTED_MESSAGE_EXTENSION_FIELD_NUMBERZ repeated_nested_enum_extensionZ+REPEATED_NESTED_ENUM_EXTENSION_FIELD_NUMBER)r r r r �testExtensionFieldNumbers s0 z+SerializationTest.testExtensionFieldNumbersc C sZ t jddddt jjdd�t jdd�t jjt jdddgd � }| j|j� � | j|jd �� | j|jd �� | j|jd �� | j|jd �� | j|jd�� | j|jd�� | j|jd�� | j|jd�� | j d|j � | j d|j � | j d|j � | j d|j � | j d|jj� | j d|jj� | j t jj|j� | j t j|j� | j dddg|j� d S )Nr? rX Ts bar)rI )rK r@ rA ) r7 r6 r� r� rV rf r� �optional_foreign_enumrB r7 r6 r� r� rV rf r� r )r r: rQ rT rR r� r^ ru r< r0 r7 r6 r� r� rV rI rf rK r� r rB )r r= r r r �testInitKwargs+ s: z SerializationTest.testInitKwargsc C s dd� }| j t|d� d S )Nc S s t jdd�} d S )N�unknown)r )r r� )Z unused_protor r r �(InitalizeEmptyMessageWithExtraKeywordArgK s z`SerializationTest.testInitArgsUnknownFieldName.<locals>.InitalizeEmptyMessageWithExtraKeywordArgz9Protocol message TestEmptyMessage has no "unknown" field.)r� r� )r r r r r � testInitArgsUnknownFieldNameJ s z.SerializationTest.testInitArgsUnknownFieldNamec C s� t jdddd�}| j|j� � | j|jd�� | j|jd�� | j|jd�� | j|jd� � | jd|j� | jd|j� | jd|j� d S )Nr? )rM r{ rK rM r{ rK �dummy2) r r< r^ ru r< r0 rM r{ rK )r r= r r r �testInitRequiredKwargsR s z(SerializationTest.testInitRequiredKwargsc C s� t jt jdddd�d�}| j|j� � | j|jd�� | j|jj� � | j|jjd�� | j|jjd�� | j|jjd�� | j|jjd� � | jt jdddd�|j� | jd|jj� | jd|jj � | jd|jj � d S ) Nr? )rM r{ rK )r| r| rM r{ rK r ) r r� r< r^ ru r< r| r0 rM r{ rK )r r= r r r � testInitRequiredForeignKwargs] s z/SerializationTest.testInitRequiredForeignKwargsc C sZ t jdddgd�}| j|j� � | jd|jd � | jd|jd � | jd|jd � d S )Nr? r@ rA )rB r )r r: r^ ru r0 rB )r r= r r r �testInitRepeatedKwargsm s z(SerializationTest.testInitRepeatedKwargsN) r& r' r( r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r r r r r r r r r r r r� D s8 6) 3! r� c @ s e Zd Zdd� Zdd� ZdS )� OptionsTestc C s< t j� }| jd|jj� j� tj� }| jd|jj� j� d S )NTF)r r� r0 r� r� r� r r: )r r= r r r �testMessageOptionsw s z OptionsTest.testMessageOptionsc C s� t j� }d|_d|_x&|j� D ]\}}| jd|j� j� q W t j� }|j j d� |j j d� x8|j� D ],\}}| jd|j� j� | jt j j|j� qfW d S )Nr? g @FT)r r: r7 r8 r� r0 r� r r� r� r{ r� r r r- r )r r= Zfield_descriptor�_r r r �testPackedOptions s z OptionsTest.testPackedOptionsN)r& r' r( r r r r r r r u s r c @ sr e Zd Zejej� dko"ej� dkd�dd� �Zdd� Z e j d�d d � �Z e j d�d d � �Z e j d�d d� �ZdS )� ClassAPITestr r@ z6C++ implementation requires a call to MakeDescriptor()c C s� t jdddd g g g g d�}t jdddd g |gg g d�}t jdddd g g g g d�}t jd d dd g ||gg g d�}tj|�}| jd|j� | jd|j� | jd|jj� d S ) NZleafzpackage.parent.child.leafr� )r r r r r r� zpackage.parent.childZsiblingzpackage.parent.sibling�parentzpackage.parent)r r r � MakeClassr� �__dict__r� )r Z leaf_descZ child_descZ sibling_descZ parent_descZ message_classr r r �!testMakeClassWithNestedDescriptor� s( z.ClassAPITest.testMakeClassWithNestedDescriptorc C s( d| d }t j� }tj||� |j� S )a Get a serialized representation of a test FileDescriptorProto. Args: name: All calls to this must use a unique message name, to avoid collisions in the cpp descriptor pool. Returns: A string containing the serialized form of a test FileDescriptorProto. zmessage_type { name: "ai " field { name: "flat" number: 1 label: LABEL_REPEATED type: TYPE_UINT32 } field { name: "bar" number: 2 label: LABEL_OPTIONAL type: TYPE_MESSAGE type_name: "Bar" } nested_type { name: "Bar" field { name: "baz" number: 3 label: LABEL_OPTIONAL type: TYPE_MESSAGE type_name: "Baz" } nested_type { name: "Baz" enum_type { name: "deep_enum" value { name: "VALUE_A" number: 0 } } field { name: "deep" number: 4 label: LABEL_OPTIONAL type: TYPE_UINT32 } } }})r r� r �Merger� )r r Zfile_descriptor_str�file_descriptorr r r � _GetSerializedFileDescriptor� s ) z)ClassAPITest._GetSerializedFileDescriptorz MakeDescriptor is not repeatablec s� t j� dkrdS tj� }|j| jd�� tj|jd �� G � fdd�dt j t j t j��}|� }d}tj||� | j|jddd g� dS ) z7Test that the generated class can parse a flat message.rw N�Ar c s e Zd Z� ZdS )zSClassAPITest.testParsingFlatClassWithExplicitClassDeclaration.<locals>.MessageClassN)r& r' r( r� r )�msg_descriptorr r � MessageClass� s r& zflat: 0 flat: 1 flat: 2 r? r@ )r ry r r� r� r# r r. r r r r r r r r r! r0 �flat)r r" r& r� �msg_strr )r% r �0testParsingFlatClassWithExplicitClassDeclaration� s z=ClassAPITest.testParsingFlatClassWithExplicitClassDeclarationc C s` t j� }|j| jd�� tj|jd �}tj|�}|� }d}t j ||� | j |j dddg� dS )z7Test that the generated class can parse a flat message.�Br zflat: 0 flat: 1 flat: 2 r? r@ N) r r� r� r# r r. r r r r r! r0 r' )r r" r% � msg_classr� r( r r r �testParsingFlatClass� s z!ClassAPITest.testParsingFlatClassc C s^ t j� }|j| jd�� tj|jd �}tj|�}|� }d}t j ||� | j |j j jd� dS )z9Test that the generated class can parse a nested message.�Cr zbar { baz { deep: 4 }}r N)r r� r� r# r r. r r r r r! r0 r� r� Zdeep)r r" r% r+ r� r( r r r �testParsingNestedClass s z#ClassAPITest.testParsingNestedClassN)r& r' r( r� r� r ry r� r r# r ZSkipReferenceLeakCheckerr) r, r. r r r r r � s 8r �__main__))r) rg rm r5 r r Z unittest2r� � ImportErrorZgoogle.protobufr r r r r r r r r� r r r r r r r r Z BaseTestCase�objectr r/ r� r� r� r� r� r� r r r& �mainr r r r �<module># sv - V! : 5
Save File
Cancel