3 ��h��@s�dZyddlZWnek r,ddlZYnXddlmZddlmZddlmZddlmZGdd�dej �Z ed kr�ej�dS) z*Tests for google.protobuf.symbol_database.�N)�unittest_pb2)� descriptor)�descriptor_pool)�symbol_databasec@sTeZdZdd�Zdd�Zdd�Zdd�Zd d �Zdd�Zd d�Z dd�Z dd�ZdS)�SymbolDatabaseTestcCs�tjrtjtj�d�}ntj�}|jtj�|j tj �|j tj j�|j tj j�|j tj j �|jtjj�|jtj jj�|jtj�|S)N)�pool)rZ_USE_C_DESCRIPTORSrZSymbolDatabaserZDefaultZRegisterFileDescriptorr� DESCRIPTORZRegisterMessage�TestAllTypes� NestedMessage� OptionalGroup� RepeatedGroupZRegisterEnumDescriptorZForeignEnumZ NestedEnumZRegisterServiceDescriptorZ_TESTSERVICE)�selfZdb�r�*/usr/lib/python3.6/symbol_database_test.py� _Database/szSymbolDatabaseTest._DatabasecCs&|j�jtjj�}|j|tjk�dS)N)rZGetPrototyperr r� assertTrue)r �instancerrr�testGetPrototypeAs z#SymbolDatabaseTest.testGetPrototypecCs(|j�jdg�}|jtj|dk�dS)Nzgoogle/protobuf/unittest.protozprotobuf_unittest.TestAllTypes)rZGetMessagesrrr )r Zmessagesrrr�testGetMessagesFs z"SymbolDatabaseTest.testGetMessagescCsj|jtj|j�jd��|jtjj|j�jd��|jtjj|j�jd��|jtjj|j�jd��dS)Nzprotobuf_unittest.TestAllTypesz,protobuf_unittest.TestAllTypes.NestedMessagez,protobuf_unittest.TestAllTypes.OptionalGroupz,protobuf_unittest.TestAllTypes.RepeatedGroup)�assertEqualrr rZ GetSymbolr rr)r rrr� testGetSymbolMsz SymbolDatabaseTest.testGetSymbolcCs8|jd|j�jjd�j�|jd|j�jjd�j�dS)Nzprotobuf_unittest.ForeignEnumz)protobuf_unittest.TestAllTypes.NestedEnum)rrrZFindEnumTypeByName� full_name)r rrr� testEnums[s zSymbolDatabaseTest.testEnumscCs8|jd|j�jjd�j�|jd|j�jjd�j�dS)Nzprotobuf_unittest.TestAllTypesz,protobuf_unittest.TestAllTypes.NestedMessage)rrrZFindMessageTypeByNamer)r rrr�testFindMessageTypeByNamefs z,SymbolDatabaseTest.testFindMessageTypeByNamecCs|jd|j�jjd�j�dS)Nzprotobuf_unittest.TestService)rrrZFindServiceByNamer)r rrr�testFindServiceByNameps z(SymbolDatabaseTest.testFindServiceByNamecCs8|jd|j�jjd�j�|jd|j�jjd�j�dS)Nzgoogle/protobuf/unittest.protoz)protobuf_unittest.TestAllTypes.NestedEnumzprotobuf_unittest.TestAllTypes)rrrZFindFileContainingSymbol�name)r rrr�testFindFileContainingSymbolvs z/SymbolDatabaseTest.testFindFileContainingSymbolcCs|jd|j�jjd�j�dS)Nzgoogle/protobuf/unittest.proto)rrrZFindFileByNamer)r rrr�testFindFileByName�s z%SymbolDatabaseTest.testFindFileByNameN)�__name__� __module__�__qualname__rrrrrrrrrrrrrr-s r�__main__) �__doc__Z unittest2Zunittest�ImportErrorZgoogle.protobufrrrrZTestCaserr�mainrrrr� s[