3
SLj
@ s d Z dZddlZddlZddlZddlZy ddlZW n ek
rP ddlZY nX ddl m
Z
ddl m Z ddl m
Z
ddl mZ ddl mZ dd l mZ dd
l mZ ddl m Z dd l mZ dd l mZ dd
l mZ ddl mZ G dd dejZG dd dejZe
jeeG dd deZG dd deZ G dd deZ G dd dejZ G dd d ejZ e
jeeG d d d eZ e!d krej" dS ) z%Test for google.protobuf.text_format.z kenton@google.com (Kenton Varda) N)_parameterized) any_test_pb2)map_unittest_pb2)unittest_mset_pb2) unittest_pb2)unittest_proto3_arena_pb2)api_implementation)message_set_extensions_pb2) test_util)descriptor_pool) text_formatc @ s e Zd Zdd ZdS )SimpleTextFormatTestsc C s$ x t jD ]}| jdt| qW d S )N )r Z_QUOTES assertEquallen)selfZquote r &/usr/lib/python3.6/text_format_test.py testQuoteMarksAreSingleCharsF s z2SimpleTextFormatTests.testQuoteMarksAreSingleCharsN)__name__
__module__ __qualname__r r r r r r
@ s r
c @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
TextFormatBasec C s4 t j| }ttkr |j S dd |D S Q R X d S )Nc S s g | ]}|j d qS )zutf-8)decode).0Z golden_liner r r
Q s z-TextFormatBase.ReadGolden..)r
Z
GoldenFilestrbytes readlines)r golden_filenamefr r r
ReadGoldenN s zTextFormatBase.ReadGoldenc C s | j |}| j|dj| d S )N )r! ZassertMultiLineEqualjoin)r textr Z golden_linesr r r CompareToGoldenFileS s
z"TextFormatBase.CompareToGoldenFilec C s | j || d S )N)r )r r$ golden_textr r r CompareToGoldenTextW s z"TextFormatBase.CompareToGoldenTextc C s> |j ddj ddj ddj dd}tjdtjjd|}|S )Nze+0ze+ze-0ze-z\.0$r" )replacerecompile MULTILINEsub)r r$ r r r RemoveRedundantZerosZ s z#TextFormatBase.RemoveRedundantZerosN)r r r r! r% r' r- r r r r r L s r 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-d. Zd/d0 Zd1d2 Zd3d4 Z d5d6 Z d7d8 Z d9d: Z d;d< Z d=S )>TextFormatTestc C sx |j }|jjd |jjd |jjd |jjd |jjd |jjd |jjd | j| jtj |d d S )
Nl l gw/^@gtQFքDg6.UnicodeSubN)r r r r r r r
UnicodeSub~ s r9 u üꜟz(repeated_string: "\303\274\352\234\237"
)sixZ text_typer/ r4 r1 r' r r5 )r r6 r9 r7 r r r testPrintExoticUnicodeSubclass| s z-TextFormatTest.testPrintExoticUnicodeSubclassc C s2 |j }|jj }d|_| jtj|ddd d S )N* T) as_one_linez"repeated_nested_message { bb: 42 })r/ repeated_nested_messageaddbbr' r r5 )r r6 r7 msgr r r testPrintNestedMessageAsOneLine s
z.TextFormatTest.testPrintNestedMessageAsOneLinec C s^ |j }|jjd |jjd |jjd |jjd |jjd | jtj|ddd d S )Nr ZGoogleZZurichT)r= zirepeated_int32: 1 repeated_int32: 1 repeated_int32: 3 repeated_string: "Google" repeated_string: "Zurich")r/ Zrepeated_int32r1 r4 r' r r5 )r r6 r7 r r r testPrintRepeatedFieldsAsOneLine s z/TextFormatTest.testPrintRepeatedFieldsAsOneLinec C s( |j }d|_| jtj|ddd d S )Nz
a
new
lineT)r= z optional_string: "a\nnew\nline")r/ optional_stringr' r r5 )r r6 r7 r r r 'testPrintNestedNewLineInStringAsOneLine s
z6TextFormatTest.testPrintNestedNewLineInStringAsOneLinec C s| |j }|jjd |jjd |jjd |jjd |jjd |jjd |jjd | j| jtj |dd d
d S ) Nl l gw/^@gtQFքDg6i
g?zoptional_float: 1.25z%optional_double: -3.45678901234568e-6zrepeated_float: -5642zrepeated_double: 7.89e-5z.15g) float_formatz&payload {{
{0}
{1}
{2}
{3}
}}
T)r= rO z payload {{ {0} {1} {2} {3} }}g/_e̾i) ZNestedTestAllTypesZpayloadoptional_floatZoptional_doubleZrepeated_floatr1 r3 r r5 r' r- format)r r6 r7 Zformatted_fieldsZ text_messager r r testPrintFloatFormat s& z#TextFormatTest.testPrintFloatFormatc C s" |j }d|_| jdt| d S )N{ zc: 123
)ZForeignMessagecr r )r r6 r7 r r r testMessageToString s z"TextFormatTest.testMessageToStringc C s |j }|jjd }|j}tjd}tj||| | jd|j |j tjd}tj
|}|j|| | jd|j |j d S )NrP Fzoptional_float: 0.0
) r/
DESCRIPTORfields_by_namerP r
TextWriterZ
PrintFieldr getvalueclose_Printer)r r6 r7 fieldvalueoutprinterr r r testPrintField s
z TextFormatTest.testPrintFieldc C s |j }|jjd }|j}tjd}tj||| | jd|j |j tjd}tj
|}|j|| | jd|j |j d S )NrP Fz0.0) r/ rV rW rP r rX ZPrintFieldValuer rY rZ r[ )r r6 r7 r\ r] r^ r_ r r r testPrintFieldValue s
z"TextFormatTest.testPrintFieldValuec C sT |j }tj| tj|}|j }tj|| | j|| |tkrPtj| | d S )N) r/ r
SetAllFieldsr r5 rI r r ExpectAllFieldsSet)r r6 r7
ascii_textrK r r r testParseAllFields" s
z!TextFormatTest.testParseAllFieldsc C s |j }tj| tj|}|jd}|j }tj|| | j|| |tkrZtj | | |j
tj || | j|| |tkrtj | | t j
r|j }d}tj || | j|jd |j
tj|| | j|jd d S )Nzutf-8u optional_string: "café"u café)r/ r
rb r r5 encoderI r r rc ClearMerger: ZPY2rE )r r6 r7 rd rK Zmsg2r$ r r r testParseAndMergeUtf8- s,
z$TextFormatTest.testParseAndMergeUtf8c C s |j }d}tj|| | jd|jd | jd|jd | jd|jd | jd|jd | jd|jd | jd
|jd | jd |jd | jd |jd | jd
|jd d S )NuU repeated_int64: -9223372036854775808
repeated_uint64: 18446744073709551615
repeated_double: 123.456
repeated_double: 1.23e+22
repeated_double: 1.23e-18
repeated_string:
"\000\001\007\010\014\n\r\t\013\\\'\""
repeated_string: "foo" 'corge' "grault"
repeated_string: "\303\274\352\234\237"
repeated_string: "\xc3\xbc"
repeated_string: "ü"
l r l gw/^@gtQFքDr g6< z
\'"Zfoocorgegraultu üꜟ ürC l )r/ r rI r r0 r2 r3 r4 )r r6 r7 r$ r r r testParseExoticH s z TextFormatTest.testParseExoticc C sv |j }d}tj|| | jd|jd | jd|jd | jd|jd | jd|jd | jd |jd d S )
Nzorepeated_int64: 100;
repeated_int64: 200;
repeated_int64: 300,
repeated_string: "one",
repeated_string: "two";
d r r i, rj onetwo)r/ r rI r r0 r4 )r r6 r7 r$ r r r testParseTrailingCommasa s z&TextFormatTest.testParseTrailingCommasc C sv |j }d}tj|| | jd|jd | jd|jd | jd|jd | jd|jd | jd |jd d S )
NzRrepeated_int64: [100, 200];
repeated_int64: 300,
repeated_string: ["one", "two"];
rm r rn r i, rj ro rp )r/ r rI r r0 r4 )r r6 r7 r$ r r r "testParseRepeatedScalarShortFormatp s z1TextFormatTest.testParseRepeatedScalarShortFormatc C sl |j }d}tj|| | jd|jd j | jd|jd j | jd|jd j | jd|jd j d S )
Nzyrepeated_nested_message: [{bb: 100}, {bb: 200}],
repeated_nested_message: {bb: 300}
repeated_nested_message [{bb: 400}];
rm r rn r i, rj i rC )r/ r rI r r> r@ )r r6 r7 r$ r r r #testParseRepeatedMessageShortFormat} s z2TextFormatTest.testParseRepeatedMessageShortFormatc C s, |j }d}tj|| | j|j | d S )Nr" )r/ r rI r )r r6 r7 r$ r r r testParseEmptyText s z!TextFormatTest.testParseEmptyTextc
C sX |j }d}| jtj}tj|| W d Q R X | j|jj d | j|jj d d S )Nzrepeated_string: "\xc3\xc3"r ) r/ assertRaisesr
ParseErrorrI r exceptionZGetLineZ GetColumn)r r6 r7 r$ er r r testParseInvalidUtf8 s z#TextFormatTest.testParseInvalidUtf8c C s( |j }d}tj| tjdtj|| d S )Nfooz?1:1 : Message type "\w+.TestAllTypes" has no field named "foo".)r/ r: assertRaisesRegexr rw rI )r r6 r7 r$ r r r testParseSingleWord s
z"TextFormatTest.testParseSingleWordc C s( |j }d}tj| tjdtj|| d S )Nzunknown_field: 8
zI1:1 : Message type "\w+.TestAllTypes" has no field named "unknown_field".)r/ r: r| r rw rI )r r6 r7 r$ r r r testParseUnknownField s
z$TextFormatTest.testParseUnknownFieldc C s( |j }d}tj| tjdtj|| d S )Nzoptional_nested_enum: BARRzG1:23 : Enum type "\w+.TestAllTypes.NestedEnum" has no value named BARR.)r/ r: r| r rw rI )r r6 r7 r$ r r r testParseBadEnumValue s
z$TextFormatTest.testParseBadEnumValuec C s( |j }d}tj| tjdtj|| d S )Nzoptional_int32: borkz#1:17 : Couldn't parse integer: bork)r/ r: r| r rw rI )r r6 r7 r$ r r r testParseBadIntValue s
z#TextFormatTest.testParseBadIntValuec C s |j }d}tj|| d}| jd|jd | j|d | d |jd | j|d | d |jd
| j|| d | | d |jd | j|| d | | d |jd | j|d
|jd d S )Na repeated_string: "\xf\x62"
repeated_string: "\\xf\\x62"
repeated_string: "\\\xf\\\x62"
repeated_string: "\\\\xf\\\\x62"
repeated_string: "\\\\\xf\\\\\x62"
repeated_string: "\x5cx20"\zbr ZxfZx62r brj rC Zx20 )r/ r rI r r4 )r r6 r7 r$ SLASHr r r testParseStringFieldUnescape s z+TextFormatTest.testParseStringFieldUnescapec C s6 |j }d}tj||}| j|| | jd|j d S )Nz%optional_int32: 42 optional_int32: 67C )r/ r rh rJ r optional_int32)r r6 r7 r$ rL r r r testMergeDuplicateScalars s
z(TextFormatTest.testMergeDuplicateScalarsc C s: |j }d}tj||}| j||k | jd|jj d S )NzCoptional_nested_message { bb: 1 } optional_nested_message { bb: 2 }rj )r/ r rh
assertTruer optional_nested_messager@ )r r6 r7 r$ rL r r r &testMergeDuplicateNestedMessageScalars s
z5TextFormatTest.testMergeDuplicateNestedMessageScalarsc C s> |j }d|_|j }tjtj|| | jd|jd d S )N oneof_uint32 oneof_field)r/ r r rI r5 r
WhichOneof)r r6 mm2r r r testParseOneof s
z TextFormatTest.testParseOneofc C s8 dj ddg}|j }tj|| | jd|jd d S )N
zoneof_uint32: 11zoneof_string: "foo"Z oneof_stringr )r# r/ r rh r r )r r6 m_stringr r r r testMergeMultipleOneof s z%TextFormatTest.testMergeMultipleOneofc
C s@ dj ddg}|j }| jtjd tj|| W d Q R X d S )Nr zoneof_uint32: 11zoneof_string: "foo"z is specified along with field )r# r/ ZassertRaisesRegexpr rw rI )r r6 r r r r r testParseMultipleOneof s
z%TextFormatTest.testParseMultipleOneofN)!r r r r8 r; rB rD rF rG rM rN rR rU r` ra re ri rl rq rr rs rt rz r} r~ r r r r r r r r r r r r r. e s<
!
r. c @ sL e Zd Zdd Zdd Zdd Zdd Zd d
Zd d Zd
d Z dd Z
dS ) OnlyWorksWithProto2RightNowTestsc C s2 t j }tj| | j| jtj|ddd d S )NT)pointy_bracketsz*text_format_unittest_data_pointy_oneof.txt)r r/ r
rb r% r- r r5 )r r7 r r r testPrintAllFieldsPointy s
z9OnlyWorksWithProto2RightNowTests.testPrintAllFieldsPointyc C sR dj | jd}tj }tj||}| j|| tj }tj| | j || d S )Nr z/text_format_unittest_data_oneof_implemented.txt)
r# r! r r/ r rI rJ r
rb r )r r& rK rL r7 r r r testParseGolden s
z0OnlyWorksWithProto2RightNowTests.testParseGoldenc C s. t j }tj| | j| jtj|d d S )Nz/text_format_unittest_data_oneof_implemented.txt)r r/ r
rb r% r- r r5 )r r7 r r r testPrintAllFields s
z3OnlyWorksWithProto2RightNowTests.testPrintAllFieldsc C sb t j }d|_d|_d|_d|j_d|j_| j| j t
j |ddd | j| j t
j |d d S )
NZ115e o r r T)Zuse_index_orderzWmy_string: "115"
my_int: 101
my_float: 111
optional_nested_message {
oo: 0
bb: 1
}
zWmy_int: 101
my_string: "115"
my_float: 111
optional_nested_message {
bb: 1
oo: 0
}
) r ZTestFieldOrderingsZ my_stringZmy_intZmy_floatr Zoor@ r' r- r r5 )r r7 r r r testPrintInIndexOrder s
z6OnlyWorksWithProto2RightNowTests.testPrintInIndexOrderc C sL | j d}tj }tj||}| j|| tj }tj| | j|| d S )Nz/text_format_unittest_data_oneof_implemented.txt) r! r r/ r Z
MergeLinesrJ r
rb r )r openedrK rL r7 r r r testMergeLinesGolden s
z5OnlyWorksWithProto2RightNowTests.testMergeLinesGoldenc C sL | j d}tj }tj||}| j|| tj }tj| | j|| d S )Nz/text_format_unittest_data_oneof_implemented.txt) r! r r/ r Z
ParseLinesrJ r
rb r )r r rK rL r7 r r r testParseLinesGolden' s
z5OnlyWorksWithProto2RightNowTests.testParseLinesGoldenc C s\ t j }d |jd < d|jd< d|jd< d|jd< d|jd< d|jd _| j t
j |d
d S )Ni rS rj " ! 123abcr r aH map_int32_int32 {
key: -123
value: -456
}
map_int64_int64 {
key: -8589934592
value: -17179869184
}
map_uint32_uint32 {
key: 123
value: 456
}
map_uint64_uint64 {
key: 8589934592
value: 17179869184
}
map_string_string {
key: "abc"
value: "123"
}
map_int32_foreign_message {
key: 111
value {
c: 5
}
}
i8il l l l l l ) r TestMapmap_int32_int32map_int64_int64map_uint32_uint32map_uint64_uint64map_string_stringmap_int32_foreign_messagerT r' r r5 )r r7 r r r testPrintMap1 s
z-OnlyWorksWithProto2RightNowTests.testPrintMapc C s| t j }x tjdd D ]}d|j|< qW x$ttjdd D ]}d|j|< q>W djdd tjD }| jtj || d S )N
Zdummyr r" c s s | ]}d |f V qdS )z3map_string_string {
key: "%c"
value: "dummy"
}
Nr )r letterr r r _ s zKOnlyWorksWithProto2RightNowTests.testMapOrderEnforcement..)
r r stringZascii_uppercaser reversedr# r' r r5 )r r7 r goldenr r r testMapOrderEnforcementY s
z8OnlyWorksWithProto2RightNowTests.testMapOrderEnforcementN) r r r r r r r r r r r r r r r r s
(r 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+S ), Proto2Testsc C sv t j }t jj}t jj}d|jj| _d|jj| _| j t
j |d t j
}t j}d|j| _| j t
j |d d S )N r{ zmessage_set {
[protobuf_unittest.TestMessageSetExtension1] {
i: 23
}
[protobuf_unittest.TestMessageSetExtension2] {
str: "foo"
}
}
ZbarzF[google.protobuf.internal.TestMessageSetExtension3] {
text: "bar"
}
)r TestMessageSetContainerTestMessageSetExtension1message_set_extensionTestMessageSetExtension2 message_set
Extensionsir r' r r5 r ZTestMessageSetZmessage_set_extension3r$ )r r7 ext1ext2Zextr r r testPrintMessageSetv s z Proto2Tests.testPrintMessageSetc C sj t jd}tj }tjj}tjj}d|jj| _ d|jj| _
t j ||dd | j |j
d |j d S )NFr r{ T)Zuse_field_numberz?1 {
1545008 {
15: 23
}
1547769 {
25: "foo"
}
}
)r rX r r r r r r r r r Z PrintMessager' rY rZ )r r^ r7 r r r r r testPrintMessageSetByFieldNumber s
z,Proto2Tests.testPrintMessageSetByFieldNumberc C sN t j }t jj}t jj}d|jj| _d|jj| _| j t
j |ddd d S )Nr r{ T)r= zmessage_set { [protobuf_unittest.TestMessageSetExtension1] { i: 23 } [protobuf_unittest.TestMessageSetExtension2] { str: "foo" } }) r r r r r r r r r r' r r5 )r r7 r r r r r testPrintMessageSetAsOneLine s z(Proto2Tests.testPrintMessageSetAsOneLinec C s t j }d}tj|| | jd|jd | jd|jd tj }d}tj|| tjj }tj
j }| jd|j j | j
| jd|j j | j d S )Nz&repeated_uint64: 1
repeated_uint64: 2
r r rj zmessage_set {
[protobuf_unittest.TestMessageSetExtension1] {
i: 23
}
[protobuf_unittest.TestMessageSetExtension2] {
str: "foo"
}
}
r r{ )r r/ r rI r r2 r r r r r r r r r )r r7 r$ r r r r r testParseMessageSet s z Proto2Tests.testParseMessageSetc C s> t j }d}tj||tj d | jtj|tj d| d S )Nzvalue {
[type.googleapis.com/google.protobuf.internal.TestAny] {
[google.protobuf.internal.TestAnyExtension1.extension1] {
i: 10
}
}
}
)r )test_extend_anyTestAnyr rh r Defaultr' r5 )r r7 r$ r r r testExtensionInsideAnyMessage s z)Proto2Tests.testExtensionInsideAnyMessagec C s t j }d}tj||dd | jd|jd | jd|jd tj }d}tj||dd tjj }tj
j }| jd|j j | j
| jd |j j | j t j }d
}tj| tjd tj|| d }tj| tjd
tj||dd d S )Nz34: 1
repeated_uint64: 2
T)Zallow_field_numberr r rj z?1 {
1545008 {
15: 23
}
1547769 {
25: "foo"
}
}
r r{ z34:1
z>1:1 : Message type "\w+.TestAllTypes" has no field named "34".z1234:1
z@1:1 : Message type "\w+.TestAllTypes" has no field named "1234".)r r/ r rI r r2 r r r r r r r r r r: r| rw )r r7 r$ r r r r r testParseMessageByFieldNumber s4
z)Proto2Tests.testParseMessageByFieldNumberc C s. t j }tj| | j| jtj|d d S )Nz(text_format_unittest_extensions_data.txt)r TestAllExtensionsr
SetAllExtensionsr% r- r r5 )r r7 r r r testPrintAllExtensions s
z"Proto2Tests.testPrintAllExtensionsc C s2 t j }tj| | j| jtj|ddd d S )NT)r z/text_format_unittest_extensions_data_pointy.txt)r r r
r r% r- r r5 )r r7 r r r testPrintAllExtensionsPointy s
z(Proto2Tests.testPrintAllExtensionsPointyc C sF dj | jd}tj }tj|| tj }tj| | j|| d S )Nr z(text_format_unittest_extensions_data.txt) r# r! r r r rI r
r r )r r& rK r7 r r r testParseGoldenExtensions s
z%Proto2Tests.testParseGoldenExtensionsc C s@ t j }tj| tj|}t j }tj|| | j|| d S )N)r r r
r r r5 rI r )r r7 rd rK r r r testParseAllExtensions s
z"Proto2Tests.testParseAllExtensionsc C sV t j }d}tj||dd d}| jtj|| t j }d}tj| tjdtj||dd t j }d}tj| tjdtj||dd t j }d }tj| tjdtj||dd t j }d
}tj| tjd tj||dd t j }d }tj| tjd
tj||dd t j }d}tj||dd t j j
}t j j
}| j d|j
j| j | j d|j
j| j d S )Nu message_set {
[unknown_extension] {
i: 23
bin: "à" [nested_unknown_ext]: {
i: 23
x: x
test: "test_string"
floaty_float: -0.315
num: -inf
multiline_str: "abc"
"def"
"xyz."
[nested_unknown_ext.ext]: <
i: 23
i: 24
pointfloat: .3
test: "test_string"
floaty_float: -0.315
num: -inf
long_string: "test" "test2"
>
}
}
[unknown_extension]: 5
}
T)Zallow_unknown_extensionzmessage_set {
}
z3message_set {
[unknown_extension] {
i:
}
}
zInvalid field value: }zGmessage_set {
[unknown_extension] {
str: "malformed string
}
}
zInvalid field value: "zSmessage_set {
[unknown_extension] {
str: "malformed
multiline
string
}
}
z7message_set {
[malformed_extension] <
i: -5
}
z5:1 : Expected ">".z)message_set {
unknown_field: true
}
zb2:3 : Message type "proto2_wireformat_unittest.TestMessageSet" has no field named "unknown_field".zmessage_set {
[protobuf_unittest.TestMessageSetExtension1] {
i: 23
}
[protobuf_unittest.TestMessageSetExtension2] {
str: "foo"
}
}
r r{ )r r r rI r' r5 r: r| rw r r r r r r r r )r r7 r$ r Z malformedr r r r r testParseAllowedUnknownExtension" sr z,Proto2Tests.testParseAllowedUnknownExtensionc
C sF t j }d}| jtj}tj|| W d Q R X | jt|jd d S )Nz#optional_nested_message { "bb": 1 }z/1:27 : Expected identifier or number, got "bb".) r r/ rv r rw rI r r rx )r r7 r$ ry r r r testParseBadIdentifier s z"Proto2Tests.testParseBadIdentifierc C sH t j }d}tj| tjdtj|| t j }tj| tjdtj|| d S )Nz[unknown_extension]: 8
z31:2 : Extension "unknown_extension" not registered.zM1:2 : Message type "protobuf_unittest.TestAllTypes" does not have extensions.)r r r: r| r rw rI r/ )r r7 r$ r r r testParseBadExtension s
z!Proto2Tests.testParseBadExtensionc C s( t j }d}tj| tjdtj|| d S )Nzoptional_nested_enum: 100zL1:23 : Enum type "\w+.TestAllTypes.NestedEnum" has no value with number 100.)r r/ r: r| r rw rI )r r7 r$ r r r testParseNumericUnknownEnum s
z'Proto2Tests.testParseNumericUnknownEnumc C s0 t j }d}tj|| | jd|jt j d S )Nza[protobuf_unittest.optional_int32_extension]: 42 [protobuf_unittest.optional_int32_extension]: 67r )r r r rh r r Zoptional_int32_extension)r r7 r$ r r r "testMergeDuplicateExtensionScalars s
z.Proto2Tests.testMergeDuplicateExtensionScalarsc C s( t j }d}tj| tjdtj|| d S )Nza[protobuf_unittest.optional_int32_extension]: 42 [protobuf_unittest.optional_int32_extension]: 67z1:96 : Message type "protobuf_unittest.TestAllExtensions" should not have multiple "protobuf_unittest.optional_int32_extension" extensions.)r r r: r| r rw rI )r r7 r$ r r r "testParseDuplicateExtensionScalars s
z.Proto2Tests.testParseDuplicateExtensionScalarsc C s( t j }d}tj| tjdtj|| d S )NzCoptional_nested_message { bb: 1 } optional_nested_message { bb: 2 }zh1:65 : Message type "protobuf_unittest.TestAllTypes.NestedMessage" should not have multiple "bb" fields.)r r/ r: r| r rw rI )r r7 r$ r r r &testParseDuplicateNestedMessageScalars s
z2Proto2Tests.testParseDuplicateNestedMessageScalarsc C s( t j }d}tj| tjdtj|| d S )Nz%optional_int32: 42 optional_int32: 67zf1:36 : Message type "protobuf_unittest.TestAllTypes" should not have multiple "optional_int32" fields.)r r/ r: r| r rw rI )r r7 r$ r r r testParseDuplicateScalars s
z%Proto2Tests.testParseDuplicateScalarsc C sD t j }d}tj| tjdtj|| d}tj| tjdtj|| d S )NzRepeatedGroup: ".zRepeatedGroup: {z1:16 : Expected "}".)r r/ r: r| r rw rI )r r7 r$ r r r testParseGroupNotClosed s z#Proto2Tests.testParseGroupNotClosedc C s\ t j }d}tj|| | j|jd |j t j }d}tj|| | j|jd d S )NzOptionalGroup: {}Z
optionalgroupzOptionalGroup: <>)r r/ r rI r ZHasFieldrg )r r7 r$ r r r testParseEmptyGroup s z Proto2Tests.testParseEmptyGroupc C s d}t j }tj|| | jd |jd | jd|jd | jd|jd | jd|jd | jd|j d | jd |j
d
j d S )NaH map_int32_int32 {
key: -123
value: -456
}
map_int64_int64 {
key: -8589934592
value: -17179869184
}
map_uint32_uint32 {
key: 123
value: 456
}
map_uint64_uint64 {
key: 8589934592
value: 17179869184
}
map_string_string {
key: "abc"
value: "123"
}
map_int32_foreign_message {
key: 111
value {
c: 5
}
}
i rS rj r r r r r r i8il l l l l l ) r r r rI r r r r r r r rT )r r$ r7 r r r testParseMap s zProto2Tests.testParseMapN)r r r r r r r r r r r r r r r r r r r r r r r r r r r r r t s* )
z
r 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 S ) Proto3Testsc C s@ t j }d|_tj }|jj| | jtj |t
j dd d S )Nr )r z[any_value {
[type.googleapis.com/protobuf_unittest.OneString] {
data: "string"
}
}
) r OneStringdatar r any_valuePackr r r5 r r )r packed_messager7 r r r testPrintMessageExpandAny s
z%Proto3Tests.testPrintMessageExpandAnyc C sR t j }tj }d|_|jj j| d|_|jj j| | jt j
|d d S )Nstring0string1zrepeated_any_value {
[type.googleapis.com/protobuf_unittest.OneString] {
data: "string0"
}
}
repeated_any_value {
[type.googleapis.com/protobuf_unittest.OneString] {
data: "string1"
}
}
) r r r r r repeated_any_valuer? r r r r5 )r r r7 r r r !testPrintMessageExpandAnyRepeated( s z-Proto3Tests.testPrintMessageExpandAnyRepeatedc C sD t j }d|_tj }|jj| tj }| j t
j ||dd d S )Nr )r zdany_value {
type_url: "type.googleapis.com/protobuf_unittest.OneString"
value: "\n\006string"
}
) r r r r r r r r DescriptorPoolr r r5 )r r r7
empty_poolr r r 2testPrintMessageExpandAnyDescriptorPoolMissingType< s z>Proto3Tests.testPrintMessageExpandAnyDescriptorPoolMissingTypec C s< t j }d|_tj }|jj| | jtj |ddd d S )Nr T)r z[any_value <
[type.googleapis.com/protobuf_unittest.OneString] <
data: "string"
>
>
)
r r r r r r r r r r5 )r r r7 r r r 'testPrintMessageExpandAnyPointyBracketsI s z3Proto3Tests.testPrintMessageExpandAnyPointyBracketsc C s< t j }d|_tj }|jj| | jtj |ddd d S )Nr T)r= zRany_value { [type.googleapis.com/protobuf_unittest.OneString] { data: "string" } })
r r r r r r r r r r5 )r r r7 r r r "testPrintMessageExpandAnyAsOneLineW s z.Proto3Tests.testPrintMessageExpandAnyAsOneLinec C sD t j }d|_tj }|jj| | jtj |ddt
j dd d S )Nr T)r= r r zRany_value < [type.googleapis.com/protobuf_unittest.OneString] < data: "string" > >) r r r r r r r r r r5 r r )r r r7 r r r 0testPrintMessageExpandAnyAsOneLinePointyBracketsd s
z t j }t j }d|_tj|}tj|| | jd|j d S )Ni )r r/ Zoptional_nested_enumr r5 rI r )r r7 Zmessage2Z text_stringr r r testUnknownEnumss s
z Proto3Tests.testUnknownEnumsc C st t j }d}tj|| tj }|jj| | jd|j |j
tj || tj }|jj| | jd|j d S )Nz[any_value {
[type.googleapis.com/protobuf_unittest.OneString] {
data: "string"
}
}
r ) r r r rh r r r Unpackr r rg rI )r r7 r$ r r r r testMergeExpandedAny{ s z Proto3Tests.testMergeExpandedAnyc C s` t j }d}tj|| tj }|jd j| | jd|j |jd j| | jd|j d S )Nzrepeated_any_value {
[type.googleapis.com/protobuf_unittest.OneString] {
data: "string0"
}
}
repeated_any_value {
[type.googleapis.com/protobuf_unittest.OneString] {
data: "string1"
}
}
r r r r )
r r r rh r r r r r r )r r7 r$ r r r r testMergeExpandedAnyRepeated s
z(Proto3Tests.testMergeExpandedAnyRepeatedc C s> t j }d}tj|| tj }|jj| | jd|j d S )Nz[any_value {
[type.googleapis.com/protobuf_unittest.OneString] <
data: "string"
>
}
r )
r r r rh r r r r r r )r r7 r$ r r r r "testMergeExpandedAnyPointyBrackets s z.Proto3Tests.testMergeExpandedAnyPointyBracketsc C s4 t j }d}tj|| tj }| jd|jj d S )NzYany_value {
[type.otherapi.com/protobuf_unittest.OneString] {
data: "string"
}
}
z-type.otherapi.com/protobuf_unittest.OneString) r r r rh r r r r Ztype_url)r r7 r$ r r r r testMergeAlternativeUrl s z#Proto3Tests.testMergeAlternativeUrlc C sR t j }d}| jtj }tj }tj|||d W d Q R X | jt |j
d d S )Nz[any_value {
[type.googleapis.com/protobuf_unittest.OneString] {
data: "string"
}
}
)r z=Type protobuf_unittest.OneString not found in descriptor pool) r r rv r rw r r rh r r rx )r r7 r$ ry r r r r -testMergeExpandedAnyDescriptorPoolMissingType s z9Proto3Tests.testMergeExpandedAnyDescriptorPoolMissingTypec C s> d}t j }tj|| tj }|jj| | jd|j d S )Nzdany_value {
type_url: "type.googleapis.com/protobuf_unittest.OneString"
value: "\n\006string"
}
r )
r r r rh r r r r r r )r r$ r7 r r r r testMergeUnexpandedAny s z"Proto3Tests.testMergeUnexpandedAnyc
C sF t j }d}| jtj}tj|| W d Q R X | jt|jd d S )NzUany_value {
[type.googleapis.com/protobuf_unittest.OneString] {
data: "string"
z3:11 : Expected "}".) r r rv r rw rh r r rx )r r7 r$ ry r r r testMergeMissingAnyEndToken s
z'Proto3Tests.testMergeMissingAnyEndTokenN)r r r r r r r r r r r r r r r r r r r r r r s
r c @ st e Zd Zdd Zdd Zdd Zdd Zd d
Zd d Zd
d Z dd Z
dd Z dd Z dd Z
dd Zdd ZdS )
TokenizerTestc T C s d}t j|j }|jdfd|jdf|jdfd|jdf|jdfd|jdf|jd fd|jd
f|jd fd|jd3f|jd
fd|jdf|jdfd|jdf|jdfdd|jdfd|jtdf|jdfd|jtd f|jdfd|jdf|jdfd|jdfd|jdfd|jd f|jd fd|jd4f|jd fd|jd5f|jd fd|jd!f|jd"fd|jd#f|jd$fd|jd%f|jd&fd|jdf|jd'fd|jdf|jd(fd|jdf|jd)fd|jdf|jd*fd|jdf|jd+fd|jdf|jd,fd|jtdf|jd-fd|jtd.f|jd/fd|jtd0fgR}d1}x|j s|| }t
|t rN|j }| j
|| |j nNt
|d2 trtj|d2 r| jtj|d1 n| j
|d2 |d1 |d27 }qW d S )6Na identifier1:"string1"
identifier2 :
123
identifier3 :'string'
identifiER_4 : 1.1e+2 ID5:-0.23 ID6:'aaaa\'bbbb'
ID7 : "aa\"bb"
ID8: {A:inf B:-inf C:true D:false}
ID9: 22 ID10: -111111111111111111 ID11: -22
ID12: 2222222222222222222 ID13: 1.23456f ID14: 1.2e+2f false_bool: 0 true_BOOL:t
true_bool1: 1 false_BOOL1:f False_bool: False True_bool: True X:iNf Y:-inF Z:nANZ identifier1:r Z identifier2rS Z identifier3r Z identifiER_4g [@ZID5gq=
ףp?ZID6z aaaa'bbbbZID7zaa"bbZID8{AinfBCTDF}ZID9 ZID10l q{U ZID11ZID12l c/Y-v ZID13g82?ZID14g ^@Z
false_boolZ true_BOOLZ
true_bool1Z false_BOOL1Z
False_boolZ True_boolXYz-infZnanr r gq=
ףpͿlq{U i)r Tokenizer
splitlinesConsumeIdentifierZ
ConsumeStringConsumeInteger ConsumeFloatfloat ConsumeBoolAtEnd
isinstancer tokenr NextTokenmathZisnanr )r r$ tokenizermethodsr r r r r r testSimpleTokenCases s
z"TokenizerTest.testSimpleTokenCasesc
C s d }d
}d|d |d f }t j|j }| jd|j | j|d |j | j|d |j | j|j d}t j|j }| jd|j | jd|j | jd|j | jd|j | jt j |j W d Q R X | jd |j
| j|j d S )Nr ? z-1 %d %dz
-0 0 0 1.2r TFg333333?l l l l ) r r r r r r r ZTryConsumeIntegerrv rw r )r int64_max
uint32_maxr$ r r r r testConsumeAbstractIntegers) s$ z)TokenizerTest.testConsumeAbstractIntegersc C s< d}d
}d|d |d f }t j|j }| jt jt j| | jt jt j| | jd t j| | jt jt j| | jt jt j| | j|d t j | | jt jt j | | j|d t j| | j
|j d}t j|j }| jdt j| | jdt j| | jdt j| | jdt j| | j
|j d S ) Nr r r z-1 %d %dz -0 -0 0 0r l l l l r ) r r r rv rw Z_ConsumeUint32Z_ConsumeUint64r Z
_ConsumeInt32Z
_ConsumeInt64r r )r r r r$ r r r r testConsumeIntegersB s2
z!TokenizerTest.testConsumeIntegersc C s d}t j|j }| jt j|j d}t j|j }| jt j|j d}t j|j }| jt j|j d}t j|j }| jt j|j d}t j|j }| jt j|j d S )Nz "string1'zstring1"z
"\xt"z
"\"z
"\x")r r r rv rw ZConsumeByteString)r r$ r r r r testConsumeByteStringb s z#TokenizerTest.testConsumeByteStringc C s& d}t j|j }| jt j|j d S )Nz
not-a-bool)r r r rv rw r )r r$ r r r r testConsumeBoolw s z TokenizerTest.testConsumeBoolc C s0 t jdj }| j|j | jt j|j d S )Nz# some comment)r r r r r rv rw ConsumeComment)r r r r r testSkipComment| s z TokenizerTest.testSkipCommentc C sB t jdj dd}| j|j | jd|j | j|j d S )Nz# some commentF)
skip_comments)r r r assertFalser r r r )r r r r r testConsumeComment s
z TokenizerTest.testConsumeCommentc C sV d}t j|j dd}| jd|j | j|j | jd|j | j|j d S )Nz # some comment
# another commentF)r z# some commentz# another comment)r r r r r r r r )r r$ r r r r testConsumeTwoComments s z$TokenizerTest.testConsumeTwoCommentsc C s d}t j|j dd}| jt j|j | jd|j | j|jd |j | jt j|j | jd|j
| j |j | jd|j | j
|j d S )Nz some_number: 4
# some commentF)r some_numberr r z# some comment)r r r rv rw r r r r r r r r r )r r$ r r r r testConsumeTrailingComment s z(TokenizerTest.testConsumeTrailingCommentc C sB t jdj dd}| j|j | jd|j | j|j d S )N# some commentF)r )Fr% )r r r r r r ConsumeCommentOrTrailingCommentr )r r r r r testConsumeLineComment s
z$TokenizerTest.testConsumeLineCommentc C sV d}t j|j dd}| jd|j | j|j | jd|j | j|j d S )Nz # some comment
# another commentF)r # some comment# another comment)Fr( )Fr) )r r r r r& r r r )r r$ r r r r testConsumeTwoLineComments s
z(TokenizerTest.testConsumeTwoLineCommentsc C s d}t j|j dd}| jt j|j | jd|j | j|jd |j | jt j|j | jd|j
| j |j | jd |j | j
|j d S )
Nz some_number: 4 # some commentF)r r# r r T# some comment)Tr+ )r r r rv rw r& r r r r r r r r )r r$ r r r r "testConsumeAndCheckTrailingComment s
z0TokenizerTest.testConsumeAndCheckTrailingCommentc C sn d}t j|j dd}| jd|j | j|jd |j | jd|j | jd |j | j |j
d S )
Nz2some_number: 4 # some comment # not a new commentF)r r# r r T"# some comment # not a new comment)Tr- ) r r r r r r r r r& r r )r r$ r r r r testHashinComment s
z TokenizerTest.testHashinCommentN)r r r r r r r r r r! r" r$ r' r* r, r. r r r r r s M
r c @ s4 e Zd Zdd Zdd Zdd Zdd Zd d
Zd S ) PrettyPrinterTestc C s< dd }|j }|jj }d|_| jtj|d|dd d S )Nc S s
~ ~~d S )Nr )r7 indentr= r r r r_ s z9PrettyPrinterTest.testPrettyPrintNoMatch..printerr< T)r= message_formatterz"repeated_nested_message { bb: 42 })r/ r> r? r@ r' r r5 )r r6 r_ r7 rA r r r testPrettyPrintNoMatch s
z(PrettyPrinterTest.testPrettyPrintNoMatchc s@ fdd} j }|jj }d|_| jtj|d|dd d S )Nc s" ~~| j jjj kr d| j S d S )NzMy lucky number is %s)rV r/
NestedMessager@ )r r0 r= )r6 r r r_ s z9PrettyPrinterTest.testPrettyPrintOneLine..printerr< T)r= r1 z1repeated_nested_message { My lucky number is 42 })r/ r> r? r@ r' r r5 )r r6 r_ r7 rA r )r6 r testPrettyPrintOneLine s
z(PrettyPrinterTest.testPrettyPrintOneLinec sX fdd} j }|jj }d|_| jtj|d|dd | jtj|d|dd d S ) Nc s6 | j jjj kr2|rdndd| }d|| jf S d S )N r zMy lucky number is:%s%s)rV r/ r3 r@ )r r0 r= Zline_deliminator)r6 r r r_ s z;PrettyPrinterTest.testPrettyPrintMultiLine..printerr< T)r= r1 z2repeated_nested_message { My lucky number is: 42 }Fz7repeated_nested_message {
My lucky number is:
42
}
)r/ r> r? r@ r' r r5 )r r6 r_ r7 rA r )r6 r testPrettyPrintMultiLine s
z*PrettyPrinterTest.testPrettyPrintMultiLinec sH fdd} j }| jtj|d|dd | jtj|d|dd d S )Nc s ~~| j jj krdS d S )NzThe is the message!)rV r/ )r r0 r= )r6 r r r_ s z?PrettyPrinterTest.testPrettyPrintEntireMessage..printerF)r= r1 zThe is the message!
TzThe is the message!)r/ r' r r5 )r r6 r_ r7 r )r6 r testPrettyPrintEntireMessage s
z.PrettyPrinterTest.testPrettyPrintEntireMessagec sb fdd} j }d|_|jj }d|_|jj }d|_|j}d|_| jtj|d|dd d S )
Nc s" ~~| j jjj kr d| j S d S )NzMy lucky number is %s)rV r/ r3 r@ )r r0 r= )r6 r r r_ s
z?PrettyPrinterTest.testPrettyPrintMultipleParts..printer= r< c r T)r= r1 zoptional_int32: 61 optional_nested_message { My lucky number is 1 } repeated_nested_message { My lucky number is 42 } repeated_nested_message { My lucky number is 99 }) r/ r r> r? r@ r r' r r5 )r r6 r_ r7 rA r )r6 r testPrettyPrintMultipleParts s
z.PrettyPrinterTest.testPrettyPrintMultiplePartsN)r r r r2 r4 r6 r7 r: r r r r r/ s
r/ __main__)#__doc__
__author__r r) r: r Z unittest2Zunittest ImportErrorZgoogle.protobuf.internalr Zgoogle.protobufr r r r r r r r r
r r ZTestCaser
r Z
Parametersr. r r r r r/ r mainr r r r ! sT
' C w]