XXc@sdZdZddlZyddlZWnek rGddlZnXddlmZddlmZddlm Z ddlm Z ddl m Z dd l m Z dd lmZdd lmZdd lmZdd lmZdZdejfdYZdefdYZdejfdYZdejfdYZdejfdYZedkrejndS(s1Unittest for google.protobuf.internal.descriptor.s#robinson@google.com (Will Robinson)iN(tunittest_custom_options_pb2(tunittest_import_pb2(t unittest_pb2(tdescriptor_pb2(tapi_implementation(t test_util(t descriptor(tdescriptor_pool(tsymbol_database(t text_formats name: 'TestEmptyMessage' tDescriptorTestcBseZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZejejdkpejdkddZRS(c Cstjdddd}|jjdd}|jjddddd tjjd tjj|jjdd }|j jdd dd |j jdddd|j jdddd|jjdd|j jdd}|j jdddddd}|j |_ |j j||j j|j|_|jj|j|_|jj|j|_|jj|j|_|jj|j|_dS(Ntnamessome/filename/some.prototpackagetprotobuf_unittestt NestedMessagetbbtnumberittypetlabelt ForeignEnumt FOREIGN_FOOit FOREIGN_BARit FOREIGN_BAZitResponseMessagetServicet CallMethodt input_types .protobuf_unittest.NestedMessaget output_types".protobuf_unittest.ResponseMessage(RtFileDescriptorProtot message_typetaddtfieldtFieldDescriptorProtot TYPE_INT32tLABEL_OPTIONALt enum_typetvaluetservicetmethodtGetDescriptorPooltpooltAddtFindFileByNameR tmy_filetmessage_types_by_namet my_messagetenum_types_by_nametmy_enumtservices_by_namet my_servicetmethods_by_namet my_method(tselft file_protot message_protot enum_protot service_protot method_proto((sJ/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/descriptor_test.pytsetUp?s:             cCs tjjS(N(RtDefaultR((R4((sJ/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/descriptor_test.pyR'cscCsV|j|jjddd|j|jjdjdj|jjdddS(NRiR(t assertEqualR-t EnumValueNameR.tvalues_by_numberR (R4((sJ/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/descriptor_test.pyttestEnumValueNamefs  cCs$|j|j|jjdjdS(Ni(R<R/tvaluesR(R4((sJ/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/descriptor_test.pyttestEnumFixupsoscCs=|j|j|jjdj|j|j|jjdS(Ni(R<R-tfieldstcontaining_typeR/(R4((sJ/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/descriptor_test.pyttestContainingTypeFixupsrs cCs|j|j|jjdS(N(R<R1R3tcontaining_service(R4((sJ/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/descriptor_test.pyttestContainingServiceFixupsvscCs|j|jjtj|j|jjdjtj|j|jjtj|j|jj djtj |j|j jtj |j|j jtjdS(Ni(R<R/t GetOptionsRt EnumOptionsR@tEnumValueOptionsR-tMessageOptionsRBt FieldOptionsR3t MethodOptionsR1tServiceOptions(R4((sJ/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/descriptor_test.pyttestGetOptionsys     cCstj}tjj}|jd}|jd}|jd}|jd}tjj}|jd}|j } tj } |j d| j | |j } tj } |j d| j | |j } tj}|j d| j |tj}|j d | j ||j }tj}|j d |j ||j }tj}|j d |j ||j }tj}|j d |j ||j }tj}|j d |j ||j }tj}|j tj|j |tjj}|j|j|j|jdS(Ntfield1tAnOneoftAnEnumt ANENUM_VAL2tFooILiI-u i*iii{IO(Rt DESCRIPTORtTestMessageWithCustomOptionstfields_by_nametoneofs_by_nameR.tenum_values_by_nametTestServiceWithCustomOptionstFindMethodByNameRGt file_opt1R<t Extensionst message_opt1t field_opt1t field_opt2t oneof_opt1t enum_opt1tenum_value_opt1t service_opt1t method_opt1tMETHODOPT1_VAL2tDummyMessageContainingEnumt assertTruet has_optionst assertFalse(R4tfile_descriptortmessage_descriptortfield_descriptortoneof_descriptortenum_descriptortenum_value_descriptortservice_descriptortmethod_descriptort file_optionsR[tmessage_optionsR]t field_optionsR^R_t oneof_optionsR`t enum_optionsRatenum_value_optionsRbtservice_optionsRctmethod_optionsRd((sJ/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/descriptor_test.pyttestSimpleCustomOptionssL                          c Csd}d}d}d}d}d}tjj}|j}|jt|jtj|j||jtj|j||jtj |jd|jtj |jd|jtj |j||jtj |j||jtj |jd|jtj|jd|jtj|j||jtj|j||jtjtjj}|j}|jt|jtj|j||jtj|j||jtj |j||jtj |j||jtj |j||jtj |j||jtj |j||jtj|j||jtj|j||jtj|j||jtjtjj}|j}|jd|jtj|jd |jtjd |jd |jtj|jd |jtj|jd |jtjtj} |j| j|jtjtjj}|j}|jd|jtjd |jd|jtjtjj}|j}|jd|jtjd |jd|jtjdS(Niii?ii i@iig$(@igYB?sHello, "World"t HelloWorldi iiifIillIillIIll(RtCustomOptionMinIntegerValuesRTRGR<tFalseR\tbool_optt int32_optt int64_optt uint32_optt uint64_optt sint32_optt sint64_optt fixed32_optt fixed64_optt sfixed32_optt sfixed64_opttCustomOptionMaxIntegerValuestTruetCustomOptionOtherValuestassertAlmostEqualt float_optt double_optt string_optt bytes_optRftTEST_OPTION_ENUM_TYPE2tenum_opttSettingRealsFromPositiveIntstSettingRealsFromNegativeInts( R4t kint32mint kint64mint kint32maxt kint64maxt kuint32maxt kuint64maxRkRst dummy_enum((sJ/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/descriptor_test.pyttestDifferentCustomOptionTypess                                      cCsYtjj}|j}|jd|jtjj|jd|jtjjtj|jd|jtjjtj j |jd|jtj j |jd|jtj jtj |jd|jtj jj|jd|jtj jjtj|jd|jtj jjtj j |jd |jtj jtjj|jd |jtj jtjjtj|jd |jtj jtjjtj j |jd |jtjjjj|jd |jtj jj|jd|jtjj |jd|jtjjj|jd|jtjjdS(Ni*iDiliiiiiiiiIiiAi ii(RtVariousComplexOptionsRTRGR<R\t complex_opt1tfootquuxtcorgetquxt complex_opt2tbaztgraulttbartgarplytComplexOptionType2tComplexOptionType4t complex_opt4twaldotfredt complex_opt3tcomplexoptiontype5tplught complexopt6txyzzy(R4Rtoptions((sJ/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/descriptor_test.pyttestComplexExtensionOptionssX                c Cstj}tjj}|jd}tjj}|jd}tjj}|jd}|jj tj }|j d|j |j d|j |j d|jj |j d|jj tj j |j d|jj tjjj |j d |jj tjj |j d |jj tjj |j d |jj tjj |j d |jj tjj |j d |jj tjj |j d|jj tjj dS(Nt fieldnametVALUEtMethodidtFileAnnotationtNestedFileAnnotationtFileExtensionAnnotationtEmbeddedMessageSetElementtMessageAnnotationtFieldAnnotationtEnumAnnotationtEnumValueAnnotationtServiceAnnotationtMethodAnnotation(RRTtAggregateMessageRVt AggregateEnumtvalues_by_nametAggregateServiceRZRGR\tfileoptR<titstsubtfiletmsettAggregateMessageSetElementtmessage_set_extensiontmsgopttfieldopttenumoptt enumvaloptt serviceoptt methodopt( R4RjRkRlRnRoRpRqRr((sJ/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/descriptor_test.pyttestAggregateOptionsFsP              cCstjjj}|jd|jjtj|jd}|jd|jjtj tjj}|j d}|jd|jjtj |j d}|jd|jjtj |jd}|jd |jjtjdS( Nit nested_fieldit NestedEnumitNESTED_ENUM_VALUEitnested_extensioni(RtNestedOptionTypeRRTR<RGR\R]RVR^R.RaRXRbtextensions_by_nameR_(R4tnested_messageRt outer_messaget nested_enumtnested_enum_valueR((sJ/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/descriptor_test.pyttestNestedOptionsws          cCs6|j|jj|j|j|jj|jdS(N(R<R/RR+R-(R4((sJ/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/descriptor_test.pyttestFileDescriptorReferencesscCse|j|jjd|j|jjd|j|jj|j|jtjjtjdS(Nssome/filename/some.protoR ( R<R+R R R(RRTRR;(R4((sJ/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/descriptor_test.pyttestFileDescriptorstcppisAImmutability of descriptors is only enforced in v2 implementationcCsxtjj}|jtd|_WdQX|jtd|jd (R+RRRRTRR-(R4tTEST_NESTED_MESSAGE_ASCII((sJ/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/descriptor_test.pyttestCopyToProto_NestedMessage&s  cCs&d}|jtjjtj|dS(Ns name: 'TestForeignNested' field: < name: 'foreign_nested' number: 1 label: 1 # Optional type: 11 # TYPE_MESSAGE type_name: '.protobuf_unittest.TestAllTypes.NestedMessage' > (R+RtTestForeignNestedRTRR-(R4tTEST_FOREIGN_NESTED_ASCII((sJ/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/descriptor_test.pyt$testCopyToProto_ForeignNestedMessage6s  cCs&d}|jtjjtj|dS(Ns name: 'ForeignEnum' value: < name: 'FOREIGN_FOO' number: 4 > value: < name: 'FOREIGN_BAR' number: 5 > value: < name: 'FOREIGN_BAZ' number: 6 > (R+RRRTRtEnumDescriptorProto(R4tTEST_FOREIGN_ENUM_ASCII((sJ/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/descriptor_test.pyttestCopyToProto_ForeignEnumGs  cCs&d}|jtjjtj|dS(Ns name: 'TestDeprecatedFields' field: < name: 'deprecated_int32' number: 1 label: 1 # Optional type: 5 # TYPE_INT32 options: < deprecated: true > > (R+RtTestDeprecatedFieldsRTRR-(R4tTEST_DEPRECATED_FIELDS_ASCII((sJ/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/descriptor_test.pyttestCopyToProto_Options]s  cCs&d}|jtjjtj|dS(Ns} name: 'TestEmptyMessageWithExtensions' extension_range: < start: 1 end: 536870912 > (R+RtTestEmptyMessageWithExtensionsRTRR-(R4t(TEST_EMPTY_MESSAGE_WITH_EXTENSIONS_ASCII((sJ/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/descriptor_test.pyttestCopyToProto_AllExtensionsps  cCs&d}|jtjjtj|dS(Ns name: 'TestMultipleExtensionRanges' extension_range: < start: 42 end: 43 > extension_range: < start: 4143 end: 4244 > extension_range: < start: 65536 end: 536870912 > (R+RtTestMultipleExtensionRangesRTRR-(R4t*TEST_MESSAGE_WITH_SEVERAL_EXTENSIONS_ASCII((sJ/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/descriptor_test.pyt!testCopyToProto_SeveralExtensions~s  cCs d}dS(NsM name: 'TestService' method: < name: 'Foo' input_type: '.protobuf_unittest.FooRequest' output_type: '.protobuf_unittest.FooResponse' > method: < name: 'Bar' input_type: '.protobuf_unittest.BarRequest' output_type: '.protobuf_unittest.BarResponse' > ((R4tTEST_SERVICE_ASCII((sJ/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/descriptor_test.pyt!testCopyToProto_ServiceDescriptors ( RRRR%R+R/R1R4R7R:R=R@RB(((sJ/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/descriptor_test.pyR s        BtMakeDescriptorTestcBs5eZdZdZdZdZdZRS(c Cs7tj}d|_|jj}|j|_|jj}d|_|jj}d|_|jj}d|_d|_|j j}d|_d|_t j j |_ t j j|_|j j}d|_d |_t j j |_ t j j|_d|_|j j}d|_d |_t j j |_ t j j|_d |_t j|}|j|jd jt j j|j|jdjt j j|j|jdjj||j|jd jd jd |j|jd jd j|jd jd |j|j|j|jd jdS(NtFoo2tSubRtBARiit uint64_fielditnested_message_fieldt bar_fields Foo2.Sub.FOOisFoo2.Sub.bar_field(RRR RRt nested_typeR#R$RRRtFieldDescriptortLABEL_REQUIREDRt TYPE_UINT64Rt TYPE_MESSAGEt type_namet TYPE_ENUMtMakeDescriptorR<RBtcpp_typetCPPTYPE_UINT64tCPPTYPE_MESSAGERCt nested_typesRt enum_typesRiRh( R4tfile_descriptor_protoRRJR#t enum_type_valRt enum_fieldtresult((sJ/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/descriptor_test.pyt"testMakeDescriptorWithNestedFieldssR                 cCs#tj}d|_|jj}|j|_|jj}d|_|jj}d|_d|_|jj}d|_d|_t j j |_ t j j |_|jj}d|_d|_t j j |_ t j j|_d |_t j|}|j|jd jt j jdS( NRSRRFiiRGiRIsFoo.FOOi(RRR RRR#R$RRRRKRLRRMRRPRORQR<RBRRRS(R4RWRR#RXRRYRZ((sJ/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/descriptor_test.pyt&testMakeDescriptorWithUnsignedIntFields.           cCsatj}tj}|jj|tj|}|j}|j d|j tj j dS(Nie( RR-RRRTR&RRQRGR<R\RR(R4tdescriptor_prototaggregate_messagetreformed_descriptorR((sJ/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/descriptor_test.pyttestMakeDescriptorWithOptionss    cCstj}d|_dddddg}dddddg}xCtt|D]/}|jj}|d|_|||_qRWtj |}x8tt|D]$}|j |j |j ||qWdS( NtBartfoo_footFooBartfooBaztfooFootfoobartfooBari( RR-R trangeRRRRRRQR<RBR(R4R]tnamestcamelcase_namesRRRZ((sJ/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/descriptor_test.pyttestCamelcaseName)s   cCstj}d|_ddddddg}dddddd g}xCtt|D]/}|jj}|d |_|||_qXWd |_t j |}x8tt|D]$}|j |j |j||qWdS( Nt TestJsonNamet field_namet fieldNamet FieldNamet _field_namet FIELD_NAMEt json_namet FIELDNAMEs@typei( RR-R RhRRRRRrRRQR<RB(R4R]Rit json_namesRRRZ((sJ/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/descriptor_test.pyt testJsonName7s      (RRR[R\R`RkRu(((sJ/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/descriptor_test.pyRCs  ,  t__main__(Rt __author__R t unittest2Rt ImportErrortgoogle.protobufRRRRtgoogle.protobuf.internalRRRRRR R.tTestCaseR RRRRCRtmain(((sJ/tmp/pip-build-h1VYrz/protobuf/google/protobuf/internal/descriptor_test.pyt!s2  gcs