__ __ __ __ _____ _ _ _____ _ _ _ | \/ | \ \ / / | __ \ (_) | | / ____| | | | | | \ / |_ __\ V / | |__) | __ ___ ____ _| |_ ___ | (___ | |__ ___| | | | |\/| | '__|> < | ___/ '__| \ \ / / _` | __/ _ \ \___ \| '_ \ / _ \ | | | | | | |_ / . \ | | | | | |\ V / (_| | || __/ ____) | | | | __/ | | |_| |_|_(_)_/ \_\ |_| |_| |_| \_/ \__,_|\__\___| |_____/|_| |_|\___V 2.1 if you need WebShell for Seo everyday contact me on Telegram Telegram Address : @jackleetFor_More_Tools:
3
���hnV � @ s( d Z dZddlZddlZddlZddlZddlZddlZddlZddl Z y ddl
Z W n e k
rp ddl Z Y nX ye
W n ek
r� dd� Z
Y nX ddlmZ ddlmZ ddlmZ dd lmZ dd
lmZ dd lmZ dd lmZ dd
lmZ ddlmZ ddlmZ ddlmZ ddlm Z ddlm Z ddlm Z ddlm Z ej �rXe!Z"dd� Z#dd� Z$dd� Z%dd � Z&e j'Z'e j(d efd ef�G d d � d e'��Z)G d!d"� d"e'�Z*G d#d$� d$e'�Z+G d%d&� d&e'�Z,G d'd(� d(e'�Z-e j.ej/� d)k�p�ej0d0k d,�G d-d.� d.e'��Z1e2d/k�r$e j3� dS )1a� Tests python protocol buffers against the golden message.
Note that the golden messages exercise every known field type, thus this
test ends up exercising and verifying nearly all of the parsing and
serialization code in the whole library.
TODO(kenton): Merge with wire_format_test? It doesn't make a whole lot of
sense to call this a test of the "message" module, which only declares an
abstract interface.
z!gps@google.com (Gregory P. Smith)� Nc C s | |k| |k S )N� )�x�yr r �"/usr/lib/python3.6/message_test.py�<lambda>> s r )�map_proto2_unittest_pb2)�map_unittest_pb2)� unittest_pb2)�unittest_proto3_arena_pb2)�descriptor_pb2)�descriptor_pool)�message_factory)� text_format)�api_implementation)�encoder)�packed_field_test_pb2)� test_util)�testing_refleaks)�message)�_parameterizedc C s | | kS )Nr )�valr r r �isnanV s r c C s t | � ot | d �S )Nr )r )r r r r �isinfY s r c C s t | �o| dkS )Nr )r )r r r r �IsPosInf\ s r c C s t | �o| dk S )Nr )r )r r r r �IsNegInf^ s r Z_proto2Z_proto3c
@ s8 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=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,dUdV� Z-dWdX� Z.dYdZ� Z/d[d\� Z0d]d^d_d`dadbe1� g i e2� g
Z3dcdd� Z4dedf� Z5dgdh� Z6didj� Z7dkdl� Z8dmdn� Z9dodp� Z:G dqdr� dre;�Z<dsdt� Z=dudv� Z>dwdx� Z?dydz� Z@d{d|� ZAd}d~� ZBdd�� ZCd�d�� ZDd�d�� ZEd�d�� ZFd�d�� ZGd]S )�� MessageTestc C sX t j� dkr| jd� tjd�}| jt��}|jj|� W d Q R X | j dt
|j �� d S )N�pythonz�Skipping testBadUtf8String, currently only the python api implementation raises UnicodeDecodeError when a string field contains bad utf-8.Zbad_utf8_stringz TestAllTypes.optional_string) r �TypeZskipTestr �GoldenFileData� assertRaises�UnicodeDecodeError� TestAllTypes�
FromString�assertIn�str� exception)�self�message_moduleZ
bad_utf8_data�contextr r r �testBadUtf8Stringj s
z MessageTest.testBadUtf8Stringc C sr |t krtjd�}n
tjd�}|j� }|j|� |t krDtj| |� | j||j� � tj |�}| j||j� � d S )NZ golden_message_oneof_implementedZgolden_message_proto3)
r r r r! �ParseFromStringZExpectAllFieldsSet� assertEqual�SerializeToString�copy�deepcopy)r& r' � golden_data�golden_message� golden_copyr r r �testGoldenMessaget s
z MessageTest.testGoldenMessagec C sh t jd�}|j� }|j|� |j� }t j|� | j||� | j||j� � tj|�}| j||j� � d S )N� golden_packed_fields_message) r r �TestPackedTypesr* ZSetAllPackedFieldsr+ r, r- r. )r& r' r/ r0 �all_setr1 r r r �testGoldenPackedMessage� s
z#MessageTest.testGoldenPackedMessagec C s� |j � }| jt|jd� | jt|jd� tjdd�}tj� dkrv| jt j
��}|j|� W d Q R X | j dt |j
�� nBtjdd��0}tjd � |j|� | j d
t |d j �� W d Q R X d S ) Nr �0� � r zUnexpected end-group tag.T)�record�alwaysz4Unexpected end-group tag: Not all data was converted���)r! r � TypeErrorr" � Exceptionr ZTagBytesr r r � DecodeErrorr+ r$ r% �warnings�catch_warnings� simplefilter)r& r' �msgZend_tagr( �wr r r �testParseErrors� s
zMessageTest.testParseErrorsc
s� d}|j � }|jjddddg� | j||jd d�� | j||jdd�� | j||jdd�� G d d
� d
t�� G � fd d �d t�}| j� �� |j|� d� W d Q R X d S )
Ns5 �
One string�
Two string�
Red string� Blue stringz
One stringz
Two stringz
Red stringz Blue string)�
deterministicFTc @ s e Zd ZdS )z:MessageTest.testDeterminismParameters.<locals>.BadArgErrorN)�__name__�
__module__� __qualname__r r r r � BadArgError� s rJ c s$ e Zd Z� fdd�Z� fdd�ZdS )z5MessageTest.testDeterminismParameters.<locals>.BadArgc s
� � �d S )Nr )r& )rJ r r �__bool__� s z>MessageTest.testDeterminismParameters.<locals>.BadArg.__bool__c s
� � �d S )Nr )r& )rJ r r rK � s N)rG rH rI rK r )rJ r r �BadArg� s rL )r! �repeated_string�extendr+ r, r> �objectr )r& r' r/ r0 rL r )rJ r �testDeterminismParameters� s" z%MessageTest.testDeterminismParametersc C s@ t jd�}|j� }|j|� tj|�}tj|�}| j||� d S )Nr0 )r r r! r* �pickle�dumps�loadsr+ )r& r' r/ r0 �pickled_message�unpickled_messager r r �testPickleSupport� s
z MessageTest.testPickleSupportc C s� |t krd}nd}|j� }|j|� | jt|j�� | jt|j�� | jt|jd �� | jt|jd �� | j ||j
� � d S )Ns ] �a �� �� �s ] �a �� �� �r ) r r! r* �
assertTruer �optional_float�optional_double�repeated_float�repeated_doubler+ r, )r& r' r/ r0 r r r �testPositiveInfinity� s
z MessageTest.testPositiveInfinityc C s� |t krd}nd}|j� }|j|� | jt|j�� | jt|j�� | jt|jd �� | jt|jd �� | j ||j
� � d S )Ns ] ��a �� ��� �s ] ��a �� ��� �r ) r r! r* rW r rX rY rZ r[ r+ r, )r& r' r/ r0 r r r �testNegativeInfinity� s
z MessageTest.testNegativeInfinityc C s� d}|j � }|j|� | jt|j�� | jt|j�� | jt|jd �� | jt|jd �� |j� }|j � }|j|� | jt|j�� | jt|j�� | jt|jd �� | jt|jd �� d S )Ns ] �a �� �� �r ) r! r* rW r rX rY rZ r[ r, )r& r' r/ r0 �
serializedr r r r �testNotANumber� s
zMessageTest.testNotANumberc C sR d}|j � }|j|� | jt|jd �� | jt|jd �� | j||j� � d S )Ns � �� �r )r4 r* rW r � packed_float�
packed_doubler+ r, )r& r' r/ r0 r r r �testPositiveInfinityPacked s
z&MessageTest.testPositiveInfinityPackedc C sR d}|j � }|j|� | jt|jd �� | jt|jd �� | j||j� � d S )Ns � ��� �r )r4 r* rW r r` ra r+ r, )r& r' r/ r0 r r r �testNegativeInfinityPacked s
z&MessageTest.testNegativeInfinityPackedc C s� d}|j � }|j|� | jt|jd �� | jt|jd �� |j� }|j � }|j|� | jt|jd �� | jt|jd �� d S )Ns � �� �r )r4 r* rW r r` ra r, )r& r' r/ r0 r^ r r r r �testNotANumberPacked% s
z MessageTest.testNotANumberPackedc C st |j � }tjdd�}||_|j|j� � | j|j|k� dtjdd� }||_|j|j� � | j|j|k� | |_|j|j� � | j|j| k� | |_|j|j� � | j|j| k� tjdd�}||_|j|j� � | j|j|k� dtjdd� }||_|j|j� � | j|j|k� | |_|j|j� � | j|j| k� | |_|j|j� � | j|j| k� d S )N� � g �?i����i����)r! �math�powrX r* r, rW )r& r' r �kMostPosExponentNoSigBits�kMostPosExponentOneSigBit�kMostNegExponentNoSigBits�kMostNegExponentOneSigBitr r r �testExtremeFloatValues3 s: z"MessageTest.testExtremeFloatValuesc C st |j � }tjdd�}||_|j|j� � | j|j|k� dtjdd� }||_|j|j� � | j|j|k� | |_|j|j� � | j|j| k� | |_|j|j� � | j|j| k� tjdd�}||_|j|j� � | j|j|k� dtjdd� }||_|j|j� � | j|j|k� | |_|j|j� � | j|j| k� | |_|j|j� � | j|j| k� d S )Nre i� g �?i���i���)r! rg rh rY r* r, rW )r& r' r ri rj rk rl r r r �testExtremeDoubleValues` s: z#MessageTest.testExtremeDoubleValuesc C s" |j � }d|_| jt|�d� d S )Ng @zoptional_float: 2.0
)r! rX r+ r$ )r& r' r r r r �testFloatPrinting� s z MessageTest.testFloatPrintingc C s> |j � }d|_tjdkr*| jt|�d� n| jt|�d� d S )Ng^!�7ݚ�?� z%optional_double: 0.12345678912345678
z optional_double: 0.123456789123
)rp )r! rY �sys� version_infor+ r$ )r& r' r r r r � testHighPrecisionFloatPrinting� s
z*MessageTest.testHighPrecisionFloatPrintingc C s< |j � }tj|� |j� }|j|j� � | jt|�d� d S )N� )r! r ZSetAllNonLazyFieldsZTestEmptyMessager* r, r+ r$ )r& r' Z populated�emptyr r r �testUnknownFieldPrinting� s
z$MessageTest.testUnknownFieldPrintingc C s� |j � }|jjdd� |jjdd� |jjdd� |jjdd� | jddddgdd� |jD �� | jddddgdd� t|j�D �� | jddddgd d� |jd d d
� D �� d S ) Nr8 )�bbre rp r9 c S s g | ]
}|j �qS r )rw )�.0�mr r r �
<listcomp>� s z@MessageTest.testRepeatedNestedFieldIteration.<locals>.<listcomp>c S s g | ]
}|j �qS r )rw )rx ry r r r rz � s c S s g | ]
}|j �qS r )rw )rx ry r r r rz � s r< )r! �repeated_nested_message�addr+ �reversed)r& r' rC r r r � testRepeatedNestedFieldIteration� s z,MessageTest.testRepeatedNestedFieldIterationc C s� |j � }|jjd� |jjd� |jjd� |jj� | j|jd d� | j|jd d� | j|jd d� | jt|j�tdddg�� |jjd� |jjd� |jjd� |jj� | j|jd d� | j|jd d� | j|jd d� |jjd� |jjd � |jjd
� |jj� | j|jd d� | j|jd d
� | j|jd d � | jt|j�tdd
d g�� |j jd � |j jd � |j jd
� |j j� | j|j d d � | j|j d d
� | j|j d d � | jt|j �td d
d g�� dS )z7Check some different types with the default comparator.r8 rp re r g�������?g�������?g333333�?�a�c�b� a� c� bN)
r! �repeated_int32�append�sortr+ r$ rZ �assertAlmostEqualrM Zrepeated_bytes)r& r' r r r r �0testSortingRepeatedScalarFieldsDefaultComparator� s@
z<MessageTest.testSortingRepeatedScalarFieldsDefaultComparatorc C s� |j � }|jjd
� |jjd � |jjd � |jjtd� | j|jd d
� | j|jd d� | j|jd d� |jjd� |jjd� |jjd� |jjtd� | j|jd d� | j|jd d� | j|jd d� d S )z2Check some different types with custom comparator.rp re r8 )�keyr �aaarw r� N������r< r< r� r� )r! r� r� r� �absr+ rM �len)r& r' r r r r �/testSortingRepeatedScalarFieldsCustomComparator� s z;MessageTest.testSortingRepeatedScalarFieldsCustomComparatorc C s� |j � }d|jj� _d|jj� _d|jj� _d|jj� _d|jj� _d|jj� _|jjtjd�d� | j|jd jd� | j|jd jd� | j|jd jd� | j|jd jd� | j|jd jd� | j|jd jd� | jt|j�d
� d S ) zECheck passing a custom comparator to sort a repeated composite field.r8 rp re � � r9 rw )r� r z0[bb: 1
, bb: 2
, bb: 3
, bb: 4
, bb: 5
, bb: 6
]N) r! r{ r| rw r� �operator�
attrgetterr+ r$ )r& r' r r r r �2testSortingRepeatedCompositeFieldsCustomComparator� s z>MessageTest.testSortingRepeatedCompositeFieldsCustomComparatorc C s� |j � }d|jj� _d|jj� _d|jj� _d|jj� _d|jj� _d|jj� _d|jj� _|jjdd � d
� | jdddddddgd d � |jD �� |j� }|j� |j|� | jdddddddgd
d � |jD �� dS )zECheck passing a custom comparator to sort a repeated composite field.� � �
�! � � �
c S s
| j d S )Nr� )rw )�zr r r r
s zFMessageTest.testSortingRepeatedCompositeFieldsStable.<locals>.<lambda>)r� c S s g | ]
}|j �qS r )rw )rx �nr r r rz
s zHMessageTest.testSortingRepeatedCompositeFieldsStable.<locals>.<listcomp>c S s g | ]
}|j �qS r )rw )rx r� r r r rz s N) r! r{ r| rw r� r+ r, �ClearZMergeFromString)r& r' r �pbr r r �(testSortingRepeatedCompositeFieldsStable� s$
z4MessageTest.testSortingRepeatedCompositeFieldsStablec C s@ |j � }tjd�}dd� }d|jj� _d|jj� _d|jj� _d|jj� _d|jj� _d |jj� _|jj|d
� | jd d � |jD �dddd ddg� |jj|d
d� | jdd � |jD �ddd dddg� tj dkr�dS |jj|d� | jdd � |jD �dddd ddg� |jj|d
d� | jdd � |jD �ddd dddg� dS )zECheck sorting a repeated composite field using list.sort() arguments.rw c S s t | j|j�S )N)�cmprw )r r� r r r r s zEMessageTest.testRepeatedCompositeFieldSortArguments.<locals>.<lambda>r8 rp re r� r� r9 )r� c S s g | ]
}|j �qS r )rw )rx �kr r r rz % s zGMessageTest.testRepeatedCompositeFieldSortArguments.<locals>.<listcomp>T)r� �reversec S s g | ]
}|j �qS r )rw )rx r� r r r rz ( s N)�
sort_functionc S s g | ]
}|j �qS r )rw )rx r� r r r rz , s )r� r� c S s g | ]
}|j �qS r )rw )rx r� r r r rz / s )rp )
r! r� r� r{ r| rw r� r+ rq rr )r& r' r Zget_bbZcmp_bbr r r �'testRepeatedCompositeFieldSortArguments s.
z3MessageTest.testRepeatedCompositeFieldSortArgumentsc C s� |j � }|jjd� |jjd� |jjd� |jjtd� | jt|j�dddg� |jjtdd� | jt|j�dddg� tjdk r�dd� }|jj|d � | jt|j�ddd g� |jj|dd
� | jt|j�d d d g� |j jd � |j jd � |j jd
� |j jt
d� | jt|j �d
d d g� |j jt
dd� | jt|j �d d d
g� tjd k �r�dd� }|j j|d � | jt|j �d
d d g� |j j|dd
� | jt|j �d d d
g� dS )!z9Check sorting a scalar field using list.sort() arguments.rp re r8 )r� T)r� r� c S s t t| �t|��S )N)r� r� )r r� r r r r > s zBMessageTest.testRepeatedScalarFieldSortArguments.<locals>.<lambda>)r� )r� r� r� rw r� c S s t t| �t|��S )N)r� r� )r r� r r r r L s Nr� r� r< r< r� r� r� r� r< )rp r< r� r� r� r� r< )rp ) r! r� r� r� r� r+ �listrq rr rM r� )r& r' r Zabs_cmpZlen_cmpr r r �$testRepeatedScalarFieldSortArguments2 s6
z0MessageTest.testRepeatedScalarFieldSortArgumentsc C sJ |j � }|j � }|jjd� |jjd� |jjd� |jjd� |jjd� |jjd� d|jj� _d|jj� _d|jj� _d|jj� _d|jj� _d|jj� _tjdkr�d S ||k }|j|jk }| jt ||�d� | jt |j|j�d� | jt |jdddg�d� | jt |j|j�d� | j
t �� t |j|jd d � � W d Q R X d S )Nr r8 re rp )rp ) r! r� r� r{ r| rw rq rr r+ r� r r= )r& r' �m1�m2�_r r r � testRepeatedFieldsComparableR s2
z(MessageTest.testRepeatedFieldsComparablec C s, |j � }| j|jtj� | j|jtj� d S )N)r! �assertIsInstancer� � collections�MutableSequencer{ )r& r' ry r r r � testRepeatedFieldsAreSequencesu s z*MessageTest.testRepeatedFieldsAreSequencesc C sL |j � }| jt�� t|j� W d Q R X | jt�� t|j� W d Q R X d S )N)r! r r= �hashr� r{ )r& r' ry r r r � testRepeatedFieldsNotHashable{ s
z)MessageTest.testRepeatedFieldsNotHashablec C s* |j � }|jjjg � | j|jd�� d S )N�payload)�NestedTestAllTypesr� r� rN rW �HasField)r& r' ry r r r �$testRepeatedFieldInsideNestedMessage� s z0MessageTest.testRepeatedFieldInsideNestedMessagec C s t ||� | j|j|�� dS )z�Make sure that a nested message object exists.
As soon as a nested message attribute is accessed, it will be present in the
_fields dict, without being marked as actually being set.
N)�getattr� assertFalser� )r& rC Z attributer r r �ensureNestedMessageExists� s
z%MessageTest.ensureNestedMessageExistsc C s, |j � }| jt|jd� | jt|jd� d S )NZno_such_oneof_fieldr )r! r �
ValueError�
WhichOneofr> )r& r' ry r r r � testOneofGetCaseNonexistingField� s z,MessageTest.testOneofGetCaseNonexistingFieldc C s� |j � }| jd |jd�� | j|jd�� d|_| jd|jd�� | j|jd�� | j|jd�� d|_| jd|jd�� | j|jd�� | j|jd�� d S )N� oneof_field� oneof_uint32r � oneof_stringrt ) r! �assertIsr� r� r� r� r+ rW r� )r& r' ry r r r �testOneofDefaultValues� s z"MessageTest.testOneofDefaultValuesc C sb |j � }| jd |jd�� d|_| jd|jd�� | j|jd�� d|_| jd|jd�� | j|jd�� | j|jd�� |j | jd|jd�� | j|jd�� | j|jd�� |j j
| jd|jd�� | j|jd�� | j|jd�� d|j _
| jd|jd�� | j|jd�� | j|jd�� d|_ | jd|jd�� | j|jd�� | j|jd�� d S ) Nr� r� r� �foor� �oneof_nested_messages bb� oneof_bytes) r! r� r� r� r+ rW r� r� r� r� rw r� )r& r' ry r r r �testOneofSemantics� s2 z MessageTest.testOneofSemanticsc C s> |j � }d|_| j|d� | jd|jd�� | jd|j� d S )Nr� r� r� r� )r! r� r� r+ r� )r& r' ry r r r �!testOneofCompositeFieldReadAccess� s
z-MessageTest.testOneofCompositeFieldReadAccessc C s� |j � }| jd |jd�� |tkr2| j|jd�� d|_| jd|jd�� |tkrb| j|jd�� d|_ | jd|jd�� |j
d� | jd |jd�� |tkr�| j|jd�� d S )Nr� r� r� s bbr� ) r! r� r� r r� r� r� r+ rW r� �
ClearField)r& r' ry r r r �testOneofWhichOneof� s
z MessageTest.testOneofWhichOneofc C sV |j � }d|_|jd� |tkr0| j|jd�� | j|jd�� | jd |jd�� d S )Nr� r� r� )r! r� r� r r� r� r� r� )r& r' ry r r r �testOneofClearField� s
z MessageTest.testOneofClearFieldc C sV |j � }d|_|jd� |tkr0| j|jd�� | j|jd�� | jd |jd�� d S )Nr� r� r� )r! r� r� r r� r� r� r� )r& r' ry r r r �testOneofClearSetField� s
z"MessageTest.testOneofClearSetFieldc C sp |j � }d|_| j|d� |jd� | jd|j� |tkrJ| j|jd�� | j|jd�� | jd|jd�� d S )Nr� r� r� r� ) r! r� r� r� r+ r rW r� r� )r& r' ry r r r �testOneofClearUnsetField� s
z$MessageTest.testOneofClearUnsetFieldc C s: |j � }d|_|j � }|j|j� � | jd|jd�� d S )Nr� r� r� )r! r� r* r, r+ r� )r& r' ry r� r r r �testOneofDeserialize s
z MessageTest.testOneofDeserializec C s6 |j � }d|_|j � }|j|� | jd|jd�� d S )Nr� r� r� )r! r� �CopyFromr+ r� )r& r' ry r� r r r �testOneofCopyFrom s
z MessageTest.testOneofCopyFromc C sb |j � }d|j_|j � }d|j_d|jj_|j|� | jd|jjd�� | jd|jjjd�� d S )Nr� s bbr� r� r� )r� r� r� r� �child� MergeFromr+ r� )r& r' ry r� r r r �testOneofNestedMergeFrom s
z$MessageTest.testOneofNestedMergeFromc C sf |j � }d|jj_d|jjj_|j � }d|j_|j|� | jd|jjd�� | jd|jjjd�� d S )Nr� � r� r� r� ) r� r� r� rw r� r� r� r+ r� )r& r' ry r� r r r �testOneofMessageMergeFrom s
z%MessageTest.testOneofMessageMergeFromc C s( |j |j j� d�}| jd|jd�� d S )N)r� r� r� )r! �
NestedMessager+ r� )r& r' ry r r r �testOneofNestedMessageInit$ s z&MessageTest.testOneofNestedMessageInitc C sB |j � }d|_|j� | j|jd�� d|_| jd|jd�� d S )Nr� r� s bbr� )r! r� r� � assertIsNoner� r� r+ )r& r' ry r r r �testOneofClear) s zMessageTest.testOneofClearc C s& |j � }td�|_| j|jtj� dS )zmAssigning a byte string to a string field should result
in the value being converted to a Unicode string.rt N)r! r$ �optional_stringr� �six� text_type)r& r' ry r r r �"testAssignByteStringToUnicodeField1 s
z.MessageTest.testAssignByteStringToUnicodeFieldc C s� |j � }|jjd� |jtd�tt|j��� }| jt|j�t|�� d|jj� _|jtd�tt|j��� }| jt|j�t|�� dS )z|It should be possible to use long-valued indicies in slices
This didn't used to work in the v2 C++ implementation.
r8 r rp N) r! r� r� �intr� r+ r{ r| rw )r& r' ry Zslr r r �testLongValuedSlice8 s z MessageTest.testLongValuedSlicec C sl |j � }| jt�� }|jjdd� td�D �� W dQ R X | jt�� }|jjdd� td�D �� W dQ R X dS )z5This didn't use to work in the v2 C++ implementation.c s s | ]
}t V qd S )N)r )rx �ir r r � <genexpr>M s zCMessageTest.testExtendShouldNotSwallowExceptions.<locals>.<genexpr>r� Nc s s | ]
}t V qd S )N)r )rx r� r r r r� P s )r! r � NameErrorr� rN �range�repeated_nested_enum)r& r' ry r� r r r �$testExtendShouldNotSwallowExceptionsI s $ z0MessageTest.testExtendShouldNotSwallowExceptionsNFr g � rt c C s^ |j � }| jg |j� x(tjD ] }|jj|� | jg |j� q W |jjg � | jg |j� dS )z,Test no-ops extending repeated int32 fields.N)r! �assertSequenceEqualr� r � FALSY_VALUESrN )r& r' ry � falsy_valuer r r �testExtendInt32WithNothingT s z&MessageTest.testExtendInt32WithNothingc C s^ |j � }| jg |j� x(tjD ] }|jj|� | jg |j� q W |jjg � | jg |j� dS )z,Test no-ops extending repeated float fields.N)r! r� rZ r r� rN )r& r' ry r� r r r �testExtendFloatWithNothinga s z&MessageTest.testExtendFloatWithNothingc C s^ |j � }| jg |j� x(tjD ] }|jj|� | jg |j� q W |jjg � | jg |j� dS )z-Test no-ops extending repeated string fields.N)r! r� rM r r� rN )r& r' ry r� r r r �testExtendStringWithNothingn s z'MessageTest.testExtendStringWithNothingc C s� |j � }| jg |j� |jjdg� | jdg|j� |jjddg� | jdddg|j� |jjddg� | jdddddg|j� dS )z7Test extending repeated int32 fields with python lists.r r8 re rp r9 N)r! r� r� rN )r& r' ry r r r � testExtendInt32WithPythonList{ s z)MessageTest.testExtendInt32WithPythonListc C s� |j � }| jg |j� |jjdg� | jdg|j� |jjddg� | jdddg|j� |jjddg� | jdddddg|j� dS )z7Test extending repeated float fields with python lists.g g �?g @g @g @N)r! r� rZ rN )r& r' ry r r r � testExtendFloatWithPythonList� s z)MessageTest.testExtendFloatWithPythonListc C s� |j � }| jg |j� |jjdg� | jdg|j� |jjddg� | jdddg|j� |jjddg� | jdddddg|j� dS )z8Test extending repeated string fields with python lists.rt Z11Z22Z33Z44N)r! r� rM rN )r& r' ry r r r � testExtendStringWithPythonList� s z*MessageTest.testExtendStringWithPythonListc C s: |j � }| jg |j� |jjd� | jdddg|j� dS )zDTest extending repeated string fields with characters from a string.�abcr r� r� N)r! r� rM rN )r& r' ry r r r �testExtendStringWithString� s z&MessageTest.testExtendStringWithStringc @ s2 e Zd ZdZd dd�Zdd� Zdd� Zd d
� ZdS ) zMessageTest.TestIterablez�This iterable object mimics the behavior of numpy.array.
__nonzero__ fails for length > 1, and returns bool(item[0]) for length == 1.
Nc C s |pg | _ d S )N)�_list)r& �valuesr r r �__init__� s z!MessageTest.TestIterable.__init__c C s8 t | j�}|dkrdS |dkr,t| jd �S td��d S )Nr Fr8 zTruth value is ambiguous.)r� r� �boolr� )r& �sizer r r rK � s
z!MessageTest.TestIterable.__bool__c C s
t | j�S )N)r� r� )r& r r r �__len__� s z MessageTest.TestIterable.__len__c C s
| j j� S )N)r� �__iter__)r& r r r r� � s z!MessageTest.TestIterable.__iter__)N)rG rH rI �__doc__r� rK r� r� r r r r � TestIterable� s
r� c C s� |j � }| jg |j� |jjtjg �� | jg |j� |jjtjdg�� | jdg|j� |jjtjddg�� | jdddg|j� |jjtjddg�� | jdddddg|j� dS )z3Test extending repeated int32 fields with iterable.r r8 re rp r9 N)r! r� r� rN r r� )r& r' ry r r r �testExtendInt32WithIterable� s z'MessageTest.testExtendInt32WithIterablec C s� |j � }| jg |j� |jjtjg �� | jg |j� |jjtjdg�� | jdg|j� |jjtjddg�� | jdddg|j� |jjtjddg�� | jdddddg|j� dS )z3Test extending repeated float fields with iterable.g g �?g @g @g @N)r! r� rZ rN r r� )r& r' ry r r r �testExtendFloatWithIterable� s z'MessageTest.testExtendFloatWithIterablec C s� |j � }| jg |j� |jjtjg �� | jg |j� |jjtjdg�� | jdg|j� |jjtjddg�� | jdddg|j� |jjtjddg�� | jdddddg|j� dS )z4Test extending repeated string fields with iterable.rt �1�2�3�4N)r! r� rM rN r r� )r& r' ry r r r � testExtendStringWithIterable� s z(MessageTest.testExtendStringWithIterablec
C sP t j� dkst j� dkr d S |j� }| jtj��}tj|jtj � W d Q R X d S )N�cppre )
r r ZVersionr! r rQ Z PickleErrorrR r� ZHIGHEST_PROTOCOL)r& r' ry r� r r r �!testPickleRepeatedScalarContainer� s z-MessageTest.testPickleRepeatedScalarContainerc C s |j � }|jj� dS )z?Exercise a scenario that has led to segfaults in the past.
N)r! r{ r� )r& r' ry r r r �'testSortEmptyRepeatedCompositeContainer� s z3MessageTest.testSortEmptyRepeatedCompositeContainerc C s, |j � }| jt��}|jd� W dQ R X dS )zBUsing HasField on a repeated field should raise an exception.
r� N)r! r r� r� )r& r' ry r� r r r �testHasFieldOnRepeatedField� s z'MessageTest.testHasFieldOnRepeatedFieldc C s� |j � }| jt��}|jj� W d Q R X |jjttd��� | jd|jj� � | jd|jjd�� | jd|jjd�� | jddg|j� d S )Nr� r9 r re r8 rp ) r! r �
IndexErrorr� �poprN r� r� r+ )r& r' ry r� r r r �testRepeatedScalarFieldPop� s z&MessageTest.testRepeatedScalarFieldPopc C s� |j � }| jt��}|jj� W d Q R X | jt��}|jjd� W d Q R X x td�D ]}|jj� }||_qTW | j d|jj� j� | j d|jjd�j� | j d|jjd�j� | j ddgdd � |jD �� d S )
Nr7 r� r9 r re r8 rp c S s g | ]
}|j �qS r )rw )rx r� r r r rz s z=MessageTest.testRepeatedCompositeFieldPop.<locals>.<listcomp>)
r! r r r{ r r= r� r| rw r+ )r& r' ry r� r� r� r r r � testRepeatedCompositeFieldPop s
z)MessageTest.testRepeatedCompositeFieldPopc C s\ |j � }x.td�D ]"}|jj||� |jj� }||_qW | j|j|j� | j|j|j� d S )Nr� ) r! r� r� �insertr{ r| rw r� r+ )r& r' ry r� r� r r r �testRepeatedCompareWithSelf s
z'MessageTest.testRepeatedCompareWithSelfc C s6 |j � }|jj� }|j}|jj� }| j|jjd� dS )zwA case that lead to a segfault when a message detached from its parent
container has itself a child container.
r N)r� Zrepeated_childr| r� r+ r� �optional_int32)r& r' ry r r r �testReleasedNestedMessages s
z&MessageTest.testReleasedNestedMessagesc C sz |j � }| jt�� g |_W d Q R X |jjd� tj� dkrZ| jt�� g |_W d Q R X n | jt�� g |_W d Q R X d S )Nr8 r )r! r �AttributeErrorr� r� r r r= )r& r' ry r r r �testSetRepeatedComposite( s z$MessageTest.testSetRepeatedComposite)HrG rH rI r) r2 r6 rE rP rV r\ r] r_ rb rc rd rm rn ro rs rv 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� � bytearray�setr� r� r� r� r� r� r� r� rO r� r� r� r� r r r r r r r r
r r r r r e s�
$ --( # #
r c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dS )�
Proto2Testc C s� t j� }| j|jd�� | j|jd�� | j|jd�� | jt�� |jd� W d Q R X | jt�� |jd� W d Q R X | jt�� |jd� W d Q R X | jd|j� | jd|j� | jd|j j
� d|_d|_d|j _
| j |jd�� | j |jd�� | j |jd�� d |_d
|_d |j _
| j |jd�� | j |jd�� | j |jd�� |j d� |j d� |j d� | j|jd�� | j|jd�� | j|jd�� | jd|j� | jd|j� | jd|j j
� d S ) Nr
�
optional_bool�optional_nested_message�field_doesnt_existr� r{ r Fr� T� )
r r! r� r� r r� r+ r
r r rw rW r� )r& r r r r �testFieldPresence: sD
z Proto2Test.testFieldPresencec C s
t j� }| jt�� }d|_W dQ R X | jt|jjd� |jjd� d|jd< | jt�� d|jd< W dQ R X tj� }d|_|jjd� |j� }t j� }|j |� | j
|j d�� | j d|j� | j dt
|j�� |j� |j |j� � | j d|j� | j d|jd � dS ) z:Assigning an invalid enum number is not allowed in proto2.i�� Nre r i@� i��t �optional_nested_enumr8 )r r! r r� r r� r� r
r, r* r� r� r+ r� r� )r& ry r� r� r^ Zm3r r r �testAssignInvalidEnumj s*
z Proto2Test.testAssignInvalidEnumc
C s6 t j� }d|jd< | jt�� d|jd< W d Q R X d S )Nr �{ r8 )r Z TestEnumMapZknown_map_fieldr r� Zunknown_map_field)r& ry r r r �testUnknownEnumMap� s
z Proto2Test.testUnknownEnumMapc C s t j� }| jtt|d� d S )N�
Extensions)r r! r r r� )r& rC r r r �testExtensionsErrors� s z Proto2Test.testExtensionsErrorsc C sh t jd�}tj� }|j|� tj� }t j|� | j||� | j||j� � tj |�}| j||j� � d S )Nr0 )
r r r ZTestAllExtensionsr* ZSetAllExtensionsr+ r, r- r. )r& r/ r0 r5 r1 r r r �testGoldenExtensions� s
z Proto2Test.testGoldenExtensionsc C sh t jd�}tj� }|j|� tj� }t j|� | j||� | j||j� � tj |�}| j||j� � d S )Nr3 )
r r r ZTestPackedExtensionsr* ZSetAllPackedExtensionsr+ r, r- r. )r& r/ r0 r5 r1 r r r �testGoldenPackedExtensions� s
z%Proto2Test.testGoldenPackedExtensionsc C sN t jdd�}tj|�}tj|�}| j||� | j|jd� | jtj |j
� d S )Nr8 )r ) r Z TestRequiredrQ rR rS r+ r r r Z EncodeErrorr, )r& r0 rT rU r r r �testPickleIncompleteProto� s
z$Proto2Test.testPickleIncompleteProtoc C s� t j� t j� t j� g}d|d _d|d _d|d _d|d _t j� }d|_d|_d|_t jj� }|jj|� |j j|� |j
j|� |j j|� |j j|� |j
j� jj|d � |j
j� jj|d � |j
j� jj|d � |jj� jj|d � |jj� jj|d � |jj� jj|d � |j� }t j� }|j|� | j|j|� | j|j|� | j|jj|� | j|jt jj |� | jt|j�d� | jt|j �d� | jt|jt jj �d� dS )zcCheck the merge behavior when a required or optional field appears
multiple times in the input.r8 r re rp ZhelloN) r r! r
�optional_int64r� ZTestParsingMergeZRepeatedFieldsGeneratorZfield1rN Zfield2Zfield3Zext1Zext2Zgroup1r| r� Zgroup2r, r* r+ Zrequired_all_typesZoptional_all_types�
optionalgroupZoptional_group_all_typesr Z optional_extr� Zrepeated_all_types�
repeatedgroupZ repeated_ext)r& ZmessagesZmerged_message� generator�dataZ
parsing_merger r r �testParsingMerge� sJ
zProto2Test.testParsingMergec C s` t jddddddiddii d dd
idd igd t jjgd
dd� }| j|t j� | jd|j� | jd|j� | jd|j� | jd|j� | jd|j j
� | j|j t jj � | jd|j j
� | j|jd�� | j|jt j� � | jt jj|j� | jdt|j�� | jd
|jd j
� | jd |jd j
� | jdt|j�� | jt jj|jd � | jt jj|jd � | jd
|j� | jd|j� | j|jd�� | jdt|j�� | jd|j � t jd d�}| jt jj|j� | j t �� t jddid� W d Q R X | j t �� t jddid� W d Q R X | j t �� t jdd� W d Q R X | j t �� t jd d � W d Q R X d S ) N�d �� g �r@� xr i� rw i� �BAZiX i� �FOOi r ) r
�optional_fixed32rX �optional_bytesr r �optional_foreign_messager r! r� �
default_int32r� r, re r r8 r �* )r ZINVALID_NESTED_FIELD� )r ZINVALID_VALUE_TYPEZ
INVALID_LABEL)r� ) r r! ZBARr� r+ r
r* rX r+ r r r r� rw rW r� r, �ForeignMessager( r r� r! r� r) r- r� r� rZ Z
default_int64r r� r= )r& r r r r �testPythonicInit� sl
zProto2Test.testPythonicInitN) rG rH rI r r r r r r r r$ r1 r r r r r 8 s 0 1r 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=d>� Z!d?d@� Z"dAS )B�
Proto3Testc C s@ t |�}x&|D ] \}}| j||| � ||= qW | ji |� d S )N)�dictr+ )r& Zmap_iterZ
dict_valuer� �vr r r �assertMapIterEquals+ s
z Proto3Test.assertMapIterEqualsc 4 C s� t j� }| jt�� |jd� W d Q R X | jt�� |jd� W d Q R X | jt�� |jd� W d Q R X | jt�� |jd� W d Q R X | j|jd�� | jt�� |jd� W d Q R X | jt�� |jd� W d Q R X | jt�� |jd� W d Q R X | jd |j� | jd |j� | jd
|j � | jd |j
� | jd |j j � d |j _ | j
|jd�� d |_d
|_d|_ d|_
d|j _ |jd� |jd� |jd� |jd� |jd� | jd |j� | jd |j� | jd
|j � | jd |j
� | jd |j j � d S )Nr
rX r� r r r r� r{ r rt Fr� g�������?r� Tr )r
r! r r� r� r� r+ r
rX r� r r rw rW r� )r& r r r r r 5 sL
z Proto3Test.testFieldPresencec C s� t j� }d|_| jd|j� |jjd� | jd|jd � d|jd< | jd|jd � |j� }t j� }|j|� | jd|j� | jd|jd � dS )zCAssigning an unknown enum value is allowed and preserves the value.i�� i��Tr i��t N)r
r! r r+ r� r� r, r* )r& ry r^ r� r r r �testAssignUnknownEnumn s
z Proto3Test.testAssignUnknownEnumc C sN t j� }| jd |jk� | jd|jk� | jd|jk� | jd|jk� | jd|jk� | jd|jk� | jd|j k� | jd|j
k� | jd|j k� | j d|jd � | j d|jd � | j d|jd � | j d|jd � | j d |jd � | j
t|jd t�� | j d|jd � | j
t|jd t�� | j d
|j d � | j d |j
d � | j d|j d � | j
d|jk� | j
d|jk� | j
d|jk� | j
d|jk� | j
d|jk� | j
d|jk� | j
d|j k� | j
d|j
k� | j
d|j k� | j|j d tj� | jt�� |j d W d Q R X | jt�� d|j k W d Q R X d S )Nr re r� Fr� �o ix r g rt r� i����l l���� l i����l l���� l i����l l���� l )r �TestMapr� �map_int32_int32�map_int64_int64�map_uint32_uint32�map_uint64_uint64�map_int32_double�
map_bool_bool�map_string_stringZmap_int32_bytes�map_int32_enumr+ rW �
isinstance�floatr� r� r� r� r r= )r& rC r r r �testScalarMapDefaults� sF z Proto3Test.testScalarMapDefaultsc C s& t j� }| j|jjd�� | jd|jjdd�� | j|jjd�� d|jd< | jd|jjd�� | jd|jjd�� | jt�� |jjd� W d Q R X | j|jjd�� | jd|jjdd�� |jd }| j ||jjd�� t
j � dk�r| jt�� |jjd� W d Q R X n| jd |jjd�� d S )Nr� r� r rt r ) r r8 r� r9 �getr+ r r= �map_int32_foreign_messager� r r )r& rC �submsgr r r �
testMapGet� s"
zProto3Test.testMapGetc ! C s4 t j� }| jdt|j�� | jd|jk� d|jd< d|jd< d|jd< d|jd< d|j d< d |j
d
< d |j d < d
|j d
< d|j
d< d|j
d< | jg |j� � | jd
t|j �� | jt�� d |j d< W d Q R X | jd
t|j �� | jt�� d|j d < W d Q R X |j� }t j� }|j|� | jt�� d |j d< W d Q R X | jt�� d|j d < W d Q R X | jd|jd � | jd |jd � | jd|jd � | jd |jd � | jd|j d � | jd |j
d
� | jd |j d � | jd
|j d
� | jd|j
d � | jd|j
d � tj� dk�r0| jdt|j�� d S )!Nr r� i� r re �" r� g333333�?gffffff
@r8 �123r� Tix r z {-123: -456}i8���i����l l���� l l���� l l i8���i����l l���� l l���� l l )r r8 r+ r� r9 r� r: r; r< Zmap_int32_floatr= r? r> r@ �FindInitializationErrorsr r= r, r* r� r r r$ )r& rC r^ �msg2r r r �
testScalarMap� sP
zProto3Test.testScalarMapc C s@ t j� }d|jd< d|jd< | j|j� d� | jd|j� � d S )Nr rt r� s
r
)r r8 r9 r? r+ �ByteSizer, )r& rC r r r � testMapEntryAlwaysSerialized s
z'Proto3Test.testMapEntryAlwaysSerializedc C sn t j� }d}|jd�}||j|< t|jj� �d \}}| j||� | j||� | j|tj � | j|tj � d S )Nu ሴ�utf8r )
r r8 �encoder? r� �itemsr+ r� r� r� )r& rC Z unicode_objZ bytes_objr� �valuer r r � testStringUnicodeConversionInMap s
z+Proto3Test.testStringUnicodeConversionInMapc C sZ t j� }| jdt|j�� | jd|jk� |jd |jjd
� | jdt|j�� | jd|j� | jd |j� | jdt|j�� | jt �� |jd W d Q R X | jt
�� |jd |jd< W d Q R X | jdt|j�� |j � }t j� }|j |� | jdt|j�� | jd|j� | jd |j� | jdt|j�� t
j� dk�rV| jd tt|j��� d S )
Nr r� r i� re rI i� r r i8���i8���i8���)r r8 r+ r� rE r� Z
get_or_creater# r r= r� r, r* r r r$ )r& rC r^ rK r r r �testMessageMap& s0
zProto3Test.testMessageMapc C s� t j� }d|jd j_|jd= d|jd j_| jdt|j�� d|jd j_| jdt|j�� |j� }t j� }|j|� ddg}x|D ] }|j|= q�W d S )Nr8 re ) r r8 �map_int32_all_typesr rw r+ r� r, r* )r& rC r^ rK �keysr� r r r � testNestedMessageMapItemDeleteO s
z)Proto3Test.testNestedMessageMapItemDeletec C sp t j� }d|jd< |j� }d|jd< | j|j� |d � d|jd _|j� }d|jd _| j|j� |d � d S )Nr8 � � )r r8 r9 rM r+ rE r� )r& rC r� r r r �testMapByteSize` s
zProto3Test.testMapByteSizec C s� t j� }d|jd< d|jd< d|jd< d|jd _d |jd
_t j� }d |jd< d |jd
< d|jd
_d|jd
_|jd
}|j|� | jd|jd � | jd|jd � | jd|jd � | jd |jd
� | jd|jd j� | jd |jd
j� | j |jd
j
d�� t j � dk�r(| jd|j� i }x.|jD ]$}| j ||k� |j| j||< �q4W | jdd d�|� |jd= | j d|jk� |jd
= | j d
|jk� t j � dk�r�| j
t�� |jd= W d Q R X n | j
t�� |jd= W d Q R X d S )NrH r� �N �8 r� � r� r7 r� �� �7 �c �X r r� �dr )r7 r^ rt )r r8 r9 r: rE r� rb r� r+ r� r� r r r r= �KeyError)r& rC rK Z
old_map_valueZas_dictr� r r r �
testMergeFroml sH
zProto3Test.testMergeFromc C s. t j� }| jtd�� |jd� W d Q R X d S )NzWParameter to MergeFrom\(\) must be instance of same class: expected .*TestMap got int\.r8 )r r8 �assertRaisesRegexr= r� )r& rC r r r �testMergeFromBadType� s
z Proto3Test.testMergeFromBadTypec C s. t j� }| jtd�� |jd� W d Q R X d S )Nz[Parameter to [A-Za-z]*From\(\) must be instance of same class: expected .*TestMap got int\.r8 )r r8 re r= r� )r& rC r r r �testCopyFromBadType� s
z Proto3Test.testCopyFromBadTypec C s� t j� }td�|jtd�< td �|jtd �< td�|jtd�< td �|jtd
�< |j� }t j� }|j|� | j d|jd � | j d|jd � | j d|jd � | j d|jd � d S )Ni� r re rH r� i8���i����l l���� l l���� l l i8���i����l l���� l l���� l l )
r r8 r� r9 r: r; r< r, r* r+ )r& rC r^ rK r r r �testIntegerMapWithLongs� s
z"Proto3Test.testIntegerMapWithLongsc C s� t j� }d|jjd< |j� }t j� }|j|� | j||� d|jjd< |j� }|j|� | j||� |jjj� |j� }|j|� | j||� d S )Ni� r i� ix )r �TestMapSubmessage�test_mapr9 r, r* r+ �clear)r& rC r^ rK r r r � testMapAssignmentCausesPresence� s
z*Proto3Test.testMapAssignmentCausesPresencec C s� t j� }d|jjd _|j� }t j� }|j|� | j||� d|jjd _|j� }|j|� | j||� |jjd j|jjd � |j� }|j|� | j||� |jjj � |j� }|j|� | j||� d S )Nr� r � ix )
r ri rj rE r� r, r* r+ r� rk )r& rC r^ rK r r r �-testMapAssignmentCausesPresenceForSubmessages� s&
z8Proto3Test.testMapAssignmentCausesPresenceForSubmessagesc C s~ t j� }t|j�}t|j�}d|jd< d|jd _| jt�� x |D ]}qDW W d Q R X | jt�� x |D ]}qhW W d Q R X d S )NrI r� r� )r r8 �iterr? rE r� r � RuntimeError)r& rC Zstring_string_iterZint32_foreign_iterr� r r r �testModifyMapWhileIterating� s
z&Proto3Test.testModifyMapWhileIteratingc
C s� t j� }|jd }| j||jd � | j|tj� d|_|j� }t j� }|j |� | j
d|jd j� | j t �� tj� |jd< W d Q R X d S )Nr7 r� ra )
r r8 rE r� r� r r0 r� r, r* r+ r r� )r& rC rF r^ rK r r r �testSubmessageMap s
z Proto3Test.testSubmessageMapc C s� t j� }x$t|jj� �D ]\}}| jd� qW d|jd< d|jd< d|jd< | jdt|j�� dddd�}| jt|jj� �|� d S )NFr9 re r� rp � )re rp r9 ) r r8 r� r9 rQ rW r+ r� r5 )r& rC r� r4 �
matching_dictr r r �testMapIteration s
zProto3Test.testMapIterationc s& t jdk �r"tj� }d|jd< d|jd< d|jd< d|jd< |j}� jdt|�� tj� }|j|j� � � fdd �}|t |j
� �t |j
� �� |t |j � �t |j � �� |t |j
� �t |j
� �� � jd|jd�� � jd |jd
�� � jd|jd�� � jd |jd�� � jdt|�� |j� \}}� jd| |� � jdt|�� |j� � jd t|�� � jt�� |j� W d Q R X � jd |jd�� � jd t|�� |j|j� � jdt|�� � jt�� |j|j|j� W d Q R X � jt�� |jd � W d Q R X � jt�� |jd
d� W d Q R X d S )Nrp r9 re r� rs r� r� c s0 � j t|�| d � � j t|�| dd � � d S )Nr r8 )r+ �nextr� )�seq�iterator)r& r r �
CheckItems9 s z-Proto3Test.testPython2Map.<locals>.CheckItemsi� r r8 r� )rR )rp )rq rr r r8 r9 r+ r� r* r, r� rQ ro rV r� rD r �popitemrk r rc �
setdefault�updater= )r& rC Z map_int32rK ry r� rR r )r& r �testPython2Map- sJ
zProto3Test.testPython2Mapc C sf t j� }d|jd< d|jd< d|jd< d|jd< d|jd< t|jj� �}t|jj� �}| j||� d S )Nrt �
local_init_op�trainable_variables� variables�init_op� summaries)r r8 r? r� rQ r+ )r& rC Zitems1Zitems2r r r � testMapItems] s
zProto3Test.testMapItemsc C s~ d}t j� }d|jd< d|jd< d|jd< d|jd < d
|jd < d
|jd < d
|jd< d|jd< d|jd< | j||jdd�� d S )Ns� r
init_opdr
item1er
item2fr
item3gr
item4QQr
local_init_opar
summarieser
trainable_variablesbr
variablescr r~ r� r r� r� rb r� �er� Zitem1�fZitem2�gZitem3ZQQZitem4T)rF )r r8 r? r+ r, )r& r/ rC r r r �!testMapDeterministicSerializationl s
z,Proto3Test.testMapDeterministicSerializationc C sR t j� }d|jd< d|jd< d|jd< t|jj� �}~dddd�}| j||� d S )Nr9 re r� rp rs )re rp r9 )r r8 r9 r� rQ r5 )r& rC �itrt r r r � testMapIterationClearMessage� s
z'Proto3Test.testMapIterationClearMessagec C sf t jddd�d�}| jd|jd � | jd|jd � t jdtjdd�id �}| jd|jd j� d S )
Nre r9 )r8 rp )r9 r8 rp r� )r� )rE )r r8 r+ r9 r r0 rE r� )r& rC r r r �testMapConstruction� s z Proto3Test.testMapConstructionc C sd t j� }|j}d|d< d|d< d|d< |jd� | jd|j� � dddd�}| jt|j� �|� d S ) Nr9 re r� rp rs r9 r� )re rp r9 ) r r8 r9 r� r+ r, r5 r� rQ )r& rC Z int32_maprt r r r � testMapValidAfterFieldCleared� s
z(Proto3Test.testMapValidAfterFieldClearedc C sL t j� }|j}d|d _|jd� | jd|j� � | jdt|j � �k� d S )Nr� re rE r� )
r r8 rE r� r� r+ r, rW r� rV )r& rC Zint32_foreign_messager r r �$testMessageMapValidAfterFieldCleared� s
z/Proto3Test.testMessageMapValidAfterFieldClearedc C s| t j� }t|j�}|jd� | jt�� x |D ]}q.W W d Q R X t|j�}|jd� | jt�� x |D ]}qfW W d Q R X d S )Nr9 rE )r r8 ro r9 r� r rp rE )r& rC r� r� r r r �"testMapIterInvalidatedByClearField� s
z-Proto3Test.testMapIterInvalidatedByClearFieldc C s� t j� }| jdt|j�� d|jd< | jdt|j�� | jt�� |jd= W d Q R X |jd= | jdt|j�� | jt�� |jd= W d Q R X d S )Nr r� r9 r8 ra � )r r8 r+ r� r9 r rc rU )r& rC r r r �
testMapDelete� s
zProto3Test.testMapDeletec C sL t j� }| j|jtj� | j|jtj� | j|jtj� | j|jtj� d S )N)r r8 r� r9 r� �Mapping�MutableMappingrE )r& rC r r r �testMapsAreMapping� s z Proto3Test.testMapsAreMappingc C sD t j� }d|jd< | j|j|j� | j|j|j� | j|jd� d S )Ni� r r i8���i����)r r8 r9 r+ rE ZassertNotEqual)r& rC r r r �testMapsCompare� s
zProto3Test.testMapsComparec C s@ t j� }d|jd< d|jd< d|jd _| jdt|j� �� d S )NrI r� �@ �# r� r� r ) r r8 r? r9 Zmap_string_foreign_messager� r+ r� rJ )r& rC r r r �(testMapFindInitializationErrorsSmokeTest� s
z3Proto3Test.testMapFindInitializationErrorsSmokeTestN)#rG rH rI r5 r r6 rC rG rL rN rS rT rW rZ rd rf rg rh rl rn rq rr ru r} r� r� r� r� r� r� r� r� r� r� r� r r r r r2 ( s@
90=) < 0
r2 c @ s e Zd Zdd� Zdd� ZdS )�ValidTypeNamesTestc s� t t|��jd�d � d| d| f}| jt� fdd�|D ��d� |f � � jd�}|d
}dj|d d � �}t||gd � d S ) N�'r8 zRepeated%sContainerzRepeated%sFieldContainerc 3 s | ]}� j |�V qd S )N)�endswith)rx r4 )�tp_namer r r� s z:ValidTypeNamesTest.assertImportFromName.<locals>.<genexpr>z%r does end with any of %r�.)�fromlistr< r< )r$ �type�splitrW �any�join�
__import__)r& rC Z base_nameZ valid_names�parts�
class_nameZ module_namer )r� r �assertImportFromName� s
z'ValidTypeNamesTest.assertImportFromNamec C s( t j� }| j|jd� | j|jd� d S )NZScalarZ Composite)r r! r� r� r{ )r& r� r r r �testTypeNamesCanBeImported
s z-ValidTypeNamesTest.testTypeNamesCanBeImportedN)rG rH rI r� r� r r r r r� � s
r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�PackedFieldTestc C s� |j jd� |jjd� |jjd� |jjd� |jjd� |jjd� |jjd� |jjd� |j jd� |j
jd� |j jd� |j jd� |j
jd� |jjd� d S )Nr8 g �?T)r� r� Zrepeated_int64Zrepeated_uint32Zrepeated_uint64Zrepeated_sint32Zrepeated_sint64Zrepeated_fixed32Zrepeated_fixed64Zrepeated_sfixed32Zrepeated_sfixed64rZ r[ Z
repeated_boolr� )r& r r r r �
setMessage s zPackedFieldTest.setMessagec C s* t j� }| j|� d}| j||j� � d S )NsH
"*2: B J R Z �?b �?jr)r r4 r� r+ r, )r& r r/ r r r �testPackedFields" s
z PackedFieldTest.testPackedFieldsc C s* t j� }| j|� d}| j||j� � d S )Ns: (0= A M Q ] �?a �?hp)r ZTestUnpackedTypesr� r+ r, )r& r r/ r r r �testUnpackedFields5 s
z"PackedFieldTest.testUnpackedFieldsN)rG rH rI r� r� r� r r r r r� s r� r re rm z;explicit tests of the C++ implementation for PY27 and abovec @ s0 e Zd Zedd� �Zdd� Zdd� Zdd� Zd S )
�OversizeProtosTestc C sF d}t j� }tj� }tj||� |j|� tj|�j |j
d��| _ d S )Na�
name: "f/f.msg2"
package: "f"
message_type {
name: "msg1"
field {
name: "payload"
number: 1
label: LABEL_OPTIONAL
type: TYPE_STRING
}
}
message_type {
name: "msg2"
field {
name: "field"
number: 1
label: LABEL_OPTIONAL
type: TYPE_MESSAGE
type_name: "msg1"
}
}
zf.msg2) r ZDescriptorPoolr ZFileDescriptorProtor ZParseZAddr
ZMessageFactoryZ GetPrototypeZFindMessageTypeByName� proto_cls)�clsZ file_descZpoolZdescr r r �
setUpClassN s
z OversizeProtosTest.setUpClassc C s( | j � | _dd | jj_| jj� | _d S )Nr� i r� r8 i i i )r� �p�fieldr� r, � p_serialized)r& r r r �setUpq s
zOversizeProtosTest.setUpc C sf ddl m} |d� | j� }y|j| j� W n4 tjk
r` } z| jt|�d� W Y d d }~X nX d S )Nr )�SetAllowOversizeProtosFzError parsing message) � google.protobuf.pyext._messager� r� r* r� r r? r+ r$ )r& r� �qr� r r r �testAssertOversizeProtov s z*OversizeProtosTest.testAssertOversizeProtoc C sB ddl m} |d� | j� }|j| j� | j| jjj|jj� d S )Nr )r� T) r� r� r� r* r� r+ r� r� r� )r& r� r� r r r �testSucceedOversizeProto s
z+OversizeProtosTest.testSucceedOversizeProtoN)rG rH rI � classmethodr� r� r� r� r r r r r� I s # r� �__main__)re rm )4r� �
__author__r� r- rg r� rQ r� rq r@ Z unittest2Zunittest� ImportErrorr� r� Zgoogle.protobufr r r r
r r r
r Zgoogle.protobuf.internalr r r r r r r ZPY3r� Zlongr r r r Z BaseTestCaseZNamedParametersr r r2 r� r� ZskipIfr rr r� rG �mainr r r r �<module>) s� W q X9: