JezK
Edit File: assertsql.pyc
� 1��fc @ 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 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_cursor_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 d e d � Z d � Z RS( c C s | | _ | | _ | | _ d S( N( t statementt paramsR ( R 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 ( ( ( 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 D 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_sqlI s c C sm | j d k r% t � } t | _ | S| j d k rD i t d 6} n i } t j j | j � j � | � Sd S( NR t postgresqlt implicit_returning( R R R t supports_default_metavalueR t URLt createt get_dialect( R R R R ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyt _compile_dialectM s c C s) | j } | j | � } d } d } d | j k rC | j d } n d } t | j t � ry | j j d | d | � } n9 | j j d | d | d | j j d | j j d | � } t j d d t j | � � } | j } | s� | j d | � g } n( g | D] } | j | d | �^ q� } | | f S( sx reconstruct the statement and params in terms of a target dialect, which for CompiledSQL is just DefaultDialect.t schema_translate_mapR t cache_keyt column_keyst for_executemanys [\n\t]R t extracted_parametersN( t contextR+ R t execution_optionst isinstancet clauseelementR t compilet compiledR. R/ R! R" R t text_typeR t construct_params( R R R1 t compare_dialectR- R0 t map_R6 t _received_statementR t _received_parameterst m( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyR; \ s2 %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( R1 R; t _all_paramsR$ R t listt dictR t enumerateR R R R t _failure_message( R R R1 R; R< 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 sZ | j rR t | j � r* | j | � } n | j } t | t � sN | g } n | Sd Sd S( N( R t callableR3 R@ R ( R R1 R ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyR? � s c C s/ d | j j d d � t | � j d d � f S( Nsz Testing for compiled statement %r partial params %s, received %%(received_statement)r with params %%(received_parameters)rt %s %%( R t replacet repr( R t expected_params( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyRC � s N( R R R R R$ R+ R; R R? RC ( ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyR C s 6 1 t RegexSQLc B s) e Z d d d � Z d � Z d � Z RS( R c C s> t j | � t j | � | _ | | _ | | _ | | _ d S( N( R R R! R5 t regext orig_regexR R ( R RQ R R ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyR � s c C s/ d | j j d d � t | � j d d � f S( Ns{ Testing for compiled statement ~%r partial params %s, received %%(received_statement)r with params %%(received_parameters)rRL s %%( RR RM RN ( R RO ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyRC � s c C s t | j j | � � S( N( t boolRQ t match( R R R# ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyR$ � s N( R R R R RC R$ ( ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyRP � s t DialectSQLc B s, e Z d � Z d � Z d � Z d � Z RS( c C s | j j S( N( R1 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 superRU R; R RX R R R1 t compiled_parameters( R R RW t received_paramsRV ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyR; � 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 R1 R t paramstyleR ( R R R# R Ra t repl( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyR$ s ( R R R+ RX R; R$ ( ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyRU � 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 Rd ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyR ! s c C s | j d 7_ d S( Ni ( Re ( 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( Rd Re 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.pyRc s t AllOfc B s e Z d � Z d � Z RS( c G s t | � | _ d S( N( t sett rules( R Rh ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyR 1 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@ Rh R R R R t discardR ( R R t rule( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyR 4 s ( R R R R ( ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyRf 0 s t EachOfc B s# e Z d � Z d � Z d � Z RS( c G s t | � | _ d S( N( R@ Rh ( R Rh ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyR F 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 ( Rh R R R R R R ( R R Rj ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyR I s c C sQ | j r. | j d j r. | j d j � n | j rM t t | � j � n d S( Ni ( Rh R R RY Rk ( R ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyR W s ( R R R R R ( ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyRk E s t Conditionalc B s e Z d � Z RS( c C s9 | r t t | � j | � n t t | � j | � d S( N( RY Rl R ( R t conditionRh t else_rules( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyR _ s ( R R R ( ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyRl ^ 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 ( Rh R R R R@ R ( R R Rj ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyR g s ( R R R ( ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyRo f s t SQLExecuteObservedc B s e Z d � Z d � Z RS( c C s= | | _ | | _ t | j t | � | � | _ g | _ d S( N( R1 R4 R t connectiont tupleR R ( R R1 R4 t multiparamsR ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyR r s c C s t | j � S( N( t strR ( R ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyt __repr__z s ( R R R Ru ( ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyRp q s t SQLCursorExecuteObservedR R R1 t executemanyc B s e Z RS( ( R R ( ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyRv ~ 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( Ry 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: %s( Rk R@ Rz R R R R R R R ( R Rh Rj t observedR ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyt assert_� s ( R R R R{ R} ( ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyRx � 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 connR4 Rs R R2 ( 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 ( Ry R1 Rp t appendR Rv ( R t cursorR R R1 Rw t obs( t asserterR� ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyt cursor_execute� s ! ( Rx R t listens_fort removeR{ ( t engineR� R� ( ( R� R� sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyt assert_engine� s !$ ( t collectionst contextlibR! R R R R� R t engine.defaultR t engine.utilR t schemaR t objectR R R R RP RU Rc Rf Rk Rl Ro Rp t namedtupleRv Rx t contextmanagerR� ( ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/assertsql.pyt <module> s4 �. "