3
SÉLj6 ã @ sü d Z dZy ddlZW n ek
r0 ddlZY nX ddlmZ ddlmZ ddlmZ ddl m
Z
ddl m Z dd l m Z dd
l m
Z
dd l mZ dd l mZ dd
l mZ ejZdd„ ZG dd„ deƒZG dd„ deƒZG dd„ deƒZedkrøejƒ dS )zJTest for preservation of unknown fields in the pure Python implementation.z!bohdank@google.com (Bohdan Koval)é N)Úunittest_mset_pb2)Ú unittest_pb2)Úunittest_proto3_arena_pb2)Úapi_implementation)Úencoder)Úmessage_set_extensions_pb2)Úmissing_enum_values_pb2)Ú test_util)Útesting_refleaks)Ú
type_checkersc C s$ t jtjƒ dkotjƒ dkdƒ| ƒS )NZcppé z7Addtional test for pure python involved protect members)ÚunittestZskipIfr ZTypeZVersion)Úfunc© r ú)/usr/lib/python3.6/unknown_fields_test.pyÚ(SkipCheckUnknownFieldIfCppImplementation; s r c @ sT e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd d
„ Zd d „ Zd
d„ Z dd„ Z
dd„ Z dS )ÚUnknownFieldsTestc C sH t jj| _t jƒ | _tj| jƒ | jjƒ | _t j ƒ | _
| j
j | jƒ d S )N) r Ú TestAllTypesÚ
DESCRIPTORÚ
descriptorÚ
all_fieldsr Ú SetAllFieldsÚSerializeToStringÚall_fields_dataÚTestEmptyMessageÚ
empty_messageÚParseFromString)Úselfr r r ÚsetUpC s
zUnknownFieldsTest.setUpc C s | j jƒ }| j|| jkƒ d S )N)r r Z
assertTruer )r Údatar r r Ú
testSerializeK s
z UnknownFieldsTest.testSerializec C sD t jƒ }|j| jƒ |r,| j| j|jƒ ƒ n| jdt|jƒ ƒƒ d S )Nr )r r r r Ú assertEqualr Úlen)r ZpreserveÚmessager r r ÚexpectSerializeProto3R s
z'UnknownFieldsTest.expectSerializeProto3c C s8 t jƒ }| j|ƒ t j| ƒ | j| ƒ t j|ƒ d S )N)r Z&GetPythonProto3PreserveUnknownsDefaultr$ Z&SetPythonProto3PreserveUnknownsDefault)r Zdefault_preserver r r ÚtestSerializeProto3Z s
z%UnknownFieldsTest.testSerializeProto3c C s | j | jjƒ | jjƒ ƒ d S )N)r! r ZByteSizer )r r r r Ú testByteSized s z UnknownFieldsTest.testByteSizec C s | j dt| jjƒ ƒƒ d S )Nr )r! r" r Z
ListFields)r r r r ÚtestListFieldsg s z UnknownFieldsTest.testListFieldsc C st t jƒ }|jjƒ }d|_tjƒ }d|_|jƒ |_ |jƒ }tj
ƒ }|j |ƒ |jƒ }t jƒ }|j |ƒ | j ||ƒ d S )Ni«¿Ýi90 )
r Z
RawMessageSetÚitemÚaddZtype_idr ZTestMessageSetExtension1Úir r# ZTestMessageSetZMergeFromStringr! )r Úrawr( Zmessage1Ú
serializedÚprotoZ reserializedZnew_rawr r r Ú1testSerializeMessageSetWireFormatUnknownExtensionk s
zCUnknownFieldsTest.testSerializeMessageSetWireFormatUnknownExtensionc C sP t jƒ }|j| jƒ | j| j|ƒ | jjdƒ |j| jjƒ ƒ | j | j|ƒ d S )NÚoptional_string)
r r r r r! r r Ú
ClearFieldr ÚassertNotEqual)r r# r r r Ú
testEqualsƒ s z UnknownFieldsTest.testEqualsc C s² | j jƒ | jd| j jƒ ƒ tjƒ }tjƒ }d|_|jj|jƒ ƒ |j j
ƒ j|jƒ ƒ | j d|jjƒ ƒ | j d|j d jƒ ƒ |jƒ | jd|jjƒ ƒ | jd|j d jƒ ƒ d S )Nó Údiscardr ) r ZDiscardUnknownFieldsr! r r r r/ Zoptional_nested_messager Zrepeated_nested_messager) r1 )r r# Z
other_messager r r ÚtestDiscardUnknownFieldsŒ s"
z*UnknownFieldsTest.testDiscardUnknownFieldsN) Ú__name__Ú
__module__Ú __qualname__r r r$ r% r& r' r. r2 r5 r r r r r A s
r c @ sH e Zd Zdd„ Zdd„ Zedd„ ƒZdd„ Zd d
„ Zd d „ Z d
d„ Z
dS )ÚUnknownFieldsAccessorsTestc C sH t jj| _t jƒ | _tj| jƒ | jjƒ | _t j ƒ | _
| j
j | jƒ d S )N) r r r r r r r r r r r r )r r r r r £ s
z UnknownFieldsAccessorsTest.setUpc
C s‚ | j j| }tj|j }tj|j|ƒ}i }xB| jj D ]6\}}||kr4t
j j | d } | |dt
|ƒ| j|ƒ q4W | j||| ƒ d S )Nr )r Úfields_by_namer ÚFIELD_TYPE_TO_WIRE_TYPEÚtyper ÚTagBytesÚnumberr Ú_unknown_fieldsr r Ú_decoders_by_tagr" r r! )
r ÚnameÚexpected_valueÚfield_descriptorÚ wire_typeÚ field_tagÚ result_dictÚ tag_bytesÚvalueÚdecoderr r r ÚCheckUnknownField± s z,UnknownFieldsAccessorsTest.CheckUnknownFieldc C st | j d| jjƒ | j d| jjƒ | j d| jjƒ | j d| jjƒ | j d| jjƒ | j d| jjƒ | j d| jjƒ d S )NÚoptional_nested_enumÚrepeated_nested_enumÚoptional_int32Úoptional_fixed32Úoptional_fixed64r/ Ú
optionalgroup) rJ r rK rL rM rN rO r/ rP )r r r r ÚtestCheckUnknownFieldValue¼ s
z5UnknownFieldsAccessorsTest.testCheckUnknownFieldValuec C s* t jƒ }|j| jƒ | j|jƒ | jƒ d S )N)r r ZCopyFromr r! r r )r r# r r r Ú testCopyFromØ s z'UnknownFieldsAccessorsTest.testCopyFromc C sœ t jƒ }d|_d|_t jƒ }|j|jƒ ƒ |jdƒ d|_d|_t jƒ }|j|jƒ ƒ |j |ƒ |j|jƒ ƒ | j
|jdƒ | j
|jdƒ | j
|jdƒ d S )Né r rM é é ) r r rM Zoptional_uint32r r r r0 Zoptional_int64Z MergeFromr! )r r# ÚsourceZ destinationr r r Ú
testMergeFromÝ s
z(UnknownFieldsAccessorsTest.testMergeFromc C s | j jƒ | j| j jƒ dƒ d S )Nr3 )r ZClearr! r )r r r r Ú testClearò s
z$UnknownFieldsAccessorsTest.testClearc C s* t jƒ }|j| jƒ | j|jƒ | jƒ d S )N)r Z TestEmptyMessageWithExtensionsr r r! r )r r# r r r ÚtestUnknownExtensions÷ s z0UnknownFieldsAccessorsTest.testUnknownExtensionsN) r6 r7 r8 r rJ r rQ rR rW rX rY r r r r r9 ¡ s r9 c @ sP e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd d
„ Zd d „ Ze d
d„ ƒZ
dd„ Z dS )ÚUnknownEnumValuesTestc C s| t jj| _t jƒ | _t jj| j_| jjjt jjt jj gƒ | jj
jt jjt jj gƒ | jj ƒ | _ t j
ƒ | _| jj| j ƒ d S )N)r ÚTestEnumValuesr r r# ZZEROrK rL ÚextendZONEÚpacked_nested_enumr Z message_dataZTestMissingEnumValuesÚmissing_messager )r r r r r ÿ s
zUnknownEnumValuesTest.setUpc
C s‚ | j j| }tj|j }tj|j|ƒ}i }xB| jj D ]6\}}||kr4t
j j | d } | |dt
|ƒ| j|ƒ q4W | j||| ƒ d S )Nr )r r: r r; r< r r= r> r^ r? r r[ r@ r" r# r! )
r rA rB rC rD rE rF rG rH rI r r r rJ s
z'UnknownEnumValuesTest.CheckUnknownFieldc C s6 t jƒ }d|_t jƒ }|j|jƒ ƒ | j|jdƒ d S )NZblahr )r Z
JustStringZdummyr[ r r r! rK )r Z just_stringZmissingr r r Ú!testUnknownParseMismatchEnumValue$ s
z7UnknownEnumValuesTest.testUnknownParseMismatchEnumValuec C sN | j | jjdƒƒ | j| jjdƒ | jjƒ }| jjdƒ | j| jjƒ |ƒ d S )NrK r )Z assertFalser^ ZHasFieldr! rK r r0 )r r, r r r ÚtestUnknownEnumValue1 s
z*UnknownEnumValuesTest.testUnknownEnumValuec C s | j g | jjƒ d S )N)r! r^ rL )r r r r Ú testUnknownRepeatedEnumValue9 s z2UnknownEnumValuesTest.testUnknownRepeatedEnumValuec C s | j g | jjƒ d S )N)r! r^ r] )r r r r ÚtestUnknownPackedEnumValue< s z0UnknownEnumValuesTest.testUnknownPackedEnumValuec C s4 | j d| jjƒ | j d| jjƒ | j d| jjƒ d S )NrK rL r] )rJ r# rK rL r] )r r r r Ú!testCheckUnknownFieldValueForEnum? s
z7UnknownEnumValuesTest.testCheckUnknownFieldValueForEnumc C s* t jƒ }|j| jjƒ ƒ | j| j|ƒ d S )N)r r[ r r^ r r! r# )r Z new_messager r r Ú
testRoundTripH s z#UnknownEnumValuesTest.testRoundTripN) r6 r7 r8 r rJ r_ r` ra rb r rc rd r r r r rZ ý s
rZ Ú__main__)Ú__doc__Ú
__author__Z unittest2r
Ú ImportErrorZgoogle.protobufr r r Zgoogle.protobuf.internalr r r r r r
r Z BaseTestCaser r r9 rZ r6 Úmainr r r r Ú! s, `\Q