Edit File: replay_fixture.cpython-37.pyc
B ��4]� � @ s� d dl mZ d dlZd dlZddlmZ ddlmZ ddlmZ ddlmZ dd lm Z dd lm Z dd l m Z G d d � d ej �ZG dd� de�ZdS )� )�dequeN� )�config)�fixtures)� profiling� )� create_engine)�MetaData)�util)�Sessionc @ s: e Zd Zejdd� �Zdd� Zdd� Zdd� Zd d � Z d S ) �ReplayFixtureTestc o s d V d S )N� )�self�arg�kwr r �T/opt/alt/python37/lib64/python3.7/site-packages/sqlalchemy/testing/replay_fixture.py� _dummy_ctx s z ReplayFixtureTest._dummy_ctxc s� t � �tjjj� � �fdd�}ttjj|dd�}t|�| _|| _ t |�| _ | � � z| j | jd� W d | �� |�� X �fdd�}ttjj|dd�}t|�| _|| _ t |�| _ | � � z| j tjd� W d | j �� |�� X d S )Nc s �� � � �S )N)�recorderr )�creator� dbapi_sessionr r r s z3ReplayFixtureTest.test_invocation.<locals>.recorderF)r Zuse_native_hstore)�ctxc s � � � S )N)�playerr )r r r r ) s z1ReplayFixtureTest.test_invocation.<locals>.player)�ReplayableSessionr Zdb�poolZ_creatorr �urlr �metadata�enginer Zsession� setup_engine� _run_stepsr �teardown_engineZdisposer Zcount_functions�close)r r r r r )r r r �test_invocation s0 z!ReplayFixtureTest.test_invocationc C s d S )Nr )r r r r r ; s z ReplayFixtureTest.setup_enginec C s d S )Nr )r r r r r > s z!ReplayFixtureTest.teardown_enginec C s t � �d S )N)�NotImplementedError)r r r r r r A s z ReplayFixtureTest._run_stepsN) �__name__� __module__� __qualname__� contextlib�contextmanagerr r! r r r r r r r r s (r c @ s� e Zd ZdZe� Ze� ZejrFe dd� e e �D ��� dd� dD ��Z n:e dd� e e �D ���dd� e�� D ��� dd� d D ��Z d d � Zd d � Zdd� ZG dd� de�ZG dd� de�ZdS )r a A simple record/playback tool. This is *not* a mock testing class. It only records a session for later playback and makes no assertions on call consistency whatsoever. It's unlikely to be suitable for anything other than DB-API recording. c C s g | ]}|� d �stt|��qS )�_)� startswith�getattr�types)�.0�tr r r � <listcomp>S s z ReplayableSession.<listcomp>c C s g | ]}t t|��qS r )r* r+ )r, r- r r r r. V s )� FunctionType�BuiltinFunctionType� MethodType�BuiltinMethodType� LambdaTypeZUnboundMethodTypec C s g | ]}|� d �stt|��qS )r( )r) r* r+ )r, r- r r r r. e s c C s" g | ]}t |t�st|�n|�qS r )� isinstance�type)r, r- r r r r. l s c C s g | ]}t t|��qS r )r* r+ )r, r- r r r r. r s )r/ r0 r1 r2 r3 c C s t � | _d S )N)r �buffer)r r r r �__init__~ s zReplayableSession.__init__c C s | � | j|�S )N)�Recorderr6 )r �baser r r r � s zReplayableSession.recorderc C s | � | j�S )N)�Playerr6 )r r r r r � s zReplayableSession.playerc @ s0 e Zd Zdd� Zdd� Zedd� �Zdd� Zd S ) zReplayableSession.Recorderc C s || _ || _d S )N)�_buffer�_subject)r r6 �subjectr r r r7 � s z#ReplayableSession.Recorder.__init__c sZ � fdd�dD �\}}|||�}t |�tjkrH|�tj� t � �||�S |�|� |S d S )Nc s g | ]}t �� |��qS r )�object�__getattribute__)r, �x)r r r r. � s z7ReplayableSession.Recorder.__call__.<locals>.<listcomp>)r< r; )r5 r �Natives�append�Callable)r �argsr r= r6 �resultr )r r �__call__� s z#ReplayableSession.Recorder.__call__c C s | j S )N)r< )r r r r � _sqla_unwrap� s z'ReplayableSession.Recorder._sqla_unwrapc s� y t �� |�S tk r Y nX � fdd�dD �\}}yt|��||�}W n" tk rn |�tj� � Y n8X t|�tjkr�|�tj� t� �||�S |�|� |S d S )Nc s g | ]}t �� |��qS r )r> r? )r, r@ )r r r r. � s z?ReplayableSession.Recorder.__getattribute__.<locals>.<listcomp>)r< r; ) r> r? �AttributeErrorr5 rB r � NoAttributerA rC )r �keyr= r6 rE r )r r r? � s z+ReplayableSession.Recorder.__getattribute__N)r# r$ r% r7 rF �propertyrG r? r r r r r8 � s r8 c @ s0 e Zd Zdd� Zdd� Zedd� �Zdd� Zd S ) zReplayableSession.Playerc C s || _ d S )N)r; )r r6 r r r r7 � s z!ReplayableSession.Player.__init__c O s* t �| d�}|�� }|tjkr"| S |S d S )Nr; )r> r? �popleftr rC )r rD r r6 rE r r r rF � s z!ReplayableSession.Player.__call__c C s d S )Nr )r r r r rG � s z%ReplayableSession.Player._sqla_unwrapc C s` y t �| |�S tk r Y nX t �| d�}|�� }|tjkrD| S |tjkrXt|��n|S d S )Nr; )r> r? rH rL r rC rI )r rJ r6 rE r r r r? � s z)ReplayableSession.Player.__getattribute__N)r# r$ r% r7 rF rK rG r? r r r r r: � s r: N)r# r$ r% �__doc__r> rC rI r Zpy2k�set�dirr+ � differencerA �union� __builtins__�valuesr7 r r r8 r: r r r r r E s( .r )� collectionsr r& r+ � r r r r r r Zormr ZTestBaser r> r r r r r �<module> s 7