T2Q4" d dl Z d dlZd dlZddlmZ ddlmZ ddlmZ ddlm Z ddl
m Z ddl m
Z
G d d
e Z G d d e Z G d
de Z G d de Z G d de Z G d de Z G d de Z G d de Z G d de Z G d d e Z G d d e Z G d d e j d g d! Z G d" d#e Z ej d$ Z dS )% N )event)util)url)DefaultDialect)_distill_params) _DDLCompilesc & e Zd ZdZdZdZd Zd ZdS )
AssertRuleFNTc d S N selfexecute_observeds p/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/sqlalchemy/testing/assertsql.pyprocess_statementz AssertRule.process_statement s c J d )NFz?All statements are complete, but pending assertion rules remainr r s r no_more_statementsz AssertRule.no_more_statements s
%
ur )__name__
__module__ __qualname__ is_consumed errormessageconsume_statementr r r r r r r sB KL
r r c e Zd ZdS ) SQLMatchRuleNr r r r r r r r $ s Dr r c e Zd ZdZddZd ZdS ) CursorSQLFNc " || _ || _ d S r
) statementparams)r r$ r% s r __init__zCursorSQL.__init__+ s " r c " |j d }| j |j k s| j 9| j |j k r)d| j d| j d|j d|j | _ d S |j d d| _ |j s d| _ d S d S )Nr zTesting for exact SQL z parameters z
received T)
statementsr$ r%
parametersr popr r )r r stmts r r zCursorSQL.process_statement/ s *1- >T^ + + K # t(F(F
NNNKKKNNNOO
' + +A . . . #D #.
.)-&&&
.
.r r
)r r r r r& r r r r r" r" ( s= . . . . .r r" c : e Zd Zd
dZd Zd Zd Zd Zd Zd Z dS ) CompiledSQLNdefaultc 0 || _ || _ || _ d S r
)r$ r% dialect)r r$ r% r1 s r r& zCompiledSQL.__init__E s " r c D t j dd| j }||k S N[\n\t] )resubr$ )r r received_statementr, s r _compare_sqlzCompiledSQL._compare_sqlJ s" viT^44!T))r c | j dk rt S | j dk rddi}ni } t j | j di |S )Nr/
postgresqlimplicit_returningTr )r1 r r URL get_dialect)r r r% s r _compile_dialectz CompiledSQL._compile_dialectN si <9 $ $!## # ||++.56374<((4466@@@@ @r c > |j }| | }t |j j t
r:|j j ||j d nO|j j ||j j |j j
|j d t j ddt j }|j }|s g}nfd|D }||fS )zxreconstruct the statement and params in terms
of a target dialect, which for CompiledSQL is just DefaultDialect.schema_translate_map)r1 rA )r1 column_keysinlinerA r4 r5 c : g | ]} | S r )construct_params).0mcompileds r
z3CompiledSQL._received_statement..u s4 $ $ $12))!,,$ $ $r )contextr?
isinstancerH r$ r compileexecution_optionsgetrB rC r6 r7 r text_typer* rE )r r rJ compare_dialect_received_statementr* _received_parametersrH s @r rQ z CompiledSQL._received_statementY sF #* //0@AA g&0, ? ? '199'%,%>%B%B*& & : HH '199' # , 8 '.%,%>%B%B*& & : H !fYDN84L4L M M%0
$,$=$=$?$?#@ $ $ $ $6@$ $ $ #$888r c 0 |j }| | \ }}| | }| || }|r|t | }t | }|ro|rmt | d } t t | D ]'\ }
} | D ]} | | vs| | | | k r n||
= n(d}n|r|m|s|rd}|rd| _ d | _ d S |
| ||dz | _ d S )Nr FT)r8 received_parameters) rJ rQ _all_paramsr9 listdictr+ enumerater r _failure_message)
r r rJ rQ rR r%
equivalent
all_params all_receivedparamidxreceived param_keys
r r z CompiledSQL.process_statement{ s "*484L4L 5
5
1 1 !!'**&&'79LMM
'! !&\\
#$8 9 9 \ !2!2 3 3E)24 3E3E)F)F
X). " "I !* 9 9#+I#6% :J#J#J % $K !-S 1 !E !& &+
- ! \ . ' '!&J #D $D $ 5 5f = =&9';A A !D r c | j rPt j | j r| | }n| j }t |t s|g}|S d S r
)r% r callablerK rV )r rJ r% s r rU zCompiledSQL._all_params s] ; }T[))
% W-- fd++
" M4r c d| j dd dt | dd dS )Nz Testing for compiled statement %%% partial params E, received %(received_statement)r with params %(received_parameters)r)r$ replacereprr expected_paramss r rY z CompiledSQL._failure_message sL &&sD1111_%%--c48888
r Nr/ )
r r r r& r9 r? rQ r rU rY r r r r. r. D s
* * * A A A 9 9 9D/ / /b
r r. c " e Zd ZddZd Zd ZdS )RegexSQLNr/ c t | t j | | _ || _ || _ || _ d S r
)r r& r6 rL regex
orig_regexr% r1 )r rp r% r1 s r r& zRegexSQL.__init__ s@ d###Z&&
r c d| j dd dt | dd dS )Nz Testing for compiled statement ~rd re rf rg )rq rh ri rj s r rY zRegexSQL._failure_message sL ''T2222_%%--c48888
r c P t | j | S r
)boolrp match)r r r8 s r r9 zRegexSQL._compare_sql s! DJ$$%788999r rl )r r r r& rY r9 r r r rn rn sF
: : : : :r rn c 0 e Zd Zd Zd Z fdZd Z xZS )
DialectSQLc |j j S r
)rJ r1 r s r r? zDialectSQL._compile_dialect s '//r c : t j dd| }||k S r3 )r6 r7 )r real_stmt
received_stmtr, s r _compare_no_spacez DialectSQL._compare_no_space s viY// $$r c t t | | \ }}|j D ] }| |j | r n t
d|z ||j j fS )NzICan't locate compiled statement %r in list of statements actually invoked) superrw rQ r) r| r$ AssertionErrorrJ compiled_parameters)r r r{ received_paramsrz __class__s r rQ z DialectSQL._received_statement s ). *
*
.
/
/ '
*4 I%%i&9=II
!.0=>
. 6 JJJr c t j dd| j }|j j j }|dk rt j dd| }n2d }|dk rd}n|dk rd }n|d
k rd }t j d|| }||k S ) Nr4 r5 pyformatz :([\w_]+)z%(\1)sqmark?formatz%snumeric)r6 r7 r$ rJ r1
paramstyle)r r r8 r, r repls r r9 zDialectSQL._compare_sql s viT^44 &-5@
# #6, 488DD DW$$x'' y((6,d33D!T))r )r r r r? r| rQ r9
__classcell__r s @r rw rw sj 0 0 0% % %K K K K K"* * * * * * *r rw c e Zd Zd Zd Zd ZdS )CountStatementsc " || _ d| _ d S Nr count_statement_count)r r s r r& zCountStatements.__init__
s
! r c & | xj dz
c_ d S )N )r r s r r z!CountStatements.process_statement s " r c X | j | j k rJ d| j | j fz d S )NFz,desired statement count %d does not match %dr r s r r z"CountStatements.no_more_statements sD :. . .
H
%L
5 / .r N)r r r r& r r r r r r r sA " " "# # # r r c e Zd Zd Zd ZdS )AllOfc . t | | _ d S r
)setrulesr r s r r& zAllOf.__init__ s ZZ
r c : t | j D ]a}d |_ | | |j r+| j | | j sd| _ d S |j s
d | _ d S bt | j d j | _ d S NTr )rV r r r r discardr r rules r r zAllOf.process_statement s $$
A
AD $D " "#3 4 4 4
""4(((z ,'+D$&
$(!
!%TZ 0 0 3 @D r N)r r r r& r r r r r r s7 A A A A Ar r c * e Zd Zd Zd Z fdZ xZS )EachOfc . t | | _ d S r
)rV r r s r r& zEachOf.__init__/ s %[[
r c | j rf| j d }| | |j r| j d n|j r |j | _ |j rn| j f| j s d| _ d S d S )Nr T)r r r r+ r r r s r r zEachOf.process_statement2 s j :a=D " "#3 4 4 4
6
q!!!!"
6$($5!%
j z $ #D $ $r c | j r3| j d j s!| j d d S | j r)t t | d S d S r )r r r r~ r )r r s r r zEachOf.no_more_statements@ sq : 5djm 7 5 JqM , , . . . . .
Z 5 &$ 2 2 4 4 4 4 4 5 5r )r r r r& r r r r s @r r r . sV ! ! ! $ $ $ 5 5 5 5 5 5 5 5 5r r c e Zd Zd ZdS )Orc | j D ](}| | |j r
d| _ d S )t | j d j | _ d S r )r r r rV r r s r r zOr.process_statementH sk J A AD " "#3 4 4 4
#'
!%TZ 0 0 3 @D r N)r r r r r r r r r G s( A A A A Ar r c e Zd Zd ZdS )SQLExecuteObservedc Z || _ || _ t || | _ g | _ d S r
)rJ
clauseelementr r* r) )r rJ r multiparamsr% s r r& zSQLExecuteObserved.__init__S s- *)+v>> r N)r r r r& r r r r r R s# r r c e Zd ZdS )SQLCursorExecuteObservedNr r r r r r Z s Dr r )r$ r* rJ executemanyc e Zd Zd Zd Zd ZdS ) SQLAsserterc g | _ d S r
) accumulatedr s r r& zSQLAsserter.__init__d s r c " | j | _ | ` d S r
)r _finalr s r _closezSQLAsserter._closeg s & r c * t | }t | j }|rJ| d }| | |j rn|j rJ |j |J|r
J d |j s| d S d S )Nr Fz Additional SQL statements remain)r rV r r+ r r r r )r r r observedr$ s r assert_zSQLAsserter.assert_k s u~ $$ 0 QI " "9 - - -
0"
0/d ///u
0 & << < <5! & # # % % % % % & &r N)r r r r& r r r r r r r c sA & & & & &r r c # K t g t j | d fd }t j | d fd } V t j | d| t j | d| d S # t j | d| t j | d| w xY w)Nbefore_executec |||fd d < d S r
r )connr r r% origs r connection_executez)assert_engine..connection_execute s f4QQQr after_cursor_executec 2 |sd S j r"j d j |u rj d }n>t |d d d }j | |j t |||| d S )Nr r r )r rJ r appendr) r ) r cursorr$ r* rJ r obsasserterr s r cursor_executez%assert_engine..cursor_execute s F
-$R(0G;;&r*CC$Wd1gtAwQHHC ' ' , , , $:w
r )r r listens_forremover )enginer r r r s @@r
assert_enginer | s }}H
D
v /005 5 5 5 105
v 566
76
*
V 3^DDD
V -/ABBB V 3^DDD
V -/ABBBs
B AC ) collections
contextlibr6 r5 r r r r engine.defaultr engine.utilr schemar objectr r r" r. rn rw r r r r r
namedtupler r contextmanagerr r r r r s + + + + + + ) ) ) ) ) ) ! ! ! ! ! !
: . . . . . . . .8}
}
}
}
}
, }
}
}
@: : : : :{ : : :.+* +* +* +* +* +* +* +*\
j
A A A A AJ A A A*5 5 5 5 5Z 5 5 52A A A A A A A A K"=== & & & & && & & &2 & & & & &r