Your IP : 216.73.216.213
B
;��] � @ s� d dl mZmZ d dlZd dlZd dlmZ d dlmZ d dl m
Z
d dl m Z d dl
mZ G dd � d e j�Zd
d d d
d gd
d igZG d
d� de�ZG dd� dej�ZdS )� )�division�absolute_importN)�dedent)�unittest)�sob)�
components)� Ephemeralc @ s e Zd ZdS )�DummyN)�__name__�
__module__� __qualname__� r
r
�H/opt/alt/python37/lib64/python3.7/site-packages/twisted/test/test_sob.pyr s r � �hello)r r c @ s e Zd ZdS )�
FakeModuleN)r
r r r
r
r
r r s r c @ sd e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dS )� PersistTestsc C sf x`t D ]X}t�|d�}xFd�� D ]:}|�|� |jd| d� t�d| |�}| �||� q W qW d S )N� z
source picklez persisttest.)�filename)�objectsr �
Persistent�split�setStyle�save�load� assertEqual)�self�o�p�style�o1r
r
r �
testStyles! s
zPersistTests.testStylesc C s� t � }d|_|�tjt�|d�� xjd�� D ]^}t�|��|� t�|�jd| d� t� d| |�}| �
|j|j� | �
t�|�j |� q,W d S )N� Zlalaz
source picklezlala.)r ) r �fooZ setComponentr Z IPersistabler r r r r r r )r r r r r
r
r �testStylesBeingSet* s z PersistTests.testStylesBeingSetc C s$ t �dd�}| jt|jddd� dS )z?
Calling save() with a passphrase is an error.
N�objectzfilename.pickle�abc)Z
passphrase)r Z
Persistant� assertRaises� TypeErrorr )r r r
r
r �testPassphraseError6 s z PersistTests.testPassphraseErrorc C s� dddg}t �|d�}x�d�� D ]z}|�|� |�� t �d|d |�}| �||� xBd�� D ]6}|�|� t �d | d
|d |�}| �||� q`W q W d S ) Nr � � r% z
source picklez object.tar z lala lolozobject-z.ta)r r r r r r r )r r r r r �tagr
r
r � testNames? s
zPersistTests.testNamesc C sB t dd��}|�d� W d Q R X t�dd�}| �|dddg� d S )Nzpersisttest.python�wz foo=[1,2,3] r# r r* r+ )�open�writer ZloadValueFromFiler )r �fr r
r
r �
testPythonM s zPersistTests.testPythonc C s� | � ttjd� | �dt�d�� | �dt�d�� | �dt�d�� | �dt�d�� | �dt�d�� | �d t�d
�� | �d t�d �� d S ) Nz file.blah�pythonzfile.pyzfile.tacz file.etac�picklezfile.tapz file.etap�sourcezfile.tasz file.etas)r' �KeyErrorr Z guessTyper )r r
r
r �testTypeGuesserT s z PersistTests.testTypeGuesserc C s� d| j _| �� }t�|� tj�|d�}dat|d��}|� t
d�� W dQ R X t � |d�}| �
|t� | �|d d� | �|d d � | �
tt� | �
tjt� bdS )
z�
L{_EverythingEphermal.__getattr__} will proxy the __main__ module as an
L{Ephemeral} object, and during load will be transparent, but after
load will return L{Ephemeral} objects from any accessed attributes.
r zpersisttest.ee_getattrNr. a�
app = []
import __main__
app.append(__main__.testMainModGetattr == 1)
try:
__main__.somethingElse
except AttributeError:
app.append(True)
else:
app.append(False)
from twisted.test import test_sob
test_sob.mainWhileLoading = __main__
r5 r zExpected attribute not set.zUnexpected attribute set.)�fakeMainZtestMainModGetattr�mktemp�os�mkdir�path�joinZmainWhileLoadingr/ r0 r r r ZassertIsInstance�listZ
assertTruer Z
somethingElse)r �dirnamer r1 �loadedr
r
r � testEverythingEphemeralGetattr^ s
z+PersistTests.testEverythingEphemeralGetattrc C s| d| j _| �� }t�|� tj�|d�}t|d��$}|�d� |�d� |�d� W dQ R X t �
|d� | � | j jd� dS ) zU
Verify that _EverythingEphemeral.__setattr__ won't affect __main__.
r zpersisttest.ee_setattrr. zimport __main__
z __main__.testMainModSetattr = 2
z app = None
Nr5 ) r8 ZtestMainModSetattrr9 r: r; r< r= r/ r0 r r r )r r? r r1 r
r
r � testEverythingEphemeralSetattr� s
z+PersistTests.testEverythingEphemeralSetattrc C sl | � � }t�|� tj�|d�}t|d��}|�d� W dQ R X | �tt j
|d� | � t t
jd �t� dS )zW
Test that an exception during load() won't cause _EE to mask __main__
zpersisttest.ee_exceptionr. zraise ValueError
Nr5 �__main__)r9 r: r; r< r= r/ r0 r' �
ValueErrorr r r �type�sys�modulesr )r r? r r1 r
r
r � testEverythingEphemeralException� s
z-PersistTests.testEverythingEphemeralExceptionc C s" t jd | _t� | _t jd< dS )zi
Replace the __main__ module with a fake one, so that it can be mutated
in tests
rC N)rF rG �realMainr r8 )r r
r
r �setUp� s zPersistTests.setUpc C s | j tjd< dS )z8
Restore __main__ to its original value
rC N)rI rF rG )r r
r
r �tearDown� s zPersistTests.tearDownN)r
r r r! r$ r) r- r2 r7 rA rB rH rJ rK r
r
r
r r s
'r )�
__future__r r r: rF �textwrapr Z
twisted.trialr Ztwisted.persistedr Ztwisted.pythonr Ztwisted.persisted.stylesr Z
Componentizedr r r% r ZTestCaser r
r
r
r �<module> s