23s2s

✏️ Editing: message_factory_test.cpython-36.opt-1.pyc

���h�%������������������@���s����d�Z�dZyddlZW�n�ek
r0���ddlZY�nX�ddlmZ�ddlmZ�ddlm	Z	�ddlm
Z
�ddlmZ�dd	lmZ�dd
lm
Z
�G�dd��dej�Zed
kr�ej���dS�)z*Tests for google.protobuf.message_factory.z"matthewtoia@google.com (Matt Toia)�����N)�descriptor_pb2)�api_implementation)�factory_test1_pb2)�factory_test2_pb2)�descriptor_database)�descriptor_pool)�message_factoryc���������������@���s4���e�Zd�Zdd��Zdd��Zdd��Zdd��Zd	d
��ZdS�)�MessageFactoryTestc�������������C���s(���t�jjtjj�|�_t�jjtjj�|�_d�S�)N)	r����FileDescriptorProto�
FromStringr����
DESCRIPTORZ
serialized_pb�factory_test1_fdr����factory_test2_fd)�self��r����*/usr/lib/python3.6/message_factory_test.py�setUp4���s����zMessageFactoryTest.setUpc�������������C���s<��|��}d|_�d|_d|j_d|j_d|j_d|jj_d|j_|jj	j
dddg��|jj	jd	��d|_d|_d|j_d|j_�d
|jj_�d|j_d|_|j	j
d	ddg��|j	jd��|j
j���d
|j
d�_d|j
d�_|j
jddd��d
|jj_�d|jjjj_�|j��}tjj|�}|j��}|�j||��|j|�}|�j||��d�S�)N�*���r���znested message value��������ZoneZtwoZthreeZfour����zone deepz	zero deepZhelloZworldZtestingZ123)�part_1�part_2����)Z	mandatoryZnested_factory_2_enumZnested_factory_2_message�valueZfactory_1_messageZfactory_1_enumZnested_factory_1_enumZnested_factory_1_messageZscalar_valueZ
list_value�extend�appendZcircular_messageZgrouped�addr���r���ZloopZSerializeToStringr����Factory2Messager����assertEqual)r����cls�msgZ
serializedZ	convertedZreserialized�resultr���r���r����_ExerciseDynamicClass:���s>����

z(MessageFactoryTest._ExerciseDynamicClassc�������������C���s~���t�j��}tj|�}|j|�j��|j|�j��tj��}|j	|j
d��}|�j|tj
k��|�j|��|j	|j
d��}|�j||k��d�S�)Nz/google.protobuf.python.internal.Factory2Message)r���ZDescriptorDatabaser����DescriptorPool�Addr
���r���r����MessageFactoryZGetPrototypeZFindMessageTypeByNameZassertFalser���r���r#����
assertTrue)r���Zdb�pool�factoryr ���Zcls2r���r���r����testGetPrototype\���s����

z#MessageFactoryTest.testGetPrototypec�������������C���s����x�t�d�D��]v}|�j|�jj|�jj��tj|�j|�jg�}|�jt	ddg�j
t	|j������|�j|d���|d�}|�jt	ddg�j
t	dd��|j
jjj|j
�D������|d���}|jjd�}|jjd�}d|j|<�d	|j|<�|�jd|j|���|�jd	|j|���|�jd�|jjd
���tj��dk�r\|�jt|j�t|j���|�jt|jjd��|�jt|jjd
��q|�jd�|jjd���|�jd�|jjd
���qW�d�S�)Nr���z/google.protobuf.python.internal.Factory2Messagez/google.protobuf.python.internal.Factory1Messagez>google.protobuf.python.internal.Factory2Message.one_more_fieldz-google.protobuf.python.internal.another_fieldc�������������s���s���|�]}|j�V��qd�S�)N)Z	full_name)�.0�extr���r���r����	<genexpr>����s���z5MessageFactoryTest.testGetMessages.<locals>.<genexpr>Ztest1Ztest2i!0��Zcppr�����)�range�assertInr
����namer����
dependencyr����GetMessagesr'����set�issubset�keysr#���r����filer(���ZFindAllExtensionsZ
ExtensionsZ_FindExtensionByNamer���Z_FindExtensionByNumberr���ZType�len�assertRaises�	TypeError)r����_ZmessagesZfactory_msg1Zmsg1Zext1Zext2r���r���r����testGetMessagesj���sN����

z"MessageFactoryTest.testGetMessagesc�������	������C���s���t�j��}tj|d�}tj��}d|_d|_|jj	��}d|_|j
j	��}d|_d|_|j
|��|j|jg�}|�jd|��tj��}d|_d|_|jjd��|jj	��}d	|_|jj	��}d
|_d|_tjj|_d	|_d|_|j
|��|j|jg�}|�jd|��tj��}d
|_d|_|jjd��|jj	��}d|_|jj	��}d
|_d|_tjj|_d|_d|_|j
|��|�jt��}|j|jg��W�d�Q�R�X�|�jt|j�ddg��d�S�)N)r(���z(google/protobuf/internal/container.protozgoogle.protobuf.python.internal�	Containerr����
���z)google.protobuf.python.internal.Containerz(google/protobuf/internal/extension.protoZ	ExtensionZextension_fieldr���z)google.protobuf.python.internal.Extensionz(google/protobuf/internal/duplicate.protoZ	Duplicatez�Extensions "google.protobuf.python.internal.Duplicate.extension_field" and "google.protobuf.python.internal.Extension.extension_field" both try to extend message type "google.protobuf.python.internal.Container" with field number 2.z!Double registration of Extensions)r���r$���r���r&���r���r
���r1����packageZmessage_typer���Zextension_range�start�endr%���r3���r0���r2���r����	extensionZnumberZFieldDescriptorProtoZLABEL_OPTIONALZlabelZ	type_nameZextendeer9����	Exception�strZ	exception)	r���r(���r)����fr!����rngZmsgsr,����cmr���r���r����testDuplicateExtensionNumber����s\����

z/MessageFactoryTest.testDuplicateExtensionNumberN)�__name__�
__module__�__qualname__r���r#���r*���r<���rH���r���r���r���r���r	���2���s
���"4r	����__main__)�__doc__�
__author__Z	unittest2Zunittest�ImportErrorZgoogle.protobufr���Zgoogle.protobuf.internalr���r���r���r���r���r���ZTestCaser	���rI����mainr���r���r���r����<module> ���s ����+

Sid Gifari From Gifari Industries - BD Cyber Security Team