o уRc=<у@s^ddlZddlZddlZddlmZddlmZddlmZddlm Z ddl mZddlm Z Gd d Дd eГZGddДdeГZGd dДdeГZGddДdeГZGddДdeГZGddДdeГZGddДdeГZGddДdeГZGddДdeГZGddДdeГZGddДdeГZGdd Дd eГZGd!d"Дd"eаd"gd#вбГZGd$d%Дd%eГZejd&d'ДГZdS)(щNщ)┌event)┌util)┌url)┌DefaultDialect)┌_distill_cursor_params)┌_DDLCompilesc@s(eZdZdZdZdZddДZddДZdS)┌ AssertRuleFNTcCsdSйNйй┌self┌execute_observedrr·>D:\Flask\env\Lib\site-packages\sqlalchemy/testing/assertsql.py┌process_statementszAssertRule.process_statementcCsJdГВ)NFz?All statements are complete, but pending assertion rules remainrйr rrr┌no_more_statementss zAssertRule.no_more_statements)┌__name__┌ __module__┌__qualname__┌is_consumed┌errormessage┌consume_statementrrrrrrr sr c@єeZdZdS)┌SQLMatchRuleNйrrrrrrrr$src@seZdZdddДZddДZdS)┌ CursorSQLNTcCє||_||_||_dSr )┌ statement┌paramsr)r rrrrrr┌__init__)є zCursorSQL.__init__cCsp|jd}|j|jks|jdur%|j|jkr%d|j|j|j|jf|_dS|jаdбd|_|js6d|_dSdS)Nrz5Testing for exact SQL %s parameters %s received %s %sT)┌ statementsrr┌ parametersr┌poprr)r r┌stmtrrrr.s& № zCursorSQL.process_statement)NTйrrrr rrrrrr(s rc@sFeZdZdddДZddДZddДZd d ДZddДZd dДZddДZ dS)┌CompiledSQLN┌defaultcCrr )rr┌dialect)r rrr)rrrr Dr!zCompiledSQL.__init__cCstаdd|jб}||kSйN·[\n\t]┌)┌re┌subr)r r┌received_statementr%rrr┌_compare_sqlIszCompiledSQL._compare_sqlcCsN|jdkr tГ}d|_|S|jdkrddi}ni}tjа|jбабdi|дОS)Nr(T┌ postgresqlZimplicit_returningr)r)rZsupports_default_metavaluer┌URL┌create┌get_dialect)r rr)rrrr┌_compile_dialectMs zCompiledSQL._compile_dialectc s╛|j}|а|б}d}dЙd|jvr|jd}nd}t|jtГr(|jj||dНЙn|jj|||jj|jj |dНЙt аddtа Ибб}|j}|sQИjИdНg}||fSЗЗfdd Д|DГ}||fS) zxreconstruct the statement and params in terms of a target dialect, which for CompiledSQL is just DefaultDialect.N┌schema_translate_map)r)r6)┌ cache_keyr)┌column_keys┌for_executemanyr6r+r,й┌extracted_parameterscsg|] }Иj|ИdНСqS)r:)┌construct_params)┌.0┌mй┌compiledr;rr┌ Йs¤ z3CompiledSQL._received_statement..)┌contextr5┌execution_options┌ isinstance┌ clauseelementr┌compiler@r8r9r-r.r┌ text_typer#r<) r rrBZcompare_dialectr7Zmap_┌_received_statementr#┌_received_parametersrr?rrH\s@ ■√ ∙№zCompiledSQL._received_statementc Csъ|j}|а|б\}}|а|б}|а||б}|r^|dur^t|Г}t|Г}|rX|rXt|аdбГ} tt|ГГD]\} }| D]}||vsH||| |krJnq:|| =nq4d}n|rX|s'|s\|r^d}|rhd|_d|_ dS|а |б||dЬ|_ dS)NrFT)r/Zreceived_parameters)rBrH┌_all_paramsr0┌list┌dictr$┌ enumeraterr┌_failure_message) r rrBrHrIrZ equivalent┌ all_paramsZall_received┌param┌idxZreceivedZ param_keyrrrrТsF ■№√ ъ ■zCompiledSQL.process_statementcCs:|jrt|jГr|а|б}n|j}t|tГs|g}|SdSr )r┌callablerDrK)r rBrrrrrJ├s zCompiledSQL._all_paramscCє"d|jаddбt|ГаddбfS)NzzTesting for compiled statement %r partial params %s, received %%(received_statement)r with params %%(received_parameters)r·%·%%)r┌replace┌reprйr Zexpected_paramsrrrrN╧є■¤ zCompiledSQL._failure_messageйNr() rrrr r0r5rHrrJrNrrrrr'Cs 61r'c@s&eZdZd ddДZddДZddДZdS) ┌RegexSQLNr(cCs,tа|бtа|б|_||_||_||_dSr )rr r-rF┌regex┌ orig_regexrr))r r\rr)rrrr ▄s zRegexSQL.__init__cCrS)Nz{Testing for compiled statement ~%r partial params %s, received %%(received_statement)r with params %%(received_parameters)rrTrU)r]rVrWrXrrrrNуrYzRegexSQL._failure_messagecCst|jа|бГSr )┌boolr\┌match)r rr/rrrr0юszRegexSQL._compare_sqlrZ)rrrr rNr0rrrrr[█s r[cs4eZdZddДZddДZЗfddДZddДZЗZS) ┌ DialectSQLcCs|jjSr )rBr)rrrrr5єszDialectSQL._compile_dialectcCstаdd|б}||kSr*)r-r.)r ┌ real_stmt┌ received_stmtr%rrr┌_compare_no_spaceЎszDialectSQL._compare_no_spacecsJtt|Га|б\}}|jD]}|а|j|бrnq td|ГВ||jjfS)NzICan't locate compiled statement %r in list of statements actually invoked) ┌superr`rHr"rcr┌AssertionErrorrBZcompiled_parameters)r rrbZreceived_paramsraй┌ __class__rrrH·s ■ zDialectSQL._received_statementcCsztаdd|jб}|jjj}|dkrtаdd|б}||kSd}|dkr%d}n |dkr,d }n|d kr2d}tаd||б}||kS)Nr+r,Zpyformatz :([\w_]+)z%(\1)sZqmark·?┌formatz%s┌numeric)r-r.rrBr)┌ paramstyle)r rr/r%rk┌replrrrr0s ўzDialectSQL._compare_sql)rrrr5rcrHr0┌ __classcell__rrrfrr`Єs r`c@є$eZdZddДZddДZddДZdS)┌CountStatementscCs||_d|_dSйNrй┌count┌_statement_count)r rrrrrr !s zCountStatements.__init__cCs|jd7_dS)Nщ)rsrrrrr%sz!CountStatements.process_statementcCs$|j|jkrJd|j|jfГВdS)NFz,desired statement count %d does not match %drqrrrrr(s■ z"CountStatements.no_more_statementsN)rrrr rrrrrrro sroc@єeZdZddДZddДZdS)┌AllOfcGєt|Г|_dSr )┌set┌rulesйr ryrrrr 1єzAllOf.__init__cCslt|jГD]%}d|_|а|б|jr!|jа|б|jsd|_dS|js*d|_dSqt|jГdj|_dSйNTr)rKryrrr┌discardйr r┌rulerrrr4s ¤zAllOf.process_statementNr&rrrrrv0srvcs,eZdZddДZddДZЗfddДZЗZS)┌EachOfcGrwr )rKryrzrrrr Fr{zEachOf.__init__cCs^|jr%|jd}|а|б|jr|jаdбn|jr|j|_|jr"n|js|js-d|_dSdS)NrT)ryrrr$rrr~rrrrIs ° zEachOf.process_statementcs@|jr|jdjs|jdабdS|jrtt|ГабdSdSrp)ryrrrdrАrrfrrrWs zEachOf.no_more_statements)rrrr rrrmrrrfrrАEsrАcseZdZЗfddДZЗZS)┌Conditionalcs,|rtt|Гj|ОdStt|Гj|ОdSr )rdrБr )r ┌ conditionryZ else_rulesrfrrr _szConditional.__init__)rrrr rmrrrfrrБ^srБc@seZdZddДZdS)┌OrcCs>|jD]}|а|б|jrd|_dSqt|jГdj|_dSr|)ryrrrKrr~rrrrgs ■zOr.process_statementN)rrrrrrrrrГfsrГc@ru)┌SQLExecuteObservedcCs*||_||_t|jt|Г|Г|_g|_dSr )rBrEr┌ connection┌tupler#r")r rBrE┌multiparamsrrrrr rs zSQLExecuteObserved.__init__cCs t|jГSr )┌strr"rrrr┌__repr__zє zSQLExecuteObserved.__repr__N)rrrr rЙrrrrrДqsrДc@r)┌SQLCursorExecuteObservedNrrrrrrЛ~srЛ)rr#rB┌executemanyc@rn)┌SQLAssertercCs g|_dSr )┌accumulatedrrrrr ИrКzSQLAsserter.__init__cCs|j|_|`dSr )rО┌_finalrrrr┌_closeЛszSQLAsserter._closecGslt|О}t|jГ}|r#|аdб}|а|б|jrn |jr!J|jГВ|s|r+Jd|ГВ|js4|абdSdS)NrFz$Additional SQL statements remain: %s)rАrKrПr$rrrr)r ryrZobservedrrrr┌assert_Пs · zSQLAsserter.assert_N)rrrr rРrСrrrrrНЗsrНc#sЦБtГЙgЙtа|dбЗfddДГ}tа|dбЗЗfddДГ}zИVWtа|d|бtа|d|бИабdStа|d|бtа|d|бИабw)NZbefore_executecs|||fИddЕ<dSr r)┌connrErЗrrC)┌origrr┌connection_executeжsz)assert_engine..connection_executeZafter_cursor_executecsj|sdSИjrИjdj|urИjd}nt|ИdИdИdГ}Иjа|б|jаt||||ГбdS)Nщ rrtr)rОrBrД┌appendr"rЛ)rТ┌cursorrr#rBrМZobsйZasserterrУrr┌cursor_executeоs ■ z%assert_engine..cursor_execute)rНrZlistens_for┌removerР)┌enginerФrЩrrШr┌ assert_engineаsА ■ rЬ) ┌collections┌ contextlibr-r,rrrЫrZengine.defaultrZengine.utilrZschemar┌objectr rrr'r[r`rorvrАrБrГrД┌ namedtuplerЛrН┌contextmanagerrЬrrrr┌s>. ■