3
SLj% @ s d Z dZy ddlZW n ek
r0 ddlZY nX ddlmZ ddlmZ ddlm Z ddlm
Z
ddlm Z dd lm Z dd
lm
Z
G d d d ejZed
krej 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
Zd S ) MessageFactoryTestc C s( t jjtjj| _t jjtjj| _d S )N) r FileDescriptorProto
FromStringr
DESCRIPTORZ
serialized_pbfactory_test1_fdr factory_test2_fd)self r */usr/lib/python3.6/message_factory_test.pysetUp4 s zMessageFactoryTest.setUpc C s< | }d|_ d|_d|j_d|j_d|j_d|jj_d|j_|jj j
dddg |jj j d d|_d|_d|j_d|j _ d
|j j _ d |j _d |_|j j
d ddg |j j d |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_1part_2 )Z mandatoryZnested_factory_2_enumZnested_factory_2_messagevalueZfactory_1_messageZfactory_1_enumZnested_factory_1_enumZnested_factory_1_messageZ scalar_valueZ
list_valueextendappendZcircular_messageZgroupedaddr r ZloopZSerializeToStringr Factory2Messager assertEqual)r clsmsgZ
serializedZ convertedZ reserializedresultr r r _ExerciseDynamicClass: s>
z(MessageFactoryTest._ExerciseDynamicClassc C s~ t j }tj|}|j| j |j| j tj }|j |j
d}| j |t j
k | j| |j |j
d}| j||k d S )Nz/google.protobuf.python.internal.Factory2Message)r ZDescriptorDatabaser DescriptorPoolAddr
r r MessageFactoryZ GetPrototypeZFindMessageTypeByNameZ assertFalser r r#
assertTrue)r Zdbpoolfactoryr Zcls2r r r testGetPrototype\ s
z#MessageFactoryTest.testGetPrototypec C s xt dD ]v}| j| jj| jj tj| j| jg}| jt ddgj
t |j | j |d |d }| jt ddgj
t dd |j
jjj|j
D |d }|jjd}|jjd}d|j|< d |j|< | jd|j| | jd |j| | jd |jjd
tj d kr\| jt|jt|j | jt|jjd | jt|jjd
q | jd |jjd | jd |jjd
q W 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).0extr r r s z5MessageFactoryTest.testGetMessages..Ztest1Ztest2i!0 Zcppr )rangeassertInr
namer
dependencyr GetMessagesr' setissubsetkeysr# r filer( ZFindAllExtensionsZ
ExtensionsZ_FindExtensionByNamer Z_FindExtensionByNumberr ZTypelen assertRaises TypeError)r _ZmessagesZ factory_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.protoz google.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 DuplicatezExtensions "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 packageZ message_typer Zextension_rangestartendr% r3 r0 r2 r extensionZnumberZFieldDescriptorProtoZLABEL_OPTIONALZlabelZ type_nameZextendeer9 ExceptionstrZ 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 s +