← Back
Editing: 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)
Save File
Cancel