�
��4]c@@ss
dd
lm
Z

ddlZddlZddlZddlmZ
ddlmZ
ddlm	Z	
ddlm
Z

dd	lmZ
d
ef
d��YZ
e
�Zd�Ze	d
��
Ze	d��
Ze	d��
Ze	d��
Zdd�
Zdef
d��YZddd�Zddd�Zdd�
Zdef
d��YZdef
d��YZeed�ZdS(i(
tabsolute_importNi
(
tconfig(
tuses_deprecated(
t	decoratori(
tevent(
tpooltConnectionKillercB@s�eZ
d�Zd
�Zd�Zd�Zd�Zd�Zd�Zd�Z	d�Z
d	�Ze�d
��
Z
e�d��
Zd�ZRS(
c

C@s.tj
�|_tj
�|_t�|_dS(
N(tweakreftWeakKeyDictionaryt
proxy_refsttesting_enginestsettconns(
tself((sM/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyt__init__s


cC@st|j
|
<dS(
N(tTrueR
(R
tengine((sM/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyt
add_engines
cC@s|jj
|
|f�

dS(
N(Rtadd(R
t
dbapi_connt
con_record((sM/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pytconnects
cC@st|j
|<dS(
N(RR	(R
t	dbapi_conRt	con_proxy((sM/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pytcheckout!s
cC@s|jj
|
|f�

dS(
N(Rtdiscard(R
RRt	exception((sM/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyt
invalidate$s
cC@s6y|
�
Wn$tk
r1
}
t
jd
|�

n
XdS(Ns5testing_reaper couldn't rollback/close connection: %s(t	Exceptiontwarningstwarn(R
tfnt
e((sM/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyt_safe's





c
C@sFx?t|j
�
D].}
|
dk	r|
jr|j|
j�

qqWdS(
N(tlistR	tNonetis_validR!trollback(R
trec((sM/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pytrollback_all/s


c
C@sFx?t|j
�
D].}
|
dk	r|
jr|j|
j�

qqWdS(
N(R"R	R#R$R!t_close(R
R&((sM/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyt	close_all4s


c
C@sIxBt|j
�
D]1\}
}|jdkr1qn|j|
j�

qWdS(
N(R"Rt
connectionR#R!R%(R
tconnR&((sM/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyt_after_test_ctx9s

c


C@s'tj
jr|j�
n
|j�
dS(
N(R
toptionstlow_connectionst_stop_test_ctx_minimalt_stop_test_ctx_aggressive(
R
((sM/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyt_stop_test_ctxHs


c
C@sP|j�
t
�|_x3t|j�
D]"}
|
tjk	r&|
j�
q&q&WdS(
N(R)RRR"R
R
tdbtdispose(R
R&((sM/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyR/Ns



c
C@sw|j�
x6t
|j�
D]%\}
}|j|
j�

d|_qWt�|_x!t
|j�
D]}|j	�
q_WdS(
N(
R)R"RR!tcloseR#R*RR
R3(R
R+R&((sM/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyR0Xs






c
C@s0x)|jD]}
|
j
r
ts(t�
q
q
WdS(
N(R	R$tFalsetAssertionError(R
R&((sM/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pytassert_all_closedcs

	
(t__name__t
__module__RRRRRR!R'R)R,R1RR/R0R7(((sM/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyRs
										
cC@s�tj
�
t|
d
�r&|
j�
ntjjjsqddlm	}
|j
ddt�

�
|j|
�

WdQXn
|j|
�

dS(NR4i
(
t
assertionssCan't sort tablestassert_(
ttesting_reaperR)thasattrR4R
R2tdialecttsupports_altertR:texpect_warningsR5tdrop_all(tmetadatatbindR:((sM/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pytdrop_all_tablesls






c
O@s#z||
|�
Wdtj
�
XdS(
N(R<R7(Rtargstkw((sM/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pytassert_conns_closedzs
c
O@s#z||
|�
Wd
tj
�
Xd
S(sBDecorator that rolls back all open connections after fn execution.N(R<R'(RRFRG((sM/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pytrollback_open_connections�s
c
O@stj
�
||
|�
d
S(s:Decorator that closes all connections before fn execution.N(R<R)(RRFRG((sM/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pytclose_first�s

c
O@s#z||
|�
Wd
tj
�
Xd
S(s9Decorator that closes all connections after fn execution.N(R<R)(RRFRG((sM/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pytclose_open_connections�s
c
c@s�d
dlj
}
xk|
jD]`}|r7||kr7qnt|
|d�}|snttd|�
j
|�}n|j�V
qWdS(Nissqlalchemy.databases.%s(tsqlalchemy.databasest	databasest__all__tgetattrR#t
__import__R>(texcludet
dtnametmod((sM/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pytall_dialects�s





tReconnectFixturecB@sAeZ
d�Zd
�Zd�Zd�Zed�
Zd�ZRS(cC@s|
|_g|_
t|_dS(
N(tdbapitconnectionsR5t
is_stopped(R
RW((sM/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyR�s
	
	
cC@st|j
|
�S(
N(RORW(R
tkey((sM/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyt__getattr__�s
c
O@st|jj
|
|�}|jr\|j|j�

|j�}|jd
�

tsptd�
�
n|j	j
|�

|SdS(Nsselect 1s%simulated connect failure didn't work(RWRRYR!R4tcursortexecuteR5R6RXtappend(R
RFtkwargsR+tcurs((sM/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyR�s
	




cC@s6y|
�
Wn$tk
r1
}
t
jd
|�

n
XdS(Ns.ReconnectFixture couldn't close connection: %s(RRR(R
RR ((sM/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyR!�s





cC@s@|
|_x't
|j�
D]}|j|j�

qWg|_dS(
N(RYR"RXR!R4(R
tstopt
c((sM/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pytshutdown�s	


c

C@s
t|_
dS(
N(R5RY(
R
((sM/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pytrestart�s
(	R8R9RR[RR!R5RcRd(((sM/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyRV�s
				c@s�|ptj
j}tj
jj}|
s0i}
nt|�
|
d
<t||
��
�
j���
fd�}�
jjj�
_	�
jjj
�
_|�
_�
S(Ntmodulec@s*�
jj
j�
t�
jj
_��
dS(
N(R>RWRcR5RY((t_disposeR(sM/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyR3�s


(R
R2turlR>RWRVttesting_engineR3Rct
test_shutdownRdttest_restart(RgR-RWR3((RfRsM/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pytreconnecting_engine�s



	


	

	
cC@s�
d
dlm
}
d
dlm}
|
s/t}n|
jdt�}|pPtjj}||�
}|
dkr�tjdks�|j
tjjj
kr�tj}
q�i}
nFtjdk	r�|j
tjjj
kr�tjj�}|j
|
�

n|||
�
}t|_t|jtj�r5
d
|j_d
|j_n|r�
tj|jdtj�
tj|jdtj�
tj|jdtj�
tj|�

n|S(	sBProduce an engine configured by --options with optional overrides.i(
t
create_engine(
tmake_urlt
use_reaperRRRN(t
sqlalchemyRltsqlalchemy.engine.urlRmRtpopR
R2RgR#t
drivernametdb_optstcopytupdatet_has_eventst
isinstanceRt	QueuePoolt_timeoutt
_max_overflowRtlistenR<RRRR(RgR-RlRmRntdefault_optR((sM/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyRh�s2

	

'
	
'


	






c
@s�d
dlm
}

|s%tjj}ng��f
d�}�f
d�}�
f
d�}|
|dddd	|�
�
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(
Rlc
@s�j|�

dS(
N(
R^(tsqlt
aRG(
tbuffer(sM/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pytexecutor'
s
c
@sJg�D]!}
tj
d
dt|
�
�^q}||ksFt|�
�
dS(Ns[\n\t]R@(tretsubtstrR6(tstmtst
strecv(
R(sM/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyt
assert_sql*
s
.
c@s)�
j�d
j
�f
d��
jD�
�
S(Ns

c
3@s'|]}
t|
j
d��
�
V
qd
S(R>N(R�tcompile(t.0R�(
RR(sM/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pys	<genexpr>0
s(R>tjointmock((
R(
RRsM/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyt	print_sql.
s
	
s://tstrategyR�R�(
RoRlR
R2RSR=R6R�R�R�(tdialect_nameRlR�R�R�((RRsM/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pytmock_engine
s


	
	
	
tDBAPIProxyCursorcB@s5eZ
dZd
�Zdd�
Zd�Zd�ZRS(suProxy a DBAPI cursor.

    Tests can provide subclasses of this to intercept
    DBAPI-level cursor operations.

    cO@s+|
|_||_
|j||�|_dS(
N(RR*R\(R
RR+RFR_((sM/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyRD
s
	
	
cK@s3|r|jj
|
||�S|jj
|
|�
SdS(
N(R\R](R
tstmtt
parametersRG((sM/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyR]I
s

cK@s|jj
|
||�S(
N(R\texecutemany(R
R�tparamsRG((sM/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyR�O
s
cC@st|j
|
�S(
N(ROR\(R
RZ((sM/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyR[R
s
N(R8R9t__doc__RR#R]R�R[(((sM/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyR�<
s
		tDBAPIProxyConnectionc
B@s2eZ
dZd
�Zd�Zd�Zd�ZRS(s}Proxy a DBAPI connection.

    Tests can provide subclasses of this to intercept
    DBAPI-level connection operations.

    cC@s/|
jj
�|_|_|
|_||_dS(
N(Rt_creatorR+t_sqla_unwrapRt
cursor_cls(R
RR�((sM/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyR^
s

	
c
O@s|j|j
|j|
|�S(
N(R�RR+(R
RFR_((sM/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyR\c
s
c


C@s|jj
�
dS(
N(R+R4(
R
((sM/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyR4f
s
cC@st|j
|
�S(
N(ROR+(R
RZ((sM/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyR[i
s
(R8R9R�RR\R4R[(((sM/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyR�V
s
			c@s&��
fd
�}tdi
|d6�
S(sIProduce an engine that provides proxy hooks for
    common methods.

    c@s�tj
�
�S(
N(R
R2((tconn_clsR�(sM/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyt	mock_connu
s
R-tcreator(
Rh(R�R�R�((R�R�sM/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pytproxying_enginem
s(t
__future__RR�RRR@R
RtutilRRRtobjectRR<RERHRIRJRKR#RURVRkRhR�R�R�R�(((sM/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/engines.pyt<module>s.





T		
	.&(