U
    4Jeø  ć                   @   sb   d Z ddlZddlmZ ddlmZ ddlmZ ddlmZ G dd dejZe	d	kr^e 
”  dS )
z*Tests for google.protobuf.symbol_database.é    N)Śunittest_pb2)Ś
descriptor)Śdescriptor_pool)Śsymbol_databasec                   @   sT   e Zd Zdd Zdd Zdd Zdd Zd	d
 Zdd Zdd Z	dd Z
dd ZdS )ŚSymbolDatabaseTestc                 C   s   t jrtjt ” d}nt ” }| tj” | 	tj
” | 	tj
j” | 	tj
j” | 	tj
j” | tjj” | tj
jj” | tj” |S )N)Śpool)r   Z_USE_C_DESCRIPTORSr   ZSymbolDatabaser   ZDefaultZRegisterFileDescriptorr   Ś
DESCRIPTORZRegisterMessageŚTestAllTypesŚNestedMessageŚOptionalGroupŚRepeatedGroupZRegisterEnumDescriptorZForeignEnumZ
NestedEnumZRegisterServiceDescriptorZ_TESTSERVICE)ŚselfŚdb© r   śQ/tmp/pip-unpacked-wheel-0oacnh65/google/protobuf/internal/symbol_database_test.pyŚ	_Database+   s    zSymbolDatabaseTest._Databasec                 C   s&   |   ”  tjj”}|  |tjk” d S )N)r   ZGetPrototyper   r	   r   Ś
assertTrue)r   Śinstancer   r   r   ŚtestGetPrototype=   s    ’z#SymbolDatabaseTest.testGetPrototypec                 C   s(   |   ”  dg”}|  tj|d k” d S )Nśgoogle/protobuf/unittest.protośprotobuf_unittest.TestAllTypes)r   ZGetMessagesr   r   r	   )r   Śmessagesr   r   r   ŚtestGetMessagesB   s    ’’’z"SymbolDatabaseTest.testGetMessagesc                 C   sj   |   tj|  ”  d”” |   tjj|  ”  d”” |   tjj|  ”  d”” |   tjj|  ”  d”” d S )Nr   ś,protobuf_unittest.TestAllTypes.NestedMessagez,protobuf_unittest.TestAllTypes.OptionalGroupz,protobuf_unittest.TestAllTypes.RepeatedGroup)ŚassertEqualr   r	   r   Z	GetSymbolr
   r   r   ©r   r   r   r   ŚtestGetSymbolI   s0     ’’ ’’ ’’ ’’z SymbolDatabaseTest.testGetSymbolc                 C   s8   |   d|  ” j d”j” |   d|  ” j d”j” d S )Nzprotobuf_unittest.ForeignEnumś)protobuf_unittest.TestAllTypes.NestedEnum)r   r   r   ZFindEnumTypeByNameŚ	full_namer   r   r   r   Ś	testEnumsW   s    
’ž
’žzSymbolDatabaseTest.testEnumsc                 C   s8   |   d|  ” j d”j” |   d|  ” j d”j” d S )Nr   r   )r   r   r   ZFindMessageTypeByNamer   r   r   r   r   ŚtestFindMessageTypeByNameb   s    
’ž
’žz,SymbolDatabaseTest.testFindMessageTypeByNamec                 C   s   |   d|  ” j d”j” d S )Nzprotobuf_unittest.TestService)r   r   r   ZFindServiceByNamer   r   r   r   r   ŚtestFindServiceByNamel   s    
’žz(SymbolDatabaseTest.testFindServiceByNamec                 C   s8   |   d|  ” j d”j” |   d|  ” j d”j” d S )Nr   r   r   )r   r   r   ZFindFileContainingSymbolŚnamer   r   r   r   ŚtestFindFileContainingSymbolr   s    
’ž
’žz/SymbolDatabaseTest.testFindFileContainingSymbolc                 C   s   |   d|  ” j d”j” d S )Nr   )r   r   r   ZFindFileByNamer"   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unittestZgoogle.protobufr   r   r   r   ZTestCaser   r%   Śmainr   r   r   r   Ś<module>   s   [