File: //lib/python3/dist-packages/twisted/spread/test/__pycache__/test_jelly.cpython-38.pyc
U
W[N � @ s2 d Z ddlmZmZ ddlZddlZddlmZ ddlm Z m
Z
ddl m Z ddl
mZ G dd � d e je�ZG d
d � d �Zd d
� ZG dd� d�ZG dd� d�ZG dd� de�ZG dd� de�ZG dd� d�ZG dd� de j�ZG dd� de j�ZG d d � d e
j e
j �Z G d d � d e
j e
j �Z G d d!� d!e j�Z dS )"z/
Test cases for L{jelly} object serialization.
� )�absolute_import�divisionN)�unicode)�jelly�pb)�unittest)�StringTransportc @ s e Zd ZdZdZddd�ZdS )�TestNodezD
An object to test jellyfying of new style class instances.
� Nc C s4 |r |j d | _ |j�| � nd| _ || _g | _d S )N� )�id�children�append�parent)�selfr � r �@/usr/lib/python3/dist-packages/twisted/spread/test/test_jelly.py�__init__ s zTestNode.__init__)N)�__name__�
__module__� __qualname__�__doc__� classAttrr r r r r r s r c @ s e Zd ZdZdd� ZdS )�A�
Dummy class.
c C s dS �z;
Method to be used in serialization tests.
Nr �r r r r �amethod) s z A.amethodN)r r r r r r r r r r $ s r c C s dS )z:
A dummy function to test function serialization.
Nr r r r r �afunc0 s r c @ s e Zd ZdZdd� ZdS )�Br c C s dS r r r r r r �bmethod< s z B.bmethodN)r r r r r r r r r r 7 s r c @ s e Zd ZdZdd� ZdS )�Cr c C s dS r r r r r r �cmethodH s z C.cmethodN)r r r r r" r r r r r! C s r! c @ s e Zd ZdZdS )�Dz
Dummy new-style class.
N)r r r r r r r r r# O s r# c @ s. e Zd ZdZdZd
dd�Zdd� Zdd � ZdS ) �Ez+
Dummy new-style class with slots.
��x�yNc C s || _ || _d S �Nr% �r r& r' r r r r ] s z
E.__init__c C s | j | jd�S )Nr% r% r r r r � __getstate__b s zE.__getstate__c C s |d | _ |d | _d S )Nr& r' r% �r �stater r r � __setstate__f s
zE.__setstate__)NN)r r r r � __slots__r r* r- r r r r r$ V s
r$ c @ s e Zd Zdd� Zdd� ZdS )�SimpleJellyTestc C s || _ || _d S r( r% r) r r r r m s zSimpleJellyTest.__init__c C s | j |j kS r( )�__dict__�r �otherr r r � isTheSameAsq s zSimpleJellyTest.isTheSameAsN)r r r r r3 r r r r r/ l s r/ c @ sN e Zd ZdZddddgdddgdddgddd ggZd
d � Zd d
� Zdd� Zdd� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
d d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zejs�d*e_d+d,� Zejs�d-e_d.d/� Zd0d1� Zd2d3� Zd4d5� Z d6d7� Z d8d9� Z d:d;� Z d<d=� Z d>d?� Z!d@dA� Z"dBdC� Z#dDdE� Z$dFdG� Z%dHdI� Z&dJdK� Z'dLS )M�
JellyTestsz�
Testcases for L{jelly} module serialization.
@cvar decimalData: serialized version of decimal data, to be used in tests.
@type decimalData: L{list}
s list� decimali� ���r �@� i������c C sJ t � |�}t �� }|�� t �||� |j�|� | �t jt j||� dS )z�
Helper test method to test security options for a type.
@param inputList: a sample input for the type.
@type inputList: L{list}
@param atom: atom identifier for the type.
@type atom: L{str}
N)r �SecurityOptionsZallowBasicTypes�unjellyZ allowedTypes�pop� assertRaises�
InsecureJelly)r � inputListZatom�c�tasterr r r �
_testSecurity� s
zJellyTests._testSecurityc C sR t � }t� }t� }|j|_||_tj}t`z| �ttj t�|�� W 5 |t_X dS )z�
If a class change after an instance has been created, L{jelly.unjelly}
shoud raise a C{TypeError} when trying to unjelly the instance.
N)
r r r! r" r �ar< � TypeErrorr r: )r rB �br? Z savecmethodr r r �test_methodsNotSelfIdentity� s z&JellyTests.test_methodsNotSelfIdentityc C s` t � }d|_t � }||_||_t�|�}t�|�}| �|t � | �|j|j� | �|jd� dS )z�
Test that a new style class can be jellied and unjellied with its
objects and attribute values preserved.
r N) r# r& �n2Zn3r r: �assertIsInstance�assertIs� assertEqual)r �nrF r? �mr r r �
test_newStyle� s
zJellyTests.test_newStylec C s> t � }d|_t�|�}t�|�}| �|t � | �|jd� dS )z�
A class defined with I{slots} can be jellied and unjellied with the
values for its attributes preserved.
r N)r$ r& r r: rG rI )r rJ r? rK r r r �test_newStyleWithSlots� s
z!JellyTests.test_newStyleWithSlotsc C s& t g}t�t�|��}| �||� dS )zn
Test that an old style class type can be jellied and unjellied
to the original type.
N)r! r r: rI �r �t�rr r r �test_typeOldStyle� s z JellyTests.test_typeOldStylec C s& t g}t�t�|��}| �||� dS )zm
Test that a new style class type can be jellied and unjellied
to the original type.
N)r# r r: rI rN r r r �test_typeNewStyle� s z JellyTests.test_typeNewStylec C s& t g}t�t�|��}| �||� dS )ze
Test that a builtin type can be jellied and unjellied to the original
type.
N)�strr r: rI rN r r r �test_typeBuiltin� s zJellyTests.test_typeBuiltinc C sP t j �� }t j �� | }||g}t�|�}t�|�}| �||� | �||� dS )z�
Jellying L{datetime.timedelta} instances and then unjellying the result
should produce objects which represent the values of the original
inputs.
N)�datetimeZnowr r: rI � assertIsNot)r ZdtnZdtdr> r? �outputr r r �
test_dateTime� s
zJellyTests.test_dateTimec C sT t �d�t �d�t �d�t �d�g}t�|�}t�|�}| �||� | �||� dS )z�
Jellying L{decimal.Decimal} instances and then unjellying the result
should produce objects which represent the values of the original
inputs.
�9.95r r7 �-78.901N)�decimal�Decimalr r: rI rV )r r> r? rW r r r � test_decimal� s �
zJellyTests.test_decimalc C s@ t �d�t �d�t �d�t �d�g}t�| j�}| �||� dS )a
Unjellying the s-expressions produced by jelly for L{decimal.Decimal}
instances should result in L{decimal.Decimal} instances with the values
represented by the s-expressions.
This test also verifies that L{decimalData} contains valid jellied
data. This is important since L{test_decimalMissing} re-uses
L{decimalData} and is expected to be unable to produce
L{decimal.Decimal} instances even though the s-expression correctly
represents a list of them.
rY r r7 rZ N)r[ r\ r r: � decimalDatarI )r ZexpectedrW r r r �test_decimalUnjelly s � z JellyTests.test_decimalUnjellyc C s t �d�g}| �|d� dS )z�
By default, C{decimal} objects should be allowed by
L{jelly.SecurityOptions}. If not allowed, L{jelly.unjelly} should raise
L{jelly.InsecureJelly} when trying to unjelly it.
rY r5 N)r[ r\ rA �r r> r r r �test_decimalSecurity s z JellyTests.test_decimalSecurityc C s< t dddg�g}t�t�|��}| �||� | �||� dS )z�
Jellying C{set} instances and then unjellying the result
should produce objects which represent the values of the original
inputs.
r � � N)�setr r: rI rV �r r> rW r r r �test_set s zJellyTests.test_setc C s< t dddg�g}t�t�|��}| �||� | �||� dS )z�
Jellying L{frozenset} instances and then unjellying the result
should produce objects which represent the values of the original
inputs.
r rb rc N)� frozensetr r: rI rV re r r r �test_frozenset, s zJellyTests.test_frozensetc C s t dddg�g}| �|d� dS )z�
By default, C{set} objects should be allowed by
L{jelly.SecurityOptions}. If not allowed, L{jelly.unjelly} should raise
L{jelly.InsecureJelly} when trying to unjelly it.
r rb rc s setN)rd rA r` r r r �test_setSecurity8 s zJellyTests.test_setSecurityc C s t dddg�g}| �|d� dS )z�
By default, L{frozenset} objects should be allowed by
L{jelly.SecurityOptions}. If not allowed, L{jelly.unjelly} should raise
L{jelly.InsecureJelly} when trying to unjelly it.
r rb rc s frozensetN)rg rA r` r r r �test_frozensetSecurityB s z!JellyTests.test_frozensetSecurityc C s� t j�dddg�g}t � |�}| �|t � tdddg�g�� t �|�}| �t|d �t|d �� tt jjkr�| �|d t jj� n| �|d t� dS )z�
Test jellying C{sets.Set}: it should serialize to the same thing as
C{set} jelly, and be unjellied as C{set} if available.
r rb rc r N)r �_sets�SetrI rd r: �listrG �r r> Z
inputJellyrW r r r � test_oldSetsL s
zJellyTests.test_oldSetsz'sets.Set is gone in Python 3 and higherc C s� t j�dddg�g}t � |�}| �|t � tdddg�g�� t �|�}| �t|d �t|d �� tt jjkr�| �|d t jj� n| �|d t� dS )z�
Test jellying C{sets.ImmutableSet}: it should serialize to the same
thing as L{frozenset} jelly, and be unjellied as L{frozenset} if
available.
r rb rc r N)r rk Z ImmutableSetrI rg r: rm rG rn r r r �test_oldImmutableSets` s
z JellyTests.test_oldImmutableSetsz1sets.ImmutableSets is gone in Python 3 and higherc C sN | � tdd��tdd��� tdd�}t�|�}t�|�}| � |�|�� dS )z%
Simplest test case.
rB rD r rb N)�
assertTruer/ r3 r r: )r rB ZcerealrD r r r � test_simplev s �
zJellyTests.test_simplec C s� g }|}|� |� |� |� | �|d |d � | �|d d |� t�|�}t�|�}| �|d |d � | �|d d |� dS )zJ
Test to make sure that objects retain identity properly.
r r N)r rH r r: )r r& r' �s�zr r r �
test_identity� s
zJellyTests.test_identityc C s< t d�}t�t�|��}| �||� | �t|�t|�� d S )NZblah)r r r: rI �typer) r r r � test_unicode� s zJellyTests.test_unicodec C sb g }d|i|f}|� |� t�|�}t�|�}| �|d d |d � | �|d d d |� d S )Nrm r r �r r r: rH )r ZrerefZ
toplevelTuplers rt r r r �test_stressReferences� s
z JellyTests.test_stressReferencesc C sF g }|f}|� |f� t�|�}t�|�}| �|d d d |� d S �Nr rx )r rB rO rs rt r r r �test_moreReferences� s
z JellyTests.test_moreReferencesc C s* t �� }t � i �}| �t jt j||� dS )z@
Test for type-level security of serialization.
N)r r9 r< r= r: )r r@ Zdctr r r �test_typeSecurity� s
z JellyTests.test_typeSecurityc C s t �t�}| �t|� d S r( )r r: r# rH )r �ujr r r �test_newStyleClasses� s
z JellyTests.test_newStyleClassesc C s� t � }t�t�|�� t�t�|j�� tdddgtd� td�ddddt tddit jg }|D ] }| �|t�t�|��� q^dS ) zA
Test for all types currently supported in jelly
r rb rc Ztestg�����L4@)r rb rc NrB )r r r: r r �boolr rI )r rB �items�ir r r �test_lotsaTypes� s �zJellyTests.test_lotsaTypesc C sx G dd� d�a t� }t |�}t |�}t ||f�}||||||d|i}t�t�|��d }| �|jd j|jd j� d S )Nc @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�
TupleStatec S s
|| _ d S r( �r2 r1 r r r r � s zTupleState.__init__c S s | j fS r( r� r r r r r* � s zTupleState.__getstate__c S s |d | _ d S rz r� r+ r r r r- � s zTupleState.__setstate__c S s
t | j�S r( )�hashr2 r r r r �__hash__� s zTupleState.__hash__N)r r r r r* r- r� r r r r r� � s r� �t3r r )r� r r r: rH r2 )r rB �t1�t2r� �dZt3primer r r �
test_setState� s zJellyTests.test_setStatec C s� t �� }|�tt� t� }t� }t� }||_||_||_||_t � ||�}t � ||�}| �
|jt j � t � |�}| � t j
t j ||� | �|j|jd� t � t|�}t � ||�}| �|td| � dS )zA
Test for class-level security of serialization.
zIdentity mismatchzA came back: %sN)r r9 ZallowInstancesOfr r r! rD r? r& r: rG Z
Unpersistabler< r= rH )r r@ rB rD r? Zfriendlyr& Zmeanr r r �test_classSecurity� s$
z JellyTests.test_classSecurityc C s\ G dd� dt j�}t �|t j� |� }d|_t �t � |��}| �|jd� | �|t j� dS )zr
Test that if Unjellyable is used to deserialize a jellied object,
state comes out right.
c @ s e Zd ZdS )z7JellyTests.test_unjellyable.<locals>.JellyableTestClassN�r r r r r r r �JellyableTestClass� s r� �valueN)r � Jellyable�setUnjellyableForClassZ UnjellyableZ attributer: rI rG )r r� �inputrW r r r �test_unjellyable� s zJellyTests.test_unjellyablec C s� i dg}|fdd�}|fdd�}t dd�}t dd�}t d d
�}||_||_||_tj||d �}tj||d �}| �|j|jj� | �|d
d� | �|j|jd� d S )Nr c S s, |d d |d< | |d |d < t |d �S )Nr r )rS )�obj�jel�perstr r r �persistentStore s z:JellyTests.test_persistentStorage.<locals>.persistentStorec S s t | �}|d | S rz )�int)ZpidstrZunjr� �pidr r r �persistentLoad s z9JellyTests.test_persistentStorage.<locals>.persistentLoadrb rc r
� � )r� )r� r z persistentStore was not called.z$Persistent storage identity failure.)r/ rD r? r r: rH rq ) r r� r� r� rB rD r? r� r& r r r �test_persistentStorage s
z!JellyTests.test_persistentStoragec C sB t � }t |�}t |� t |� t�|�}t�|�}| �||� d S r( )r r r: �_check_newstyle)r rJ Zn1r� rK r r r � test_newStyleClassesAttributes" s
z)JellyTests.test_newStyleClassesAttributesc C sl | � |j|j� | � |jd� | � |jd� | � t|j�t|j�� t|j|j�D ]\}}| �||� qRd S )Nr
)rI r r �lenr
�zipr� )r rB rD r& r' r r r r� . s zJellyTests._check_newstylec C s` t �� }t �� }|�t� � tj||d�}t �� }|�t� � tj||d�}| �|j|j � dS )a
A L{pb.Referenceable} instance jellies to a structure which unjellies to
a L{pb.RemoteReference}. The C{RemoteReference} has a I{luid} that
matches up with the local object key in the L{pb.Broker} which sent the
L{Referenceable}.
�ZinvokerN)
r Z
ReferenceableZBrokerZmakeConnectionr r r: ZassertInZluidZ localObjects)r �refZ jellyBroker�jZ
unjellyBrokerr} r r r �test_referenceable7 s z JellyTests.test_referenceableN)(r r r r r^ rA rE rL rM rQ rR rT rX r] r_ ra rf rh ri rj ro r rk �skiprp rr ru rw ry r{ r| r~ r� r� r� r� r� r� r� r� r r r r r4 v sT �
�
r4 c @ s e Zd ZdZdd� Zdd� ZdS )�JellyDeprecationTestsz+
Tests for deprecated Jelly things
c C sP t j | �| jg�}| �t|�d� | �|d d d� | �|d d t� dS )zD
L{jelly.instance_atom} is deprecated since 15.0.0.
r r �messagezltwisted.spread.jelly.instance_atom was deprecated in Twisted 15.0.0: instance_atom is unused within Twisted.�categoryN)r Z
instance_atom�
flushWarnings�test_deprecatedInstanceAtomrI r� �DeprecationWarning�r �warningsr r r r� P s
�
�z1JellyDeprecationTests.test_deprecatedInstanceAtomc C sZ t �dddgdgg� | �� }| �t|�d� | �|d d d� | �|d d t� d
S ) zI
Unjellying the instance atom is deprecated with 15.0.0.
�instance�classz twisted.spread.test.test_jelly.AZ
dictionaryr r r� ztUnjelly support for the instance atom is deprecated since Twisted 15.0.0. Upgrade peer for modern instance support.r� N)r r: r� rI r� r� r� r r r �%test_deprecatedUnjellyingInstanceAtom` s ��
�
�z;JellyDeprecationTests.test_deprecatedUnjellyingInstanceAtomN)r r r r r� r� r r r r r� K s r� c @ s e Zd Zdd� ZdS )�ClassAc C s t | �| _d S r( )�ClassBr� r r r r r u s zClassA.__init__N�r r r r r r r r r� t s r� c @ s e Zd Zdd� ZdS )r� c C s
|| _ d S r( )r� )r r� r r r r { s zClassB.__init__Nr� r r r r r� z s r� c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �CircularReferenceTestszN
Tests for circular references handling in the jelly/unjelly process.
c C s@ t �tt� t �tt� t �t � t� ��}| �|jj|d� d S )N�,Identity not preserved in circular reference)r r� r� r� r: rH r� )r rB r r r �test_simpleCircle� s �z(CircularReferenceTests.test_simpleCirclec C sf G dd� d�}|� }d |_ t� }t�tt� t�tt� tj||d�}t�|�}| �|jj|d� d S )Nc @ s e Zd ZdS )zHCircularReferenceTests.test_circleWithInvoker.<locals>.DummyInvokerClassNr� r r r r �DummyInvokerClass� s r� r� r� )ZserializingPerspectiver� r r� r� r: ZfailUnlessIdenticalr� )r r� Z dummyInvokerZa0r� Za1r r r �test_circleWithInvoker� s
�z-CircularReferenceTests.test_circleWithInvokerc C sP t � }t|d�}|�|� t�t�|��}| �|jt � | �t|j�|g� dS )z�
Check that a C{set} can contain a circular reference and be serialized
and unserialized without losing the reference.
N) rd r/ �addr r: rG r& rI rm )r rs rB �resr r r rf � s
z CircularReferenceTests.test_setc C sP t dd�}t|g�}||_t�t�|��}| �|jt� | �t|j�|g� dS )z�
Check that a L{frozenset} can contain a circular reference and be
serialized and unserialized without losing the reference.
N)r/ rg r& r r: rG rI rm )r rB rs r� r r r rh � s
z%CircularReferenceTests.test_frozensetN)r r r r r� r� rf rh r r r r r� � s
r� )!r Z
__future__r r rU r[ Ztwisted.python.compatr Ztwisted.spreadr r Z
twisted.trialr Ztwisted.test.proto_helpersr r� �objectr r r r r! r# r$ r/ ZTestCaser4 r� ZCopyableZ
RemoteCopyr� r� r� r r r r �<module> s.
X)