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