3 S╔LjЮ9у@sцdZdZyddlZWnek r0ddlZYnXddlmZddlmZddlm Z ddlm Z ddlmZdd lmZdd lm Z ddlmZddlmZdd lmZdZGddДdejГZGddДdejГZedkrтejГdS)zрUnittest that directly tests the output of the pure-Python protocol compiler. See //google/protobuf/internal/reflection_test.py for a test which further ensures that we can use Python protocol message objects as we expect. z#robinson@google.com (Will Robinson)щN)┌test_bad_identifiers_pb2)┌unittest_custom_options_pb2)┌unittest_import_pb2)┌unittest_import_public_pb2)┌unittest_mset_pb2)┌unittest_mset_wire_format_pb2)┌ unittest_no_generic_services_pb2)┌unittest_pb2)┌service)┌symbol_databasei c@s╝eZdZddДZddДZddДZddДZd d ДZddДZd dДZ ddДZ ddДZddДZddДZ ddДZddДZddДZddДZdd ДZd!d"ДZd#d$ДZd%d&ДZd'd(ДZd)d*ДZd+d,ДZd-S).┌ GeneratorTestcCs(d}tj}|j|jj|jj|jГdS)N┌optional_nested_message)r ┌TestAllTypes┌assertEqual┌ NestedMessage┌ DESCRIPTOR┌fields_by_nameZmessage_type)┌self┌ field_nameZ proto_typeйr·$/usr/lib/python3.6/generator_test.py┌testNestedMessageDescriptor@s z)GeneratorTest.testNestedMessageDescriptorcCsР|jdtjГ|jdtjГ|jdtjГtjГ}|jd|jГ|jdtjjГ|jd|jГ|jdtjjГ|jd|jГ|jdtjjГdS)Nщщщщщщ) rr ZFOREIGN_FOOZFOREIGN_BARZFOREIGN_BAZr┌FOOZBARZBAZ)r┌protorrr┌ testEnumsGszGeneratorTest.testEnumscs╬tjГ}ddДЙЗfddД}|j||jГГ|j|jdkГ|j||jГГ|j|jdkГ|jИ|jГГ|j||jГГ|j|jdkГ|j||jГГ|j|jdkГ|jИ|jГГ|j d|j ГdS)NcSs||kS)Nr)┌valrrr┌isnan\sz5GeneratorTest.testExtremeDefaultValues..isnancsИ|ГoИ|dГS)Nrr)r!)r"rr┌isinf_sz5GeneratorTest.testExtremeDefaultValues..isinfrz? ? ?? ?? ??? ??/ ??-)r ZTestExtremeDefaultValues┌ assertTrueZ inf_doubleZneg_inf_doubleZ nan_doubleZ inf_floatZ neg_inf_floatZ nan_floatrZcpp_trigraph)r┌messager#r)r"r┌testExtremeDefaultValuesWsz&GeneratorTest.testExtremeDefaultValuescs>tjj}dddddЬЙtЗfddД|jDГГ}|jИ|ГdS)NFT)Zoptional_int32Zrepeated_int32r Z default_int32cs"g|]}|jИkr|j|jfСqSr)┌nameZhas_default_value)┌.0┌f)┌expected_has_default_by_namerr· {sz6GeneratorTest.testHasDefaultValues..)r rr┌dict┌fieldsr)r┌descZhas_default_by_namer)r*r┌testHasDefaultValuesps z"GeneratorTest.testHasDefaultValuescCs.|jtjjtjjГ|jtjjjtjjГdS)N)rr ┌optional_int32_extension┌containing_type┌TestAllExtensionsr┌TestRequired┌single)rrrr┌'testContainingTypeBehaviorForExtensionsАs z5GeneratorTest.testContainingTypeBehaviorForExtensionscCs*|jtjjdГ|jtjjjtjjГdS)N)rr r0Zextension_scoper3r4r)rrrr┌testExtensionScopeЖs z GeneratorTest.testExtensionScopecCsB|jtjjГ|jtjjjГtjj}|jd}|j|jГdS)N┌a)r$r r0Zis_extensionr3r4rr)rZmessage_descriptorZnon_extension_descriptorrrr┌testIsExtensionМs zGeneratorTest.testIsExtensioncCstjГ}|j|jjГjГdS)N)rZTestMessageSetr$r┌ GetOptionsZmessage_set_wire_format)rrrrr┌testOptionsФszGeneratorTest.testOptionscCs*tjГ}|jjdjГ}|j|dk ГdS)NZAnEnum)rZTestMessageWithCustomOptionsr┌enum_types_by_namer9r$)rrZenum_optionsrrr┌testMessageWithCustomOptionsШsz*GeneratorTest.testMessageWithCustomOptionscCs\|jttjjjГttjjjtjjjtjjjgГГ|jtj jjgГ|jtjjjjgГdS)N) r┌setr rrZnested_typesr┌ OptionalGroup┌ RepeatedGroup┌TestEmptyMessage)rrrr┌testNestedTypesЯszGeneratorTest.testNestedTypescCst|jtjjjdkГ|jtjjjdkГ|jtjjjjtjjГ|jtjjjjtjjГ|jtjjjjtjjГdS)N) r$r r@rr1rrrr?)rrrr┌testContainingTypeлs z GeneratorTest.testContainingTypecCs*|jtjjdkГ|jtjjtjjГdS)N)r$r ┌_FOREIGNENUMr1r┌_TESTALLTYPES_NESTEDENUMrr)rrrr┌"testContainingTypeInEnumDescriptor║s z0GeneratorTest.testContainingTypeInEnumDescriptorcCs||jtjjjjdГtjjj}|j|jjdГ|jtjjjjdГ|jtj jjdГ|jtj jjdГ|jtjjjdГdS)N┌protobuf_unittestZprotobuf_unittest_import)rr rr┌file┌packagerrZ ImportMessagerCrDZ_IMPORTENUM)rr.rrr┌testPackage┐s zGeneratorTest.testPackagecCsJ|jtjjjgГ|jtjjjdtfgГ|jtjjjdddtfgГdS) Nrщ*щ+щ/щФi)rJrK)rLrM)rr rrZextension_rangesr2┌ MAX_EXTENSIONZTestMultipleExtensionRanges)rrrr┌testExtensionRange╥sz GeneratorTest.testExtensionRangecCsr|jtjjdГ|jtjjdГ|jtjjdkГ|jtjjtjgГ|jtjjt jgГ|jtjj t jgГdS)Nzgoogle/protobuf/unittest.protorF)rr rr'rH┌assertFalseZ serialized_pbZdependenciesrrZpublic_dependencies)rrrr┌testFileDescriptor▄s z GeneratorTest.testFileDescriptorcCsR|jttdГГ|jttdГГ|jttdГГttdГrN|jttjtjГГdS)NZTestMessagerZtest_extension┌TestService)r$┌hasattrrrP┌ issubclassrRr ZService)rrrr┌testNoGenericServicesчs z#GeneratorTest.testNoGenericServicescCs6tj}|jtj|jtjjГ|jtjj|jkГdS)N)r rrZ _TESTALLTYPESZmessage_types_by_namer'rPZ_TESTALLTYPES_NESTEDMESSAGE)r┌ file_typerrr┌testMessageTypesByNameЄsz$GeneratorTest.testMessageTypesByNamecCs"tj}|jtj|jtjjГdS)N)r rrrCr;r')rrVrrr┌testEnumTypesByName■sz!GeneratorTest.testEnumTypesByNamecCs"tj}|jtj|jtjjГdS)N)r rrZmy_extension_stringZextensions_by_namer')rrVrrr┌testExtensionsByNamesz"GeneratorTest.testExtensionsByNamecCsDtjГ}|jd|jjГtjГ}|jd|jГ|jtjtjkГdS)Nr) r rrZoptional_public_import_message┌erZPublicImportMessager$r)rZall_type_protoZpublic_import_protorrr┌testPublicImports szGeneratorTest.testPublicImportscCs\tjГ}|j|jtjdГ|j|jtjdГ|j|jtjdГ|j|jtjdГdS)NZfooZbarZbazZqux)rZTestBadIdentifiersrZ Extensionsr%Z descriptorZ reflectionr )rr%rrr┌testBadIdentifierssz GeneratorTest.testBadIdentifierscCsцtjj}|jdt|jГГ|jd|jdjГ|jd|jdjГ|j||jdj Г|j|jd|j dГtddddgГ}|j|tdd Д|jdjDГГГx@|j jГD]2\}}||kr╥|j|jd|jГqм|j|jГqмWdS) NrZoneof_fieldrZoneof_uint32Zoneof_nested_messageZoneof_stringZoneof_bytescSsg|] }|jСqSr)r')r(Zfieldrrrr+-sz+GeneratorTest.testOneof..)r rrr┌lenZoneofsr'┌indexZassertIsr1Zoneofs_by_namer=r-r┌itemsZcontaining_oneofZassertIsNone)rr.Znested_namesrZ field_descrrr┌ testOneof"s zGeneratorTest.testOneofN)┌__name__┌ __module__┌__qualname__rr r&r/r5r6r8r:r<rArBrErIrOrQrUrWrXrYr[r\r`rrrrr>s, rc@s(eZdZdZddДZddДZddДZdS) ┌SymbolDatabaseRegistrationTestz?Checks that messages, enums and files are correctly registered.cCsО|jtjtjГjdГГ|jtjjtjГjdГГ|jtГПtjГjdГWdQRX|jtjj tjГjdГГ|jtjj tjГjdГГdS)Nzprotobuf_unittest.TestAllTypesz,protobuf_unittest.TestAllTypes.NestedMessagezprotobuf_unittest.NestedMessagez,protobuf_unittest.TestAllTypes.OptionalGroupz,protobuf_unittest.TestAllTypes.RepeatedGroup)rr rr┌DefaultZ GetSymbolrZassertRaises┌KeyErrorr>r?)rrrr┌ testGetSymbol8s"z,SymbolDatabaseRegistrationTest.testGetSymbolcCs8|jdtjГjjdГjГ|jdtjГjjdГjГdS)Nzprotobuf_unittest.ForeignEnumz)protobuf_unittest.TestAllTypes.NestedEnum)rrre┌poolZFindEnumTypeByNameZ full_name)rrrrr Ks z(SymbolDatabaseRegistrationTest.testEnumscCs|jdtjГjjdГjГdS)Nzgoogle/protobuf/unittest.proto)rrrerhZFindFileByNamer')rrrr┌testFindFileByNameUs z1SymbolDatabaseRegistrationTest.testFindFileByNameN)rarbrc┌__doc__rgr rirrrrrd5s rd┌__main__)rj┌ __author__Z unittest2Zunittest┌ImportErrorZgoogle.protobuf.internalrZgoogle.protobufrrrrrrr r rrNZTestCaserrdra┌mainrrrr┌'s*x&