__ __ __ __ _____ _ _ _____ _ _ _ | \/ | \ \ / / | __ \ (_) | | / ____| | | | | | \ / |_ __\ V / | |__) | __ ___ ____ _| |_ ___ | (___ | |__ ___| | | | |\/| | '__|> < | ___/ '__| \ \ / / _` | __/ _ \ \___ \| '_ \ / _ \ | | | | | | |_ / . \ | | | | | |\ V / (_| | || __/ ____) | | | | __/ | | |_| |_|_(_)_/ \_\ |_| |_| |_| \_/ \__,_|\__\___| |_____/|_| |_|\___V 2.1 if you need WebShell for Seo everyday contact me on Telegram Telegram Address : @jackleetFor_More_Tools:
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