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

ddlZddlZddlZddlmZ
ddlmZ
ddlmZ
ddlm	Z

ddlmZ
dd	l	m
Z

d
dlmZ
d
dlmZ
d
d
lmZ
d
dlmZ
d
dlmZ
d
dlm	Z	
d
dlmZ
d
dlmZ
d
dl	mZ
d
dl	mZ
d�Zejd��
Zd�Zd�Zd�Zd�Z eje!e!e"d��
Z#d�Z$da%d�Z&dd�
Z(dd�
Z)dd�
Z*dd �
Z+dd!�
Z,dd"�
Z-dd#�
Z.dd$�
Z/dd%�
Z0dd&�
Z1dd'�
Z2dd(�
Z3d)�Z4d*�Z5d+e6f
d,��YZ7d-e6f
d.��YZ8d/e6f
d0��YZ9dS(1i(
tabsolute_importNi
(
t	assertsql(
tconfig(
tmock(
tutil(
tdb_spec(
tfaili(
texc(
torm(
tpool(
tschema(
ttypes(
tdefault(
turl(
tcompat(
t	decoratorcO@stt
j||
�S(
s�
Context manager which expects one or more warnings.

    With no arguments, squelches all SAWarnings emitted via
    sqlalchemy.util.warn and sqlalchemy.util.warn_limited.   Otherwise
    pass string expressions that will match selected warnings via regex;
    all non-matching warnings are sent through.

    The expect version **asserts** that the warnings were in fact seen.

    Note that the test suite sets SAWarning warnings to raise exceptions.

    (t_expect_warningstsa_exct	SAWarning(tmessagestkw((sP/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pytexpect_warnings s
c
o@sUt|�
}t
|tj�r6|tj�
r6d
V
nt|
|��

d
V
Wd
QXd
S(s�Context manager which expects one or more warnings on specific
    dialects.

    The expect version **asserts** that the warnings were in fact seen.

    N(Rt
isinstanceRtstring_typesRt_currentR(tdbRRtspec((sP/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pytexpect_warnings_on0s
"

c@st�f
d
��
}
|
S(s�Decorator form of expect_warnings().

    Note that emits_warning does **not** assert that the warnings
    were in fact seen.

    c
@s*td
t
��
�
||
|�SWdQXdS(Ntassert_(RtFalse(tfntargsR(
R(sP/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pytdecorateIs
(
R(RR ((
RsP/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyt
emits_warningAscO@stt
j||
�S(
N(RRtSADeprecationWarning(RR((sP/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pytexpect_deprecatedQs
c
@st��
fd
��
}|S(sV
Mark a test as emitting a warning on a specific dialect.

    With no arguments, squelches all SAWarning failures.  Or pass one or more
    strings; these will be matched to the root of the warning description by
    warnings.filterwarnings().

    Note that emits_warning_on does **not** assert that the warnings
    were in fact seen.

    c
@s-t�d
t
�
�

�
||
|�SWdQXdS(NR(RR(RRR(RR(sP/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyR as
(
R(RRR ((RRsP/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pytemits_warning_onUsc@st�f
d
��
}
|
S(s+Mark a test as immune from fatal deprecation warnings.

    With no arguments, squelches all SADeprecationWarning failures.
    Or pass one or more strings; these will be matched to the root
    of the warning description by warnings.filterwarnings().

    As a special case, you may pass a function name prefixed with //
    and it will be re-written as needed to match the standard warning
    verbiage emitted by the sqlalchemy.util.deprecated decorator.

    Note that uses_deprecated does **not** assert that the warnings
    were in fact seen.

    c
@s*td
t
��
�
||
|�SWdQXdS(NR(R#R(RRR(
R(sP/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyR ys
(
R(RR ((
RsP/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pytuses_deprecatedisc#@s��r8g|
D]"}tj
|tjtjB�^q
�
n|
�
t�
�
�tj���
���fd
�}tjd|��

dV
WdQX|r�|s�t	j
r�ndS(Nc
@s�t|��r$|}t
|�
}n|
r7|
d
}nd}|sTt|��rd�||
|�
S�
sndSxW�
D]?}�r�|j|�
s��ru||kru�j|�

PququW�||
|�

dS(Ni(RtstrtNonet
issubclasstmatchtdiscard(tmsgtargRt	exceptiontfilter_(texc_clstfilterst	real_warntregextseen(sP/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pytour_warn�s 













s
warnings.warn(tretcompilet
It
StsettwarningstwarnRtpatchRtpy3k(R/RR2Rtpy2konlyR+R4((R/R0R1R2R3sP/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyR�s
2	

c
C@st�
d
S(s�Check things that have to be finalized at the end of a test suite.

    Hardcoded at the moment, a modular system can be built here
    to support things like PG prepared transactions, tables all
    dropped, etc.

    N(
t!_assert_no_stray_pool_connections(((sP/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pytglobal_cleanup_assertions�sc
C@s�tj
�
tjr<td
7adttj�
GHtj�
ntjr{ttj�
}tjj�
datj	d|�

ntdkr�dandS(Ni
s2Encountered a stray connection in test cleanup: %sis8Stray connection refused to leave after gc.collect(): %si
(
ttestutiltlazy_gcR	t_refst_STRAY_CONNECTION_FAILURESR&t
gc_collecttclearR:R;(
terr((sP/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyR?�s
	


	




c
C@sdS(
N((t
at
bR+((sP/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyteq_regex�s
c
C@sd
S(s.Assert a == b, with repr messaging on failure.N((RHRIR+((sP/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyteq_�sc
C@sd
S(s.Assert a != b, with repr messaging on failure.N((RHRIR+((sP/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pytne_�sc
C@sd
S(s.Assert a <= b, with repr messaging on failure.N((RHRIR+((sP/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pytle_�scC@st|t
d
|
�

dS(NR+(tis_tTrue(RHR+((sP/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pytis_true�s
cC@st|t
d
|
�

dS(NR+(RNR(RHR+((sP/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pytis_false�s
c
C@sd
S(s.Assert a is b, with repr messaging on failure.N((RHRIR+((sP/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyRN
sc
C@sd
S(s2Assert a is not b, with repr messaging on failure.N((RHRIR+((sP/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pytis_not_
sc
C@sd
S(s.Assert a in b, with repr messaging on failure.N((RHRIR+((sP/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pytin_
sc
C@sd
S(s2Assert a in not b, with repr messaging on failure.N((RHRIR+((sP/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pytnot_in_
sc
C@sd
S(s>Assert a.startswith(fragment), with repr messaging on failure.N((RHtfragmentR+((sP/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pytstartswith_
scC@sXtj
d
d|�}tj
dd|�}tj
d
d|
�}
tj
dd|
�}
dS(Ns^\s+?|\nts {2,}t
 (R5tsub(RHRIR+((sP/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyteq_ignore_whitespace
s




cO@s5y|
||�
t}Wn|k
r0


t
}n
XdS(
N(RRO(t
except_clst	callable_RRtsuccess((sP/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyt
assert_raises'
s








cO@sBy|||�
Wn*|k
r=
}
tj
|�
jd
�
GHn
XdS(Nsutf-8(Rt	text_typetencode(R[R+R\Rtkwargst
e((sP/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pytassert_raises_message2
s





tAssertsCompiledSQLc
B@s,eZ
d
d
d
d
d
eeed
d�	ZRS(cC@s|rtj
�}n�|	r$d}n�|dkrEt|d
d�}n|dkr`tjj}nc|dkr{tj
�}nH|dkr�tj�}n-t|t	j
�r�tj|�
j
��}ni}i}
|r�||d<n|dk	r

t|�
|d<n|
r
t|
d<nt|
tj�rJ
|
j�}t|j_|j}
nUt|
tjj�r�
tjj|
d��&}|
j�
|jdd	d}
WdQXn|
r�
|
|d
<n|
jd||�
}tt|di��
}t	jr%|jd
�
j dd�}dt	j!|�
|jd
�
GHndt	j!|�
jd
�
|GHt"j#ddt	j!|�
�}t$||d|||f�
|dk	r�t$|j%|�
|�
n|dk	r�|j%|�
}t$t&g|j'D]}||^q��
|�
n|dk	rt$|j(|�
ndS(Nt__dialect__Rtdefault_enhancedtschema_translate_maptcolumn_keyst
literal_bindst
_execute_stmtii
tcompile_kwargstdialecttparamssutf-8tasciitignores

SQL String:
s[\n\t]RWs%r != %r on dialect %r()RtDefaultDialectR'tgetattrRRRltStrCompileDialectRRRR
tURLtget_dialecttlistRORtQueryt_compile_contextt	statementt
use_labelstpersistencetBulkUDRR<tobjecttexec_t
mock_callsR6treprR=R`tdecodeR_R5RYRKtconstruct_paramsttupletpositiontuptprefetch(tselftclausetresultRmtcheckparamsRltcheckpositionaltcheck_prefetchtuse_default_dialecttallow_dialect_selectRiRgRRktcontextt	stmt_mockt
ct	param_strtcct
pt
x((sP/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pytassert_compile?
s`


	























	





0

N(t__name__t
__module__R'RR�(((sP/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyRd>
s







tComparesTablescB@seZ
ed�
Zd
�ZRS(cC@s�x�t|
j
|j
�D]�\}}t|j|j�
t|j|j�
t|j|j�
|rjd
}n|j||�
t|jt	j
�r�t|jj|jj�
ntd�|jD�
d�|jD�
�
|j
rqqWx|
jD]}q�WdS(Ns)Type '%s' doesn't correspond to type '%s'c
S@sh|]}
|
jj
�qS((tcolumntname(t.0t
f((sP/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pys	<setcomp>�
s	c
S@sh|]}
|
jj
�qS((R�R�(R�R�((sP/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pys	<setcomp>�
s	(tzipR�RKR�tprimary_keytnullabletassert_types_baseRttypetsqltypestStringtlengthtforeign_keystserver_default(R�ttabletreflected_tabletstrict_typesR�treflected_cR+((sP/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pytassert_tables_equal�
s""






	

c
C@sdS(
N((R�tc1tc2((sP/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyR��
s
(R�R�RR�R�(((sP/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyR��
s
!tAssertsExecutionResultscB@sweZ
d�Zd
�Zd�Zd�Zdd�
Zd�Zd�Z	d�Z
d�Zej
d	��
Zd
�ZRS(cG@s.t|
�
}
t
|
�
GH|j|
||�
dS(
N(RuRtassert_list(R�R�tclass_tobjects((sP/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyt
assert_result�
s


cC@sl|jt
|
�
t
|�
kd
d|j�
x8tdt
|�
�D]!}|j||
|||�
qCWdS(Ns/result list is not the same size as test list, s
for class i(RtlenR�tranget
assert_row(R�R�R�tlist_t
i((sP/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyR��
s



cC@s�|j|j
|
kd
t|
�
�
x�|j�D]�\}}t|t�r�t|dt�r�|jt||�|d|d�
q�|j	|dt||�|d�
q0|jt||�|kd|t||�|f�
q0WdS(Nsitem class is not i
is'attribute %s value %s does not match %s(
Rt	__class__RtitemsRR�RuR�RqR�(R�R�trowobjtdesctkeytvalue((sP/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyR��
s




''


c	@s:
d
tf
d��Y�t
j|
�
}�f
d�|D�
}xCt
j�
f
d�|�D]&}tdt|�
j�
jf�

qWWt|�
t|�
kr�tdt|�
t|�
f�

nt����fd�}x\|D]T}xK|D]&}|||�r�|j	|�

Pq�q�Wtd�
jt
|�
f�

q�WtS(	s�As assert_result, but the order of objects is not considered.

        The algorithm is very expensive but not a big deal for the small
        numbers of rows that the test suite manipulates.
        t
immutabledictc
B@seZ
d�ZRS(
c

S@s
t|�
S(
N(
tid(
R�((sP/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyt__hash__�
s
(R�R�R�(((sP/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyR��
s
c
@sh|]}
�|
�
�qS(((R�Rb(
R�(sP/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pys	<setcomp>�
s	c

@s
t|��S(
N(
R(
t
o(
tcls(sP/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyt<lambda>�
RWs#Unexpected type "%s", expected "%s"s+Unexpected object count "%s", expected "%s"c@s�x�|
j�D]w\}}t
|t�rhy(�
jt||�|d
|d�
Wq�tk
rd


tSXq
t||��|kr
tSq
WtS(Nii
(R�RR�tassert_unordered_resultRqtAssertionErrorRRO(tobjRR�R�(tNOVALUER�(sP/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyt
_compare_item�
s




"



s2Expected %s instance with attributes %s not found.(tdictRtIdentitySettitertools_filterfalseRR�R�R�R|tremoveRRO(	R�R�R�texpectedtfoundtwrongR�t
expected_itemt
found_item((R�R�R�R�sP/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyR��
s.
	





	








cC@s,|
dkrd
dl
m}

ntj|
�
S(Ni
(
R(R'RWRR
t
assert_engine(R�R((sP/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pytsql_execution_asserters

cG@s2|j|
�
�}|�}WdQX|j
|�
|S(
N(R�R(R�RR\trulestasserterR�((sP/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pytassert_sql_executions




c	C@s�g}xv|D]n}t|t
�r_tjg|j�D]\}}tj||�^q5�}ntj|�}|j|�

q
W|j|
||�S(
N(RR�R
tAllOfR�tCompiledSQLtappendR�(	R�RR\R�tnewrulestrulet
kt
vtnewrule((sP/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyt
assert_sql#s




7
cC@s |j|
|t
j|�
�
dS(
N(R�R
tCountStatements(R�RR\tcount((sP/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pytassert_sql_count1s

c
C@s�gt|
|�D]$\}}|j
|�
||f^q}g}x*|D]"\}}}|j|j��

qGWz|�SWdxOt||�D]>\}	\}}}|jddd�
|	jtj|�
�

q�WXdS(
N(	R�R�R�t	__enter__t__exit__R'RR
R�(
R�tdbsR\tcountsRR�trecst	asserterstctxR�((sP/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pytassert_multiple_sql_count6s
7



%

cg@s.|j|
�
�}dV
WdQX|j
|�
dS(
N(R�R(R�RR�R�((sP/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pytassert_executionEs

cC@s|j|
t
j|�
�S(
N(R�R
R�(R�RR�((sP/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pytassert_statement_countKs
N(R�R�R�R�R�R�R'R�R�R�R�R�t
contextlibtcontextmanagerR�R�(((sP/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyR��
s
		
		8				(:t
__future__RR�R5R:RWR
RRRRAt
exclusionsRRRRRR	R
RR�tengineRR
RRRR�RR!R#R$R%RORRR@RDR?R'RJRKRLRMRPRQRNRRRSRTRVRZR^RcR|RdR�R�(((sP/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyt<module>s^
















					,		*			W)