Edit File: engines.cpython-37.pyc
B ��4]�( � @ s d dl mZ d dlZd dlZd dlZddlmZ ddlmZ ddlm Z ddlm Z dd lm Z G d d � d e �Z e � Zd d � Ze dd� �Ze dd� �Ze dd� �Ze dd� �Zd&dd�ZG dd� de �Zd'dd�Zd(d d �Zd)d d �ZG d d!� d!e �ZG d"d#� d#e �Zeefd$d%�ZdS )*� )�absolute_importN� )�config)�uses_deprecated)� decorator� )�event)�poolc @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z e � dd� �Ze � dd� �Zdd� ZdS ) �ConnectionKillerc C s t �� | _t �� | _t� | _d S )N)�weakref�WeakKeyDictionary� proxy_refs�testing_engines�set�conns)�self� r �M/opt/alt/python37/lib64/python3.7/site-packages/sqlalchemy/testing/engines.py�__init__ s zConnectionKiller.__init__c C s d| j |< d S )NT)r )r �enginer r r � add_engine s zConnectionKiller.add_enginec C s | j �||f� d S )N)r �add)r Z dbapi_conn� con_recordr r r �connect s zConnectionKiller.connectc C s d| j |< d S )NT)r )r � dbapi_conr Z con_proxyr r r �checkout! s zConnectionKiller.checkoutc C s | j �||f� d S )N)r �discard)r r r � exceptionr r r � invalidate$ s zConnectionKiller.invalidatec C s@ y |� W n0 t k r: } zt�d| � W d d }~X Y nX d S )Nz5testing_reaper couldn't rollback/close connection: %s)� Exception�warnings�warn)r �fn�er r r �_safe' s zConnectionKiller._safec C s2 x,t | j�D ] }|d k r |jr | �|j� q W d S )N)�listr �is_validr$ �rollback)r �recr r r � rollback_all/ s z ConnectionKiller.rollback_allc C s2 x,t | j�D ] }|d k r |jr | �|j� q W d S )N)r% r r& r$ �_close)r r( r r r � close_all4 s zConnectionKiller.close_allc C s4 x.t | j�D ] \}}|jd kr q | �|j� q W d S )N)r% r � connectionr$ r' )r �connr( r r r �_after_test_ctx9 s z ConnectionKiller._after_test_ctxc C s t jjr| �� n| �� d S )N)r �optionsZlow_connections�_stop_test_ctx_minimal�_stop_test_ctx_aggressive)r r r r �_stop_test_ctxH s z ConnectionKiller._stop_test_ctxc C s: | � � t� | _x$t| j�D ]}|tjk r |�� q W d S )N)r+ r r r% r r �db�dispose)r r( r r r r0 N s z'ConnectionKiller._stop_test_ctx_minimalc C sZ | � � x(t| j�D ]\}}| �|j� d |_qW t� | _xt| j�D ] }|�� qFW d S )N) r+ r% r r$ �closer, r r r4 )r r- r( r r r r1 X s z*ConnectionKiller._stop_test_ctx_aggressivec C s" x | j D ]}|jrdst�qW d S )NF)r r& �AssertionError)r r( r r r �assert_all_closedc s z"ConnectionKiller.assert_all_closedN)�__name__� __module__� __qualname__r r r r r r$ r) r+ r. r2 r r0 r1 r7 r r r r r s r c C sd t �� t|d�r|�� tjjjsVddlm } |j ddd�� | � |� W d Q R X n | � |� d S )Nr5 r )� assertionszCan't sort tablesF)�assert_) �testing_reaperr+ �hasattrr5 r r3 �dialectZsupports_alter� r; Zexpect_warningsZdrop_all)�metadata�bindr; r r r �drop_all_tablesl s rC c O s z| ||� W d t �� X d S )N)r= r7 )r" �args�kwr r r �assert_conns_closedz s rF c O s z| ||� W dt �� X dS )zBDecorator that rolls back all open connections after fn execution.N)r= r) )r" rD rE r r r �rollback_open_connections� s rG c O s t �� | ||� dS )z:Decorator that closes all connections before fn execution.N)r= r+ )r" rD rE r r r � close_first� s rH c O s z| ||� W dt �� X dS )z9Decorator that closes all connections after fn execution.N)r= r+ )r" rD rE r r r �close_open_connections� s rI c c s\ dd l m} xJ|jD ]@}| r&|| kr&qt||d �}|sJttd| �j|�}|�� V qW d S )Nr zsqlalchemy.databases.%s)Zsqlalchemy.databasesZ databases�__all__�getattr� __import__r? )�exclude�d�name�modr r r � all_dialects� s rQ c @ s> e Zd Zdd� Zdd� Zdd� Zdd� Zdd d �Zd d � ZdS )�ReconnectFixturec C s || _ g | _d| _d S )NF)�dbapi� connections� is_stopped)r rS r r r r � s zReconnectFixture.__init__c C s t | j|�S )N)rK rS )r �keyr r r � __getattr__� s z ReconnectFixture.__getattr__c O sT | j j||�}| jr@| �|j� |�� }|�d� dsPtd��n| j� |� |S d S )Nzselect 1Fz%simulated connect failure didn't work) rS r rU r$ r5 �cursor�executer6 rT �append)r rD �kwargsr- Zcursr r r r � s zReconnectFixture.connectc C s@ y |� W n0 t k r: } zt�d| � W d d }~X Y nX d S )Nz.ReconnectFixture couldn't close connection: %s)r r r! )r r" r# r r r r$ � s zReconnectFixture._safeFc C s0 || _ x t| j�D ]}| �|j� qW g | _d S )N)rU r% rT r$ r5 )r �stop�cr r r �shutdown� s zReconnectFixture.shutdownc C s d| _ d S )NF)rU )r r r r �restart� s zReconnectFixture.restartN)F) r8 r9 r: r rW r r$ r^ r_ r r r r rR � s rR c sj | p t jj} t jjj}|s i }t|�|d< t| |���j� � �fdd�}�jjj�_ �jjj �_ |�_�S )N�modulec s �j j�� d�j j_� � d S )NF)r? rS r^ rU r )�_disposer r r r4 � s z$reconnecting_engine.<locals>.dispose) r r3 �urlr? rS rR �testing_enginer4 r^ Z test_shutdownr_ Z test_restart)rb r/ rS r4 r )ra r r �reconnecting_engine� s rd c C s ddl m} ddlm} |s"d}n |�dd�}| p8tjj} || �} |dkrrtjdksd| jtjjjkrltj }q�i }n.tjdk r�| jtjjjkr�tj � � }|� |� || f|�}d|_ t |jtj�r�d|j_d|j_|�rt�|jdtj� t�|jdtj� t�|jd tj� t�|� |S ) zBProduce an engine configured by --options with optional overrides.r )� create_engine)�make_urlT� use_reaperNr r r )� sqlalchemyre Zsqlalchemy.engine.urlrf �popr r3 rb Z drivernameZdb_opts�copy�updateZ _has_events� isinstancer Z QueuePool�_timeoutZ _max_overflowr �listenr= r r r r )rb r/ re rf rg Z default_optr r r r rc � s2 rc c sv ddl m} | stjj} g � � fdd�}� fdd�}�fdd�}|| d d |d ��t�d �r`t�� �_|�_|�_ �S ) a' Provides a mocking engine based on the current testing.db. This is normally used to test DDL generation flow as emitted by an Engine. It should not be used in other cases, as assert_compile() and assert_sql_execution() are much better choices with fewer moving parts. r )re c s � � | � d S )N)rZ )Zsql�arE )�bufferr r �executor' s z mock_engine.<locals>.executorc s" dd� � D �}|| ks t |��d S )Nc S s g | ]}t �d dt|���qS )z[\n\t]r@ )�re�sub�str)�.0�sr r r � <listcomp>+ s z3mock_engine.<locals>.assert_sql.<locals>.<listcomp>)r6 )Zstmts�recv)rp r r � assert_sql* s z mock_engine.<locals>.assert_sqlc s �j � d�� fdd��jD ��S )N� c 3 s | ]}t |j� d ��V qdS ))r? N)rt �compile)ru rv )rN r r � <genexpr>0 s z1mock_engine.<locals>.print_sql.<locals>.<genexpr>)r? �join�mockr )r )rN r � print_sql. s z mock_engine.<locals>.print_sqlz://r~ )Zstrategyrq ) rh re r r3 rO r> r6 r~ ry r )Z dialect_namere rq ry r r )rp r r � mock_engine s r� c @ s2 e Zd ZdZdd� Zd dd�Zdd� Zd d � ZdS ) �DBAPIProxyCursorzuProxy a DBAPI cursor. Tests can provide subclasses of this to intercept DBAPI-level cursor operations. c O s || _ || _|j||�| _d S )N)r r, rX )r r r- rD r[ r r r r D s zDBAPIProxyCursor.__init__Nc K s* |r| j j||f|�S | j j|f|�S d S )N)rX rY )r �stmt� parametersrE r r r rY I s zDBAPIProxyCursor.executec K s | j j||f|�S )N)rX � executemany)r r� �paramsrE r r r r� O s z DBAPIProxyCursor.executemanyc C s t | j|�S )N)rK rX )r rV r r r rW R s z DBAPIProxyCursor.__getattr__)N)r8 r9 r: �__doc__r rY r� rW r r r r r� < s r� c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) �DBAPIProxyConnectionz}Proxy a DBAPI connection. Tests can provide subclasses of this to intercept DBAPI-level connection operations. c C s" |j �� | _| _|| _|| _d S )N)r Z_creatorr- Z _sqla_unwrapr � cursor_cls)r r r� r r r r ^ s z DBAPIProxyConnection.__init__c O s | j | j| jf|�|�S )N)r� r r- )r rD r[ r r r rX c s zDBAPIProxyConnection.cursorc C s | j �� d S )N)r- r5 )r r r r r5 f s zDBAPIProxyConnection.closec C s t | j|�S )N)rK r- )r rV r r r rW i s z DBAPIProxyConnection.__getattr__N)r8 r9 r: r� r rX r5 rW r r r r r� V s r� c s � �fdd�}t d|id�S )zIProduce an engine that provides proxy hooks for common methods. c s � t j��S )N)r r3 r )�conn_clsr� r r � mock_connu s z"proxying_engine.<locals>.mock_connZcreator)r/ )rc )r� r� r� r )r� r� r �proxying_enginem s r� )N)NN)NN)N) � __future__r rr r r r@ r r �utilr r r �objectr r= rC rF rG rH rI rQ rR rd rc r� r� r� r� r r r r �<module> s. T . & (