3
SLj{ @ s d Z dZddlZddlZy ddlZW n ek
r@ ddlZY nX ddlmZ ddlm Z ddlm
Z
ddlm Z ddl m
Z
dd l mZ dd
l mZ dd l mZ dd l mZ dd
l mZ ddl mZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ G dd deZG dd deejZ G dd deejZ G dd deejZ 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#G d&d' d'eZ$G d(d) d)eZ%G d*d+ d+ejZ&e d,d-e!e dmdnge!e dodpge!d6e dqdrgid;e"d/d6d7fdd;e"d/d6d4fdd;e"d/d6dKfdd;e"d/d6d^fdgoogle.protobuf.python.internal.Factory2Message.one_more_fieldz protobuf_unittest.TestServicez google/protobuf/unittest.proto)
r FindFileContainingSymbolr r
r r r r r r )r r r Z
file_desc3Z
file_desc4Z
file_desc5r! r! r" testFindFileContainingSymbolT s: z3DescriptorPoolTestBase.testFindFileContainingSymbolc C s& | j t | jjd W d Q R X d S )NzDoes not exist)r$ r% r r' )r r! r! r" #testFindFileContainingSymbolFailurey s z:DescriptorPoolTestBase.testFindFileContainingSymbolFailurec C s | j jd}| j|tj | jd|j | jd|j | jd |j | j |j
|j d }| jd|j | j||j |j d }| jd|j | j||j | j||j
d j | j||j
d j | j jd}| j|tj | jd |j | jd|j | j|j |j d }| jd
|j | j||j |j d }| jd |j | j||j | j||j
d j | j||j
d
j | j|j
d j | jd|j
d j | j|j
d j | jd|j
d j | j|j
d j | jd|j
d j | j|j
d j | j |j
d j | j|j
d j | jd|j
d j | j jd}| j|| | j|j
d j | jd|j
d j | jdt|j | jdt|j | jdt|jd j xBd D ]:}| j|jd |j
| j | j|j
| |jd j qW d S ) Nz/google.protobuf.python.internal.Factory1Messager r ZNestedFactory1MessageZNestedFactory1EnumZnested_factory_1_messageZnested_factory_1_enumz/google.protobuf.python.internal.Factory2Messager ZNestedFactory2MessageZNestedFactory2EnumZnested_factory_2_messageZnested_factory_2_enumZint_with_defaulti Zdouble_with_defaultg{G#@Zstring_with_defaultz hello worldZbool_with_defaultZenum_with_default zEgoogle.protobuf.python.internal.Factory2Message.NestedFactory2MessageZbytes_with_defaults a c oneof_int oneof_string)r, r- )r FindMessageTypeByNamer r
Z
Descriptorr r full_namecontaining_type assertFalse has_optionsZ nested_typesZ
enum_typesfields_by_name message_type enum_typeZ assertIsNone
assertTruehas_default_value
default_valuelenZoneofsoneofs_by_nameZfieldsZcontaining_oneofr ) r Zmsg1Z nested_msg1 nested_enum1Zmsg2Z nested_msg2 nested_enum2Zmsg3r r! r! r" testFindMessageTypeByName} s
z0DescriptorPoolTestBase.testFindMessageTypeByNamec C s | j t| jjd tj dkrH| j t| jjd | j t| jjd t}nt}| j || jj d | j || jj
d | j || jj d | j || jj d | j || jj
d | j || jjd | j || jjd tj dkrt}| j || jjd d S )N cppr python)r$ TypeErrorr FindExtensionByNumberr TypeZFindMethodByNamer% AttributeErrorr. FindFieldByNameFindExtensionByNameFindEnumTypeByNameFindOneofByNameFindServiceByNamer' r )r Z
error_typer! r! r" testFindTypeErrors s z)DescriptorPoolTestBase.testFindTypeErrorsc C s& | j t | jjd W d Q R X d S )NzDoes not exist)r$ r% r r. )r r! r! r" testFindMessageTypeByNameFailure s z7DescriptorPoolTestBase.testFindMessageTypeByNameFailurec C s | j jd}| j|tj | jd|jd j | jd|jd j | j|j | j jd}| j|tj | jd|jd j | jd|jd j | j jd }| j|tj | jd|jd
j | jd|jd j | j jd }| j|tj | jd|jd
j | jd|jd j d S )Nz,google.protobuf.python.internal.Factory1Enumr ZFACTORY_1_VALUE_0r* ZFACTORY_1_VALUE_1zBgoogle.protobuf.python.internal.Factory1Message.NestedFactory1EnumZNESTED_FACTORY_1_VALUE_0ZNESTED_FACTORY_1_VALUE_1z,google.protobuf.python.internal.Factory2EnumZFACTORY_2_VALUE_0ZFACTORY_2_VALUE_1zBgoogle.protobuf.python.internal.Factory2Message.NestedFactory2EnumZNESTED_FACTORY_2_VALUE_0ZNESTED_FACTORY_2_VALUE_1)
r rG r r
ZEnumDescriptorr values_by_namenumberr1 r2 )r Zenum1r; Zenum2r< r! r! r" testFindEnumTypeByName s2 z-DescriptorPoolTestBase.testFindEnumTypeByNamec C s& | j t | jjd W d Q R X d S )NzDoes not exist)r$ r% r rG )r r! r! r" testFindEnumTypeByNameFailure s z4DescriptorPoolTestBase.testFindEnumTypeByNameFailurec C sv t | trtj dkrd S | jjd}| j|jd | j|j|j | j
|j | j t
| jjd W d Q R X d S )Nr? z:google.protobuf.python.internal.Factory1Message.list_valueZ
list_valuezDoes not exist)
isinstance#SecondaryDescriptorFromDescriptorDBr rC r rE r r labelZLABEL_REPEATEDr1 r2 r$ r% )r fieldr! r! r" testFindFieldByName
s
z*DescriptorPoolTestBase.testFindFieldByNamec C sZ t | trtj dkrd S | jjd}| j|jd | jt | jjd W d Q R X d S )Nr? z;google.protobuf.python.internal.Factory2Message.oneof_field oneof_fieldzDoes not exist)
rP rQ r rC r rH r r r$ r% )r Zoneofr! r! r" testFindOneofByName s
z*DescriptorPoolTestBase.testFindOneofByNamec C s t | trtj dkrd S | jjd}| j|jd | jjd}| j|jd | j|jd | j t
| jj d W d Q R X d S )Nr? z>google.protobuf.python.internal.Factory2Message.one_more_fieldone_more_fieldz-google.protobuf.python.internal.another_field
another_fieldi zDoes not exist) rP rQ r rC r rF r r rM r$ r% rE )r extensionr! r! r" testFindExtensionByName% s
z.DescriptorPoolTestBase.testFindExtensionByNamec C s | j jd}| j jd}|jd }| j j| | j jd}|jd }| j j| | j j|}t||g}| j|t| |jd | j j|}| j|t| d S )Nz/google.protobuf.python.internal.Factory1Messagez/google.protobuf.python.internal.Factory2MessagerW z,google/protobuf/internal/factory_test2.protorX Zunexpected_element) r r. extensions_by_nameAddExtensionDescriptorr ZFindAllExtensionssetr append)r factory1_messagefactory2_messagerW
factory_test2rX
extensionsZexpected_extension_numbersr! r! r" testFindAllExtensions7 s
z,DescriptorPoolTestBase.testFindAllExtensionsc
C s | j jd}| j jd}|jd }| j j| | j jd}|jd }| j j| | j j|d}| j|jd | j j|d}| j|jd | jt | j j|d}W d Q R X d S ) Nz/google.protobuf.python.internal.Factory1Messagez/google.protobuf.python.internal.Factory2MessagerW z,google/protobuf/internal/factory_test2.protorX i i i )
r r. r[ r\ r rB r r r$ r% )r r_ r` rW ra rX rY r! r! r" testFindExtensionByNumberO s
z0DescriptorPoolTestBase.testFindExtensionByNumberc C sj | j t | jjd W d Q R X | j t | jjd W d Q R X | j t | jjd W d Q R X d S )Nz-google.protobuf.python.internal.another_fieldz>google.protobuf.python.internal.Factory2Message.one_more_fieldz:google.protobuf.python.internal.Factory1Message.list_value)r$ r% r rE rF )r r! r! r" testExtensionsAreNotFieldsf s z1DescriptorPoolTestBase.testExtensionsAreNotFieldsc C s@ | j jd}| j|jd | jt | j jd W d Q R X d S )Nz protobuf_unittest.TestServicezDoes not exist)r rI r r/ r$ r% )r servicer! r! r" testFindServicep s z&DescriptorPoolTestBase.testFindServicec C s8 t j }tj|| _|j| j |j| j | j d S )N) r DescriptorDatabaser DescriptorPoolr Addfactory_test1_fdfactory_test2_fdr= )r dbr! r! r" testUserDefinedDBv s
z(DescriptorPoolTestBase.testUserDefinedDBc C sT t | trtj dkrd S tj | _| jj| jj | jj| j
j | j d S )Nr? ) rP rQ r rC r ri r AddSerializedFilerk SerializeToStringrl r= )r r! r! r" testAddSerializedFile} s
z,DescriptorPoolTestBase.testAddSerializedFilec s fdd}|t j tj jjd} j|t j || t trVt j
dkrVdS t j j
t jj} jj| jjd}|| dS )z:Test the default value of enums which don't start at zero.c s& | j d jd j} j|tjj d S )NDescriptorPoolTest1 nested_enum)r r3 r8 r r rr BETA)file_descriptorr8 )r r! r" _CheckDefaultValue s
zGDescriptorPoolTestBase.testEnumDefaultValue.._CheckDefaultValuez4google/protobuf/internal/descriptor_pool_test1.protor? N)r
DESCRIPTORr Defaultr r assertIsrP rQ r rC r FileDescriptorProto
FromString
serialized_pbrj )r rv ru Zdescriptor_protor! )r r" testEnumDefaultValue s
z+DescriptorPoolTestBase.testEnumDefaultValuec s fdd fdd}|t j tj }|jtjjtj j
|jtjjt j j
|jtjjt j j
t j
|j|jt jj j}|| dS )z0Check the value returned by non-existent fields.c s j | | j| | d S )N)r r )valueZexpected_valueZ
expected_type)r r! r" _CheckValueAndType s zTDescriptorPoolTestBase.testDefaultValueForCustomMessages.._CheckValueAndTypec s yt }W n tk
r t}Y nX yt}W n tk
rB t}Y nX | jdt | jd|tf | jdttf | j dttf | j
dt | j d| | j
dt | j| jt d S )Nr Fr> )Zlong NameErrorintZunicodestroptional_int32Zoptional_uint64Zoptional_floatfloatZoptional_doubleZ
optional_boolboolZoptional_stringZoptional_bytesbytesZoptional_nested_enumZFOO)msgZint64Z unicode_type)r r! r" _CheckDefaultValues s
zUDescriptorPoolTestBase.testDefaultValueForCustomMessages.._CheckDefaultValuesN)r TestAllTypesr ri rj r rz r{ r rw r| r r ZMessageFactoryZ GetPrototyper. r/ )r r r Z
message_classr! )r r r" !testDefaultValueForCustomMessages s
z8DescriptorPoolTestBase.testDefaultValueForCustomMessagesc C sF t | trtj dkrd S tjdd}| jj| | jj|j d S )Nr? zsome/file.proto)r )
rP rQ r rC r rz r rj ro rp )r file_descr! r! r" testAddFileDescriptor s
z,DescriptorPoolTestBase.testAddFileDescriptorc C s t | trtj dkrd S tjjtjj }tjjt
jj }tjjt jj }| j j
| | j j
| | j j
| tj| | j tj| | j d S )Nr? )rP rQ r rC r rz r{ r rw r| r r r rj
TEST1_FILE CheckFile
TEST2_FILE)r Zmore_messages_descZ
test1_descZ
test2_descr! r! r" testComplexNesting s
z)DescriptorPoolTestBase.testComplexNestingN)__name__
__module__ __qualname__r# r& r( r) r= rJ rK rN rO rT rV rZ rc rd re rg rn rq r} r r r r! r! r! r" r ? s, %R
& r c @ s e Zd Zdd Zdd ZdS )DefaultDescriptorPoolTestc C s2 t j | _tjjtjj| _ tjjt
jj| _ d S )N) r rx r r rz r{ r rw r| rk r
rl )r r! r! r" setUp s
z DefaultDescriptorPoolTest.setUpc C s | j | jjdtj | j | jjdtjj | j | jjdtjjjd | j | jj dtj
j | j | jj dtjj d | j | jj
dtjjjd | j | jjd
tjjd d S ) Nz google/protobuf/unittest.protoz protobuf_unittest.TestAllTypesz-protobuf_unittest.TestAllTypes.optional_int32r z protobuf_unittest.ForeignEnumz*protobuf_unittest.optional_int32_extensionZoptional_int32_extensionz*protobuf_unittest.TestAllTypes.oneof_fieldrU z protobuf_unittest.TestServiceZ TestService)ry r r r rw r. r rE r3 rG ForeignEnumrF r[ rH r: rI Zservices_by_name)r r! r! r" testFindMethods s.
z)DefaultDescriptorPoolTest.testFindMethodsN)r r r r r r! r! r! r" r s r c @ s e Zd Zdd ZdS )CreateDescriptorPoolTestc C s t j | _tjjtjj| _ tjjt
jj| _ | jj | j | jj | j | jj tjjt
jj | jj tjjtjj | jj tjjtjj d S )N)r ri r r rz r{ r rw r| rk r
rl rj r r r )r r! r! r" r s
z CreateDescriptorPoolTest.setUpN)r r r r r! r! r! r" r s r c @ s e Zd Zdd ZdS )rQ c C s t jjtjj| _t jjtjj| _t j
}|j | j |j | j |j t jjt jj |j t jjt
jj |j t jjtjj tj|d| _d S )N)Z
descriptor_db)r rz r{ r rw r| rk r
rl r rh rj r r r r ri r )r rm r! r! r" r ( s
z)SecondaryDescriptorFromDescriptorDB.setUpN)r r r r r! r! r! r" rQ % s rQ c @ s e Zd ZdddZdd ZdS ) ProtoFileNc C s* || _ || _|| _|pg | _|p"g | _d S )N)r r messages dependenciespublic_dependencies)r r r r r r r! r! r" __init__; s
zProtoFile.__init__c C s |j | j}|j| j|j |j| j|j dd |jD }|j| j| dd |jD }|j| j| x&| jj D ]\}}|j|d || qtW d S )Nc S s g | ]
}|j qS r! )r ).0fr! r! r"
G s z'ProtoFile.CheckFile..c S s g | ]
}|j qS r! )r )r r r! r! r" r I s ) r r r r r r r items CheckType)r testr r Zdependencies_namesZpublic_dependencies_namesr Zmsg_typer! r! r" r C s zProtoFile.CheckFile)NN)r r r r r r! r! r! r" r 9 s
r c @ s e Zd Zdd Zdd ZdS )EnumTypec C s
|| _ d S )N)values)r r r! r! r" r Q s zEnumType.__init__c C s |j | }|j||j dj|j|g}|j||j |j||j |j||j xjt| jD ]\\}\}} |j | }
|j||
j |j||
j
|j| |
j |j||
j |j
||j q^W d S )N.)enum_types_by_namer r joinr/ r0 file enumerater rL indexrM typer Zenum_values_by_name) r r msg_descr r enum_descZexpected_enum_full_namer r~ rM Z
value_descr! r! r" r T s
zEnumType.CheckTypeN)r r r r r r! r! r! r" r O s r c @ s e Zd ZdddZdd ZdS ) MessageTypeFNc C s || _ || _|| _|pg | _d S )N) type_dict
field_list
is_extendablerb )r r r r rb r! r! r" r f s zMessageType.__init__c
C s |d kr$|j | }dj|j|g}n|j| }dj|j|g}|j||j |j||j |j||j |j|j| |j| j |j x&| j
j D ]\}}|j |||| qW x,t
| jD ] \}\}} | j||||| qW x,t
| jD ] \}\}} | j||||| qW d S )Nr )r r r nested_types_by_namer/ r r r0 r r r r r r r
CheckFieldrb )
r r Zcontaining_type_descr r ZdescZexpected_full_namesubtyper rS r! r! r" r m s
zMessageType.CheckType)FN)r r r r r r! r! r! r" r d s
r c @ s e Zd Zdd Zdd ZdS ) EnumFieldc C s || _ || _|| _d S )N)rM type_namer8 )r rM r r8 r! r! r" r s zEnumField.__init__c C s |j | }|j| j }|j||j dj|j|g}|j||j |j||j |j| j|j |jt j
j |j |jt j
j
|j |j|j |j|j| j j|j |j|j| j j |j||j |j||j |j||j d S )Nr )r3 r r r r r r/ r rM r
FieldDescriptorZ TYPE_ENUMr Z CPPTYPE_ENUMcpp_typer6 r7 rL r8 r1 r2 r0 r5 r ) r r r r r r
field_descr expected_field_full_namer! r! r" r s"
zEnumField.CheckFieldN)r r r r r r! r! r! r" r s r c @ s e Zd Zdd Zdd ZdS ) MessageFieldc C s || _ || _d S )N)rM r )r rM r r! r! r" r s zMessageField.__init__c C s |j | }|j| j }|j||j dj|j|g}|j||j |j||j |j| j|j |jt j
j |j |jt j
j
|j |j|j |j||j |j||j |j||j tj dkr|jtt|d d S )Nr r? r8 )r3 r r r r r r/ r rM r
r TYPE_MESSAGEr CPPTYPE_MESSAGEr r1 r7 r0 r4 r r rC r$ NotImplementedErrorgetattr) r r r r r r r Zfield_type_descr r! r! r" r s"
zMessageField.CheckFieldN)r r r r r r! r! r! r" r s r c @ s e Zd Zdd Zdd ZdS ) StringFieldc C s || _ || _d S )N)rM r8 )r rM r8 r! r! r" r s zStringField.__init__c C s |j | }|j||j dj|j|g}|j||j |j||j |j| j|j |jtjj |j
|jtjj |j |j
|j |j| j|j |j||j d S )Nr )r3 r r r r/ r rM r
r Z TYPE_STRINGr ZCPPTYPE_STRINGr r6 r7 r8 r )r r r r r r r r r! r! r" r s
zStringField.CheckFieldN)r r r r r r! r! r! r" r s r c @ s e Zd Zdd Zdd ZdS )ExtensionFieldc C s || _ || _d S )N)rM
extended_type)r rM r r! r! r" r s zExtensionField.__init__c C s |j | }|j||j dj|j|g}|j||j |j| j|j |j||j |jtjj |j
|jtjj |j |j
|j |j|j |j||j |j||j |j| j|jj |j||j d S )Nr )r[ r r r r/ rM r r
r r r r r r1 r7 r6 Z is_extensionZextension_scoper4 r r0 r )r r r r r r r r r! r! r" r s
zExtensionField.CheckFieldN)r r r r r r! r! r! r" r s r c @ s e Zd Zdd Zejej dkddd Zdd Z ejej dkdd d
Z
ejej dkdd d Z ejej dkdd
d Z dd Z
dd Zdd Zdd ZdS )AddDescriptorTestc C s t j }|jtjj | jd|j|d j | j t
|j|d W d Q R X |jtjj j | jd|j|d j | jd|j dj
| jd|j|d j
d S )Nz protobuf_unittest.TestAllTypesz,protobuf_unittest.TestAllTypes.NestedMessagez google/protobuf/unittest.proto)r ri
AddDescriptorr r rw r r. r/ r$ r%
NestedMessager r r' )r prefixr r! r! r" _TestMessage s,
z AddDescriptorTest._TestMessager? z7With the cpp implementation, Add() must be called firstc C s | j d | j d d S )Nr> r )r )r r! r! r" testMessage s
z AddDescriptorTest.testMessagec C s t j }|jtjj | jd|j|d j | j t
|j|d W d Q R X |jtj j j | jd|j|d j | jd|j
dj | jd|j|d j d S )Nz protobuf_unittest.ForeignEnumz(protobuf_unittest.ForeignEnum.NestedEnumz)protobuf_unittest.TestAllTypes.NestedEnumz google/protobuf/unittest.proto)r ri AddEnumDescriptorr r rw r rG r/ r$ r% r
NestedEnumr r r' )r r r r! r! r" _TestEnum s,
zAddDescriptorTest._TestEnumc C s | j d | j d d S )Nr> r )r )r r! r! r" testEnum, s
zAddDescriptorTest.testEnumc C sL t j }| jt |jd W d Q R X |jtj | jd|jdj d S )Nz protobuf_unittest.TestService)
r ri r$ r% rI AddServiceDescriptorr Z _TESTSERVICEr r/ )r r r! r! r" testService2 s z AddDescriptorTest.testServicec C sL t j }|jtj | jd|jdj | jt |j
d W d Q R X d S )Nz google/protobuf/unittest.protoz protobuf_unittest.TestAllTypes) r ri AddFileDescriptorr rw r r r r$ r% r' )r r r! r! r" testFile= s
zAddDescriptorTest.testFilec C sR t j }tjj}| jt|j| tj }tjj | |j
| | j |j| d S )N) r ri r rw r r$ r% r rz Z CopyToProtorj r6 )r r Zproto_file_nameru r! r! r" testEmptyDescriptorPoolM s
z)AddDescriptorTest.testEmptyDescriptorPoolc C s
t j }tjddd}|jjdd |j| | j|jdj d | j|j
dj d tjdd}|jjdd}|j jd d
tj j
tj jd |jjd d}|jjd
dd |jjdd t j }|j| | jd|j
dj | jd |jd j | jd|jdj d S )Nzsome/file.protor )r r ZMessage)r zpackage.Messagez some/filename/container.protoZ
TopMessageZbbr* )r rM r rR ZTopEnumZ FOREIGN_FOO )r rM Z
TopService)r ri r rz r4 addrj r r r r. rS ZFieldDescriptorProtoZ
TYPE_INT32ZLABEL_OPTIONALr5 r~ rf rG rI )r r r Z
file_protoZ
message_protoZ
enum_protor! r! r" testCustomDescriptorPoolY s6
z*AddDescriptorTest.testCustomDescriptorPoolc C sp t j }d}tj|d}tj}d|jj| _|j | |j
|}|j }| j d|j| j | j
||j d S )Nz9file_descriptor_options_with_custom_descriptor_pool.proto)r Zfoo)r ri r rz r Z foo_optionsoptionsZ
ExtensionsZfoo_namerj r Z
GetOptionsr ry )r r file_nameZfile_descriptor_protoZ extension_idru r r! r! r" 1testFileDescriptorOptionsWithCustomDescriptorPoolx s
zCAddDescriptorTest.testFileDescriptorOptionsWithCustomDescriptorPoolc % C s t j }| jt |jd W d Q R X | jt |jd W d Q R X | jt |jd W d Q R X | jt |jd W d Q R X | jt |jd W d Q R X d S )Nr ) r ri r$ rA r r r r\ r )r r r! r! r" testAddTypeError s z"AddDescriptorTest.testAddTypeErrorN)r r r r unittestZskipIfr rC r r r r r r r r r r! r! r! r" r s
r z4google/protobuf/internal/descriptor_pool_test1.protoz google.protobuf.python.internalALPHAr* rt r+ EPSILON ZETA r ETA THETA rs Z nested_fieldZtheta)r DeepNestedMessageZbetaZdeep_nested_message r )r r Znested_messager T)r GAMMADELTAr IOTA KAPPA
LAMBDA MU lambdaZdelta)rr ZDescriptorPoolTest2z4google/protobuf/internal/descriptor_pool_test2.protoZDescriptorPoolTest3NU
XI OMICRON PI RHO SIGMA ZsigmaZnuZdescriptor_pool_testi rr )rb z,google/protobuf/internal/more_messages.proto)r r __main__)r r* )rt 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 )+__doc__
__author__ossysZ unittest2r ImportErrorZgoogle.protobufr r r r Zgoogle.protobuf.internalr r r r r
r r r
r r r r objectr ZTestCaser r rQ r r r r r r r r r r r mainr! r! r! r" s 3# 0