3
SLj @ s d Z y ddlZW n ek
r, ddlZY nX ddlmZ ddlmZ ddlmZ ddlmZ G dd dej Z
e d krej dS )
z*Tests for google.protobuf.symbol_database. N) unittest_pb2)
descriptor)descriptor_pool)symbol_databasec @ sT e Zd Zdd Zdd Zdd Zdd Zd d
Zd d Zd
d Z dd Z
dd Z dS )SymbolDatabaseTestc C s t jrtjtj d}ntj }|jtj |j tj
|j tj
j |j tj
j |j tj
j
|jtjj |jtj
jj |jtj |S )N)pool)r Z_USE_C_DESCRIPTORSr ZSymbolDatabaser ZDefaultZRegisterFileDescriptorr
DESCRIPTORZRegisterMessage TestAllTypes
NestedMessage
OptionalGroup
RepeatedGroupZRegisterEnumDescriptorZ ForeignEnumZ
NestedEnumZRegisterServiceDescriptorZ _TESTSERVICE)selfZdb r */usr/lib/python3.6/symbol_database_test.py _Database/ s z SymbolDatabaseTest._Databasec C s& | j jtjj}| j|tjk d S )N)r Z GetPrototyper r r
assertTrue)r
instancer r r testGetPrototypeA s
z#SymbolDatabaseTest.testGetPrototypec C s( | j jdg}| jtj|d k d S )Nz google/protobuf/unittest.protoz protobuf_unittest.TestAllTypes)r Z GetMessagesr r r )r
Zmessagesr r r testGetMessagesF s
z"SymbolDatabaseTest.testGetMessagesc C sj | j tj| j jd | j tjj| j jd | j tjj| j jd | j tjj| j jd d S )Nz protobuf_unittest.TestAllTypesz,protobuf_unittest.TestAllTypes.NestedMessagez,protobuf_unittest.TestAllTypes.OptionalGroupz,protobuf_unittest.TestAllTypes.RepeatedGroup) assertEqualr r r Z GetSymbolr
r r )r
r r r
testGetSymbolM s z SymbolDatabaseTest.testGetSymbolc C s8 | j d| j jjdj | j d| j jjdj d S )Nz protobuf_unittest.ForeignEnumz)protobuf_unittest.TestAllTypes.NestedEnum)r r r ZFindEnumTypeByName full_name)r
r r r testEnums[ s
z SymbolDatabaseTest.testEnumsc C s8 | j d| j jjdj | j d| j jjdj d S )Nz protobuf_unittest.TestAllTypesz,protobuf_unittest.TestAllTypes.NestedMessage)r r r ZFindMessageTypeByNamer )r
r r r testFindMessageTypeByNamef s
z,SymbolDatabaseTest.testFindMessageTypeByNamec C s | j d| j jjdj d S )Nz protobuf_unittest.TestService)r r r ZFindServiceByNamer )r
r r r testFindServiceByNamep s
z(SymbolDatabaseTest.testFindServiceByNamec C s8 | j d| j jjdj | j d| j jjdj d S )Nz google/protobuf/unittest.protoz)protobuf_unittest.TestAllTypes.NestedEnumz protobuf_unittest.TestAllTypes)r r r ZFindFileContainingSymbolname)r
r r r testFindFileContainingSymbolv s
z/SymbolDatabaseTest.testFindFileContainingSymbolc C s | j d| j jjdj d S )Nz google/protobuf/unittest.proto)r r r ZFindFileByNamer )r
r r r testFindFileByName s
z%SymbolDatabaseTest.testFindFileByNameN) __name__
__module__ __qualname__r r r r r r r r r r r r r r - s
r __main__)
__doc__Z unittest2Zunittest ImportErrorZgoogle.protobufr r r r ZTestCaser r mainr r r r s [