SLjc @ s d Z d Z d d l Z y d d l Z Wn e k
rG d d l Z n Xd d l m Z d d l m Z d d l m Z d d l m
Z
d d l m Z d d l m
Z
d d
l m Z d d l m Z d d l m Z d d
l m Z d Z d e j f d YZ d e f d YZ d e j f d YZ d e j f d YZ d e j f d YZ e d k re j n d S( s1 Unittest for google.protobuf.internal.descriptor.s# robinson@google.com (Will Robinson)iN( t unittest_custom_options_pb2( t unittest_import_pb2( t unittest_pb2( t descriptor_pb2( t api_implementation( t test_util( t
descriptor( t descriptor_pool( t symbol_database( t text_formats
name: 'TestEmptyMessage'
t DescriptorTestc B s e Z d Z d Z d Z d Z d Z d Z d Z d Z d Z
d Z d
Z d Z
d Z d
Z e j e j d k p e j d k d d Z RS( c C s t j d d d d } | j j d d } | j j d d d d d t j j d
t j j | j j d d } | j j d d d d
| j j d d d d | j j d d d d | j j d d | j
j d d } | j j d d d d d d } | j | _
| j
j | | j
j | j | _ | j j | j | _ | j j | j | _ | j j | j | _ | j j | j | _ d S( Nt names some/filename/some.protot packaget protobuf_unittestt
NestedMessaget bbt numberi t typet labelt ForeignEnumt FOREIGN_FOOi t FOREIGN_BARi t FOREIGN_BAZi t ResponseMessaget Servicet
CallMethodt
input_types .protobuf_unittest.NestedMessaget output_types" .protobuf_unittest.ResponseMessage( R t FileDescriptorProtot message_typet addt fieldt FieldDescriptorProtot
TYPE_INT32t LABEL_OPTIONALt enum_typet valuet servicet methodt GetDescriptorPoolt poolt Addt FindFileByNameR t my_filet message_types_by_namet
my_messaget enum_types_by_namet my_enumt services_by_namet
my_servicet methods_by_namet my_method( t selft
file_protot
message_protot
enum_protot
service_protot method_proto( ( sL /usr/lib/python2.7/site-packages/google/protobuf/internal/descriptor_test.pyt setUp> s:
c C s
t j j S( N( R t DefaultR( ( R4 ( ( sL /usr/lib/python2.7/site-packages/google/protobuf/internal/descriptor_test.pyR' b s c C s | j | j j d d d | j | j j d j d j | j j d d | j t | j j d d Wd QX| j t | j j d d Wd QX| j t | j j Wd QXd S( NR i R i t NoneEnum( t assertEqualR- t
EnumValueNameR. t values_by_numberR t assertRaisest KeyErrort TypeError( R4 ( ( sL /usr/lib/python2.7/site-packages/google/protobuf/internal/descriptor_test.pyt testEnumValueNamee s c C s$ | j | j | j j d j d S( Ni ( R= R/ t valuesR ( R4 ( ( sL /usr/lib/python2.7/site-packages/google/protobuf/internal/descriptor_test.pyt testEnumFixupst s c C s= | j | j | j j d j | j | j | j j d S( Ni ( R= R- t fieldst containing_typeR/ ( R4 ( ( sL /usr/lib/python2.7/site-packages/google/protobuf/internal/descriptor_test.pyt testContainingTypeFixupsw s c C s | j | j | j j d S( N( R= R1 R3 t containing_service( R4 ( ( sL /usr/lib/python2.7/site-packages/google/protobuf/internal/descriptor_test.pyt testContainingServiceFixups{ s c C s | j | j j t j | j | j j d j t j | j | j j t j | j | j j d j t j
| j | j j t j | j | j
j t j d S( Ni ( R= R/ t
GetOptionsR t EnumOptionsRD t EnumValueOptionsR- t MessageOptionsRF t FieldOptionsR3 t
MethodOptionsR1 t ServiceOptions( R4 ( ( sL /usr/lib/python2.7/site-packages/google/protobuf/internal/descriptor_test.pyt testGetOptions~ s
c C st t j } t j j } | j d } | j d } | j d } | j d } | j d } t j j } | j d } | j }
t j
} | j d |
j | | j } t j
}
| j d | j |
| j } t j } | j d | j | t j } | j d
| j | | j } t j } | j d | j | | j } t j } | j d | j | | j } t j } | j d
| j | | j } t j } | j d | j | | j } t j } | j t j | j | t j j } | j | j | j | j | j | j | j | j | j | j | j | j | j | j d S( Nt field1t AnOneoft AnEnumt ANENUM_VAL2t ANENUM_VAL1t Fool ` il -a i* iii{ l` ( R t
DESCRIPTORt TestMessageWithCustomOptionst fields_by_namet oneofs_by_nameR. t enum_values_by_namet TestServiceWithCustomOptionst FindMethodByNameRK t file_opt1R= t
Extensionst message_opt1t
field_opt1t
field_opt2t
oneof_opt1t enum_opt1t enum_value_opt1t service_opt1t method_opt1t METHODOPT1_VAL2t DummyMessageContainingEnumt
assertTruet has_optionst assertFalse( R4 t file_descriptort message_descriptort field_descriptort oneof_descriptort enum_descriptort enum_value_descriptort other_enum_value_descriptort service_descriptort method_descriptort file_optionsR` t message_optionsRb t
field_optionsRc Rd t
oneof_optionsRe t enum_optionsRf t enum_value_optionsRg t service_optionsRh t method_optionsRi ( ( sL /usr/lib/python2.7/site-packages/google/protobuf/internal/descriptor_test.pyt testSimpleCustomOptions s\
c
C s d } d } d } d } d } d } t j j } | j } | j t | j t j | j | | j t j | j | | j t j | j d | j t j
| j d | j t j | j | | j t j | j | | j t j
| j d | j t j | j d | j t j | j | | j t j | j | | j t j t j j } | j } | j t | j t j | j | | j t j | j | | j t j | j | | j t j
| j | | j t j | j | | j t j | j | | j t j
| j | | j t j | j | | j t j | j | | j t j | j | | j t j t j j } | j } | j d | j t j | j d | j t j d
| j d | j t j | j d | j t j | j d
| j t j t j } | j | j | j t j t j j } | j } | j d | j t j d
| j d | j t j t j j } | j } | j d | j t j d
| j d | j t j d S( Ni i i? i i i@ i ig$(@i gYB?s Hello, "World"s Hello Worldi i iifl l l l l l l l l l l l ( R t CustomOptionMinIntegerValuesRY RK R= t FalseRa t bool_optt int32_optt int64_optt
uint32_optt
uint64_optt
sint32_optt
sint64_optt fixed32_optt fixed64_optt sfixed32_optt sfixed64_optt CustomOptionMaxIntegerValuest Truet CustomOptionOtherValuest assertAlmostEqualt float_optt
double_optt
string_optt bytes_optRk t TEST_OPTION_ENUM_TYPE2t enum_optt SettingRealsFromPositiveIntst SettingRealsFromNegativeInts(
R4 t kint32mint kint64mint kint32maxt kint64maxt
kuint32maxt
kuint64maxRp Ry t
dummy_enum( ( sL /usr/lib/python2.7/site-packages/google/protobuf/internal/descriptor_test.pyt testDifferentCustomOptionTypes s c C sY t j j } | j } | j d | j t j j | j d | j t j j t j | j d | j t j j t j j
| j d | j t j j | j d | j t j j t j
| j d | j t j j j | j d | j t j j j t j | j d | j t j j j t j j
| j d | j t j j t j j | j d
| j t j j t j j t j | j d | j t j j t j j t j j
| j d | j t j j j j | j d
| j t j j j | j d | j t j j
| j d | j t j j j | j d | j t j j d S( Ni* iD il i i i i i i i iI i iA i i i ( R t VariousComplexOptionsRY RK R= Ra t complex_opt1t foot quuxt corget quxt complex_opt2t bazt graultt bart garplyt ComplexOptionType2t ComplexOptionType4t complex_opt4t waldot fredt complex_opt3t complexoptiontype5t plught complexopt6t xyzzy( R4 R t options( ( sL /usr/lib/python2.7/site-packages/google/protobuf/internal/descriptor_test.pyt testComplexExtensionOptions sX
c C s t j } t j j } | j d } t j j } | j d } t j j } | j d } | j j t j
} | j d | j | j d | j
| j d | j j
| j d | j j t j
j
| j d | j j t j j j
| j d | j j t j j
| j d
| j j t j j
| j d | j j t j j
| j d | j j t j j
| j d
| j j t j j
| j d | j j t j j
d S( Nt fieldnamet VALUEt Methodid t FileAnnotationt NestedFileAnnotationt FileExtensionAnnotationt EmbeddedMessageSetElementt MessageAnnotationt FieldAnnotationt EnumAnnotationt EnumValueAnnotationt ServiceAnnotationt MethodAnnotation( R RY t AggregateMessageR[ t
AggregateEnumt values_by_namet AggregateServiceR_ RK Ra t fileoptR= t it st subt filet msett AggregateMessageSetElementt message_set_extensiont msgoptt fieldoptt enumoptt
enumvaloptt
serviceoptt methodopt( R4 Ro Rp Rq Rs Rt Rv Rw Rx ( ( sL /usr/lib/python2.7/site-packages/google/protobuf/internal/descriptor_test.pyt testAggregateOptionsR sP
c C s t j j j } | j d | j j t j | j d } | j d | j j t j t j j } | j
d } | j d | j j t j | j d } | j d | j j t j
| j d } | j d | j j t j d S(
Ni t nested_fieldi t
NestedEnumi t NESTED_ENUM_VALUEi t nested_extensioni ( R t NestedOptionTypeR RY R= RK Ra Rb R[ Rc R. Rf R] Rg t extensions_by_nameRd ( R4 t nested_messageR t
outer_messaget nested_enumt nested_enum_valueR ( ( sL /usr/lib/python2.7/site-packages/google/protobuf/internal/descriptor_test.pyt testNestedOptions s
c C s6 | j | j j | j | j | j j | j d S( N( R= R/ R R+ R- ( R4 ( ( sL /usr/lib/python2.7/site-packages/google/protobuf/internal/descriptor_test.pyt testFileDescriptorReferences s c C s | j | j j d | j | j j d | j | j j | j | j | j j | j d | j j t j } | j j
| | j | j j | j | j t
j j t j d S( Ns some/filename/some.protoR
t proto2( R= R+ R R R( Rn Rm t syntaxR R t CopyToProtot
serialized_pbt SerializeToStringR RY R R; ( R4 R5 ( ( sL /usr/lib/python2.7/site-packages/google/protobuf/internal/descriptor_test.pyt testFileDescriptor s
t cppi sA Immutability of descriptors is only enforced in v2 implementationc C s} t j } t j j } | j d } | j d } | j d } | j t d | _ Wd QX| j t d | j d
( RQ R R R RY R RS ( R4 t TEST_NESTED_MESSAGE_ASCII( ( sL /usr/lib/python2.7/site-packages/google/protobuf/internal/descriptor_test.pyt testCopyToProto_NestedMessage s
c C s& d } | j t j j t j | d S( Ns
name: 'TestForeignNested'
field: <
name: 'foreign_nested'
number: 1
label: 1 # Optional
type: 11 # TYPE_MESSAGE
type_name: '.protobuf_unittest.TestAllTypes.NestedMessage'
>
( RQ R t TestForeignNestedRY R RS ( R4 t TEST_FOREIGN_NESTED_ASCII( ( sL /usr/lib/python2.7/site-packages/google/protobuf/internal/descriptor_test.pyt$ testCopyToProto_ForeignNestedMessage s
c C s& d } | j t j j t j | d S( Ns
name: 'ForeignEnum'
value: <
name: 'FOREIGN_FOO'
number: 4
>
value: <
name: 'FOREIGN_BAR'
number: 5
>
value: <
name: 'FOREIGN_BAZ'
number: 6
>
( RQ R R RY R t EnumDescriptorProto( R4 t TEST_FOREIGN_ENUM_ASCII( ( sL /usr/lib/python2.7/site-packages/google/protobuf/internal/descriptor_test.pyt testCopyToProto_ForeignEnum s
c C s& d } | j t j j t j | d S( Ns
name: 'TestDeprecatedFields'
field: <
name: 'deprecated_int32'
number: 1
label: 1 # Optional
type: 5 # TYPE_INT32
options: <
deprecated: true
>
>
( RQ R t TestDeprecatedFieldsRY R RS ( R4 t TEST_DEPRECATED_FIELDS_ASCII( ( sL /usr/lib/python2.7/site-packages/google/protobuf/internal/descriptor_test.pyt testCopyToProto_Options s
c C s& d } | j t j j t j | d S( Ns}
name: 'TestEmptyMessageWithExtensions'
extension_range: <
start: 1
end: 536870912
>
( RQ R t TestEmptyMessageWithExtensionsRY R RS ( R4 t( TEST_EMPTY_MESSAGE_WITH_EXTENSIONS_ASCII( ( sL /usr/lib/python2.7/site-packages/google/protobuf/internal/descriptor_test.pyt testCopyToProto_AllExtensions s
c C s& d } | j t j j t j | d S( Ns
name: 'TestMultipleExtensionRanges'
extension_range: <
start: 42
end: 43
>
extension_range: <
start: 4143
end: 4244
>
extension_range: <
start: 65536
end: 536870912
>
( RQ R R7 RY R RS ( R4 t* TEST_MESSAGE_WITH_SEVERAL_EXTENSIONS_ASCII( ( sL /usr/lib/python2.7/site-packages/google/protobuf/internal/descriptor_test.pyt! testCopyToProto_SeveralExtensions s
c C s+ d d d } | j t j t j | d S( Nsd
name: 'google/protobuf/unittest_import.proto'
package: 'protobuf_unittest_import'
dependency: 'google/protobuf/unittest_import_public.proto'
message_type: <
name: 'ImportMessage'
field: <
name: 'd'
number: 1
label: 1 # Optional
type: 5 # TYPE_INT32
>
>
s enum_type: <
name: 'ImportEnum'
value: <
name: 'IMPORT_FOO'
number: 7
>
value: <
name: 'IMPORT_BAR'
number: 8
>
value: <
name: 'IMPORT_BAZ'
number: 9
>
>
enum_type: <
name: 'ImportEnumForMap'
value: <
name: 'UNKNOWN'
number: 0
>
value: <
name: 'FOO'
number: 1
>
value: <
name: 'BAR'
number: 2
>
>
options: <
java_package: 'com.google.protobuf.test'
optimize_for: 1 # SPEED
sG
cc_enable_arenas: true
>
public_dependency: 0
( RQ R RY R R ( R4 t% UNITTEST_IMPORT_FILE_DESCRIPTOR_ASCII( ( sL /usr/lib/python2.7/site-packages/google/protobuf/internal/descriptor_test.pyt testCopyToProto_FileDescriptor s 0c C s& d } | j t j j t j | d S( NsM
name: 'TestService'
method: <
name: 'Foo'
input_type: '.protobuf_unittest.FooRequest'
output_type: '.protobuf_unittest.FooResponse'
>
method: <
name: 'Bar'
input_type: '.protobuf_unittest.BarRequest'
output_type: '.protobuf_unittest.BarResponse'
>
( RQ R RA RY R t ServiceDescriptorProto( R4 t TEST_SERVICE_ASCII( ( sL /usr/lib/python2.7/site-packages/google/protobuf/internal/descriptor_test.pyt! testCopyToProto_ServiceDescriptorN s
R s It is not implemented in python.c C s5 d } t j j j d } | j | t j | d S( Ns}
name: 'Foo'
input_type: '.protobuf_unittest.FooRequest'
output_type: '.protobuf_unittest.FooResponse'
RX ( R RA RY R_ RQ R t MethodDescriptorProto( R4 RJ Rw ( ( sL /usr/lib/python2.7/site-packages/google/protobuf/internal/descriptor_test.pyt testCopyToProto_MethodDescriptora s s! Pure python does not raise error.c C s t j } | j t t j j j | | j t t j j j | | j t t j j j | t j
} | j t t j j | d S( N( R R R@ RB R RR RY R R RA RS R ( R4 R5 t proto( ( sL /usr/lib/python2.7/site-packages/google/protobuf/internal/descriptor_test.pyt testCopyToProto_TypeErrorr s ( R R R RL RQ RU RW RZ R] R` Rc Re Rg Rj R R R R Rl Rn ( ( ( sL /usr/lib/python2.7/site-packages/google/protobuf/internal/descriptor_test.pyRF s$ ; t MakeDescriptorTestc B s5 e Z d Z d Z d Z d Z d Z RS( c C sr t j } d | _ | j j } | j | _ | j j } d | _ | j j } d | _ | j j } d | _ d | _ | j j } d | _ d | _ t
j j | _
t
j j | _ | j j } d | _ d | _ t
j j | _
t
j j | _ d | _ | j j } d | _ d
| _ t
j j | _
t
j j | _ d | _ t
j | } | j | j d j t
j j | j | j d j t
j j | j | j d j j | | j | j d j d j d
| j | j d j d j | j d j d | j | j | j | j d j t j d k rn| j! t" t# | j d _ Wd QXn d S( Nt Foo2t SubR= t BARi i t uint64_fieldi t nested_message_fieldt bar_fields Foo2.Sub.FOOi s Foo2.Sub.bar_fieldR ($ R R R R R t nested_typeR# R$ R R R t FieldDescriptort LABEL_REQUIREDR t TYPE_UINT64R t TYPE_MESSAGEt type_namet TYPE_ENUMt MakeDescriptorR= RF t cpp_typet CPPTYPE_UINT64t CPPTYPE_MESSAGERG t nested_typesR R Rn Rm R R R@ R R ( R4 t file_descriptor_protoR Rv R# t
enum_type_valR t
enum_fieldt result( ( sL /usr/lib/python2.7/site-packages/google/protobuf/internal/descriptor_test.pyt" testMakeDescriptorWithNestedFields sX
c C s# t j } d | _ | j j } | j | _ | j j } d | _ | j j } d | _ d | _ | j j } d | _ d | _ t j
j | _ t j
j
| _ | j j } d | _ d | _ t j
j | _ t j
j | _ d | _ t j | } | j | j d
j t j
j d S( NRX R= Rr i i Rs i Ru s Foo.FOOi ( R R R R R R# R$ R R R Rw Rx R Ry R R| R{ R} R= RF R~ R ( R4 R R R# R R R R ( ( sL /usr/lib/python2.7/site-packages/google/protobuf/internal/descriptor_test.pyt& testMakeDescriptorWithUnsignedIntField s. c C sa t j } t j } | j j | t j | } | j } | j d | j
t j j d S( Nie (
R RS R R RY R R R} RK R= Ra R R ( R4 t descriptor_protot aggregate_messaget reformed_descriptorR ( ( sL /usr/lib/python2.7/site-packages/google/protobuf/internal/descriptor_test.pyt testMakeDescriptorWithOptions s c C s t j } d | _ d d d d d g } d d d d d g } xC t t | D]/ } | j j } | d | _ | | | _ qR Wt j | } x8 t t | D]$ } | j
| j | j | | q Wd S( Nt Bart foo_foot FooBart fooBazt fooFoot foobart fooBari (
R RS R t rangeR R R R R R} R= RF R ( R4 R t namest camelcase_namesR R R ( ( sL /usr/lib/python2.7/site-packages/google/protobuf/internal/descriptor_test.pyt testCamelcaseName s
c C s t j } d | _ d d d d d d g } d d d d d d g } xC t t | D]/ } | j j } | d
| _ | | | _ qX Wd | _ t j
| } x8 t t | D]$ } | j | j | j | | q Wd S( Nt TestJsonNamet
field_namet fieldNamet FieldNamet _field_namet
FIELD_NAMEt json_namet FIELDNAMEs @typei (
R RS R R R R R R R R R} R= RF ( R4 R R t
json_namesR R R ( ( sL /usr/lib/python2.7/site-packages/google/protobuf/internal/descriptor_test.pyt testJsonName s
( R R R R R R R ( ( ( sL /usr/lib/python2.7/site-packages/google/protobuf/internal/descriptor_test.pyRo s
/
t __main__( R t
__author__R+ t unittest2R t ImportErrort google.protobufR R R R t google.protobuf.internalR R R R R R RT t TestCaseR
R R RF Ro R t main( ( ( sL /usr/lib/python2.7/site-packages/google/protobuf/internal/descriptor_test.pyt s2
v