3
SLj* @ sp d Z dZy ddlZW n ek
r0 ddlZY nX ddlmZ ddlmZ G dd dej Z
e dkrlej dS ) z.Test for google.protobuf.internal.wire_format.z#robinson@google.com (Will Robinson) N)message) wire_formatc @ s< e Zd Zdd Zdd Zdd Zdd Zd d
Zd d Zd
S )WireFormatTestc C sR d}d}| j |d> |B tj|| tj}| jtj||d | jtj||d d S )Ni
) assertEqualr PackTag assertRaisesr EncodeError)self field_numberZtag_typer r &/usr/lib/python3.6/wire_format_test.py testPackTag/ s zWireFormatTest.testPackTagc C s xLd D ]D}x>t dD ]2}tjtj||\}}| j|| | j|| qW qW | jttjd | jttjd | jttjd | jttjt d S )
Nr r abcg )r r r r r )ranger Z UnpackTagr r
r TypeErrorobject)r Zexpected_field_numberZexpected_wire_typer Z wire_typer r r
testUnpackTag: s
z WireFormatTest.testUnpackTagc C s t j}| jd|d | jd|d | jd|d | jd|d | jd|d | jd|d | jd|d | jd
|d | jd |d | jt|d | jt|d | jt|d | jt|t d S )Nr r r r l il l l l l l abcdg r i l )r Z ZigZagEncoder
r r r )r Zr r r testZigZagEncodeH s z WireFormatTest.testZigZagEncodec C s t j}| jd|d | jd|d | jd|d | jd|d | jd|d | jd|d | jd|d | jd
|d | jd|d
| jt|d | jt|d | jt|d | jt|t d S )Nr r r r r il l l l l l l r g r r i l )r Z ZigZagDecoder
r r r )r r r r r testZigZagDecodeY s z WireFormatTest.testZigZagDecodec
C sD x>d
D ]6\}}|| }|||}| j ||d|||||f qW d S )Nr r r r r zFbyte_size_fn: %s, field_number: %d, value: %r
Expected: %d, Actual: %dr r r r r" r r# r )r$ r% r& r' )r
)r byte_size_fnvalueZexpected_value_sizer Z tag_bytesZ
expected_sizeZ actual_sizer r r NumericByteSizeTestHelperj s
z(WireFormatTest.NumericByteSizeTestHelperc 6 C s t jddgt jddgt jddgt jd dgt jddgt jddgt jddgt jd dgt jddgt jddgt jddgt jt jdgt jddgt jddgt jddgt jt jdgt jddgt jd dgt jddgt jd dgt jddgt jd dgt jd dgt jddgt jd!dgt jddgt jd"dgt jddgt jd#dgt jd dgt j dd
gt j t jd
gt j
dd gt j
t jd gt j dd
gt j t j d
gt j t j
d
gt jdd gt jt jd gt jt jd gt jd d
gt jd
d
gt jd$d
gt jd d gt jd
d gt jd%d gt jddgt jddgt jddgt jddgt jddgt jt jdgg4}x|D ]}| j| q$W xPt jt jfD ]@}| jd|dd | jd|dd | jd|ddd qFW | jdt jddjd G dd dt}d}||d}| jd| t jd| | jd
| t jd| | jd|j t j d| | jd|j t j d| d|_ | jd
|j t j d| d|_ | j|j d t j d| d|_ | j|j d t j d| | j|j d t j d| | j t j!t jddd> d S )&Nr r r
? @ r g g eAFTr r r as Тестzutf-8c @ s e Zd Zdd Zdd ZdS )z9WireFormatTest.testByteSizeFunctions..MockMessagec S s
|| _ d S )N) byte_size)r r4 r r r __init__ s zBWireFormatTest.testByteSizeFunctions..MockMessage.__init__c S s | j S )N)r4 )r r r r ByteSize s zBWireFormatTest.testByteSizeFunctions..MockMessage.ByteSizeN)__name__
__module__ __qualname__r5 r6 r r r r MockMessage s r: )r4 r r r r iir iig eg e)"r Z
Int32ByteSizeZ
Int64ByteSizeZUInt32ByteSizeZ
UINT32_MAXZUInt64ByteSizeZ
UINT64_MAXZSInt32ByteSizeZSInt64ByteSizeZFixed32ByteSizeZFixed64ByteSizeZSFixed32ByteSizeZ INT32_MINZ INT32_MAXZSFixed64ByteSizeZ INT64_MINZ INT64_MAXZ
FloatByteSizeZDoubleByteSizeZ BoolByteSizeZ EnumByteSizer* ZStringByteSizeZ
BytesByteSizer
decoder Z
GroupByteSizer4 ZMessageByteSizeZMessageSetItemByteSizer r r
)r Z NUMERIC_ARGSargsr( r: Zmessage_byte_sizeZ mock_messager r r testByteSizeFunctionst s
z$WireFormatTest.testByteSizeFunctionsN) r7 r8 r9 r r r r! r* r> r r r r r - s
r __main__)
__doc__
__author__Z unittest2Zunittest ImportErrorZgoogle.protobufr Zgoogle.protobuf.internalr ZTestCaser r7 mainr r r r s S