File: //lib/python3/dist-packages/twisted/test/__pycache__/test_sob.cpython-38.pyc
U
W[ � @ 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 �7/usr/lib/python3/dist-packages/twisted/test/test_sob.pyr s r � �hello)r r c @ s e Zd ZdS )�
FakeModuleNr
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 s^ t D ]T}t�|d�}d�� D ]:}|�|� |jd| d� t�d| |�}| �||� q qd S )N� �
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�� d�� D ]^}t�|��|� t�|�jd| d� t� d| |�}| �
|j|j� | �
t�|�j |� q*d S )N� Zlalar zlala.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�}d�� D ]v}|�|� |�� t �d|d |�}| �||� d�� D ]6}|�|� t �d | d
|d |�}| �||� q\q d S ) Nr � � r( r z 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 5 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 5 Q 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_getattrNr1 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__
r8 r zExpected attribute not set.zUnexpected attribute set.)�fakeMainZtestMainModGetattr�mktemp�os�mkdir�path�joinZmainWhileLoadingr2 r3 r r r ZassertIsInstance�listZ
assertTruer Z
somethingElse)r �dirnamer r4 Zloadedr r r � testEverythingEphemeralGetattr^ s
z+PersistTests.testEverythingEphemeralGetattrc C s| d| j _| �� }t�|� tj�|d�}t|d��$}|�d� |�d� |�d� W 5 Q R X t �
|d� | � | j jd� dS ) zU
Verify that _EverythingEphemeral.__setattr__ won't affect __main__.
r zpersisttest.ee_setattrr1 zimport __main__
z __main__.testMainModSetattr = 2
z app = None
r8 N) r< ZtestMainModSetattrr= r> r? r@ rA r2 r3 r r r �r rC r r4 r r r � testEverythingEphemeralSetattr� s
z+PersistTests.testEverythingEphemeralSetattrc C sl | � � }t�|� tj�|d�}t|d��}|�d� W 5 Q 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_exceptionr1 zraise ValueError
r8 �__main__N)r= r> r? r@ rA r2 r3 r* �
ValueErrorr r r �type�sys�modulesr rE 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
rG N)rJ rK �realMainr r< r: r r r �setUp� s zPersistTests.setUpc C s | j tjd< dS )z8
Restore __main__ to its original value
rG N)rM rJ rK r: r r r �tearDown� s zPersistTests.tearDownN)r r r
r$ r' r, r0 r5 r; rD rF rL rN rO r r r r r s
'r )Z
__future__r r r> rJ �textwrapr Z
twisted.trialr Ztwisted.persistedr Ztwisted.pythonr Ztwisted.persisted.stylesr Z
Componentizedr r r( r ZTestCaser r r r r �<module> s �