Current File : //opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyo
�
��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� 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.warn( t ret compilet It St sett warningst warnR t patchR t py3k( 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� 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
(
t testutilt lazy_gcR t _refst _STRAY_CONNECTION_FAILURESR& t
gc_collectt clearR: R; ( t err( ( sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyR? � s
c C@ s d S( N( ( t at bR+ ( ( sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyt eq_regex� s c C@ s d S( s. Assert a == b, with repr messaging on failure.N( ( RH RI R+ ( ( sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyt eq_� s c C@ s d S( s. Assert a != b, with repr messaging on failure.N( ( RH RI R+ ( ( sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyt ne_� s c C@ s d S( s. Assert a <= b, with repr messaging on failure.N( ( RH RI 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( RH 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+ ( RN R ( RH R+ ( ( sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyt is_false� s c C@ s d S( s. Assert a is b, with repr messaging on failure.N( ( RH RI R+ ( ( sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyRN s c C@ s d S( s2 Assert a is not b, with repr messaging on failure.N( ( RH RI R+ ( ( sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyt is_not_ s c C@ s d S( s. Assert a in b, with repr messaging on failure.N( ( RH RI R+ ( ( sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyt in_ s c C@ s d S( s2 Assert a in not b, with repr messaging on failure.N( ( RH RI R+ ( ( sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyt not_in_ s c C@ s d S( s> Assert a.startswith(fragment), with repr messaging on failure.N( ( RH t fragmentR+ ( ( sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyt startswith_ s c C@ sX t j d d | � } t j d d | � } t j d d | � } t j d d | � } d S( Ns ^\s+?|\nt s {2,}t ( R5 t sub( RH RI R+ ( ( sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyt eq_ignore_whitespace s
c O@ s5 y | | | � t } Wn | k
r0 t } n Xd S( N( R RO ( 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@ sB y | | | � Wn* | k
r= } t j | � j d � GHn Xd S( Ns utf-8( R t text_typet encode( R[ R+ R\ 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]RW s %r != %r on dialect %r() R t DefaultDialectR' t getattrR R Rl t StrCompileDialectR R R R
t URLt get_dialectt listRO R t Queryt _compile_contextt statementt
use_labelst persistencet BulkUDR R< t objectt exec_t
mock_callsR6 t reprR= R` t decodeR_ R5 RY RK t construct_paramst tuplet positiontupt prefetch( t selft clauset resultRm t checkparamsRl t checkpositionalt check_prefetcht use_default_dialectt allow_dialect_selectRi Rg R Rk 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.pyRd > s t ComparesTablesc B@ s e Z e d � Z d � Z RS( c C@ s� x� t | j | j � D]� \ } } t | j | j � t | j | j � t | j | j � | rj d } n | j | | � t | j t j
� r� t | j j | j j � n t d � | j D� d � | j D� � | j
r q q Wx | j D] } q� Wd S( Ns) Type '%s' doesn't correspond to type '%s'c S@ s h | ] } | j j � q S( ( t columnt name( t .0t 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� ( R� R� ( ( sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pys <setcomp>� s ( t zipR� RK R� t primary_keyt nullablet assert_types_baseR t typet sqltypest Stringt lengtht foreign_keyst server_default( 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 d S( N( ( 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( Ru 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 t lenR� 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� Ru R� Rq 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( ( ( 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( t cls( sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyt <lambda>� RW 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_resultRq t AssertionErrorR RO ( 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 RO ( 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' RW 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 R\ 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 R\ 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 R\ 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 dbsR\ 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� R5 R: RW R R R R RA t
exclusionsR R R R R R R
R R� t engineR R
R R R R� R R! R# R$ R% RO R R R@ RD R? R' RJ RK RL RM RP RQ RN RR RS RT RV RZ R^ Rc R| Rd R� R� ( ( ( sP /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertions.pyt <module> s^ , * W)
Mini Shell