3 ��h��@s�dZyddlmZWn ek r4ddlmZYnXyddlZWnek r^ddlZYnXddlmZddlm Z ddlm Z ddlmZGdd �d ej�Z ed kr�ej�dS)z(Tests for google.protobuf.proto_builder.�)�OrderedDictN)�descriptor_pb2)�descriptor_pool)� proto_builder)�text_formatc@s,eZdZdd�Zdd�Zdd�Zdd�Zd S) �ProtoBuilderTestcCs.tdtjjfdtjjfg�|_t|j�|_dS)N�foo�bar)rrZFieldDescriptorProtoZ TYPE_INT64ZTYPE_STRING�ordered_fields�dict�_fields)�self�r�(/usr/lib/python3.6/proto_builder_test.py�setUp3s zProtoBuilderTest.setUpcCs8tj|jdd�}|�}d|_d|_|jdtj|��dS)z&Test that we can create a proto class.z0net.proto2.python.public.proto_builder_test.Test)� full_namei90�asdfzbar: "asdf" foo: 12345 N)r�MakeSimpleProtoClassrrr �assertMultiLineEqualr�MessageToString)r � proto_cls�protorrr�testMakeSimpleProtoClass:sz)ProtoBuilderTest.testMakeSimpleProtoClasscCs8tj|jdd�}|�}d|_d|_|jdtj|��dS)zBTest that the field order is maintained when given an OrderedDict.z7net.proto2.python.public.proto_builder_test.OrderedTest)ri90rzfoo: 12345 bar: "asdf" N)rrr rr rrr)r rrrrr�testOrderedFieldsEsz"ProtoBuilderTest.testOrderedFieldscCs@tj�}tj|jd|d�}tj|jd|d�}|j|j|j�dS)z%Test that the DescriptorPool is used.z0net.proto2.python.public.proto_builder_test.Test)r�poolN)rZDescriptorPoolrrrZassertIsZ DESCRIPTOR)r rZ proto_cls1Z proto_cls2rrr�testMakeSameProtoClassTwicePsz,ProtoBuilderTest.testMakeSameProtoClassTwiceN)�__name__� __module__�__qualname__rrrrrrrrr1sr�__main__)�__doc__�collectionsr�ImportErrorZordereddictZ unittest2ZunittestZgoogle.protobufrrrrZTestCaserr�mainrrrr� s-