← Back
Editing: generator_test.cpython-36.pyc
3 ���h�9 � @ s� d Z dZy ddlZW n ek r0 ddlZY nX ddlmZ ddlmZ ddlm Z ddlm Z ddlm Z dd lm Z dd lm Z dd lmZ dd lmZ dd lmZ dZG dd� dej�ZG dd� dej�Zedkr�ej� dS )z�Unittest that directly tests the output of the pure-Python protocol compiler. See //google/protobuf/internal/reflection_test.py for a test which further ensures that we can use Python protocol message objects as we expect. z#robinson@google.com (Will Robinson)� N)�test_bad_identifiers_pb2)�unittest_custom_options_pb2)�unittest_import_pb2)�unittest_import_public_pb2)�unittest_mset_pb2)� unittest_mset_wire_format_pb2)� unittest_no_generic_services_pb2)� unittest_pb2)�service)�symbol_databasei c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-S ).� GeneratorTestc C s( d}t j}| j|jj|jj| j� d S )N�optional_nested_message)r � TestAllTypes� assertEqual� NestedMessage� DESCRIPTOR�fields_by_nameZ message_type)�self� field_nameZ proto_type� r �$/usr/lib/python3.6/generator_test.py�testNestedMessageDescriptor@ s z)GeneratorTest.testNestedMessageDescriptorc C s� | j dtj� | j dtj� | j dtj� tj� }| j d|j� | j dtjj� | j d|j� | j dtjj� | j d|j� | j dtjj� d S )N� � � � � � ) r r Z FOREIGN_FOOZ FOREIGN_BARZ FOREIGN_BAZr �FOOZBARZBAZ)r �protor r r � testEnumsG s zGeneratorTest.testEnumsc s� t j� }dd� � � fdd�}| j||j�� | j|jdk� | j||j�� | j|jdk � | j� |j�� | j||j�� | j|jdk� | j||j�� | j|jdk � | j� |j�� | j d|j � d S )Nc S s | | kS )Nr )�valr r r �isnan\ s z5GeneratorTest.testExtremeDefaultValues.<locals>.isnanc s � | � o� | d �S )Nr r )r! )r" r r �isinf_ s z5GeneratorTest.testExtremeDefaultValues.<locals>.isinfr z? ? ?? ?? ??? ??/ ??-) r ZTestExtremeDefaultValues� assertTrueZ inf_doubleZneg_inf_doubleZ nan_doubleZ inf_floatZ neg_inf_floatZ nan_floatr Z cpp_trigraph)r �messager# r )r" r �testExtremeDefaultValuesW s z&GeneratorTest.testExtremeDefaultValuesc s> t jj}ddddd�� t� fdd�|jD ��}| j� |� d S )NFT)Zoptional_int32Zrepeated_int32r Z default_int32c s" g | ]}|j � kr|j |jf�qS r )�nameZhas_default_value)�.0�f)� expected_has_default_by_namer r � <listcomp>{ s z6GeneratorTest.testHasDefaultValues.<locals>.<listcomp>)r r r �dict�fieldsr )r �descZhas_default_by_namer )r* r �testHasDefaultValuesp s z"GeneratorTest.testHasDefaultValuesc C s. | j tjjtjj� | j tjjjtjj� d S )N)r r �optional_int32_extension�containing_type�TestAllExtensionsr � TestRequired�single)r r r r �'testContainingTypeBehaviorForExtensions� s z5GeneratorTest.testContainingTypeBehaviorForExtensionsc C s* | j tjjd � | j tjjjtjj� d S )N)r r r0 Zextension_scoper3 r4 r )r r r r �testExtensionScope� s z GeneratorTest.testExtensionScopec C sB | j tjj� | j tjjj� tjj}|jd }| j |j � d S )N�a)r$ r r0 Z is_extensionr3 r4 r r )r Zmessage_descriptorZnon_extension_descriptorr r r �testIsExtension� s z GeneratorTest.testIsExtensionc C s t j� }| j|jj� j� d S )N)r ZTestMessageSetr$ r � GetOptionsZmessage_set_wire_format)r r r r r � testOptions� s zGeneratorTest.testOptionsc C s* t j� }|jjd j� }| j|d k � d S )NZAnEnum)r Z TestMessageWithCustomOptionsr �enum_types_by_namer9 r$ )r r Z enum_optionsr r r � testMessageWithCustomOptions� s z*GeneratorTest.testMessageWithCustomOptionsc C s\ | j ttjjj�ttjjjtjjjtjjjg�� | j tj jjg � | j tjjjjg � d S )N) r �setr r r Z nested_typesr � OptionalGroup� RepeatedGroup�TestEmptyMessage)r r r r �testNestedTypes� s z GeneratorTest.testNestedTypesc C st | j tjjjd k� | j tjjjd k� | jtjjjjtjj� | jtjjjjtjj� | jtjjjjtjj� d S )N) r$ r r@ r r1 r r r r? )r r r r �testContainingType� s z GeneratorTest.testContainingTypec C s* | j tjjd k� | jtjjtjj� d S )N)r$ r � _FOREIGNENUMr1 r �_TESTALLTYPES_NESTEDENUMr r )r r r r �"testContainingTypeInEnumDescriptor� s z0GeneratorTest.testContainingTypeInEnumDescriptorc C s| | j tjjjjd� tjjj}| j |jjd� | j tjjjjd� | j tj jjd� | j tj jjd� | j tj jjd� d S )N�protobuf_unittestZprotobuf_unittest_import) r r r r �file�packager r Z ImportMessagerC rD Z _IMPORTENUM)r r. r r r � testPackage� s zGeneratorTest.testPackagec C sJ | j tjjjg � | j tjjjdtfg� | j tjjjdddtfg� d S ) Nr �* �+ �/ � i )rJ rK )rL rM )r r r r Zextension_rangesr2 � MAX_EXTENSIONZTestMultipleExtensionRanges)r r r r �testExtensionRange� s z GeneratorTest.testExtensionRangec C sr | j tjjd� | j tjjd� | jtjjd k� | j tjjtjg� | j tjjt jg� | j tjj t jg� d S )Nz google/protobuf/unittest.protorF ) r r r r' rH � assertFalseZ serialized_pbZ dependenciesr r Zpublic_dependencies)r r r r �testFileDescriptor� s z GeneratorTest.testFileDescriptorc C sR | j ttd�� | j ttd�� | j ttd�� ttd�rN| jttjtj�� d S )NZ TestMessager Ztest_extension� TestService)r$ �hasattrr rP � issubclassrR r ZService)r r r r �testNoGenericServices� s z#GeneratorTest.testNoGenericServicesc C s6 t j}| jt j|jt jj � | jt jj|jk� d S )N)r r r Z _TESTALLTYPESZmessage_types_by_namer' rP Z_TESTALLTYPES_NESTEDMESSAGE)r � file_typer r r �testMessageTypesByName� s z$GeneratorTest.testMessageTypesByNamec C s" t j}| jt j|jt jj � d S )N)r r r rC r; r' )r rV r r r �testEnumTypesByName� s z!GeneratorTest.testEnumTypesByNamec C s" t j}| jt j|jt jj � d S )N)r r r Zmy_extension_stringZextensions_by_namer' )r rV r r r �testExtensionsByName s z"GeneratorTest.testExtensionsByNamec C sD t j� }| jd|jj� tj� }| jd|j� | jtjtjk� d S )Nr ) r r r Z optional_public_import_message�er ZPublicImportMessager$ r )r Zall_type_protoZpublic_import_protor r r �testPublicImports s z GeneratorTest.testPublicImportsc C s\ t j� }| j|jt j d� | j|jt j d� | j|jt j d� | j|jt j d� d S )NZfooZbarZbazZqux)r ZTestBadIdentifiersr Z Extensionsr% Z descriptorZ reflectionr )r r% r r r �testBadIdentifiers s z GeneratorTest.testBadIdentifiersc C s� t jj}| jdt|j�� | jd|jd j� | jd|jd j� | j||jd j � | j|jd |j d � t ddddg�}| j|t dd � |jd j D ��� xDt |jj� �D ]2\}}||kr�| j|jd |j� q�| j|j� q�W d S ) Nr Z oneof_fieldr Z oneof_uint32Zoneof_nested_messageZ oneof_stringZ oneof_bytesc S s g | ] }|j �qS r )r' )r( Zfieldr r r r+ - s z+GeneratorTest.testOneof.<locals>.<listcomp>)r r r r �lenZoneofsr' �indexZassertIsr1 Zoneofs_by_namer= r- �listr �itemsZcontaining_oneofZ assertIsNone)r r. Z nested_namesr Z field_descr r r � testOneof" s zGeneratorTest.testOneofN)�__name__� __module__� __qualname__r r r& r/ r5 r6 r8 r: r< rA rB rE rI rO rQ rU rW rX rY r[ r\ ra r r r r r > s, r c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) � SymbolDatabaseRegistrationTestz?Checks that messages, enums and files are correctly registered.c C s� | j tjtj� jd�� | j tjjtj� jd�� | jt�� tj� jd� W d Q R X | j tjj tj� jd�� | j tjj tj� jd�� d S )Nz protobuf_unittest.TestAllTypesz,protobuf_unittest.TestAllTypes.NestedMessagez protobuf_unittest.NestedMessagez,protobuf_unittest.TestAllTypes.OptionalGroupz,protobuf_unittest.TestAllTypes.RepeatedGroup) r r r r �DefaultZ GetSymbolr Z assertRaises�KeyErrorr> r? )r r r r � testGetSymbol8 s" z,SymbolDatabaseRegistrationTest.testGetSymbolc C s8 | j dtj� jjd�j� | j dtj� jjd�j� d S )Nz protobuf_unittest.ForeignEnumz)protobuf_unittest.TestAllTypes.NestedEnum)r r rf �poolZFindEnumTypeByNameZ full_name)r r r r r K s z(SymbolDatabaseRegistrationTest.testEnumsc C s | j dtj� jjd�j� d S )Nz google/protobuf/unittest.proto)r r rf ri ZFindFileByNamer' )r r r r �testFindFileByNameU s z1SymbolDatabaseRegistrationTest.testFindFileByNameN)rb rc rd �__doc__rh r rj r r r r re 5 s re �__main__)rk � __author__Z unittest2Zunittest� ImportErrorZgoogle.protobuf.internalr Zgoogle.protobufr r r r r r r r r rN ZTestCaser re rb �mainr r r r �<module>' s* x&
Save File
Cancel