File: //usr/lib/python3/dist-packages/twisted/test/__pycache__/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