XXc@s|dZdZddlZddlZddlZddlZddlZddlZddlZyddl Z Wne k rddl Z nXddl m Z ddl mZddl mZddl mZddl mZdd l mZdd l mZdd lmZdd lmZdd lmZddlmZddl mZddlmZejrreZndZdZdZ dZ!ej"Z"ej#defdefde"fdYZ$de"fdYZ%de"fdYZ&de"fdYZ'de"fd YZ(e j)ej*d!kd"d#e"fd$YZ+e,d%krxe j-ndS(&sTests python protocol buffers against the golden message. Note that the golden messages exercise every known field type, thus this test ends up exercising and verifying nearly all of the parsing and serialization code in the whole library. TODO(kenton): Merge with wire_format_test? It doesn't make a whole lot of sense to call this a test of the "message" module, which only declares an abstract interface. s!gps@google.com (Gregory P. Smith)iN(tmap_unittest_pb2(t unittest_pb2(tunittest_proto3_arena_pb2(tdescriptor_pb2(tdescriptor_pool(tmessage_factory(t text_format(tapi_implementation(tpacked_field_test_pb2(t test_util(ttesting_refleaks(tmessage(t_parameterizedcCs ||kS(N((tval((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pytisnanPscCst| ot|dS(Ni(R(R ((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pytisinfSscCst|o|dkS(Ni(R(R ((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pytIsPosInfVscCst|o|dkS(Ni(R(R ((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pytIsNegInfXst_proto2t_proto3t MessageTestc BsIeZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZ dZ!d Z"d!Z#d"Z$d#Z%d$Z&d%Z'd&Z(d'Z)d(Z*d)Z+d?e-d*d+d,d-e.gie/g Z0d.Z1d/Z2d0Z3d1Z4d2Z5d3Z6d4Z7d5e8fd6YZ9d7Z:d8Z;d9Z<d:Z=d;Z>d<Z?d=Z@d>ZARS(@cCsvtjdkr"|jdntjd}|jt}|jj|WdQX|j dt |j dS(NtpythonsSkipping testBadUtf8String, currently only the python api implementation raises UnicodeDecodeError when a string field contains bad utf-8.tbad_utf8_stringsTestAllTypes.optional_string( RtTypetskipTestR tGoldenFileDatat assertRaisestUnicodeDecodeErrort TestAllTypest FromStringtassertIntstrt exception(tselftmessage_modulet bad_utf8_datatcontext((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyttestBadUtf8Stringds cCs|tkrtjd}ntjd}|j}|j||tkretj||n|j||jtj |}|j||jdS(Nt golden_message_oneof_implementedtgolden_message_proto3( RR RRtParseFromStringtExpectAllFieldsSett assertEqualtSerializeToStringtcopytdeepcopy(R!R"t golden_datatgolden_messaget golden_copy((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyttestGoldenMessagens     cCstjd}|j}|j||j}tj||j|||j||jtj|}|j||jdS(Ntgolden_packed_fields_message( R RtTestPackedTypesR(tSetAllPackedFieldsR*R+R,R-(R!R"R.R/tall_setR0((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyttestGoldenPackedMessages    cCsZtjd}|j}|j|tj|}tj|}|j||dS(NR/(R RRR(tpickletdumpstloadsR*(R!R"R.R/tpickled_messagetunpickled_message((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyttestPickleSupports   cCs|tkrd}nd}|j}|j||jt|j|jt|j|jt|jd|jt|jd|j ||j dS(Ns]as ]ai( RRR(t assertTrueRtoptional_floattoptional_doubletrepeated_floattrepeated_doubleR*R+(R!R"R.R/((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyttestPositiveInfinitys    cCs|tkrd}nd}|j}|j||jt|j|jt|j|jt|jd|jt|jd|j ||j dS(Ns]as ]ai( RRR(R=RR>R?R@RAR*R+(R!R"R.R/((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyttestNegativeInfinitys    cCsd}|j}|j||jt|j|jt|j|jt|jd|jt|jd|j}|j}|j||jt|j|jt|j|jt|jd|jt|jddS(Ns]ai( RR(R=RR>R?R@RAR+(R!R"R.R/t serializedR ((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyttestNotANumbers     cCsmd}|j}|j||jt|jd|jt|jd|j||jdS(Nsi(R3R(R=Rt packed_floatt packed_doubleR*R+(R!R"R.R/((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyttestPositiveInfinityPackeds   cCsmd}|j}|j||jt|jd|jt|jd|j||jdS(Nsi(R3R(R=RRFRGR*R+(R!R"R.R/((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyttestNegativeInfinityPackeds   cCsd}|j}|j||jt|jd|jt|jd|j}|j}|j||jt|jd|jt|jddS(Nsi(R3R(R=RRFRGR+(R!R"R.R/RDR ((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyttestNotANumberPackeds     cCs|j}tjdd}||_|j|j|j|j|kdtjdd}||_|j|j|j|j|k| |_|j|j|j|j| k| |_|j|j|j|j| ktjdd}||_|j|j|j|j|kdtjdd}||_|j|j|j|j|k| |_|j|j|j|j| k| |_|j|j|j|j| kdS(Niig?i(RtmathtpowR>R(R+R=(R!R"R tkMostPosExponentNoSigBitstkMostPosExponentOneSigBittkMostNegExponentNoSigBitstkMostNegExponentOneSigBit((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyttestExtremeFloatValuess:         cCs|j}tjdd}||_|j|j|j|j|kdtjdd}||_|j|j|j|j|k| |_|j|j|j|j| k| |_|j|j|j|j| ktjdd}||_|j|j|j|j|kdtjdd}||_|j|j|j|j|k| |_|j|j|j|j| k| |_|j|j|j|j| kdS(Niig?i(RRKRLR?R(R+R=(R!R"R RMRNRORP((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyttestExtremeDoubleValues!s:         cCs/|j}d|_|jt|ddS(Ng@soptional_float: 2.0 (RR>R*R(R!R"R ((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyttestFloatPrintingNs  cCsW|j}d|_tjdkr=|jt|dn|jt|ddS(Ng^!7ݚ?is%optional_double: 0.12345678912345678 s optional_double: 0.123456789123 (i(RR?tsyst version_infoR*R(R!R"R ((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyttestHighPrecisionFloatPrintingSs   cCsR|j}tj||j}|j|j|jt|ddS(Nt(RR tSetAllNonLazyFieldstTestEmptyMessageR(R+R*R(R!R"t populatedtempty((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyttestUnknownFieldPrinting[s    cCs|j}|jjdd|jjdd|jjdd|jjdd|jddddgg|jD]}|j^qw|jddddggt|jD]}|j^q|jddddgg|jdddD]}|j^qdS(Ntbbiiiii(Rtrepeated_nested_messagetaddR*R]treversed(R!R"tmsgtm((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyt testRepeatedNestedFieldIterationbs  &cCs|j}|jjd|jjd|jjd|jj|j|jdd|j|jdd|j|jdd|jjd|jjd|jjd|jj|j|jdd|j|jdd|j|jdd|jjd|jjd |jjd |jj|j|jdd|j|jdd |j|jdd |jjd|jjd |jjd |jj|j|jdd|j|jdd |j|jdd d S( s7Check some different types with the default comparator.iiiig?g?g333333?tatctbN( Rtrepeated_int32tappendtsortR*R@tassertAlmostEqualtrepeated_stringtrepeated_bytes(R!R"R ((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyt0testSortingRepeatedScalarFieldsDefaultComparatorps:     cCs |j}|jjd|jjd|jjd|jjdt|j|jdd|j|jdd|j|jdd|jjd|jjd |jjd |jjdt|j|jdd |j|jdd |j|jddd S( s2Check some different types with custom comparator.iiitkeyiiitaaaR]ReN(RRgRhRitabsR*Rktlen(R!R"R ((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyt/testSortingRepeatedScalarFieldsCustomComparators cCs4|j}d|jj_d|jj_d|jj_d|jj_d|jj_d|jj_|jjdtjd|j|jd jd|j|jdjd|j|jdjd|j|jdjd|j|jdjd|j|jdjdd S( sECheck passing a custom comparator to sort a repeated composite field.iiiiiiRnR]iN(RR^R_R]Ritoperatort attrgetterR*(R!R"R ((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyt2testSortingRepeatedCompositeFieldsCustomComparators cCsC|j}d|jj_d|jj_d|jj_d|jj_d|jj_d|jj_d|jj_|jjdd |jdddddddgg|jD]}|j^q|j}|j|j||jdddddddgg|jD]}|j^q)d S( sECheck passing a custom comparator to sort a repeated composite field.iii i!i ii RncSs |jdS(Ni (R](tz((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pytsN( RR^R_R]RiR*R+tCleartMergeFromString(R!R"R tntpb((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyt(testSortingRepeatedCompositeFieldsStables$     cCs|j}tjd}d}d|jj_d|jj_d|jj_d|jj_d|jj_d|jj_|jjd ||jg|jD]}|j^qddddddg|jjd |d t|jg|jD]}|j^qddddddgt j dkrEd S|jjd ||jg|jD]}|j^qhddddddg|jjd |d t|jg|jD]}|j^qddddddgd S(sECheck sorting a repeated composite field using list.sort() arguments.R]cSst|j|jS(N(tcmpR](RdRf((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyRwsiiiiiiRntreverseNt sort_functionR}(i( RRsRtR^R_R]RiR*tTrueRTRU(R!R"R tget_bbtcmp_bbtk((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyt'testRepeatedCompositeFieldSortArgumentss.  """"cCsf|j}|jjd|jjd|jjd|jjdt|jt|jdddg|jjdtdt|jt|jdddgtj dkr7d}|jjd||jt|jdddg|jjd |dt|jt|jdddgn|j jd |j jd |j jd |j jdt |jt|j d d d g|j jdt dt|jt|j d d d gtj dkrbd }|j jd||jt|j d d d g|j jd |dt|jt|j d d d gndS(s9Check sorting a scalar field using list.sort() arguments.iiiRnR~icSstt|t|S(N(R}Rp(RdRf((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyRwsRR}RoR]RecSstt|t|S(N(R}Rq(RdRf((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyRwsN(i(i( RRgRhRiRpR*tlistRRTRURkRq(R!R"R tabs_cmptlen_cmp((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyt$testRepeatedScalarFieldSortArgumentss6 "" "%"" "cCs|j}|j}|jjd|jjd|jjd|jjd|jjd|jjdd|jj_d|jj_d|jj_d|jj_d|jj_d|jj_tjdkrdS||k}|j|jk}|jt ||d|jt |j|jd|jt |jdddgd|jt |j|jd|j t t |j|jWdQXdS(Niiii(i( RRgRhR^R_R]RTRUR*R}Rt TypeError(R!R"tm1tm2t_((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyttestRepeatedFieldsComparables2   %cCs<|j}|j|jtj|j|jtjdS(N(RtassertIsInstanceRgt collectionstMutableSequenceR^(R!R"Rb((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyttestRepeatedFieldsAreSequences1s  cCs't|||j|j|dS(sMake sure that a nested message object exists. As soon as a nested message attribute is accessed, it will be present in the _fields dict, without being marked as actually being set. N(tgetattrt assertFalsetHasField(R!Rat attribute((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pytensureNestedMessageExists7s cCs&|j}|jt|jddS(Ntno_such_oneof_field(RRt ValueErrort WhichOneof(R!R"Rb((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyt testOneofGetCaseNonexistingField@s cCs|j}|jd|jd|j|jdd|_|jd|jd|j|jd|j|jdd|_ |jd|jd|j|jd|j|jddS(Nt oneof_fieldt oneof_uint32it oneof_stringRW( RtassertIstNoneRRRRR*R=R(R!R"Rb((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyttestOneofDefaultValuesDs   cCs|j}|jd|jdd|_|jd|jd|j|jdd|_|jd|jd|j |jd|j|jd|j |jd|jd|j|jd|j |jd|j j |jd|jd|j|jd|j |jdd|j _ |jd|jd|j |jd|j|jdd|_ |jd|jd|j |jd|j|jddS( NRi RufooRtoneof_nested_messageR]t oneof_bytes( RRRRRR*R=RRRRR]R(R!R"Rb((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyttestOneofSemanticsTs2      cCsU|j}d|_|j|d|jd|jd|jd|jdS(Ni RRR(RRRR*R(R!R"Rb((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyt!testOneofCompositeFieldReadAccessws   cCs|j}|jd|jd|tkrJ|j|jdnd|_|jd|jd|tkr|j |jdnd|_ |jd|jd|j d|jd|jd|tkr|j|jdndS(NRi RR]R( RRRRRRRRR*R=Rt ClearField(R!R"Rb((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyttestOneofWhichOneofs       cCsz|j}d|_|jd|tkrG|j|jdn|j|jd|jd|jddS(Ni RR( RRRRRRRRR(R!R"Rb((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyttestOneofClearFields    cCsz|j}d|_|jd|tkrG|j|jdn|j|jd|jd|jddS(Ni RR( RRRRRRRRR(R!R"Rb((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyttestOneofClearSetFields    cCs|j}d|_|j|d|jd|jd|j|tkrj|j|jdn|j|jd|jd|jddS(Ni RRR( RRRRR*RR=RR(R!R"Rb((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyttestOneofClearUnsetFields    cCsQ|j}d|_|j}|j|j|jd|jddS(Ni RR(RRR(R+R*R(R!R"RbR((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyttestOneofDeserializes    cCsK|j}d|_|j}|j||jd|jddS(Ni RR(RRtCopyFromR*R(R!R"RbR((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyttestOneofCopyFroms     cCs|j}d|j_|j}d|j_d|jj_|j||jd|jjd|jd|jjjddS(Ni R]RRR(tNestedTestAllTypestpayloadRRtchildt MergeFromR*R(R!R"RbR((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyttestOneofNestedMergeFroms     cCs|j}d|jj_d|jjj_|j}d|j_|j||jd|jjd|jd|jjjddS(Ni i i RR( RRRR]RRRR*R(R!R"RbR((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyttestOneofMessageMergeFroms      cCs8|jd|jj}|jd|jddS(NRR(Rt NestedMessageR*R(R!R"Rb((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyttestOneofNestedMessageInits cCs[|j}d|_|j|j|jdd|_|jd|jddS(Ni RR]R(RRRxt assertIsNoneRRR*(R!R"Rb((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyttestOneofClears     cCs5|j}td|_|j|jtjdS(smAssigning a byte string to a string field should result in the value being converted to a Unicode string.RWN(RRtoptional_stringRtsixt text_type(R!R"Rb((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyt"testAssignByteStringToUnicodeFields cCs|j}|jjd|jtdtt|j!}|jt|jt|d|jj_|jtdtt|j!}|jt|jt|dS(s|It should be possible to use long-valued indicies in slices This didn't used to work in the v2 C++ implementation. iiiN( RRgRhtlongRqR*R^R_R](R!R"Rbtsl((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyttestLongValuedSlices %%c Cs|j}|jt'}|jjdtdDWdQX|jt'}|jjdtdDWdQXdS(s5This didn't use to work in the v2 C++ implementation.css|] }tVqdS(N(Rd(t.0ti((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pys si Ncss|] }tVqdS(N(Rd(RR((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pys s(RRt NameErrorRgtextendtrangetrepeated_nested_enum(R!R"RbR((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyt$testExtendShouldNotSwallowExceptionss  & igRWucCs}|j}|jg|jx4tjD])}|jj||jg|jq)W|jjg|jg|jdS(s,Test no-ops extending repeated int32 fields.N(RtassertSequenceEqualRgRt FALSY_VALUESR(R!R"Rbt falsy_value((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyttestExtendInt32WithNothings cCs}|j}|jg|jx4tjD])}|jj||jg|jq)W|jjg|jg|jdS(s,Test no-ops extending repeated float fields.N(RRR@RRR(R!R"RbR((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyttestExtendFloatWithNothings cCs}|j}|jg|jx4tjD])}|jj||jg|jq)W|jjg|jg|jdS(s-Test no-ops extending repeated string fields.N(RRRkRRR(R!R"RbR((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyttestExtendStringWithNothings cCs|j}|jg|j|jjdg|jdg|j|jjddg|jdddg|j|jjddg|jdddddg|jdS(s7Test extending repeated int32 fields with python lists.iiiiiN(RRRgR(R!R"Rb((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyttestExtendInt32WithPythonList*s cCs|j}|jg|j|jjdg|jdg|j|jjddg|jdddg|j|jjddg|jdddddg|jdS(s7Test extending repeated float fields with python lists.gg?g@g@g@N(RRR@R(R!R"Rb((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyttestExtendFloatWithPythonList5s cCs|j}|jg|j|jjdg|jdg|j|jjddg|jdddg|j|jjddg|jdddddg|jdS(s8Test extending repeated string fields with python lists.RWt11t22t33t44N(RRRkR(R!R"Rb((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyttestExtendStringWithPythonList@s cCsO|j}|jg|j|jjd|jdddg|jdS(sDTest extending repeated string fields with characters from a string.tabcRdRfReN(RRRkR(R!R"Rb((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyttestExtendStringWithStringKs t TestIterablecBs5eZdZddZdZdZdZRS(sThis iterable object mimics the behavior of numpy.array. __nonzero__ fails for length > 1, and returns bool(item[0]) for length == 1. cCs|p g|_dS(N(t_list(R!tvalues((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyt__init__YscCsLt|j}|dkrtS|dkr<t|jdStddS(NiisTruth value is ambiguous.(RqRtFalsetboolR(R!tsize((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyt __nonzero__\s   cCs t|jS(N(RqR(R!((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyt__len__dscCs |jjS(N(Rt__iter__(R!((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyRgsN(t__name__t __module__t__doc__RRRRR(((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyRRs    cCs|j}|jg|j|jjtjg|jg|j|jjtjdg|jdg|j|jjtjddg|jdddg|j|jjtjddg|jdddddg|jdS(s3Test extending repeated int32 fields with iterable.iiiiiN(RRRgRRR(R!R"Rb((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyttestExtendInt32WithIterablejs cCs|j}|jg|j|jjtjg|jg|j|jjtjdg|jdg|j|jjtjddg|jdddg|j|jjtjddg|jdddddg|jdS(s3Test extending repeated float fields with iterable.gg?g@g@g@N(RRR@RRR(R!R"Rb((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyttestExtendFloatWithIterablews cCs|j}|jg|j|jjtjg|jg|j|jjtjdg|jdg|j|jjtjddg|jdddg|j|jjtjddg|jdddddg|jdS(s4Test extending repeated string fields with iterable.RWt1t2t3t4N(RRRkRRR(R!R"Rb((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyttestExtendStringWithIterables cCsitjdks$tjdkr(dS|j}|jtj}tj|jtj WdQXdS(Ntcppi( RRtVersionRRR7t PickleErrorR8RgtHIGHEST_PROTOCOL(R!R"RbR((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyt!testPickleRepeatedScalarContainers  cCs|j}|jjdS(s?Exercise a scenario that has led to segfaults in the past. N(RR^Ri(R!R"Rb((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyt'testSortEmptyRepeatedCompositeContainers cCs5|j}|jt}|jdWdQXdS(sBUsing HasField on a repeated field should raise an exception. RgN(RRRR(R!R"RbR((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyttestHasFieldOnRepeatedFields cCs|j}|jt}|jjWdQX|jjtd|jd|jj|jd|jjd|jd|jjd|jddg|jdS(Niiiiii(RRt IndexErrorRgtpopRRR*(R!R"RbR((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyttestRepeatedScalarFieldPops cCs|j}|jt}|jjWdQXx,tdD]}|jj}||_q>W|jd|jjj|jd|jjdj|jd|jjdj|jddgg|jD]}|j^qdS(Niiiiii( RRRR^RRR_R]R*(R!R"RbRRRz((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyttestRepeatedCompositeFieldPops  N(BRRR%R1R6R<RBRCRERHRIRJRQRRRSRVR\RcRmRrRuR|RRRRRRRRRRRRRRRRRRRRRRRRt bytearraytsetRRRRRRRRtobjectRRRRRRRRR(((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyR_sv      - -     %     #    #         *     t Proto2TestcBsGeZdZdZdZdZdZdZdZRS(cCs\tj}|j|jd|j|jd|j|jd|jt|jdWdQX|jt|jdWdQX|jt|jdWdQX|jd|j|jt|j |jd|j j d|_t|_ d|j _ |j |jd|j |jd|j |jdd|_t |_ d |j _ |j |jd|j |jd|j |jd|jd|jd|jd|j|jd|j|jd|j|jd|jd|j|jt|j |jd|j j dS( Ntoptional_int32t optional_booltoptional_nested_messagetfield_doesnt_existRgR^iii(RRRRRRR*RRRRR]R=RR(R!R ((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyttestFieldPresencesD          cCsJtj}|jt}d|_WdQX|jt|jjddS(sPIt should not be possible to assign an invalid enum number to an enum field.iN(RRRRtoptional_nested_enumRRh(R!RbR((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyttestAssignInvalidEnums cCstjd}tj}|j|tj}tj||j|||j||jtj |}|j||jdS(NR/( R RRtTestAllExtensionsR(tSetAllExtensionsR*R+R,R-(R!R.R/R5R0((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyttestGoldenExtensionss    cCstjd}tj}|j|tj}tj||j|||j||jtj |}|j||jdS(NR2( R RRtTestPackedExtensionsR(tSetAllPackedExtensionsR*R+R,R-(R!R.R/R5R0((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyttestGoldenPackedExtensionss    cCsmtjdd}tj|}tj|}|j|||j|jd|jtj |j dS(NRdi( Rt TestRequiredR7R8R9R*RdRR t EncodeErrorR+(R!R/R:R;((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyttestPickleIncompleteProtos cCs`tjtjtjg}d|d_d|d_d|d_d|d_tj}d|_d|_d|_tjj}|jj||j j||j j||j j||j j||j jjj|d|j jjj|d|j jjj|d|jjjj|d|jjjj|d|jjjj|d|j}tj}|j||j|j||j|j||j|jj||j|jtjj||jt|jd|jt|jd|jt|jtjjddS(scCheck the merge behavior when a required or optional field appears multiple times in the input.iiiithelloN(RRRtoptional_int64RtTestParsingMergetRepeatedFieldsGeneratortfield1Rtfield2tfield3text1text2tgroup1R_Rtgroup2R+R(R*trequired_all_typestoptional_all_typest optionalgrouptoptional_group_all_typest Extensionst optional_extRqtrepeated_all_typest repeatedgroupt repeated_ext(R!tmessagestmerged_messaget generatortdatat parsing_merge((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyttestParsingMerge'sJ               cCsKtjddddddddd id d 6d id d6didddidd 6idd 6gddtjjgdddd }|j|tj|jd|j|jd|j|jd|j|jd|j|jd |j j |j|j tjj |jd |j j |j|jd|j|jtj|jtjj|j|jdt|j|jd|jdj |jd|jdj |jdt|j|jtjj|jd|jtjj|jd|jd|j|jd|j|j|jd|jdt|j|jd|jtjdd }|jtjj|j|jttjd id!d"6WdQX|jttjd id#d6WdQX|jttjdd$WdQX|jttjddWdQXdS(%NRidtoptional_fixed32iR>gr@toptional_bytestxRiRdRiR]toptional_foreign_messageRtBAZRiXiRtFOOt default_int32i RtyiiiRi*uBAZitINVALID_NESTED_FIELDtINVALID_VALUE_TYPEt INVALID_LABEL( RRtBARRR*RR!R>R"RRdRRR]R=RR$tForeignMessageR%RRqRRR&R'RRR@t default_int64RRR(R!R ((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyttestPythonicInitXsl             ( RRRRRRRR R/(((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyRs 4  1t Proto3TestcBseZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdZdZdZdZdZdZdZdZRS(cCsRt|}x/|D]'\}}|j|||||=qW|ji|dS(N(tdictR*(R!tmap_itert dict_valueRtv((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pytassertMapIterEqualss   c Csrtj}|jt|jdWdQX|jt|jdWdQX|jt|jdWdQX|jt|jdWdQX|j|jd|jt|jdWdQX|jt|jdWdQX|jt|jdWdQX|jd |j|jd |j|jd |j |jt |j |jd |j j d |j _ |j|jdd |_d |_d |_ t|_ d|j _ |jd|jd|jd|jd|jd|jd |j|jd |j|jd |j |jt |j |jd |j j dS(NRR>RRRRRgR^iRWig?Ri(RRRRRRR*RR>RRRRR]R=RR(R!R ((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyRsL            cCstj}d|_|jd|j|jjd|jd|jdd|jd<|jd|jd|j}tj}|j||jd|j|jd|jddS(sCAssigning an unknown enum value is allowed and preserves the value.iiTiitN(RRRR*RRhR+R((R!RbRDR((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyttestAssignUnknownEnums      cCstj}|jd|jk|jd |jk|jd|jk|jd |jk|jd|jk|jt|j k|jd|j k|jd|j k|jd|j k|j d|jd|j d|jd|j d|jd|j d|jd|j d |jd|jt|jdt|j t|j t|jt|j tt|j d |j d|j d |j d|j d|j d|jd|jk|jd|jk|jd|jk|jd|jk|jd|jk|jt|j k|jd|j k|jd|j k|jd|j k|j|j dtj|jt|j dWdQX|jtd|j kWdQXdS(Niii!i{RioixigRWIIIIIIIII(RtTestMapRtmap_int32_int32tmap_int64_int64tmap_uint32_uint32tmap_uint64_uint64tmap_int32_doubleRt map_bool_booltmap_string_stringtmap_int32_bytestmap_int32_enumR*R=t isinstancetfloatRRRRRR(R!Ra((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyttestScalarMapDefaultssF cCstj}|j|jjd|jd|jjdd|j|jjdd|jd<|jd|jjd|j|jjd|jd|jjdd|jd}|j||jjddS(Nii i(RR7RR8tgetR*tmap_int32_foreign_messageR(R!Ratsubmsg((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyt testMapGet%s   c Cstj}|jdt|j|jd|jkd|jdR@tFindInitializationErrorsRRR+R((R!RaRDtmsg2((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyt testScalarMap7s:          cCstj}d}|jd}||j|RtitemsR*RRR(R!Rat unicode_objt bytes_objRntvalue((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyt testStringUnicodeConversionInMapgs  cCstj}|jdt|j|jd|jk|jd|jjd|jdt|j|jd|j|jd|j|jdt|j|jt |jdWdQX|jt |jd|jdREReRt RuntimeError(R!Ratstring_string_itertint32_foreign_iterRn((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyttestModifyMapWhileIteratings     cCstj}|jd}|j||jd|j|tjd|_|j}tj}|j ||j d|jdj|j t tj|jdRNR*(R!Ratitems1titems2((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyt testMapItemsNs      cCsttj}d|jdR8tmap_string_foreign_messageReR*RqRI(R!Ra((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyt(testMapFindInitializationErrorsSmokeTests    (RRR5RR6RCRGRKRRRTRXRZR[R\R`RaRfRgRiRqRsRtRvRyRzR|RR(((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyR0s4 9  0  0  $ .              tValidTypeNamesTestcBseZdZdZRS(cstt|jddd|d|f}|jtfd|Dd|fjd}|d}dj|d }t|d |gdS( Nt'isRepeated%sContainersRepeated%sFieldContainerc3s|]}j|VqdS(N(tendswith(RR4(ttp_name(sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pys ss%r does end with any of %rt.itfromlist(RttypetsplitR=tanytjoint __import__(R!Rat base_namet valid_namestpartst class_namet module_name((RsG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pytassertImportFromNames  cCs6tj}|j|jd|j|jddS(NtScalart Composite(RRRRgR^(R!R{((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyttestTypeNamesCanBeImporteds (RRRR(((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyRs tPackedFieldTestcBs#eZdZdZdZRS(cCs|jjd|jjd|jjd|jjd|jjd|jjd|jjd|jjd|j jd|j jd|j jd|j jd|j jt|jjddS(Nig?(RgRhtrepeated_int64trepeated_uint32trepeated_uint64trepeated_sint32trepeated_sint64trepeated_fixed32trepeated_fixed64trepeated_sfixed32trepeated_sfixed64R@RAt repeated_boolRR(R!R ((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyt setMessagescCs9tj}|j|d}|j||jdS(NsH "*2:BJRZ?b?jr(RR3RR*R+(R!R R.((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyttestPackedFieldss  cCs9tj}|j|d}|j||jdS(Ns: (0=AMQ]?a?hp(RtTestUnpackedTypesRR*R+(R!R R.((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyttestUnpackedFieldss  (RRRRR(((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyRs  Rs(explicit tests of the C++ implementationtOversizeProtosTestcBs2eZedZdZdZdZRS(cCscd}tj}tj}tj|||j|tj|j |j d|_ dS(Ns name: "f/f.msg2" package: "f" message_type { name: "msg1" field { name: "payload" number: 1 label: LABEL_OPTIONAL type: TYPE_STRING } } message_type { name: "msg2" field { name: "field" number: 1 label: LABEL_OPTIONAL type: TYPE_MESSAGE type_name: "msg1" } } sf.msg2( RtDescriptorPoolRtFileDescriptorProtoRtParsetAddRtMessageFactoryt GetPrototypetFindMessageTypeByNamet proto_cls(tclst file_desctpooltdesc((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyt setUpClasss   cCs8|j|_dd|jj_|jj|_dS(NReii@iiii(RtptfieldRR+t p_serialized(R!((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pytsetUp9scCsmddlm}|t|j}y|j|jWn,tjk rh}|jt |dnXdS(Ni(tSetAllowOversizeProtossError parsing message( tgoogle.protobuf.pyext._messageRRRR(RR t DecodeErrorR*R(R!Rtqte((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyttestAssertOversizeProto>s  cCsYddlm}|t|j}|j|j|j|jjj |jj dS(Ni(R( RRRRR(RR*RRR(R!RR((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyttestSucceedOversizeProtoGs   (RRt classmethodRRRR(((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyRs#  t__main__(.Rt __author__RR,RKRsR7RRTt unittest2tunittestt ImportErrortgoogle.protobufRRRRRRRtgoogle.protobuf.internalRRR R R R tPY3tintRRRRRt BaseTestCasetNamedParametersRRR0RRtskipIfRRRtmain(((sG/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/message_test.pyt*sd                e.9 :