File: //lib/python3.6/site-packages/google/protobuf/internal/__pycache__/text_format_test.cpython-36.pyc
3
���h�� �
@ 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� dej�ZG dd� dej�Ze
jee�G dd� de��ZG dd� de�Z G dd� de�Z G dd� dej�Z G dd � d ej�Z e
jee�G d d � d e��Z e!d k�r�ej"� 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� assertEqual�len)�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 �
<listcomp>Q s z-TextFormatBase.ReadGolden.<locals>.<listcomp>)r
Z
GoldenFile�str�bytes� readlines)r �golden_filename�fr r r �
ReadGoldenN s zTextFormatBase.ReadGoldenc C s | j |�}| j|dj|�� d S )N� )r! ZassertMultiLineEqual�join)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 dd�j dd�j dd�j dd�}tjdtj�jd|�}|S )Nze+0ze+ze-0ze-z\.0$r" )�replace�re�compile� MULTILINE�sub)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��/�^@g�t�QFքDg���6<z
\'"u üꜟz�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: "\303\274\352\234\237"
l���� )
� TestAllTypes�repeated_int64�append�repeated_uint64�repeated_double�repeated_stringr' r- r �MessageToString)r �message_module�messager r r �testPrintExotich s z TextFormatTest.testPrintExoticc C s@ G dd� dt j�}|j� }|jj|d�� | jtj|�d� d S )Nc @ s e Zd ZdS )zATextFormatTest.testPrintExoticUnicodeSubclass.<locals>.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|dd�d� d S )N�* T)� as_one_linez"repeated_nested_message { bb: 42 })r/ �repeated_nested_message�add�bbr' r r5 )r r6 r7 �msgr r r � testPrintNestedMessageAsOneLine� s
z.TextFormatTest.testPrintNestedMessageAsOneLinec C s^ |j � }|jjd� |jjd� |jjd� |jjd� |jjd� | jtj|dd�d� 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|dd�d� 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��/�^@g�t�QFքDg���6<z
\'"u üꜟT)r= z�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: "\303\274\352\234\237"l���� )
r/ r0 r1 r2 r3 r4 r' r- r r5 )r r6 r7 r r r �testPrintExoticAsOneLine� s
z'TextFormatTest.testPrintExoticAsOneLinec C s� |j � }|jjd � |jjd� |jjd� |jjd� |jjd� |jjd� |jjd� tj|dd d
�}|j � }tj||�}| j ||� | j
||� tj|ddd
�}|j � }tj||�}| j ||� | j
||d ||f � d S )
Nl l ���� gw��/�^@g�t�QFքDg���6<z
\'"u üꜟTF)r= �as_utf8z
%s != %sl���� ) r/ r0 r1 r2 r3 r4 r r5 �Parse�assertIsr )r r6 r7 Z wire_text�parsed_message�rr r r � testRoundTripExoticAsOneLine� s. z+TextFormatTest.testRoundTripExoticAsOneLinec C s\ |j � }|jjd� tj|dd�}| j|d� |j � }tj||� | j||d||f � d S )Nu üꜟT)rH u repeated_string: "üê"
z
%s != %s)r/ r4 r1 r r5 r' rI r )r r6 r7 r$ rK r r r �testPrintRawUtf8String� s z%TextFormatTest.testPrintRawUtf8Stringc C s� |j � }d|j_d|j_|jjjd� |jjjd� ddddg}tj|d d
�}| j | j
|�d j |� � tj|d d d
�}| j | j
|�dj |� � d S )Ng �?g�/_�e��>i
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��) ZNestedTestAllTypesZpayload�optional_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
)ZForeignMessage�cr 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/ �
DESCRIPTOR�fields_by_namerP r �
TextWriterZ
PrintFieldr �getvalue�close�_Printer)r r6 r7 �field�value�out�printerr 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||� |tkr�tj | |� 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 �Clear�Merger: 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��/�^@g�t�QFքDr g���6<� 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 �one�two)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 )N�fooz?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: 67�C )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 �m�m2r 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/ r| r 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|dd��d� 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 |dd��d� | j| j t
j |��d � d S )
NZ115�e �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 �" �! �123�abcr� 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
}
}
i8���i����l l���� l l���� l l ) r �TestMap�map_int32_int32�map_int64_int64�map_uint32_uint32�map_uint64_uint64�map_string_string�map_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 � <genexpr>_ s zKOnlyWorksWithProto2RightNowTests.testMapOrderEnforcement.<locals>.<genexpr>)
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{ z�message_set {
[protobuf_unittest.TestMessageSetExtension1] {
i: 23
}
[protobuf_unittest.TestMessageSetExtension2] {
str: "foo"
}
}
ZbarzF[google.protobuf.internal.TestMessageSetExtension3] {
text: "bar"
}
)r �TestMessageSetContainer�TestMessageSetExtension1�message_set_extension�TestMessageSetExtension2� message_set�
Extensions�ir r' r r5 r ZTestMessageSetZmessage_set_extension3r$ )r r7 �ext1�ext2Zextr 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 |dd�d� d S )Nr� r{ T)r= z�message_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 z�message_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 )Nz�value {
[type.googleapis.com/google.protobuf.internal.TestAny] {
[google.protobuf.internal.TestAnyExtension1.extension1] {
i: 10
}
}
}
)r )�test_extend_any�TestAnyr 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|dd��d� 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".z�message_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|j�d� 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]: 67z�1: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: <z1:16 : Expected ">".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� i8���i����l 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 � d�d� d S )Nr� )r z[any_value {
[type.googleapis.com/protobuf_unittest.OneString] {
data: "string"
}
}
) r � OneString�datar r� � any_value�Packr 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 )N�string0�string1z�repeated_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 ||d�d� 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 |dd�d� 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 |dd�d� 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 � d�d� 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<Proto3Tests.testPrintMessageExpandAnyAsOneLinePointyBracketsc C s> 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 )Nz�repeated_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|j�d� 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|jtd�f|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|jtd�f|jd-fd|jtd.�f|jd/fd|jtd0�fgR}d1}x�|j � �s�|| }t
|t ��rN|j }| j
||� |j� nNt
|d2 t��r�tj|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�{�A�inf�B�CT�DF�}ZID9� ZID10l �q��{U ZID11ZID12l �c/Y-�v ZID13g82�����?ZID14g ^@Z
false_boolZ true_BOOLZ
true_bool1Z false_BOOL1Z
False_boolZ True_bool�X�Yz-inf�Z�nanr r gq=
ףpͿl�����q��{U i��)r � Tokenizer�
splitlines�ConsumeIdentifierZ
ConsumeString�ConsumeInteger� ConsumeFloat�float� ConsumeBool�AtEnd�
isinstancer �tokenr � NextToken�mathZisnanr� )r r$ � tokenizer�methodsr� 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|d�d� d S )Nc S s
~ ~~d S )Nr )r7 �indentr= r r r r_ � s z9PrettyPrinterTest.testPrettyPrintNoMatch.<locals>.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|d�d� 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.<locals>.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|d�d� | jtj|d|d�d� 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.<locals>.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|d�d� | jtj|d|d�d� 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.<locals>.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|d�d � 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.<locals>.printer�= r< �c r T)r= r1 z�optional_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 �<module>! sT
' C w]