3
SLj @ 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 deZ G dd deZ G dd deZ G dd d eZ!G d d d eZ"G d d d eZ#G d!d" d"eZ$G d#d$ d$eZ%G d%d& d&eZ&e'd'krej( 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)selfbytes 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 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 Zdydz Z?d{d| Z@d}d~ ZAdd ZBdd ZCdd ZDdd ZEdd ZFdd ZGdd ZHdd ZIdd ZJdd ZKdd ZLdd ZMdd ZNdd ZOdd ZPdd ZQdddZRdd ZSeTjUeVjW dkpeVjX dkddd ZYdd ZZdd Z[dd Z\dd Z]dd Z^dd Z_dS )ReflectionTestc C sB | j t|t| x(tt|D ]}| j || || q"W d S )N) assertEqualr$ range)r valuesZothersir 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_int32optional_doubler6 optional_floatr9 )r TestAllTypesr0 r7 r8 r6 assertFalseHasField)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 gGz?g +)K@TFr6 )repeated_int32repeated_double
repeated_boolrepeated_stringrepeated_float) r r: r0 listrB rC rD rE rF )r r= r r r testRepeatedScalarConstructor s
z,ReflectionTest.testRepeatedScalarConstructorc C s t jt jjt jjdt jjt jjdgt jd
dt jddt jddgt jj t jjddt jjddgd }| jt jjt jjdt jjt jjdgt|j | jt jd dt jddt jddgt|j
| jt jj t jjddt jjddgt|j d S ) N)bb+ )ci r? )ar@ )repeated_nested_messagerepeated_foreign_message
repeatedgroupii) r r:
NestedMessageFOOBARForeignMessageZ
RepeatedGroupr0 rG rN rO rP )r r= r r r testRepeatedCompositeConstructor s6
z/ReflectionTest.testRepeatedCompositeConstructorc C s t jddddgdddgt jjt jjdt jjt jjdgt jdd t jd
d t jd d gd d }| jd|j | jd|j | jddgt |j
| jdddgt |j | jt jjt jjdt jjt jjdgt |j | jt jdd t jd
d t jd d gt |j
| j|jd
d S )Nr5 r6 gGz?g +)K@TF)rI rJ )rK i rL )r7 r6 rC rD rN rO optional_nested_messagerV ii)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 ) Nfoo)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..TestCompositeHasBits
optionalgrouprM rV rI optional_foreign_messagerK Zoptional_import_messagedr )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_ gccollect)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 proto1proto2r r r (testDisconnectingNestedMessageAfterMerge s
z7ReflectionTest.testDisconnectingNestedMessageAfterMergec C s8 t j dkrd S tj }d|j_|jd ~tj d S )Npythonrp 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_int32appendr0 r^ r< r_ Z%nestedmessage_repeated_foreignmessageadd)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
DESCRIPTORfields_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 barbazc s s | ]}t |V qd S )N)str).0xr r r s z8ReflectionTest.testRepeatedListFields..r@ r7 rB repeated_fixed32rE 01)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_extensionoptional_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_extensionrepeated_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 ሴii)*r r: r0 r7 optional_int64optional_uint32optional_uint64optional_sint32optional_sint64r optional_fixed64optional_sfixed32optional_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 )Nnonexistent_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 xtdD ]} jj qW t j } j| | jt fdd d S )N
c s
j d S )Nrp )repeated_messager )r= r r Q s z9ReflectionTest.testClearRemovesChildren..)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 tjddkrp|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 ParseFromStringSerializeToString)
field_namevalueZ
expected_typer= ru )
integer_fnr r r TestGetAndDeserialize} s z@ReflectionTest.assertIntegerTypes..TestGetAndDeserializer7 r? r Lr r < r Ni @i @l l i @l i @l )intZlong 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..)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 )N2Zmy_errorrp ) r r: rY rZ r ZassertRaisesRegexp 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 )zVerifies 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..TestMinAndMaxIntegersr7 r? r r r l r ? r l Nl i l il 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..)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 abcindexr )
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 ForeignEnumNamer 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 dD |jdd < | jddddd g|j dddg|jdd < | jddddd g|j g }x|jD ]}|j| qW | 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..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| qW | 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= m0m1m2Zm3Zm4r 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 numbertypeZcpp_typelabelZ
default_valuecontaining_type message_typeZ enum_typeZ is_extensionZextension_scopeoptionsZMyProtoZignored) r r
filenamer nested_types
enum_typesfields
extensionsr c s e Zd Z ZdS )z>ReflectionTest.testHandWrittenReflection..MyProtoClassN)r& r' r( r r ) mydescriptorr r MyProtoClassH s r ri )r
ry r FieldDescriptor
TYPE_INT64Z
CPPTYPE_INT64LABEL_OPTIONALr Z FieldOptions
DescriptorZMessageOptionssixwith_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|| jdjk | jdjk | jdjk | 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_indexfieldr| r r r r FieldDescriptorProtor r
)r= r Z
field_type new_field)AddDescriptorFieldr r r% W s
zEReflectionTest.testDescriptorProtoSupport..AddDescriptorFieldr ZCarr year automaticpricer? ownersc s e Zd Z ZdS )z=ReflectionTest.testDescriptorProtoSupport..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 operatorsetitem)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 TestRequiredsingler^ r0 r r0 rM r1 )r r2 r3 Zrequiredr r r testNestedExtensions s
z#ReflectionTest.testNestedExtensionsc C sT t jj}| j|jt jjd | j|jdjt 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_extensionrepeated_int_extensionr{ optional_message_extensionforeign_message_intr$ repeated_message_extensionr| )r toplevelforeignr 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 message1message2r 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 copydeepcopyr0 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 ZSetAllNonLazyFieldsClearr0 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_sizeerrorsr 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.testIsInitializedcppr@ 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 |ddt
|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 expectedassertInr 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|jtj tj d|_| j|jt
d t
d|_| j|jtj d | jt t|dd dj d}||_d|_d|_d S ) Nr zunicode objectZTestingr6 s aau Тест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| jtj | 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
TestMessageSetTestMessageSetExtension2message_set_extensionr r r r r0 r\ r$ r
RawMessageSetMergeFromStringitemtype_idr^ r endswithr r r replaceUnicodeDecodeErrorr )
r r= Zextension_messager3 Z test_utf8Ztest_utf8_bytes
serializedraw
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_package9 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~ unittestskipIfr
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 ZExtendedMessageextended_protoTestPackedTypes packed_protoTestPackedExtensionspacked_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 sz fdd}|dd |dd x6t tdddtddD ]\}}|d|> d | q:W |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..Testr r? rp r i' r r@ r r l l )zipr1 )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 gX9v?@ 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_0foreign_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
dkr| 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 }xtt|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.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 jddddd}| 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 dkddd Zdd Z e
j dd d
Z e
j dd d Z
e
j dd
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.siblingparentzpackage.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_strfile_descriptorr r r _GetSerializedFileDescriptor s ) z)ClassAPITest._GetSerializedFileDescriptorz MakeDescriptor is not repeatablec s t j dkrdS tj }|j| jd tj|jd G fdddt j
t j t
j}| }d}tj|| | j|jddd g dS )
z7Test that the generated class can parse a flat message.rw NAr c s e Zd Z ZdS )zSClassAPITest.testParsingFlatClassWithExplicitClassDeclaration..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 BaseTestCaseobjectr r/ r r r r r r r r r& mainr r r r # sv - V! : 5