File: //usr/lib/python3/dist-packages/twisted/spread/test/__pycache__/test_banana.cpython-38.pyc
U
W[`< � @ s d dl mZmZ d dlZd dlmZ d dlmZ d dlm Z d dl
m Z d dl m
Z
d dlmZmZmZmZ d d lmZmZ d d
lmZ er�d Zn d d lmZ G d
d� de j�Z dd� Z dd� Z G dd� de j�Z G dd� de �Z G dd� de �Z!G dd� de j�Z"dS )� )�absolute_import�divisionN)�partial)�BytesIO)�unittest)�banana)�failure)�long� iterbytes� _bytesChr�_PY3)�protocol�main)�StringTransportl ���� )�maxintc @ s e Zd Zdd� ZdS )� MathTestsc C sp t tdd��t tdd�� t tdd�� dg }|D ]6}t� }t�||j� |�� }t�|�}| �||� q4d S )Nr �d i� iL i@B i�B l ) �list�ranger r Zint2b128�write�getvalueZb1282int� assertEqual)�selfZ funkylist�i�x�v�y� r �A/usr/lib/python3/dist-packages/twisted/spread/test/test_banana.py�
test_int2b128 s ��
zMathTests.test_int2b128N)�__name__�
__module__� __qualname__r r r r r r s r c C s | � |� dS )z�
Dictate a Banana dialect to use.
@param protocol: A L{banana.Banana} instance which has not yet had a
dialect negotiated.
@param dialect: A L{bytes} instance naming a Banana dialect to select.
N)Z_selectDialect)r
Zdialectr r r �
selectDialect$ s r# c C s0 t � }| � }|�|� |�� |�|� |�� S )a�
Banana encode an object using L{banana.Banana.sendEncoded}.
@param bananaFactory: A no-argument callable which will return a new,
unconnected protocol instance to use to do the encoding (this should
most likely be a L{banana.Banana} instance).
@param obj: The object to encode.
@type obj: Any type supported by Banana.
@return: A L{bytes} instance giving the encoded form of C{obj}.
)r �makeConnection�clear� sendEncoded�value)Z
bananaFactory�objZ transportr r r r �encode4 s
r) c @ s. e Zd ZdZejZdd� Zdd� Zdd� Z dS ) �BananaTestBasezn
The base for test classes. It defines commonly used things and sets up a
connection for testing.
c C sN t � | _| �� | _| j�t�| j�� t| jd� | j| j_ t
t | j�| _ d S )Ns none) r �io�encClass�encr$ r
Z FileWrapperr# � putResultZexpressionReceivedr r) �r r r r �setUpR s
zBananaTestBase.setUpc C s
|| _ dS )z�
Store an expression received by C{self.enc}.
@param result: The object that was received.
@type result: Any type supported by Banana.
N)�result)r r1 r r r r. [ s zBananaTestBase.putResultc C s | j �t�tj�� | ` d S �N)r- ZconnectionLostr ZFailurer ZCONNECTION_DONEr/ r r r �tearDowne s zBananaTestBase.tearDownN)
r r! r" �__doc__r �Bananar, r0 r. r3 r r r r r* K s
r* c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
e rNde
_ dd� Z
dd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Z d1d2� Z d3S )4� BananaTestsz
General banana tests.
c C s0 | j �d� | j �| j�� � | jdks,t�d S )Ns hello)r- r& � dataReceivedr+ r r1 �AssertionErrorr/ r r r � test_stringp s zBananaTests.test_stringc C s" t r| �dd� n | �dd� dS )z�
Banana does not support unicode. ``Banana.sendEncoded`` raises
``BananaError`` if called with an instance of ``unicode``.
Zhelloz builtins.strz__builtin__.unicodeN)r �_unsupportedTypeTestr/ r r r �test_unsupportedUnicodev s z#BananaTests.test_unsupportedUnicodec C s" t r| �td� n | �td� dS )z�
Banana does not support arbitrary builtin types like L{type}.
L{banana.Banana.sendEncoded} raises L{banana.BananaError} if called
with an instance of L{type}.
z
builtins.typez__builtin__.typeN)r r: �typer/ r r r �test_unsupportedBuiltinType� s z'BananaTests.test_unsupportedBuiltinTypec C s | � t� td � dS )z�
Banana does not support arbitrary user-defined types (such as those
defined with the ``class`` statement). ``Banana.sendEncoded`` raises
``BananaError`` if called with an instance of such a type.
z
.MathTestsN)r: r r r/ r r r �test_unsupportedUserType� s z$BananaTests.test_unsupportedUserTypec C s. | � tj| jj|�}| �d�|�t|�� dS )a�
Assert that L{banana.Banana.sendEncoded} raises L{banana.BananaError}
if called with the given object.
@param obj: Some object that Banana does not support.
@param name: The name of the type of the object.
@raise: The failure exception is raised if L{Banana.sendEncoded} does
not raise L{banana.BananaError} or if the message associated with the
exception is not formatted to include the type of the unsupported
object.
z Banana cannot send {0} objectsN)� assertRaisesr � BananaErrorr- r& ZassertIn�format�str)r r( �name�excr r r r: � s
z BananaTests._unsupportedTypeTestc C sP dt d�fD ]>}| j�|� | j�| j�� � | �| jd� | �| jt � q dS )z�
A positive integer less than 2 ** 32 should round-trip through
banana without changing value and should come out represented
as an C{int} (regardless of the type which was encoded).
i�' N)
r r- r& r7 r+ r r r1 �assertIsInstance�int)r r' r r r �test_int� s
zBananaTests.test_intc C s� dD ]�}dD ]x}d| | }|| d fD ]X}| j �| �|�� | �| j|� |tksd|t d k rt| �| jt� q*| �| jt� q*q qdS )a
Integers greater than 2 ** 32 and less than -2 ** 32 should
round-trip through banana without changing value and should
come out represented as C{int} instances if the value fits
into that type on the receiving platform.
)� �@ � � )r � � rL N) r- r7 r) r r1 �_maxintrE r rF )r Zexp�add�m�nr r r �test_largeLong� s zBananaTests.test_largeLongz@Python 3 has unified int/long into an int type of unlimited sizec C s( | j j}|d }d| d }|d }|S )N� rM rL )r- � prefixLimit)r �bytes�bits�largest�smallestr r r � _getSmallest� s
zBananaTests._getSmallestc C s | � � }| �tj| jj|� dS )zz
Test that a long above the implementation-specific limit is rejected
as too large to be encoded.
N)rY r? r r@ r- r& )r rX r r r �test_encodeTooLargeLong� s z#BananaTests.test_encodeTooLargeLongc C sj | � � }| j�| jjd � | j�|� | j�� }| j�d� | j�| jjd � | �t j
| jj |� dS )zz
Test that a long above the implementation specific limit is rejected
as too large to be decoded.
rM r N) rY r- �setPrefixLimitrT r& r+ r �truncater? r r@ r7 )r rX �encodedr r r �test_decodeTooLargeLong� s
z#BananaTests.test_decodeTooLargeLongc C s
| � � S r2 )rY r/ r r r � _getLargest� s zBananaTests._getLargestc C s | � � }| �tj| jj|� dS )z�
Test that a negative long below the implementation-specific limit is
rejected as too small to be encoded.
N)r_ r? r r@ r- r& )r rW r r r �test_encodeTooSmallLong� s z#BananaTests.test_encodeTooSmallLongc C sj | � � }| j�| jjd � | j�|� | j�� }| j�d� | j�| jjd � | �t j
| jj |� dS )z�
Test that a negative long below the implementation specific limit is
rejected as too small to be decoded.
rM r N) r_ r- r[ rT r& r+ r r\ r? r r@ r7 )r rW r] r r r �test_decodeTooSmallLong� s
z#BananaTests.test_decodeTooSmallLongc C s0 | j �d� | j �| j�� � | �| jd� d S )Ni� �r- r& r7 r+ r r r1 r/ r r r � test_integer
s zBananaTests.test_integerc C s0 | j �d� | j �| j�� � | �| jd� d S )Ni ���rb r/ r r r �
test_negative s zBananaTests.test_negativec C s0 | j �d� | j �| j�� � | �| jd� d S )Ng ��@rb r/ r r r �
test_float s zBananaTests.test_floatc C sX ddddgddgddd d
d ggd gg g}| j �|� | j �| j�� � | �| j|� d S )
NrL rM � � � �>@皙���D@� � six� seven� eight� �
rb �r Zfoor r r � test_list s � zBananaTests.test_listc C sl ddddgddgddd d
d ggd gt jd t jd t jd
g
}| j�|� | �| j�� � | �| j|� dS )zj
Test feeding the data byte per byte to the receiver. Normally
data is not split.
rL rM rf rg rh ri rj rk rl rm rn ro ���N) �sys�maxsizer- r& �feedr+ r r r1 rp r r r � test_partial' s � zBananaTests.test_partialc C s t |�D ]}| j�|� qdS )z�
Feed the data byte per byte to the receiver.
@param data: The bytes to deliver.
@type data: L{bytes}
N)r
r- r7 )r �dataZbyter r r ru 5 s zBananaTests.feedc C s d}| � tj| j|� d S )Ns ��r? r r@ ru �r rw r r r �test_oversizedList@ s z BananaTests.test_oversizedListc C s d}| � tj| j|� d S )Ns �rx ry r r r �test_oversizedStringF s z BananaTests.test_oversizedStringc C s0 d}z| j �|� W n tjk
r* Y nX d S )Ns �)r- r7 r r@ )r Z crashStringr r r �test_crashStringL s
z BananaTests.test_crashStringc C s0 | j �d� | j �| j�� � | �| jd� d S )N� �rb r/ r r r �test_crashNegativeLong_ s z"BananaTests.test_crashNegativeLongc C sP d}d}d}| � | �|d �d| � | � | �|d �d| � | � | �|d �d | � d
}| � | �|d �d | � | � | �|d �d | � | � | �|d
�d| � d}| � | �|d �d| � | � | �|d �d | � | � | �|d �d� d}| � | �|d �d | � | � | �|d �d| � | � | �|d
�d| � dS )z�
Test that integers below the maximum C{INT} token size cutoff are
serialized as C{INT} or C{NEG} and that larger integers are
serialized as C{LONGINT} or C{LONGNEG}.
i���r} s �rM � }rL � ~r � s �� � rf � s �s �s �� N)r r) )r Z baseIntInZ baseNegInZ
baseIntOutZbaseLongIntOutZ
baseNegOutZbaseLongNegOutr r r �test_sizedIntegerTypesl s$ z"BananaTests.test_sizedIntegerTypesN) r r! r" r4 r9 r; r= r> r: rG rR r �skiprY rZ r^ r_ r` ra rc rd re rq rv ru rz r{ r| r~ r� r r r r r6 k s8
�
r6 c @ sX e Zd ZdZdZeejje �ej Z
ed�ej Z dd� Z dd� Z
dd � Zd
d � Zd S )
� DialectTestsz2
Tests for Banana's handling of dialects.
s remote�z c C s | � t| jj| j� dS )z|
If no dialect has been selected and a PB VOCAB item is received,
L{NotImplementedError} is raised.
N)r? �NotImplementedErrorr- r7 � legalPbItemr/ r r r �test_dialectNotSet� s
�z DialectTests.test_dialectNotSetc C s. t | jd� | j�| j� | �| j| j� dS )zS
If the PB dialect has been selected, a PB VOCAB item is accepted.
� pbN)r# r- r7 r� r r1 �vocabr/ r r r �test_receivePb� s zDialectTests.test_receivePbc C s$ t | jd� | �t| jj| j� dS )z�
If the PB dialect has been selected and an unrecognized PB VOCAB item
is received, L{banana.Banana.dataReceived} raises L{KeyError}.
r� N)r# r- r? �KeyErrorr7 �
illegalPbItemr/ r r r �test_receiveIllegalPb� s z"DialectTests.test_receiveIllegalPbc C s2 t | jd� | j�| j� | �| j| j�� � dS )zl
if pb dialect is selected, the sender must be able to send things in
that dialect.
r� N)r# r- r& r� r r� r+ r r/ r r r � test_sendPb� s zDialectTests.test_sendPbN)r r! r" r4 r� �chrr r5 ZoutgoingVocabularyZVOCABr� r� r� r� r� r� r r r r r� � s
r� c @ s e Zd ZdZdd� ZdS )�GlobalCoderTestszM
Tests for the free functions L{banana.encode} and L{banana.decode}.
c C s0 d}| � tjtj|� d}| �t�|�d� dS )zJ
Calls to L{banana.decode} are independent of each other.
sB �s �rL N)r? r r@ �decoder )r Z undecodableZ decodabler r r �test_statelessDecode� s z%GlobalCoderTests.test_statelessDecodeN)r r! r" r4 r� r r r r r� � s r� )#Z
__future__r r rs � functoolsr r+ r Z
twisted.trialr Ztwisted.spreadr Ztwisted.pythonr Ztwisted.python.compatr r
r r� r Ztwisted.internetr
r Ztwisted.test.proto_helpersr rN r ZTestCaser r# r) r* r6 r� r� r r r r �<module> s*
"/