File: //lib/python3/dist-packages/twisted/spread/test/__pycache__/test_jelly.cpython-310.pyc
o
�boJ � @ s" d Z ddlZddlZddlmZmZmZ ddlmZ ddl m
Z
ddl m Z G dd� dej
�ZG d d
� d
�Zd d � ZG d
d� d�ZG dd� d�ZG dd� d�ZG dd� d�ZG dd� d�Zdd� ZG dd� de �ZG dd � d e �ZG d d � d ejej�ZG d d � d ejej�Z G d!d"� d"e �Z dS )#z/
Test cases for L{jelly} object serialization.
� N)�banana�jelly�pb)�StringTransport)�unittest)�TestCasec @ 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 @ � e Zd ZdZdd� ZdS )�A�
Dummy class.
c C � dS �z;
Method to be used in serialization tests.
Nr �r r r r �amethod( � z A.amethodN)r r r r r r r r r r # � r c C r )z:
A dummy function to test function serialization.
Nr r r r r �afunc. r r" c @ r )�Br c C r r r r r r r �bmethod9 r z B.bmethodN)r r r r r$ r r r r r# 4 r! r# c @ r )�Cr c C r r r r r r r �cmethodD r z C.cmethodN)r r r r r& r r r r r% ? r! r% c @ s e Zd ZdZdS )�Dz
Dummy new-style class.
N)r r r r r r r r r' J s r' c @ s. e Zd ZdZdZd
dd�Zdd� Zdd � ZdS ) �Ez+
Dummy new-style class with slots.
��x�yNc C � || _ || _d S r r) �r r* r+ r r r r W �
z
E.__init__c C s | j | jd�S )Nr) r) r r r r � __getstate__[ � zE.__getstate__c C s |d | _ |d | _d S )Nr* r+ r) �r �stater r r � __setstate__^ s
zE.__setstate__)NN)r r r r � __slots__r r/ r3 r r r r r( P s
r( c @ s e Zd Zdd� Zdd� ZdS )�SimpleJellyTestc C r, r r) r- r r r r d r. zSimpleJellyTest.__init__c C s | j |j kS r )�__dict__�r �otherr r r � isTheSameAsh s zSimpleJellyTest.isTheSameAsN)r r r r r9 r r r r r5 c s r5 c C s8 t � |�}t�|�}t�|�}t �|�}| �||� dS )zt
Verify that the given object round-trips through jelly & banana and comes
out equivalent to the input.
N)r r �encode�decode�unjelly� assertEqual)�testCase� toSerialize�jellied�encoded�decoded� unjelliedr r r �jellyRoundTripl s
rD c @ s e Zd ZdZdg d�g d�g d�g d�gZdd� Zd d
� Zd d � Zd
d� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Zd3d4� Zd5d6� Z d7d8� Z d9d:� Z d;d<� Z d=d>� Z d?d@� Z!dAdB� Z"dCS )D�
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� ���)rF r r )rF �@� r )rF 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 �SecurityOptions�allowBasicTypesr< � allowedTypes�pop� assertRaises�
InsecureJelly)r � inputList�atom�c�tasterr r r �
_testSecurity� s
zJellyTests._testSecurityc C sV t � }t� }t� }|j|_||_tj}t`z| �ttj t�|�� W |t_dS |t_w )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$ �arN � TypeErrorr r< )r rU �brR � 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* �n2�n3r r< �assertIsInstance�assertIsr= )r �nrZ rR �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< r\ r= )r r^ rR r_ r r r �test_newStyleWithSlots� s
z!JellyTests.test_newStyleWithSlotsc C �& 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< r= �r �t�rr r r �test_typeNewStyle� � z JellyTests.test_typeNewStylec C rb )ze
Test that a builtin type can be jellied and unjellied to the original
type.
N)�strr r< r= rc r r r �test_typeBuiltin� rg 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)�datetime�nowr r< r= � assertIsNot)r �dtn�dtdrP rR �outputr r r �
test_dateTime� s
zJellyTests.test_dateTimec C sj t �ddd�}t �dddd�}t j �||�}|t � ddd� }t| |� t| |� t| |� t| |� d S )
z�
Jellying L{datetime.time}, L{datetime.timedelta}, L{datetime.datetime},
and L{datetime.date} objects should result in jellied objects which can
be serialized and unserialized with banana.
i� � � r
� � iX � N)rj �date�time�combinerD )r �
sampleDate�
sampleTime�sampleDateTime�sampleTimeDeltar r r �test_bananaTimeTypes� s
z JellyTests.test_bananaTimeTypesc 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 rH �-78.901N)�decimal�Decimalr r< r= rl )r rP rR ro 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.
r~ r rH r N)r� r� r r< � decimalDatar= )r �expectedro 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.
r~ rF N)r� r� rT �r rP r r r �test_decimalSecurity s z JellyTests.test_decimalSecurityc C s6 h d�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
� ru N)r r< r= rl �r rP ro r r r �test_set& s
zJellyTests.test_setc C s: t g d��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
r� ru N)� frozensetr r< r= rl r� r r r �test_frozenset1 s zJellyTests.test_frozensetc C s h d�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� s setN)rT r� r r r �test_setSecurity< s
zJellyTests.test_setSecurityc C s t g d��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� s frozensetN)r� rT r� r r r �test_frozensetSecurityE s z!JellyTests.test_frozensetSecurityc C sN | � tdd��tdd��� tdd�}t�|�}t�|�}| � |�|�� dS )z%
Simplest test case.
rU rW r
r� N)�
assertTruer5 r9 r r< )r rU �cerealrW r r r � test_simpleN 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
r] r r< )r r* r+ �s�zr r r �
test_identityZ s
zJellyTests.test_identityc C s8 d}t �t � |��}| �||� | �t|�t|�� d S )N�blah)r r< r= �typer- r r r �test_stri s zJellyTests.test_strc C sb g }d|i|f}|� |� t�|�}t�|�}| �|d d |d � | �|d d d |� d S )N�listr r
�r
r r< r] )r �reref�
toplevelTupler� r� r r r �test_stressReferenceso s
z JellyTests.test_stressReferencesc C sF g }|f}|� |f� t�|�}t�|�}| �|d d d |� d S �Nr r� )r rU rd r� r� r r r �test_moreReferencesx s
z JellyTests.test_moreReferencesc C s* t �� }t � i �}| �t jt j||� dS )z@
Test for type-level security of serialization.
N)r rJ rN rO r< )r rS �dctr r r �test_typeSecurity� s
z JellyTests.test_typeSecurityc C s t �t�}| �t|� d S r )r r< r' r] )r �ujr r r �test_newStyleClasses� s
z JellyTests.test_newStyleClassesc C s~ t � }t�t�|�� t�t�|j�� tg d�td� td�ddddt tddit jg }|D ]}| �|t�t�|��� q.dS )zA
Test for all types currently supported in jelly
r� r
�testg�����L4@NrU )r r r< r r" �boolr r= )r rU �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 �
|| _ d S r �r8 r7 r r r r � �
zTupleState.__init__c S s | j fS r r� r r r r r/ � s zTupleState.__getstate__c S s |d | _ d S r� r� r1 r r r r3 � r0 zTupleState.__setstate__c S s
t | j�S r )�hashr8 r r r r �__hash__� r� zTupleState.__hash__N)r r r r r/ r3 r� r r r r r� � s
r� �t3r r
)r� r r r< r] r8 )r rU �t1�t2r� �d�t3primer 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 rJ �allowInstancesOfr r# r% rW rR r* r< r\ �
UnpersistablerN rO r] )r rS rU rW rR �friendlyr* �meanr 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 @ � e Zd ZdS )z7JellyTests.test_unjellyable.<locals>.JellyableTestClassN�r r r r r r r �JellyableTestClass� � r� �valueN)r � Jellyable�setUnjellyableForClass� Unjellyable� attributer< r= r\ )r r� �inputro 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 )rh )�obj�jel�perstr r r �persistentStore� s z:JellyTests.test_persistentStorage.<locals>.persistentStorec S s t | �}|d | S r� )�int)�pidstr�unjr� �pidr r r �persistentLoad� s z9JellyTests.test_persistentStorage.<locals>.persistentLoadr� ru r rt � )r� )r� r z persistentStore was not called.z$Persistent storage identity failure.)r5 rW rR r r< r] r� ) r r� r� r� rU rW rR 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 r^ �n1r� r_ r r r � test_newStyleClassesAttributes s
z)JellyTests.test_newStyleClassesAttributesc C sl | � |j|j� | � |jd� | � |jd� | � t|j�t|j�� t|j|j�D ]
\}}| �||� q)d S )Nr )r= r r �lenr �zipr� )r rU rW 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}.
��invokerN)
r �
Referenceable�Broker�makeConnectionr r r< �assertIn�luid� localObjects)r �ref� jellyBroker�j�
unjellyBrokerr� r r r �test_referenceable s z JellyTests.test_referenceableN)#r r r r r� rT rY r` ra rf ri rp 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 rE x sL � rE 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 �
instance_atom�
flushWarnings�test_deprecatedInstanceAtomr= r� �DeprecationWarning�r �warningsr r r r� . 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.A�
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� r= r� r� r� r r r �%test_deprecatedUnjellyingInstanceAtom< s �
�z;JellyDeprecationTests.test_deprecatedUnjellyingInstanceAtomN)r r r r r� r� r r r r r� ) s r� c @ � e Zd Zdd� ZdS )�ClassAc C s t | �| _d S r )�ClassBr� r r r r r N r0 zClassA.__init__N�r r r r r r r r r� M � r� c @ r� )r� c C r� r )r� )r r� r r r r S r� zClassB.__init__Nr� r r r r r� R r� 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< r] r� )r rU 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 @ r� )zHCircularReferenceTests.test_circleWithInvoker.<locals>.DummyInvokerClassNr� r r r r �DummyInvokerClassc r� r� r� r� )�serializingPerspectiver� r r� r� r< �failUnlessIdenticalr� )r r� � dummyInvoker�a0r� �a1r r r �test_circleWithInvokerb 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) �setr5 �addr r< r\ r* r= r� )r r� rU �resr r r r� q 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)r5 r� r* r r< r\ r= r� )r rU r� r r r r r� } s
z%CircularReferenceTests.test_frozensetN)r r r r r� r� r� r� r r r r r� W s r� ) r rj r� �twisted.spreadr r r �twisted.test.proto_helpersr �
twisted.trialr �twisted.trial.unittestr r� r r r" r# r% r' r( r5 rD rE r� �Copyable�
RemoteCopyr� r� r� r r r r �<module> s0 4$