/
opt
/
alt
/
python27
/
lib64
/
python2.7
/
site-packages
/
sqlalchemy
/
testing
/
up file
home
� 1��fc @@ s� d d l m Z d d l Z d d l Z d d l Z d d l Z d d l m Z d d l m Z d d l m Z d d l m Z d d l m Z d d l m Z d e f d � � YZ e � Z e d � � Z e d � � Z e d � � Z e d � � Z d d � Z d e f d � � YZ d d d � Z d d d e e e d � Z d d � Z d e f d � � YZ d e f d � � YZ e e d � Z d S( i ( t absolute_importNi ( t config( t decorator( t gc_collecti ( t event( t pool( t await_onlyt ConnectionKillerc B@ s� e Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z RS( c C@ s1 t j � | _ t j t � | _ t � | _ d S( N( t weakreft WeakKeyDictionaryt proxy_refst collectionst defaultdictt sett testing_enginest dbapi_connections( t self( ( sM /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyt __init__ s c C@ s\ t j | d | j � t j | d | j � t j | d | j � t j | d | j � d S( Nt checkoutt checkint closet close_detached( R t listent _add_connt _remove_conn( R R ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyt add_pool s c C@ s! | j j | � t | j | <d S( N( R t addt TrueR ( R t dbapi_cont con_recordt con_proxy( ( sM /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyR % s c G@ s | j j | � d S( N( R t discard( R t dbapi_connt arg( ( sM /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyR ) s c C@ s: | j | j � | d k s" t � | j | j | � d S( Nt classt globalt functiont fixture( R" R# R$ R% ( R R t AssertionErrorR R ( R t enginet scope( ( sM /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyt add_engine, s c C@ s6 y | � Wn$ t k r1 } t j d | � n Xd S( Ns5 testing_reaper couldn't rollback/close connection: %s( t Exceptiont warningst warn( R t fnt e( ( sM /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyt _safe2 s c C@ sF x? t | j � D]. } | d k r | j r | j | j � q q Wd S( N( t listR t Nonet is_validR/ t rollback( R t rec( ( sM /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyt rollback_all: s c C@ s� xR t | j � D]A } | d k r | j r | j j | j � | j | j � q q Wx! | j D] } | j | j � q_ W| j j � d S( N( R0 R R1 R2 R R t dbapi_connectionR/ t _checkinR3 t clear( R R4 t con( ( sM /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyt checkin_all? s c C@ s | j � d S( N( R: ( R ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyt close_allN s c C@ s7 t j s d Sd d l m } | j | j j | � d S( Ni ( t provision( R t bootstrapped_as_sqlalchemyt R< t prepare_for_drop_tablesR' t url( R t connectionR< ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyR? Q s c C@ s� | j | } x� t | � D]� } xc t | j � D]R } | d k r0 | j r0 | j d k r� | j | j k r� | j | j � q� q0 q0 Wt | d � r� t | j � � q | j � q W| j � d S( Nt sync_engine( R R0 R R1 R2 t _poolR R/ R7 t hasattrR t disposeR8 ( R R( t engR4 t proxy_ref( ( sM /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyt _drop_testing_enginesZ s c C@ s | j d � d S( NR$ ( RH ( R ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyt after_testj s c C@ si t j s d S| j j r d S| j � d d l m } t j j � � } | j | j j | � Wd QXd S( Ni ( R< ( R R= t __class__t" __leave_connections_for_teardown__R: R> R< t dbt connectR? R' R@ ( R t testR< t conn( ( sM /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyt after_test_outside_fixturesm s c C@ s( | j � | j d � | j d � d S( NR$ R" ( R: RH ( R ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyt stop_test_class_inside_fixtures s c C@ sa t j j r] t � t j j r] t t j j � } t j j j � t sZ t d | � � q] n d S( Ns/ %d connection recs not cleared after test suite( R t baset _strong_ref_connection_recordsR t lenR8 t FalseR& ( R t ln( ( sM /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyt stop_test_class_outside_fixtures� s c C@ s/ | j � x | j D] } | j | � q Wd S( N( R: R RH ( R R( ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyt final_cleanup� s c C@ s0 x) | j D] } | j r t s( t � q q Wd S( N( R R2 RU R& ( R R4 ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyt assert_all_closed� s ( t __name__t __module__R R R R R) R/ R5 R: R; R? RH RI RP RQ RW RX RY ( ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyR s" c O@ s# z | | | � Wd t j � Xd S( N( t testing_reaperRY ( R- t argst kw( ( sM /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyt assert_conns_closed� s c O@ s# z | | | � Wd t j � Xd S( sB Decorator that rolls back all open connections after fn execution.N( R\ R5 ( R- R] R^ ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyt rollback_open_connections� s c O@ s t j � | | | � d S( s: Decorator that closes all connections before fn execution.N( R\ R: ( R- R] R^ ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyt close_first� s c O@ s# z | | | � Wd t j � Xd S( s9 Decorator that closes all connections after fn execution.N( R\ R: ( R- R] R^ ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyt close_open_connections� s c c@ s� d d l j } xk | j D]` } | r7 | | k r7 q n t | | d � } | sn t t d | � j | � } n | j � Vq Wd S( Ni s sqlalchemy.dialects.%s( t sqlalchemy.dialectst dialectst __all__t getattrR1 t __import__t dialect( t excludet dt namet mod( ( sM /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyt all_dialects� s t ReconnectFixturec B@ sA e Z d � Z d � Z d � Z d � Z e d � Z d � Z RS( c C@ s | | _ g | _ t | _ d S( N( t dbapit connectionsRU t is_stopped( R Ro ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyR � s c C@ s t | j | � S( N( Rf Ro ( R t key( ( sM /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyt __getattr__� s c O@ st | j j | | � } | j r\ | j | j � | j � } | j d � t sp t d � � n | j j | � | Sd S( Ns select 1s% simulated connect failure didn't work( Ro RM Rq R/ R t cursort executeRU R& Rp t append( R R] t kwargsRO t curs( ( sM /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyRM � s c C@ s6 y | � Wn$ t k r1 } t j d | � n Xd S( Ns. ReconnectFixture couldn't close connection: %s( R* R+ R, ( R R- R. ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyR/ � s c C@ s@ | | _ x' t | j � D] } | j | j � q Wg | _ d S( N( Rq R0 Rp R/ R ( R t stopt c( ( sM /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyt shutdown� s c C@ s t | _ d S( N( RU Rq ( R ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyt restart� s ( RZ R[ R Rs RM R/ RU R{ R| ( ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyRn � s c @ s� | p t j j } t j j j } | s0 i } n t | � | d <t | | � � � j � � � f d � } � j j j � _ � j j j � _ | � _ � S( Nt modulec @ s* � j j j � t � j j _ � � d S( N( Rh Ro R{ RU Rq ( ( t _disposeR' ( sM /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyRE s ( R RL R@ Rh Ro Rn t testing_engineRE R{ t test_shutdownR| t test_restart( R@ t optionsRo RE ( ( R~ R' sM /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyt reconnecting_engine� s c C@ s� | r& | s t � d d l m } nJ | sM t j r` t j j r` | t k r` d d l m } n d d l m } d d l m } | s� t } d } t } n6 | j d t � } | j d d � } | j d t � } | p� t j j } | | � } | d k r7t j d k s"| j t j j j k r.t j } q}i } nF t j d k r}| j t j j j k r}t j j � } | j | � n | | | � } | r�| j d k r�t j | d � d � � } t j | d � d � � } n | r6d d l m } t j d k r6t t j j | � r6t } | j j t j j � q6n | d k rc| rWt | j _ qct | _ n t | j t j � r�d | j _ d | j _ n | r�t! j" | | � n | S( sB Produce an engine configured by --options with optional overrides.i ( t create_async_engine( t create_engine( t make_urlR$ t use_reaperR( t sqlite_savepointt sqliteRM c S@ s d | _ d S( N( R1 t isolation_level( R6 t connection_record( ( sM /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyt do_connectA s t beginc S@ s | j d � d S( Nt BEGIN( t exec_driver_sql( RO ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyt do_beginE s ( t StaticPoolR# N(# R& t sqlalchemy.ext.asyncioR� R RL t _is_futureRU t sqlalchemy.futureR� t sqlalchemyt sqlalchemy.engine.urlR� R t popR@ R1 t drivernamet db_optst copyt updateRk R t listens_fort sqlalchemy.poolR� t isinstanceR t _transfer_fromRB t _has_eventst QueuePoolt _timeoutt _max_overflowR\ R) ( R@ R� t futuret asynciot transfer_staticpoolt _sqlite_savepointR� R� R� R( R� t default_optR' R� R� R� ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyR sV ! ' ' $ c @ s� d d l m } | s% t j j } n g � � f d � } � f d � } � f d � } | | d | � � t � d � s� t � � � _ | � _ | � _ � S( s' 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. i ( t create_mock_enginec @ s � j | � d S( N( Rv ( t sqlt aR^ ( t buffer( sM /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyt executort s c @ sJ g � D]! } t j d d t | � � ^ q } | | k sF t | � � d S( Ns [\n\t]R> ( t ret subt strR& ( t stmtst st recv( R� ( sM /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyt assert_sqlw s .c @ s) � j � d j � f d � � j D� � S( Ns c 3@ s' | ] } t | j d � � � Vq d S( Rh N( R� t compile( t .0R� ( Rj ( sM /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pys <genexpr>} s ( Rh t joint mock( ( R' ( Rj sM /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyt print_sql{ s s ://R� ( R� R� R RL Rk RD R& R� R� R� ( t dialect_nameR� R� R� R� ( ( R� R' sM /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyt mock_enginea s t DBAPIProxyCursorc B@ s> e Z d Z d � Z d d � Z d � Z d � Z d � Z RS( su Proxy a DBAPI cursor. Tests can provide subclasses of this to intercept DBAPI-level cursor operations. c O@ s+ | | _ | | _ | j | | � | _ d S( N( R' RA Rt ( R R' RO R] Rw ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyR � s c K@ s3 | r | j j | | | � S| j j | | � Sd S( N( Rt Ru ( R t stmtt parametersR^ ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyRu � s c K@ s | j j | | | � S( N( Rt t executemany( R R� t paramsR^ ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyR� � s c C@ s t | j � S( N( t iterRt ( R ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyt __iter__� s c C@ s t | j | � S( N( Rf Rt ( R Rr ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyRs � s N( RZ R[ t __doc__R R1 Ru R� R� Rs ( ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyR� � s t DBAPIProxyConnectionc B@ s2 e Z d Z d � Z d � Z d � Z d � Z RS( s} Proxy a DBAPI connection. Tests can provide subclasses of this to intercept DBAPI-level connection operations. c C@ s( | j j � | _ | | _ | | _ d S( N( R t _creatorRO R' t cursor_cls( R R' R� ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyR � s c O@ s | j | j | j | | � S( N( R� R' RO ( R R] Rw ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyRt � s c C@ s | j j � d S( N( RO R ( R ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyR � s c C@ s t | j | � S( N( Rf RO ( R Rr ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyRs � s ( RZ R[ R� R Rt R Rs ( ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyR� � s c @ s6 � � f d � } d � } t d i | d 6| d 6� S( sI Produce an engine that provides proxy hooks for common methods. c @ s � t j � � S( N( R RL ( ( t conn_clsR� ( sM /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyt mock_conn� s c @ s � f d � } | S( Nc @ s � | j � S( N( RO ( R ( t do_on_connect( sM /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyt go� s ( ( R� R� ( ( R� sM /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyt _wrap_do_on_connect� s R� t creatorR� ( R ( R� R� R� R� ( ( R� R� sM /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyt proxying_engine� s ( t __future__R R R� R+ R R> R t utilR R R R R t objectR R\ R_ R` Ra Rb R1 Rm Rn R� RU R R� R� R� R� ( ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyt <module> s: � , J &