3
SLj @ s d Z dZddlZddlZddlZddlmZ ddlmZ ye W n e
k
rX e Z Y nX dd Z dd Z
d
d Zd d
Zdd Zdd Zdd Zdd Zdd Zdd Zdd Zd d ZG d d d ejZdS ) zpUtilities for Python proto2 tests.
This is intentionally modeled on C++ code in
//google/protobuf/test_util.*.
z#robinson@google.com (Will Robinson) N)unittest_import_pb2) unittest_pb2c C s | j jdkS )NZproto2)Z
DESCRIPTORZsyntax)message r /usr/lib/python3.6/test_util.pyIsProto27 s r c C s d| _ d| _d| _d| _d| _d| _d| _d| _d | _d
| _ d | _
d | _ d
| _ d| _
d| _t| rjd| j_d| j_d| j_d| j_d| j_tjj | _ tj | _ t| rt j!| _"d| _#d| _$| j%j&d | j'j&d | j(j&d | j)j&d | j*j&d | j+j&d | j,j&d | j-j&d | j.j&d | j/j&d | j0j&d! | j1j&d" | j2j&d
| j3j&d# | j4j&d$ t| rd%| j5j6 _d&| j7j6 _d'| j8j6 _d(| j9j6 _d)| j:j6 _| j;j&tjj< | j=j&tj> t| r| j?j&t j@ | jAj&d* | jBj&d+ | j%j&d, | j'j&d, | j(j&d, | j)j&d, | j*j&d, | j+j&d, | j,j&d, | j-j&d, | j.j&d, | j/j&d, | j0j&d, | j1j&d, | j2j&d
| j3j&d- | j4j&d. d/| j%d0< d1| j'd0< d2| j(d0< d3| j)d0< d4| j*d0< d5| j+d0< d6| j,d0< d7| j-d0< d8| j.d0< d9| j/d0< d:| j0d0< d;| j1d0< d<| j2d0< d=| j3d0< d>| j4d0< t| r`d?| j5j6 _d@| j7j6 _dA| j8j6 _dB| j9j6 _dC| j:j6 _| j;j&tjj< tjj | j;d0< | j=j&tj t| r| j?j&t j! | jAj&dD | jBj&dE t| rvdF| _CdG| _DdH| _EdI| _FdJ| _GdK| _HdL| _IdM| _JdN| _KdO| _LdP| _MdQ| _Nd<| _OdR| _PdS| _QtjjR| _StjT| _Ut jV| _WdT| _XdU| _YdV| _ZdW| j[_dX| _\dY| _]dZS )[zmSets every non-lazy field in the message to a unique value.
Args:
message: A TestAllTypes instance.
e f g h i j k l m n o p T115s 116u v w x ~ 124125 215s 216 224225r 0 0i- i. i/ i0 i1 i2 i3 i4 i5 i6 i7 i8 F315s 316i= i> i? i@ iG 324325i i i i i i i i i i i i 415s 416424425iY iZ 603s 604N)^optional_int32optional_int64optional_uint32optional_uint64optional_sint32optional_sint64optional_fixed32optional_fixed64optional_sfixed32optional_sfixed64optional_floatoptional_double
optional_booloptional_stringoptional_bytesr
optionalgroupaoptional_nested_messagebboptional_foreign_messagecoptional_import_messaged optional_public_import_messageer TestAllTypesBAZoptional_nested_enum FOREIGN_BAZoptional_foreign_enumr
IMPORT_BAZoptional_import_enumoptional_string_piece
optional_cordrepeated_int32appendrepeated_int64repeated_uint32repeated_uint64repeated_sint32repeated_sint64repeated_fixed32repeated_fixed64repeated_sfixed32repeated_sfixed64repeated_floatrepeated_double
repeated_boolrepeated_stringrepeated_bytes
repeatedgroupaddrepeated_nested_messagerepeated_foreign_messagerepeated_import_messagerepeated_lazy_messagerepeated_nested_enumBARrepeated_foreign_enum FOREIGN_BARrepeated_import_enum
IMPORT_BARrepeated_string_piece
repeated_cord
default_int32
default_int64default_uint32default_uint64default_sint32default_sint64default_fixed32default_fixed64default_sfixed32default_sfixed64
default_floatdefault_double default_booldefault_string
default_bytesFOOdefault_nested_enum FOREIGN_FOOdefault_foreign_enum
IMPORT_FOOdefault_import_enumZdefault_string_pieceZ default_cordZ oneof_uint32Zoneof_nested_messageZ oneof_stringZ oneof_bytes)r r r r SetAllNonLazyFields; s
r c C s t | d| j_d S )N )r optional_lazy_messagerL )r r r r SetAllFields s r c C sb | j }t}t}d||j< d||j< d||j< d||j< d||j< d||j< d||j < d||j
< d ||j < d
||j < d ||j
< d ||j< d
||j< d||j< d||j< d||j _d||j _d||j _d||j _d||j _d||j _|j j ||j < |j j ||j < |j ||j!< |j"||j#< d||j$< d||j%< ||j& j'd ||j( j'd ||j) j'd ||j* j'd ||j+ j'd ||j, j'd ||j- j'd ||j. j'd ||j/ j'd ||j0 j'd! ||j1 j'd" ||j2 j'd# ||j3 j'd
||j4 j'd$ ||j5 j'd% d&||j6 j7 _d'||j8 j7 _d(||j9 j7 _d)||j: j7 _d*||j; j7 _||j< j'|j j= ||j> j'|j? ||j@ j'|jA ||jB j'd+ ||jC j'd, ||j& j'd- ||j( j'd. ||j) j'd/ ||j* j'd0 ||j+ j'd1 ||j, j'd2 ||j- j'd3 ||j. j'd4 ||j/ j'd5 ||j0 j'd6 ||j1 j'd7 ||j2 j'd8 ||j3 j'd9 ||j4 j'd: ||j5 j'd; d<||j6 j7 _d=||j8 j7 _d>||j9 j7 _d?||j: j7 _d@||j; j7 _||j< j'|j j ||j> j'|j ||j@ j'|j" ||jB j'dA ||jC j'dB dC||jD< dD||jE< dE||jF< dF||jG< dG||jH< dH||jI< dI||jJ< dJ||jK< dK||jL< dL||jM< dM||jN< dN||jO< d9||jP< dO||jQ< dP||jR< |j jS||jT< |jU||jV< |jW||jX< dQ||jY< dR||jZ< dS||j[< dT||j\ _dU||j]< dV||j^< dWS )XzzSets every extension in the message to a unique value.
Args:
message: A unittest_pb2.TestAllExtensions instance.
r r r
r r r
r r r r r r Tr s 116r r r r r r r r r r r r r r! r" r# r$ r% r& r' r( s 216r) r* r+ r, r- r. r/ i- i. i/ i0 i1 i2 i3 i4 i5 i6 i7 i8 Fr3 s 316i= i> i? i@ iG r4 r5 i i i i i i i i i i i i r6 s 416r7 r8 iY iZ r9 s 604N)_
Extensionsr r Zoptional_int32_extensionZoptional_int64_extensionZoptional_uint32_extensionZoptional_uint64_extensionZoptional_sint32_extensionZoptional_sint64_extensionZoptional_fixed32_extensionZoptional_fixed64_extensionZoptional_sfixed32_extensionZoptional_sfixed64_extensionZoptional_float_extensionZoptional_double_extensionZoptional_bool_extensionZoptional_string_extensionZoptional_bytes_extensionZoptionalgroup_extensionrJ Z!optional_nested_message_extensionrL Z"optional_foreign_message_extensionrN Z!optional_import_message_extensionrP Z(optional_public_import_message_extensionrR Z optional_lazy_message_extensionrS rT Z optional_nested_enum_extensionrV Z optional_foreign_enum_extensionrX Z optional_import_enum_extensionZ optional_string_piece_extensionZoptional_cord_extensionZrepeated_int32_extensionr] Zrepeated_int64_extensionZrepeated_uint32_extensionZrepeated_uint64_extensionZrepeated_sint32_extensionZrepeated_sint64_extensionZrepeated_fixed32_extensionZrepeated_fixed64_extensionZrepeated_sfixed32_extensionZrepeated_sfixed64_extensionZrepeated_float_extensionZrepeated_double_extensionZrepeated_bool_extensionZrepeated_string_extensionZrepeated_bytes_extensionZrepeatedgroup_extensionrm Z!repeated_nested_message_extensionZ"repeated_foreign_message_extensionZ!repeated_import_message_extensionZ repeated_lazy_message_extensionZ repeated_nested_enum_extensionrs Z repeated_foreign_enum_extensionru Z repeated_import_enum_extensionrw Z repeated_string_piece_extensionZrepeated_cord_extensionZdefault_int32_extensionZdefault_int64_extensionZdefault_uint32_extensionZdefault_uint64_extensionZdefault_sint32_extensionZdefault_sint64_extensionZdefault_fixed32_extensionZdefault_fixed64_extensionZdefault_sfixed32_extensionZdefault_sfixed64_extensionZdefault_float_extensionZdefault_double_extensionZdefault_bool_extensionZdefault_string_extensionZdefault_bytes_extensionr Z default_nested_enum_extensionr Z default_foreign_enum_extensionr Z default_import_enum_extensionZ default_string_piece_extensionZdefault_cord_extensionZoneof_uint32_extensionZ oneof_nested_message_extensionZoneof_string_extensionZoneof_bytes_extension)r
extensionspb2Z
import_pb2r r r SetAllExtensions s
r c C s. d| _ d| _d| _d| jtj< d| jtj< dS )zSets every field and extension in the message to a unique value.
Args:
message: A unittest_pb2.TestAllExtensions message.
r2 foog ? barN)my_int my_stringmy_floatr r my_extension_intmy_extension_string)r r r r SetAllFieldsAndExtensionsm s
r c C s t j}t j}g }t j }d|_|j|j |j d|j|< |j|j |j d|_ |j|j |j d|j|< |j|j |j d|_
|j|j |j dj |}|| krt d|| f dS ) zEnsures that serialized is the serialization we expect for a message
filled with SetAllFieldsAndExtensions(). (Specifically, ensures that the
serialization is in canonical, tag-number order).
r2 r r r g ? zExpected %r, found %rN)
r r r ZTestFieldOrderingsr r] ZSerializeToStringZClearr r r join
ValueError)Z
serializedr r Zexpected_stringsr Zexpectedr r r #ExpectAllFieldsAndExtensionsInOrderz s,
r c C s
| j |jd | j |jd | j |jd | j |jd | j |jd | j |jd | j |jd | j |jd | j |jd | j |jd
| j |jd | j |jd | j |jd
| j |jd | j |jd t|r
| j |jd | j |jd | j |jd | j |jd | j |jjd | j |jjd | j |jjd | j |jjd | j |jd | j |jd t|r| j |jd | j |jd | j |jd | jd |j | jd |j | jd |j
| jd |j | jd!|j | jd"|j
| jd#|j | jd$|j | jd%|j | jd&|j | jd'|j | jd(|j | jd)|j | jd*|j | jd+|j t|r| jd,|jj | jd-|jj | jd.|jj | jd/|jj | jd0|jj | jd1|j j | jt j j |j! | jt j"|j# t|rT| jt$j%|j& | jd2t'|j( | jd2t'|j) | jd2t'|j* | jd2t'|j+ | jd2t'|j, | jd2t'|j- | jd2t'|j. | jd2t'|j/ | jd2t'|j0 | jd2t'|j1 | jd2t'|j2 | jd2t'|j3 | jd2t'|j4 | jd2t'|j5 | jd2t'|j6 t|r~| jd2t'|j7 | jd2t'|j8 | jd2t'|j9 | jd2t'|j: | jd2t'|j; | jd2t'|j< t|r| jd2t'|j= | jd2t'|j> | jd2t'|j? | jd3|j(d4 | jd5|j)d4 | jd6|j*d4 | jd7|j+d4 | jd8|j,d4 | jd9|j-d4 | jd:|j.d4 | jd;|j/d4 | jd<|j0d4 | jd=|j1d4 | jd>|j2d4 | jd?|j3d4 | jd)|j4d4 | jd@|j5d4 | jdA|j6d4 t|rD| jdB|j7d4 j | jdC|j8d4 j | jdD|j9d4 j | jdE|j:d4 j | jdF|j@d4 j | jt j jA|j;d4 | jt jB|j r? r@ rA rB rC rD rE rF rG rH rI rK rM rO rJ rL rN rP rU rW rY rZ r[ r r r
r r r
r r r r r r Tr s 116r r r r r r r r r r r r r! r" r# r$ r% r& r' r( s 216r) r* r+ r, r- i- r2 i. i/ i0 i1 i2 i3 i4 i5 i6 i7 i8 Fr3 s 316i= i> i? i@ iG rz r{ r| r} r~ r r r r r r r r r r r r r i i i i i i i i i i i i r6 s 416N)YZ
assertTrueZHasFieldr rI rK rM rO Z assertEqualr: r; r< r= r> r? r@ rA rB rC rD rE rF rG rH rJ rL rN rP rQ rR r r rS rT rU rV rW r rX rY lenr\ r^ r_ r` ra rb rc rd re rf rg rh ri rj rk rl rn ro rp rr rt rv rx ry rq rs ru rw rz r{ r| r} r~ r r r r r r r r r r r r r r r r )Z test_caser r r r ExpectAllFieldsSet sj
r c C s d}xNt jj|rRt jjt jj|drBt jj|d| }t|dS t jj|d}qW d}t jj|d| }t jj|r~t|dS tddS ) zFFinds the given golden file and returns a file object representing it..zsrc/google/protobufz src/google/protobuf/testdatarbz..z'third_party/py/google/protobuf/testdatazCould not find golden files. This test must be run from within the protobuf source package so that it can read test data files from the C++ source tree.N)ospathexistsr open RuntimeError)filenamer full_pathr r r
GoldenFilei s
r c C s t |
}|j S Q R X dS )z5Finds the given golden file and returns its contents.N)r read)r fr r r GoldenFileData s
r c C s | j jddg | jjddg | jjddg | jjddg | jjd d
g | jjd d g | jjd
dg | jjddg | j jddg | j
jddg | j jddg | j jddg | j
jddg | jjtjtjg dS ) zgSets every field in the message to a unique value.
Args:
message: A TestPackedTypes instance.
iY i iZ i i[ i i\ i i] i i^ i i_ i i` i ia i ib i g @g 8@g @g @@TFN)Z packed_int32extendZ packed_int64Z
packed_uint32Z
packed_uint64Z
packed_sint32Z
packed_sint64Zpacked_fixed32Zpacked_fixed64Zpacked_sfixed32Zpacked_sfixed64Z packed_floatZ
packed_doubleZ packed_boolZ packed_enumr ru rV )r r r r SetAllPackedFields s
r c C s* | j }t}||j jddg ||j jddg ||j jddg ||j jddg ||j jd d
g ||j jd d g ||j jd
dg ||j
jddg ||j jddg ||j jddg ||j
jddg ||j jddg ||j jddg ||j jtjtjg dS ) z}Sets every extension in the message to a unique value.
Args:
message: A unittest_pb2.TestPackedExtensions instance.
iY i iZ i i[ i i\ i i] i i^ i i_ i i` i ia i ib i g @g 8@g @g @@TFN)r r Zpacked_int32_extensionr Zpacked_int64_extensionZpacked_uint32_extensionZpacked_uint64_extensionZpacked_sint32_extensionZpacked_sint64_extensionZpacked_fixed32_extensionZpacked_fixed64_extensionZpacked_sfixed32_extensionZpacked_sfixed64_extensionZpacked_float_extensionZpacked_double_extensionZpacked_bool_extensionZpacked_enum_extensionru rV )r r r r r r SetAllPackedExtensions s" r c C s | j jddg | jjddg | jjddg | jjddg | jjd d
g | jjd d g | jjd
dg | jjddg | j jddg | j
jddg | j jddg | j jddg | j
jddg | jjtjtjg dS ) zvSets every field in the message to a unique value.
Args:
message: A unittest_pb2.TestUnpackedTypes instance.
iY i iZ i i[ i i\ i i] i i^ i i_ i i` i ia i ib i g @g 8@g @g @@TFN)Zunpacked_int32r Zunpacked_int64Zunpacked_uint32Zunpacked_uint64Zunpacked_sint32Zunpacked_sint64Zunpacked_fixed32Zunpacked_fixed64Zunpacked_sfixed32Zunpacked_sfixed64Zunpacked_floatZunpacked_doubleZ
unpacked_boolZ
unpacked_enumr ru rV )r r r r SetAllUnpackedFields s
r c @ sZ e Zd ZdZdUddZdd 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=d> Z!d?d@ Z"dAdB Z#dCdD Z$dEdF Z%dGdH Z&dIdJ Z'dKdL Z(dMdN Z)dOdP Z*dQdR Z+dSdT Z,dS )VNonStandardIntegera An integer object that does not subclass int.
This is used to verify that both C++ and regular proto systems can handle
integer others than int and long and that they handle them in predictable
ways.
NonStandardInteger is the minimal legal specification for a custom Integral.
As such, it does not support 0 < x < 5 and it is not hashable.
Note: This is added here instead of relying on numpy or a similar library
with custom integers to limit dependencies.
Nc C s t |tr|j}|| _|| _d S )N)
isinstancer valerror_string_on_conversion)selfr r r r r __init__ s
zNonStandardInteger.__init__c C s | j rt| j t| jS )N)r r longr )r r r r __long__ s
zNonStandardInteger.__long__c C s t tj| jS )N)r operatorabsr )r r r r __abs__ s zNonStandardInteger.__abs__c C s t tj| j|S )N)r r rm r )r yr r r __add__ s zNonStandardInteger.__add__c C s t tj| j|S )N)r r divr )r r r r r __div__ s zNonStandardInteger.__div__c C s t j| j|S )N)r eqr )r r r r r __eq__ s zNonStandardInteger.__eq__c C s t tj| j|S )N)r r floordivr )r r r r r __floordiv__ s z NonStandardInteger.__floordiv__c C s t tj| j|S )N)r r truedivr )r r r r r __truediv__ s z NonStandardInteger.__truediv__c C s t tj| jS )N)r r invertr )r r r r
__invert__ s z NonStandardInteger.__invert__c C s t tj| j|S )N)r r modr )r r r r r __mod__ s zNonStandardInteger.__mod__c C s t tj| j|S )N)r r mulr )r r r r r __mul__ s zNonStandardInteger.__mul__c C s t tj| jS )N)r r negr )r r r r __neg__ s zNonStandardInteger.__neg__c C s t tj| jS )N)r r posr )r r r r __pos__ s zNonStandardInteger.__pos__c C s t tj| j|S )N)r r powr )r r r r r __pow__ s zNonStandardInteger.__pow__c C s
t | jS )N)intr )r r r r __trunc__ s z NonStandardInteger.__trunc__c C s t tj|| jS )N)r r rm r )r r r r r __radd__ s zNonStandardInteger.__radd__c C s t tj|| jS )N)r r r r )r r r r r __rdiv__ s zNonStandardInteger.__rdiv__c C s t tj|| jS )N)r r r r )r r r r r __rmod__ s zNonStandardInteger.__rmod__c C s t tj|| jS )N)r r r r )r r r r r __rmul__ s zNonStandardInteger.__rmul__c C s t tj|| jS )N)r r r r )r r r r r __rpow__ s zNonStandardInteger.__rpow__c C s t tj|| jS )N)r r r r )r r r r r
__rfloordiv__! s z NonStandardInteger.__rfloordiv__c C s t tj|| jS )N)r r r r )r r r r r __rtruediv__$ s z NonStandardInteger.__rtruediv__c C s t tj| j|S )N)r r lshiftr )r r r r r
__lshift__' s z NonStandardInteger.__lshift__c C s t tj| j|S )N)r r rshiftr )r r r r r
__rshift__* s z NonStandardInteger.__rshift__c C s t tj|| jS )N)r r r r )r r r r r __rlshift__- s z NonStandardInteger.__rlshift__c C s t tj|| jS )N)r r r r )r r r r r __rrshift__0 s z NonStandardInteger.__rrshift__c C s t |tr|j}tj| j|S )N)r r r r le)r r r r r __le__3 s
zNonStandardInteger.__le__c C s t |tr|j}tj| j|S )N)r r r r lt)r r r r r __lt__8 s
zNonStandardInteger.__lt__c C s t tj| j|S )N)r r and_r )r r r r r __and__= s zNonStandardInteger.__and__c C s t tj| j|S )N)r r or_r )r r r r r __or__@ s zNonStandardInteger.__or__c C s t tj| j|S )N)r r xorr )r r r r r __xor__C s zNonStandardInteger.__xor__c C s t tj|| jS )N)r r r r )r r r r r __rand__F s zNonStandardInteger.__rand__c C s t tj|| jS )N)r r r r )r r r r r __ror__I s zNonStandardInteger.__ror__c C s t tj|| jS )N)r r r r )r r r r r __rxor__L s zNonStandardInteger.__rxor__c C s | j S )N)r )r r r r __bool__O s zNonStandardInteger.__bool__c C s | j S )N)r )r r r r __nonzero__R s z NonStandardInteger.__nonzero__c C s | S )Nr )r r r r __ceil__U s zNonStandardInteger.__ceil__c C s | S )Nr )r r r r __floor__X s z NonStandardInteger.__floor__c C s | j rt| j t| jS )N)r r r r )r r r r __int__[ s
zNonStandardInteger.__int__c C s | S )Nr )r r r r __round__` s z NonStandardInteger.__round__c C s
d| j S )NzNonStandardInteger(%s))r )r r r r __repr__c s zNonStandardInteger.__repr__)N)-__name__
__module__ __qualname____doc__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 r r r r r r r r r r r r r r r r r sT
r )r
__author__Znumbersr Zos.pathr Zgoogle.protobufr r r NameErrorr r r r r r r r r r r r r ZIntegralr r r r r # s4
R