Current File : //proc/227033/root/opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyc
�
��4]c @ s� 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 d e f d � � YZ d
e f d � � YZ d e f d � � YZ d e f d � � YZ d e f d � � YZ d e f d � � YZ d e f d � � YZ d e f d � � YZ d e f d � � YZ d e f d � � YZ d e j d d d! d" d# g � f d$ � � YZ d% e f d& � � YZ e j d' � � Z d S(( i����Ni ( t event( t util( t url( t DefaultDialect( t _distill_params( t _DDLCompilest
AssertRulec B s, e Z e Z d Z e Z d � Z d � Z RS( c C s d S( N( ( t selft execute_observed( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyt process_statement s c C s t s t d � � d S( Ns? All statements are complete, but pending assertion rules remain( t Falset AssertionError( R ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyt no_more_statements s N(
t __name__t
__module__R
t is_consumedt Nonet errormessaget Truet consume_statementR R ( ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyR s
t SQLMatchRulec B s e Z RS( ( R
R ( ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyR $ s t CursorSQLc B s# e Z e Z d d � Z d � Z RS( c C s | | _ | | _ d S( N( t statementt params( R R R ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyt __init__+ s c C s� | j d } | j | j k s@ | j d k rh | j | j k rh d | j | j | j | j f | _ n. | j j d � t | _ | j s� t | _ n d S( Ni s5 Testing for exact SQL %s parameters %s received %s %s(
t
statementsR R R t
parametersR t popR R R ( R R t stmt( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyR / s
! N( R
R R
R R R R ( ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyR ( s t CompiledSQLc B sM e Z d d d � Z d � Z d � Z d � Z d � Z d � Z d � Z RS( t defaultc C s | | _ | | _ | | _ d S( N( R R t dialect( R R R R ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyR E s c C s" t j d d | j � } | | k S( Ns [\n\t]t ( t ret subR ( R R t received_statementR ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyt _compare_sqlJ s c C s[ | j d k r t � S| j d k r5 i t d 6} n i } t j | j � j � | � Sd S( NR t
postgresqlt implicit_returning( R R R R t URLt get_dialect( R R R ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyt _compile_dialectN s c
C s | j } | j | � } t | j j t � rZ | j j j d | d | j j d � � } nB | j j j d | d | j j d | j j
d | j j d � � } t j d d t
j | � � } | j } | s� | j � g } n" g | D] } | j | � ^ q� } | | f S( sx reconstruct the statement and params in terms
of a target dialect, which for CompiledSQL is just DefaultDialect.R t schema_translate_mapt column_keyst inlines [\n\t]R ( t contextR) t
isinstancet compiledR R t compilet execution_optionst getR+ R, R! R" R t text_typeR t construct_params( R R R- t compare_dialectR/ t _received_statementR t _received_parameterst m( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyR6 Y s& "c
C sV | j } | j | � \ } } | j | � } | j | | � } | r| d k rt | � } t | � } x� | r� | r� t | j d � � } xd t t | � � D]I \ }
} x: | D]* } | | k s� | | | | k r� Pq� q� W| |
=Pq� Wt } Pql W| s| rt } qqn | r.t
| _ d | _ n$ | j
| � i | d 6| d 6| _ d S( Ni R# t received_parameters( R- R6 t _all_paramsR$ R t listt dictR t enumerateR
R R R t _failure_message(
R R R- R6 R7 R t
equivalentt
all_paramst all_receivedt paramt idxt receivedt param_key( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyR { s8
c C s] | j rU t j | j � r- | j | � } n | j } t | t � sQ | g } n | Sd Sd S( N( R R t callableR. R; R ( R R- R ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyR: � s c C s d | j j d d � | f S( Nsz Testing for compiled statement %r partial params %r, received %%(received_statement)r with params %%(received_parameters)rt %s %%( R t replace( R t expected_params( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyR> � s N(
R
R R R R$ R) R6 R R: R> ( ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyR D s " 1 t RegexSQLc B s& e Z d d � Z d � Z d � Z RS( c C s> t j | � t j | � | _ | | _ | | _ d | _ d S( NR ( R R R! R0 t regext
orig_regexR R ( R RK R ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyR � s
c C s d | j | f S( Ns{ Testing for compiled statement ~%r partial params %r, received %%(received_statement)r with params %%(received_parameters)r( RL ( R RI ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyR> � s c C s t | j j | � � S( N( t boolRK t match( R R R# ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyR$ � s N( R
R R R R> R$ ( ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyRJ � s t
DialectSQLc B s, e Z d � Z d � Z d � Z d � Z RS( c C s
| j j S( N( R- R ( R R ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyR) � s c C s t j d d | � } | | k S( Ns [\n\t]R ( R! R" ( R t real_stmtt
received_stmtR ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyt _compare_no_space� s c C sk t t | � j | � \ } } x: | j D] } | j | j | � r( Pq( q( Wt d | � � | | j j f S( NsI Can't locate compiled statement %r in list of statements actually invoked( t superRO R6 R RR R R R- t compiled_parameters( R R RQ t received_paramsRP ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyR6 � s
c C s� t j d d | j � } | j j j } | d k rK t j d d | � } nZ d } | d k rf d } n* | d k r{ d } n | d
k r� d } n t j d | | � } | | k S( Ns [\n\t]R t pyformats :([\w_]+)s %(\1)st qmarkt ?t formats %st numeric( R! R" R R- R t
paramstyleR ( R R R# R R[ t repl( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyR$ � s ( R
R R) RR R6 R$ ( ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyRO � s t CountStatementsc B s# e Z d � Z d � Z d � Z RS( c C s | | _ d | _ d S( Ni ( t countt _statement_count( R R^ ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyR s c C s | j d 7_ d S( Ni ( R_ ( R R ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyR s c C s; | j | j k r7 t s7 t d | j | j f � � n d S( Ns, desired statement count %d does not match %d( R^ R_ R
R ( R ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyR
s ( R
R R R R ( ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyR] s t AllOfc B s e Z d � Z d � Z RS( c G s t | � | _ d S( N( t sett rules( R Rb ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyR s c C s� x� t | j � D]d } d | _ | j | � | j r^ | j j | � | j sZ t | _ n Pq | j s d | _ Pq q Wt | j � d j | _ d S( Ni ( R; Rb R R R R t discardR ( R R t rule( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyR s
( R
R R R ( ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyR` s t EachOfc B s# e Z d � Z d � Z d � Z RS( c G s t | � | _ d S( N( R; Rb ( R Rb ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyR ( s c C s� xh | j rj | j d } | j | � | j rB | j j d � n | j rZ | j | _ n | j r Pq q W| j s� t | _ n d S( Ni ( Rb R R R R R R ( R R Rd ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyR + s
c C sQ | j r. | j d j r. | j d j � n | j rM t t | � j � n d S( Ni ( Rb R R RS Re ( R ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyR 9 s ( R
R R R R ( ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyRe ' s t Orc B s e Z d � Z RS( c C sT xM | j D]) } | j | � | j r
t | _ Pq
q
Wt | j � d j | _ d S( Ni ( Rb R R R R; R ( R R Rd ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyR A s
( R
R R ( ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyRf @ s t SQLExecuteObservedc B s e Z d � Z RS( c C s1 | | _ | | _ t | | � | _ g | _ d S( N( R- t
clauseelementR R R ( R R- Rh t multiparamsR ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyR L s ( R
R R ( ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyRg K s t SQLCursorExecuteObservedR R R- t executemanyc B s e Z RS( ( R
R ( ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyRj S s t SQLAsserterc B s# e Z d � Z d � Z d � Z RS( c C s
g | _ d S( N( t accumulated( R ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyR ] s c C s | j | _ | ` d S( N( Rm t _final( R ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyt _close` s c G s� t | � } t | j � } xT | rq | j d � } | j | � | j rM Pq | j r t sn t | j � � q q W| r� t s� t d � � n | j s� | j � n d S( Ni s Additional SQL statements remain(
Re R; Rn R R R R R
R R ( R Rb Rd t observedR ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyt assert_d s
( R
R R Ro Rq ( ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyRl \ s c # s� t � � g � t j | d � � f d � � } t j | d � � � f d � � } z � VWd t j | d | � t j | d | � � j � Xd S( Nt before_executec s | | | f � (d S( N( ( t connRh Ri R ( t orig( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyt connection_execute{ s t after_cursor_executec s� | s
d S� j r9 � j d j | k r9 � j d } n1 t | � d � d � d � } � j j | � | j j t | | | | � � d S( Ni����i i i ( Rm R- Rg t appendR Rj ( Rs t cursorR R R- Rk t obs( t asserterRt ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyt cursor_execute� s ! ( Rl R t listens_fort removeRo ( t engineRu R{ ( ( Rz Rt sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyt
assert_engineu s !$ ( t collectionst
contextlibR! R R R R~ R t engine.defaultR t engine.utilR t schemaR t objectR R R R RJ RO R] R` Re Rf Rg t
namedtupleRj Rl t contextmanagerR ( ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyt <module> s2 }. "
Mini Shell