Current File : //opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyc
�
��4]c @@ s� d d l m 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
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
d l m Z d
d l m Z d
d l m Z d
d l m Z d � Z e j d � � Z d � Z d � Z d � Z d � Z e j e! e! e" d � � Z# d � Z$ d a% d � Z& d d � Z( d d � Z) d d � Z* d d � Z+ d d! � Z, d d" � Z- d d# � Z. d d$ � Z/ d d% � Z0 d d& � Z1 d d' � Z2 d d( � Z3 d) � Z4 d* � Z5 d+ e6 f d, � � YZ7 d- e6 f d. � � YZ8 d/ e6 f d0 � � YZ9 d S(1 i ( t absolute_importNi ( t assertsql( t config( t mock( t util( t db_spec( t faili ( t exc( t orm( t pool( t schema( t types( t default( t url( t compat( t decoratorc O@ s t t 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_warningst sa_exct SAWarning( t messagest kw( ( sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyt expect_warnings s
c o@ sU t | � } t | t j � r6 | t j � r6 d Vn t | | � �
d VWd 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( R t
isinstanceR t string_typesR t _currentR ( t dbR R t spec( ( sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyt expect_warnings_on0 s
"c @ s t � 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* t d t � � � | | | � SWd QXd S( Nt assert_( R t False( t fnt argsR ( R ( sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyt decorateI s ( R ( R R ( ( R sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyt
emits_warningA s c O@ s t t j | | � S( N( R R t SADeprecationWarning( R R ( ( sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyt expect_deprecatedQ s c @ s t � � f d � � } | 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 � �� | | | � SWd QXd S( NR ( R R ( R R R ( R R ( sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyR a s ( R ( R R R ( ( R R sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyt emits_warning_onU s c @ s t � 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* t d t � � � | | | � SWd QXd S( NR ( R# R ( R R R ( R ( sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyR y s ( R ( R R ( ( R sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyt uses_deprecatedi s c #@ s� � r8 g | D]" } t j | t j t j B� ^ q
� n | � t � � � t j � � � � � � f d � } t j d | � �
d VWd QX| r� | s� t j
r� � s� t d d j � f d � � D� � � � n d S( Nc @ s� t | � � r$ | } t | � } n | r7 | d } n d } | sT t | � � rd � | | | � S� sn d SxW � D]? } � r� | j | � s� � ru | | k ru � j | � Pqu qu W� | | | � d S( Ni ( R t strt Nonet
issubclasst matcht discard( t msgt argR t exceptiont filter_( t exc_clst filterst real_warnt regext seen( sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyt our_warn� s
s
warnings.warns Warnings were not seen: %ss , c 3@ s( | ] } d � r | j n | Vq d S( s %rN( t pattern( t .0t s( R2 ( sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pys <genexpr>� s (
t ret compilet It St sett warningst warnR t patchR t py3kt AssertionErrort join( R/ R R2 R t py2konlyR+ R4 ( ( R/ R0 R1 R2 R3 sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyR � s 2 c C@ s t � 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.pyt global_cleanup_assertions� s c C@ s� t j � t j r< t d 7a d t t j � GHt j � n t j r{ t t j � } t j j � d a t j d | � n' t d k r� t
s� t d � � d a n d S( Ni s2 Encountered a stray connection in test cleanup: %si s8 Stray connection refused to leave after gc.collect(): %si
s* Encountered more than 10 stray connections( t testutilt lazy_gcR t _refst _STRAY_CONNECTION_FAILURESR& t
gc_collectt clearR= R> R RA ( t err( ( sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyRD � s
c C@ s2 t j | | � s. t | p( d | | f � � d S( Ns %r !~ %r( R8 R) RA ( t at bR+ ( ( sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyt eq_regex� s c C@ s, | | k s( t | p" d | | f � � d S( s. Assert a == b, with repr messaging on failure.s %r != %rN( RA ( RM RN R+ ( ( sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyt eq_� s c C@ s, | | k s( t | p" d | | f � � d S( s. Assert a != b, with repr messaging on failure.s %r == %rN( RA ( RM RN R+ ( ( sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyt ne_� s c C@ s, | | k s( t | p" d | | f � � d S( s. Assert a <= b, with repr messaging on failure.s %r != %rN( RA ( RM RN R+ ( ( sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyt le_� s c C@ s t | t d | �d S( NR+ ( t is_t True( RM R+ ( ( sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyt is_true� s c C@ s t | t d | �d S( NR+ ( RS R ( RM R+ ( ( sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyt is_false� s c C@ s, | | k s( t | p" d | | f � � d S( s. Assert a is b, with repr messaging on failure.s %r is not %rN( RA ( RM RN R+ ( ( sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyRS s c C@ s, | | k s( t | p" d | | f � � d S( s2 Assert a is not b, with repr messaging on failure.s %r is %rN( RA ( RM RN R+ ( ( sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyt is_not_ s c C@ s, | | k s( t | p" d | | f � � d S( s. Assert a in b, with repr messaging on failure.s %r not in %rN( RA ( RM RN R+ ( ( sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyt in_ s c C@ s, | | k s( t | p" d | | f � � d S( s2 Assert a in not b, with repr messaging on failure.s %r is in %rN( RA ( RM RN R+ ( ( sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyt not_in_ s c C@ s/ | j | � s+ t | p% d | | f � � d S( s> Assert a.startswith(fragment), with repr messaging on failure.s %r does not start with %rN( t
startswithRA ( RM t fragmentR+ ( ( sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyt startswith_ s c C@ s� t j d d | � } t j d d | � } t j d d | � } t j d d | � } | | k s| t | pv d | | f � � d S( Ns ^\s+?|\nt s {2,}t s %r != %r( R8 t subRA ( RM RN R+ ( ( sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyt eq_ignore_whitespace s
c O@ sG y | | | � t } Wn | k
r0 t } n X| sC t d � � d S( Ns# Callable did not raise an exception( R RT RA ( t
except_clst callable_R R t success( ( sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyt
assert_raises' s
c O@ s� y# | | | � t s" t d � � Wna | k
r� } t j | t j | � t j � sl t d | | f � � t j | � j d � GHn Xd S( Ns# Callable did not raise an exceptions %r !~ %ss utf-8( R RA R8 t searchR t text_typet UNICODEt encode( Ra R+ Rb R t kwargst e( ( sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyt assert_raises_message2 s
'
t AssertsCompiledSQLc
B@ s, e Z d d d d d e e e d d � Z RS( c C@ s | r t j � } n� | r$ d } n� | d k rE t | d d � } n | d k r` t j j } nc | d k r{ t j � } nH | d k r� t j � } n- t | t j
� r� t j | � j
� � } n i } i }
| r� | | d <n | d k rt | � | d <n |
rt |
d <n t | t j � rJ| j � } t | j _ | j } nU t | t j j � r�t j j | d � �& } | j � | j d d d } Wd QXn |
r�|
| d
<n | j d | | � } t t | d i � � } t j r%| j d
� j d d � } d t j! | � | j d
� GHn d t j! | � j d
� | GHt" j# d d t j! | � � } t$ | | d | | | f � | d k r�t$ | j% | � | � n | d k r�| j% | � } t$ t& g | j' D] } | | ^ q�� | � n | d k rt$ | j( | � n d S( Nt __dialect__R t default_enhancedt schema_translate_mapt column_keyst
literal_bindst
_execute_stmti i t compile_kwargst dialectt paramss utf-8t asciit ignores
SQL String:
s [\n\t]R] s %r != %r on dialect %r() R t DefaultDialectR' t getattrR R Rt t StrCompileDialectR R R R
t URLt get_dialectt listRT R t Queryt _compile_contextt statementt
use_labelst persistencet BulkUDR R? t objectt exec_t
mock_callsR9 t reprR@ Rh t decodeRf R8 R_ RP t construct_paramst tuplet positiontupt prefetch( t selft clauset resultRu t checkparamsRt t checkpositionalt check_prefetcht use_default_dialectt allow_dialect_selectRq Ro R Rs t contextt stmt_mockt ct param_strt cct pt x( ( sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyt assert_compile? s`
0 N( t __name__t
__module__R' R R� ( ( ( sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyRl > s t ComparesTablesc B@ s e Z e d � Z d � Z RS( c C@ s� t | j � t | j � k s$ t � xMt | j | j � D]6\ } } t | j | j � | | j | j k su t � t | j | j � t | j | j � | r� d } t | j t | j � � s� t | | j | j f � � n | j
| | � t | j t j � r"t | j j
| j j
� n t d � | j D� d � | j D� � | j r: t | j t j � spt � q: q: Wt | j � t | j � k s�t � x0 | j D]% } | j j | j d k s�t � q�Wd S( Ns) Type '%s' doesn't correspond to type '%s'c S@ s h | ] } | j j � q S( ( t columnt name( R6 t f( ( sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pys <setcomp>� s c S@ s h | ] } | j j � q S( ( R� R� ( R6 R� ( ( sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pys <setcomp>� s ( t lenR� RA t zipRP R� t primary_keyt nullableR t typet assert_types_baset sqltypest Stringt lengtht foreign_keyst server_defaultR
t FetchedValuet columnsR' ( R� t tablet reflected_tablet strict_typesR� t reflected_cR+ ( ( sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyt assert_tables_equal� s. $" ! $c C@ s; | j j | j � s7 t d | j | j | j f � � d S( Ns7 On column %r, type '%s' doesn't correspond to type '%s'( R� t _compare_type_affinityRA R� ( R� t c1t c2( ( sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyR� � s ( R� R� R R� R� ( ( ( sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyR� � s !t AssertsExecutionResultsc B@ sw e Z d � Z d � Z d � Z d � Z d d � Z d � Z d � Z d � Z
d � Z e j
d � � Z d
� Z RS( c G@ s. t | � } t | � GH| j | | | � d S( N( R} R� t assert_list( R� R� t class_t objects( ( sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyt
assert_result� s c C@ sl | j t | � t | � k d d | j � x8 t d t | � � D]! } | j | | | | | � qC Wd S( Ns/ result list is not the same size as test list, s
for class i ( R R� R� t ranget
assert_row( R� R� R� t list_t i( ( sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyR� � s
c C@ s� | j | j | k d t | � � x� | j � D]� \ } } t | t � r� t | d t � r� | j t | | � | d | d � q� | j | d t | | � | d � q0 | j t | | � | k d | t | | � | f � q0 Wd S( Ns item class is not i i s' attribute %s value %s does not match %s(
R t __class__R� t itemsR R� R} R� Ry R� ( R� R� t rowobjt desct keyt value( ( sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyR� � s ''c @ s: d t f d � � Y� t j | � } � f d � | D� } xC t j � f d � | � D]& } t d t | � j � j f � qW Wt | � t | � k r� t d t | � t | � f � n t � � � � f d � } x\ | D]T } xK | D]& } | | | � r� | j | � Pq� q� Wt d � j t
| � f � q� Wt S( 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@ s e Z d � Z RS( c S@ s
t | � S( N( t id( 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 @ s h | ] } � | � � q S( ( ( R6 Rj ( R� ( sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pys <setcomp>� s c @ s
t | � � S( N( R ( t o( t cls( sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyt <lambda>� R] s# Unexpected type "%s", expected "%s"s+ Unexpected object count "%s", expected "%s"c @ s� x� | j � D]w \ } } t | t � rh y( � j t | | � | d | d � Wq� t k
rd t SXq
t | | � � | k r
t Sq
Wt S( Ni i ( R� R R� t assert_unordered_resultRy RA R RT ( t objR R� R� ( t NOVALUER� ( sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyt
_compare_item� s "
s2 Expected %s instance with attributes %s not found.( t dictR t IdentitySett itertools_filterfalseR R� R� R� R� t removeR� RT ( R� R� R� t expectedt foundt wrongR� t
expected_itemt
found_item( ( R� R� R� R� sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyR� � s.
c C@ s, | d k r d d l m } n t j | � S( Ni ( R ( R' R] R R t
assert_engine( R� R ( ( sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyt sql_execution_asserter s c G@ s2 | j | � � } | � } Wd QX| j | � | S( N( R� R ( R� R Rb t rulest asserterR� ( ( sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyt assert_sql_execution s
c C@ s� g } xv | D]n } t | t � r_ t j g | j � D] \ } } t j | | � ^ q5 � } n t j | � } | j | � q
W| j | | | � S( N( R R� R t AllOfR� t CompiledSQLt appendR� ( R� R Rb R� t newrulest rulet kt vt newrule( ( sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyt
assert_sql# s
7c C@ s | j | | t j | � � d S( N( R� R t CountStatements( R� R Rb t count( ( sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyt assert_sql_count1 s c
C@ s� g t | | � D]$ \ } } | j | � | | f ^ q } g } x* | D]" \ } } } | j | j � � qG Wz | � SWd xO t | | � D]> \ } \ } } } | j d d d � | j t j | � � q� WXd S( N( R� R� R� t __enter__t __exit__R' R R R� (
R� t dbsRb t countsR R� t recst asserterst ctxR� ( ( sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyt assert_multiple_sql_count6 s 7 %c g@ s. | j | � � } d VWd QX| j | � d S( N( R� R ( R� R R� R� ( ( sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyt assert_executionE s c C@ s | j | t j | � � S( N( R� R R� ( R� R R� ( ( sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyt assert_statement_countK s N( R� R� R� R� R� R� R' R� R� R� R� R� t
contextlibt contextmanagerR� R� ( ( ( sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyR� � s
8 (: t
__future__R R� R8 R= R] R R R R RF t
exclusionsR R R R R R R
R R� t engineR R
R R R R� R R! R# R$ R% RT R R RE RI RD R' RO RP RQ RR RU RV RS RW RX RY R\ R` Rd Rk R� Rl R� R� ( ( ( sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyt <module> s^ , * W)
Mini Shell