← Back
Editing: test_persisted.cpython-38.pyc
U ��W[ 9 � @ s< d dl mZmZ d dlZd dlmZ z d dlZW n ek rL d dlZY nX d dl Z zd dl m Z W n ek r~ dZ Y nX dZ z d dlZW n d dlZY nX d dlmZmZmZ d dlmZ G dd � d ej�ZG d d � d ej�ZG d d � d ej�ZG dd� de�ZG dd� dee�Z G dd� dej�Z G dd� dej �Z G dd� dej�Z G dd� d�Z!G dd� de"�Z#G d d � d e"�Z$G d d � d e"�Z%d d!� Z&e�e$e&� G d"d#� d#�Z'G d$d%� d%�Z(d&d'� Z)G d(d)� d)ej�Z*G d*d+� d+ej�Z+G d,d-� d-�Z,G d.d/� d/�Z-G d0d1� d1ej�Z.G d2d3� d3ej�Z/ee e*gZ0dS )4� )�division�absolute_importN)�unittest)�StringIOzNo cStringIO available.)�styles�aot�crefutil)�_PY3c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) � VersionTestsc C sT G dd� dt �at�t� �}G dd� dtjt �at�|�}t�� |jsPt d��d S )Nc @ s e Zd Zdd� ZdS )� NullVersionedc S s d| _ d S )Nr ��ok��self� r �=/usr/lib/python3/dist-packages/twisted/test/test_persisted.py�__init__( s zNullVersioned.__init__N��__name__� __module__� __qualname__r r r r r r ' s r c @ s e Zd ZdZdd� ZdS )r � c S s d| _ d S �Nr r r r r r �upgradeToVersion1- s z NullVersioned.upgradeToVersion1N)r r r �persistenceVersionr r r r r r + s zinitial upgrade not run!) �objectr �pickle�dumpsr � Versioned�loads� doUpgrader �AssertionError)r ZpkclZmnvr r r �test_nullVersionUpgrade% s z$VersionTests.test_nullVersionUpgradec C s� G dd� dt j�at� }|js$|jr,td��t�|�}dt_t� |�}t � � |js\td��|jsjtd��t�|�}t� |�}t � � |jdks�td��|jdks�td��d S ) Nc @ s6 e Zd ZdZdgZdZdZdd� Zdd� Zdd � Z d S ) � MyVersioned� � garbagedatar c S s d| _ dd� | _d S )NZxxxc S s dS )Nz cant persistr )�qr r r �<lambda>= � z&MyVersioned.__init__.<locals>.<lambda>)Zsomedatar% r r r r r ; s zMyVersioned.__init__c S s | j d7 _ d S r )�v3r r r r �upgradeToVersion3? s z MyVersioned.upgradeToVersion3c S s | j d7 _ d S r )�v4r r r r �upgradeToVersion4B s z MyVersioned.upgradeToVersion4N) r r r r ZpersistenceForgetsr) r+ r r* r, r r r r r# 5 s r# zhasn't been upgraded yet� zdidn't do version 3 upgradezdidn't do version 4 upgrader zupgraded unnecessarily) r r r# r) r+ r! r r r r r )r Zmv�pickl�objr r r �test_versionUpgrade3 s z VersionTests.test_versionUpgradec C s� G dd� dt j�atdd�}tdd�}t�||f�}~~dt_dd� t_t�|�\}}t �� | � |j d� | � |j d� | � |j � | � |j � d S ) Nc @ s e Zd Zdd� Zdd� ZdS )�ClassWithCustomHashc S s || _ || _d S �N)�unique�hash)r r3 r4 r r r r U s z ClassWithCustomHash.__init__c S s | j S r2 )r4 r r r r �__hash__X s z ClassWithCustomHash.__hash__N)r r r r r5 r r r r r1 T s r1 �v1r �v2r c S s t | dd�S �N�upgradedT��setattrr r r r r' a r( z3VersionTests.test_nonIdentityHash.<locals>.<lambda>) r r r1 r r r r r r � assertEqualr3 � assertTruer9 )r r6 r7 Zpklr r r �test_nonIdentityHashR s z!VersionTests.test_nonIdentityHashc s� G dd� dt j�aG dd� dt j�at� }t� }t�|�t�|� }� ~~dt_� fdd�}|t_dt_dd � t_t�|�}t � � | � |j j � d S ) Nc @ s e Zd ZdS )� ToyClassAN�r r r r r r r r? k s r? c @ s e Zd ZdS )� ToyClassBNr@ r r r r rA m s rA r c s t �� �| _t�� d S r2 )r r �yr r r �ZpklBr r r t s zWVersionTests.test_upgradeDeserializesObjectsRequiringUpgrade.<locals>.upgradeToVersion1c S s t | dd�S r8 r: r r r r r' y r( zNVersionTests.test_upgradeDeserializesObjectsRequiringUpgrade.<locals>.<lambda>) r r r? rA r r r r r r r= rB r9 )r �xrB ZpklAr r rC r �/test_upgradeDeserializesObjectsRequiringUpgradei s z<VersionTests.test_upgradeDeserializesObjectsRequiringUpgradeN)r r r r" r0 r> rE r r r r r $ s r c @ s e Zd ZdS )�VersionedSubClassNr@ r r r r rF � s rF c @ s e Zd ZdS )�SecondVersionedSubClassNr@ r r r r rG � s rG c @ s e Zd ZdS )�VersionedSubSubClassNr@ r r r r rH � s rH c @ s e Zd ZdS )�VersionedDiamondSubClassNr@ r r r r rI � s rI c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) � AybabtuTestsz� L{styles._aybabtu} gets all of classes in the inheritance hierarchy of its argument that are strictly between L{Versioned} and the class itself. c C s | � t�tj�g � dS )zM L{styles._aybabtu} of L{Versioned} itself is an empty list. N)r< r �_aybabtur r r r r �test_aybabtuStrictEmpty� s z$AybabtuTests.test_aybabtuStrictEmptyc C s | � t�t�g � dS )z� There are no classes I{between} L{VersionedSubClass} and L{Versioned}, so L{styles._aybabtu} returns an empty list. N)r< r rK rF r r r r �test_aybabtuStrictSubclass� s z'AybabtuTests.test_aybabtuStrictSubclassc C s | � t�t�tg� dS )z� With a sub-sub-class of L{Versioned}, L{styles._aybabtu} returns a list containing the intervening subclass. N)r< r rK rH rF r r r r �test_aybabtuSubsubclass� s �z$AybabtuTests.test_aybabtuSubsubclassc C s | � t�t�tttg� dS )z� For a diamond-shaped inheritance graph, L{styles._aybabtu} returns a list containing I{both} intermediate subclasses. N)r< r rK rI rH rF rG r r r r �test_aybabtuStrict� s �z AybabtuTests.test_aybabtuStrictN)r r r �__doc__rL rM rN rO r r r r rJ � s rJ c @ s e Zd Zdd� ZdS )� MyEphemeralc C s || _ d S r2 �rD �r rD r r r r � s zMyEphemeral.__init__Nr r r r r rQ � s rQ c @ s e Zd Zdd� ZdS )�EphemeralTestsc C s\ t d�}| �|jt � | �|jd� t�|�}t�|�}| �|jtj� | � t |d�� d S )N� rD ) rQ r< � __class__rD r r r r � EphemeralZ assertFalse�hasattr)r �or. r r r �test_ephemeral� s z EphemeralTests.test_ephemeralN)r r r rZ r r r r rT � s rT c @ s e Zd Zdd� Zdd� ZdS )� Pickleablec C s || _ d S r2 rR rS r r r r � s zPickleable.__init__c C s | j S r2 rR r r r r �getX� s zPickleable.getXN)r r r r r\ r r r r r[ � s r[ c @ s e Zd ZdZdd� ZdS )� NotPickleablez) A class that is not pickleable. c C s t d��dS )z9 Raise an exception instead of pickling. zNot serializable.N)� TypeErrorr r r r � __reduce__� s zNotPickleable.__reduce__N)r r r rP r_ r r r r r] � s r] c @ s e Zd ZdZdd� ZdS )�CopyRegisteredzb A class that is pickleable only because it is registered with the C{copyreg} module. c C s t � | _dS )zE Ensure that this object is normally not pickleable. N)r] Z notPickleabler r r r r � s zCopyRegistered.__init__N)r r r rP r r r r r r` � s r` c @ s e Zd ZdZdS )�CopyRegisteredLoadedz2 L{CopyRegistered} after unserialization. N)r r r rP r r r r ra � s ra c C s t dfS )z� Externally implement C{__reduce__} for L{CopyRegistered}. @param cr: The L{CopyRegistered} instance. @return: a 2-tuple of callable and argument list, in this case L{CopyRegisteredLoaded} and no arguments. r )ra )Zcrr r r �reduceCopyRegistered� s rb c @ s e Zd ZdZdd� ZdS )�A� dummy class c C s d S r2 r r r r r �amethod s z A.amethodN)r r r rP re r r r r rc s rc c @ s e Zd ZdZdd� ZdS )�Brd c C s d S r2 r r r r r �bmethod s z B.bmethodN)r r r rP rg r r r r rf s rf c C s d S r2 r r r r r �funktion s rh c @ sJ e Zd ZdZdd� Zdd� Zejdkr,de_dd � Z d d � Z e rFe e _d S ) � PicklingTestsz$Test pickling of extra object types.c C s$ t �t�}t �|�}| �|t� d S r2 )r r r r r< �r r. rY r r r � test_module! s zPicklingTests.test_modulec C s( t �tj�}t �|�}| �|tj� dS )zt After importing L{twisted.persisted.styles}, it is possible to pickle classmethod objects. N)r r r[ r\ r r< rj r r r �test_classMethod' s z PicklingTests.test_classMethod)rU r- z_As of Python 3.4 it is no longer possible to globally change the behavior of function pickling.c C sF t d�}t�|j�}t�|�}| �|� d� | �t|�t|j�� d S )Nr- )r[ r r r\ r r< �type)r r/ r. rY r r r �test_instanceMethod7 s z!PicklingTests.test_instanceMethodc C sL t � }|�d� t�|�}t�|�}| �t|�t|�� | �|�� d� d S )N�abc)�_oldStyleCStringIO�writer r r r< rm �getvalue)r �fr. rY r r r � test_stringIO> s zPicklingTests.test_stringION) r r r rP rk rl �sys� version_info�skiprn rt � skipStringIOr r r r ri s �ri c @ s e Zd ZdZesdZdd� ZdS )�StringIOTransitionTestsz� When pickling a cStringIO in Python 2, it should unpickle as a BytesIO or a StringIO in Python 3, depending on the type of its contents. z4In Python 2 we can still unpickle cStringIO as such.c C s0 d}t �|�}| �|tj� | �|�� d� dS )zg A cStringIO pickled with bytes in it will yield an L{io.BytesIO} on python 3. sE ctwisted.persisted.styles unpickleStringI p0 (S'test' p1 I0 tp2 Rp3 .ZtestN)r r �assertIsInstance�ior r< rr )r ZpickledStringIWithTextZloadedr r r �test_unpickleBytesIOU s � z,StringIOTransitionTests.test_unpickleBytesION)r r r rP r rw r| r r r r ry K s ry c @ s e Zd Zdd� ZdS )� EvilSourcerorc C s | | _ | | j _|| j j_d S r2 )�a�b�crS r r r r e s zEvilSourceror.__init__Nr r r r r r} d s r} c @ s e Zd Zdd� Zdd� ZdS )� NonDictStatec C s | j S r2 ��stater r r r � __getstate__k s zNonDictState.__getstate__c C s || _ d S r2 r� )r r� r r r � __setstate__m s zNonDictState.__setstate__N)r r r r� r� r r r r r� j s r� c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dS )�AOTTestsc C sF ddddt ddd�ddtjd d tf }t�t�|��}| �||� d S ) Nr g @y @Tr$ rU �hello�world)�sliceru �maxsize�Ellipsisr �unjellyFromSource� jellyToSourcer< )r r/ ZrtObjr r r �test_simpleTypesq s �zAOTTests.test_simpleTypesc C sL t � }t� }|j|_||_t�t�|��jj}| �t�|�j t� |�� d S r2 ) rc rf rg r~ r r� r� r< Z _selfOfMethodrV Z_classOfMethod)r r~ r Zim_r r r �test_methodSelfIdentityx s �z AOTTests.test_methodSelfIdentityc C sL t � }t� }|j|_||_tj}t`z| �ttjt�|�� W 5 |t_X dS )z� If a class change after an instance has been created, L{aot.unjellyFromSource} shoud raise a C{TypeError} when trying to unjelly the instance. N) rc rf rg r~ � assertRaisesr^ r r� r� )r r~ r Z savedbmethodr r r �test_methodNotSelfIdentity� s �z#AOTTests.test_methodNotSelfIdentityc C s& G dd� dt �}| �ttj|� � dS )z� L{aot.jellyToSource} should raise a C{TypeError} when trying to jelly an unknown type without a C{__dict__} property or C{__getstate__} method. c @ s e Zd Zedd� �ZdS )z2AOTTests.test_unsupportedType.<locals>.UnknownTypec S s t � �d S r2 )�AttributeErrorr r r r �__dict__� s z;AOTTests.test_unsupportedType.<locals>.UnknownType.__dict__N)r r r �propertyr� r r r r � UnknownType� s r� N)r r� r^ r r� )r r� r r r �test_unsupportedType� s z AOTTests.test_unsupportedTypec C s� t �� j}d|d�}dddddddd d tt j|tg }t|�}|�|� |�|� |�|� t �t �||g��}|d |d ks�t �|d d d � |d d � ks�t �d S ) Nr� )r� �methodr r$ rU zhe llo "x world!u goodbye တ world!g �?�d r � ) r Z AOTJellierZ jellyToAOr rh �tuple�appendr� r� r! )r Zaj�d�l�t�ujr r r �test_basicIdentity� s( � zAOTTests.test_basicIdentityc C s, t � }d|_t�t�|��j|jks(t�d S )Nz meringue!)r� r� r r� r� r! )r r~ r r r �test_nonDictState� s zAOTTests.test_nonDictStatec C s" t �t �t� ��}| �|t� dS )z� L{aot.jellyToSource} and L{aot.unjellyFromSource} honor functions registered in the pickle copy registry. N)r r� r� r` rz ra )r r� r r r � test_copyReg� s zAOTTests.test_copyRegc C sV t t g ��}t�|�}t�|�}|j|ks.t�|jj|jks@t�|j|jjk sRt�d S r2 )r} r � jellyToAOT�unjellyFromAOTr~ r! r r� )r rY �j1�ojr r r �test_funkyReferences� s z AOTTests.test_funkyReferencesc C sf g }|df}|� |� t�|�}t�|�}| �|d t� | �|d d |� | �|d d d� dS )zS L{aot.jellyToAOT} can persist circular references through tuples. i� r r N)r� r r� r� rz r� ZassertIsr< )r r� r� r� r� r r r �test_circularTuple� s zAOTTests.test_circularTupleN) r r r r� r� r� r� r� r� r� r� r� r r r r r� p s r� c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd S ) � CrefUtilTestsz Tests for L{crefutil}. c C s t �i �}| �t|jdd� dS )zO L{crefutil._DictKeyAndValue} only support keys C{0} and C{1}. r$ rU N)r Z_DictKeyAndValuer� � RuntimeError� __setitem__�r r� r r r �test_dictUnknownKey� s z!CrefUtilTests.test_dictUnknownKeyc C s& t �� }d|d< | �t|jdd� dS )zI L{crefutil._Defer} can be assigned a key only one time. r r N)r Z_Deferr� r� r� r� r r r �test_deferSetMultipleTimes� s z(CrefUtilTests.test_deferSetMultipleTimesc C s( t �dddgt�}| �t|ji d� dS )z� A L{crefutil._Container} where all of its elements are known at construction time is nonsensical and will result in errors in any call to addDependant. r r$ rU z ignore-meN)r Z _Container�listr� r! Z addDependant)r Z containerr r r �&test_containerWhereAllElementsAreKnown� s �z4CrefUtilTests.test_containerWhereAllElementsAreKnownc C s | � tt� jt�� d� dS )z� If a dictionary key contains a circular reference (which is probably a bad practice anyway) it will be resolved by a L{crefutil._DictKeyAndValue}, not by placing a L{crefutil.NotKnown} into a dictionary key. �valueN)r� r! �dictr� r �NotKnownr r r r �.test_dontPutCircularReferencesInDictionaryKeys� s �z<CrefUtilTests.test_dontPutCircularReferencesInDictionaryKeysc C s | � tt�dt�� t�� dS )z� L{crefutil._InstanceMethod} raises L{AssertionError} to indicate it should not be called. This should not be possible with any of its API clients, but is provided for helping to debug. Zno_nameN)r� r! r Z_InstanceMethodr� rm r r r r �*test_dontCallInstanceMethodsThatArentReady s ��z8CrefUtilTests.test_dontCallInstanceMethodsThatArentReadyN) r r r rP r� r� r� r� r� r r r r r� � s r� )1Z __future__r r ru Z twisted.trialr ZcPickler � ImportErrorr{ � cStringIOr rp rx �copyregZcopy_regZtwisted.persistedr r r Ztwisted.python.compatr ZTestCaser r rF rG rH rI rJ rW rQ rT r[ r r] r` ra rb rc rf rh ri ry r} r� r� r� Z testCasesr r r r �<module> sT ]) -n8
Save File
Cancel