[hf4<ddlmZddlZddlZddlZddlmZddlmZddlmZddlZddl Z ddl m Z ddl m Z dd l mZd d l mZd d l mZd d l mZd dlmZejrd dlmZd dlmZd dlmZGddZeZe dZe dZe dZe dZ d2dZ!GddZ"d3dZ#ej$ d4d5d'Z%ej$ d6d7d+Z% d8d,Z%d2d-Z&Gd.d/Z'Gd0d1Z(dS)9) annotationsN)Any)Dict)Optional)config) decorator) gc_collect)event)pool) await_only)Literal)Engine)URL) AsyncEnginecteZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdS)ConnectionKillerctj|_tjt |_t |_dSN)weakrefWeakKeyDictionary proxy_refs collections defaultdictsettesting_enginesdbapi_connectionsselfs i/var/www/api.easyaligner.net/htdocs/venv_linux/lib/python3.11/site-packages/sqlalchemy/testing/engines.py__init__zConnectionKiller.__init__%s7!355*6s;;!$ctj|d|jtj|d|jtj|d|jtj|d|jdS)Ncheckoutcheckincloseclose_detached)r listen _add_conn _remove_conn)r r s r!add_poolzConnectionKiller.add_pool*sd T:t~666 T9d&7888 T7D$5666 T+T->?????r#cN|j|d|j|<dS)NT)raddr)r dbapi_con con_record con_proxys r!r*zConnectionKiller._add_conn2s+ ""9---%) """r#c:|j|dSr)rdiscard)r dbapi_connargs r!r+zConnectionKiller._remove_conn6s &&z22222r#c||j|dvsJ|j||dS)N)classglobalfunctionfixture)r,r rr.)r enginescopes r! add_enginezConnectionKiller.add_engine9sJ fk"""BBBBB U#''/////r#cz |dS#t$r"}tjd|zYd}~dSd}~wwxYw)Nz5testing_reaper couldn't rollback/close connection: %s Exceptionwarningswarnr fnes r!_safezConnectionKiller._safe?so  BDDDDD    MG!K            :5:czt|jD]%}|!|jr||j&dSr)listris_validrFrollbackr recs r! rollback_allzConnectionKiller.rollback_allGsE(( ) )C3< 3<((( ) )r#c2t|jD]D}|@|jr9|j|j||jE|jD]}||j|j dSr) rIrrJrr3dbapi_connectionrF_checkinrKclear)r rMcons r! checkin_allzConnectionKiller.checkin_allLs(( ) )C3<&..s/CDDD 3<((() % %C JJs| $ $ $ $ $$&&&&&r#c.|dSr)rTrs r! close_allzConnectionKiller.close_all[s r#cntjsdSddlm}||jj|dSNr) provision)rbootstrapped_as_sqlalchemyrYprepare_for_drop_tablesr;url)r connectionrYs r!r\z(ConnectionKiller.prepare_for_drop_tables^sF0  F))**;*?LLLLLr#c|j|}t|D]}t|jD]:}|6|jr/|j(|j|jur||j;t|dr"t| | | dS)N sync_engine) rrIrrJ_poolr rFrQhasattrrdisposerR)r r<engrM proxy_refs r!_drop_testing_enginesz&ConnectionKiller._drop_testing_enginesgs"5)99  C!$/22 7 7 (Y-?(!3%Osx77 9#5666sM** 3;;==))))  r#c0|ddS)Nr9)rfrs r! after_testzConnectionKiller.after_testxs "":.....r#c tjsdS|jjrdS|ddlm}tj5}| |j j |ddddS#1swxYwYdSrX) rrZ __class__"__leave_connections_for_teardown__rTr[rYdbconnectr\r;r])r testrYconns r!after_test_outside_fixturesz,ConnectionKiller.after_test_outside_fixtures{s0  F > <  F   Y    ED  - -dkot D D D E E E E E E E E E E E E E E E E E Es!BB Bc||d|ddS)Nr9r7)rTrfrs r!stop_test_class_inside_fixturesz0ConnectionKiller.stop_test_class_inside_fixturessB  "":... ""7+++++r#ctjjrnttjjrQt tjj}tjj Jd|zdSdS)NFz/%d connection recs not cleared after test suite)r base_strong_ref_connection_recordsr lenrR)r lns r! stop_test_class_outside_fixturesz1ConnectionKiller.stop_test_class_outside_fixturess 9 3 L LLLy7 LABB 8>>@@@LDKLLL  L L L Lr#cl||jD]}||dSr)rTrrf)r r<s r! final_cleanupzConnectionKiller.final_cleanupsG ) . .E  & &u - - - - . .r#c,|jD] }|jrJ dSr)rrJrLs r!assert_all_closedz"ConnectionKiller.assert_all_closeds1?  C|     r#N)__name__ __module__ __qualname__r"r,r*r+r=rFrNrTrVr\rfrhrprrrxrzr|r#r!rr$s''' @@@***333000 ))) ' ' 'MMM"///EEE$,,, L L L... r#rc ||i|tdS#twxYwr)testing_reaperr|rDargskws r!assert_conns_closedrsH+ DB((*****((**** %Ac ||i|tdS#twxYw)zBDecorator that rolls back all open connections after fn execution.N)rrNrs r!rollback_open_connectionsrsH& DB##%%%%%##%%%%rcHt||i|dS)z:Decorator that closes all connections before fn execution.NrrTrs r! close_firstrs+   BOOOOOOr#c ||i|tdS#twxYw)z9Decorator that closes all connections after fn execution.Nrrs r!close_open_connectionsrsH% DB""$$$$$""$$$$rc#Kddlm}|jD]W}|r||vr t||d}|s%tt d|zj|}|VXdS)Nrzsqlalchemy.dialects.%s)sqlalchemy.dialectsdialects__all__getattr __import__dialect)excludednamemods r! all_dialectsrs######    tw at$$ 3d:;;DdCkkmm  r#c4eZdZdZdZdZdZd dZdZdS) ReconnectFixturec0||_g|_d|_dSNF)dbapi connections is_stopped)r rs r!r"zReconnectFixture.__init__s r#c,t|j|Sr)rrr keys r! __getattr__zReconnectFixture.__getattr__stz3'''r#c|jj|i|}|jrM||j|}|dJd|j||S)Nzselect 1Fz%simulated connect failure didn't work) rrmrrFr'cursorexecuterappend)r rkwargsrocurss r!rmzReconnectFixture.connects!tz!42622 ?  JJtz " " ";;==D LL $ $ $ AA A A A   # #D ) ) )Kr#cz |dS#t$r"}tjd|zYd}~dSd}~wwxYw)Nz.ReconnectFixture couldn't close connection: %sr?rCs r!rFzReconnectFixture._safesb P BDDDDD P P P MJQN O O O O O O O O O PrGFc||_t|jD]}||jg|_dSr)rrIrrFr')r stopcs r!shutdownzReconnectFixture.shutdownsJ d&''  A JJqw    r#cd|_dSr)rrs r!restartzReconnectFixture.restarts r#N)F) r}r~rr"rrmrFrrrr#r!rrsv   (((   PPP         r#rc(|ptjj}tjjj}|si}t ||d<t ||jfd}jjj_ jjj _ |_S)Nmoduleczjjdjj_dSr)rrrr)_disposer;sr!rcz$reconnecting_engine..disposes6%%'''*/' r#) rrlr]rrrtesting_enginercr test_shutdownr test_restart)r]optionsrrcrr;s @@r!reconnecting_enginer s C I  #E (//GH C ) )F~H ">/8F ..6FFN Mr#Fr] Optional[URL]rOptional[Dict[str, Any]]asyncioLiteral[False]transfer_staticpoolboolreturnrcdSrrr]rrrs r!rrs Sr#T Literal[True]rcdSrrrs r!rr's #r#cr|r |rJddlm}nddlm}ddlm}|sd}d} d} nB|dd}|d d} |d d} |ptjj }||}tj|j tjj j kr+tj rtj } ni} || |||fi| } | rG| jd kr.do_connect\s/3  , , ,r#beginc0|ddS)NBEGIN)exec_driver_sql)ros r!do_beginz testing_engine..do_begin`s   ) ) ) ) )r#) StaticPoolr8r pool_timeout max_overflow)sqlalchemy.ext.asyncior sqlalchemyrsqlalchemy.engine.urlrpoprrlr] drivernamedb_optscopyupdaterr listens_forsqlalchemy.poolr isinstancer _transfer_fromr` _has_events QueuePool_timeout _max_overflowrr=)r]rrr share_pool_sqlite_savepointrrrr<r use_optionsr;rrrs r!rr0s-$$$$        -,,,,,...... B  [[t44  GZ00";;'95AA C (3--C  S^vy}/GGG .Hn))++  7### ]3 . .+ . .F*FK833  69 - - 4 4 . - 4  67 + + * * , + *%...... 9 Z  %K%K J K & &vy~ 6 6 6 %in    -1F  * *   6;//& + % % + - - + - -  $% !1!!&%000 Mr#cddlm}|stjj}gfd}fd}fd}||dz|t drJ_|_|_S)a'Provides a mocking engine based on the current testing.db. This is normally used to test DDL generation flow as emitted by an Engine. It should not be used in other cases, as assert_compile() and assert_sql_execution() are much better choices with fewer moving parts. r)create_mock_enginec2|dSr)r)sqlarbuffers r!executorzmock_engine..executors cr#c@dD}||ks J|dS)Nc Tg|]%}tjddt|&S)z[\n\t]r[)resubstr).0ss r! z3mock_engine..assert_sql..s,>>>!y"c!ff-->>>r#r)stmtsrecvrs r! assert_sqlzmock_engine..assert_sqls->>v>>>u}}}d}}}}}r#c`jdfdjDS)N c3^K|]'}t|V(dS))rN)rcompile)rrrs r! z1mock_engine..print_sql..s9HHqQYYqY1122HHHHHHr#)rjoinmock)rr;s@r! print_sqlzmock_engine..print_sqls4 NyyHHHHFKHHHHHHr#z://r) rrrrlrrbrrr) dialect_namerrrrrr;s @@r! mock_enginers.----- &y~ F#####IIIII  u 4h ? ?Fvv&&&&&FK"F F Mr#c2eZdZdZdZddZdZdZdZdS) DBAPIProxyCursorzuProxy a DBAPI cursor. Tests can provide subclasses of this to intercept DBAPI-level cursor operations. cF||_||_|j|i||_dSr)r;r^r)r r;rorrs r!r"zDBAPIProxyCursor.__init__s, !dk42622 r#Nc T|r|jj||fi|S|jj|fi|Sr)rr)r stmt parametersrs r!rzDBAPIProxyCursor.executesF  3&4;&tZ>>2>> >&4;&t22r22 2r#c *|jj||fi|Sr)r executemany)r r paramsrs r!rzDBAPIProxyCursor.executemanys!&t{&tV::r:::r#c*t|jSr)iterrrs r!__iter__zDBAPIProxyCursor.__iter__sDK   r#c,t|j|Sr)rrrs r!rzDBAPIProxyCursor.__getattr__st{C(((r#r) r}r~r__doc__r"rrrrrr#r!r r sn333 3333 ;;;!!!)))))r#r c*eZdZdZdZdZdZdZdS)DBAPIProxyConnectionz}Proxy a DBAPI connection. Tests can provide subclasses of this to intercept DBAPI-level connection operations. c0||_||_||_dSr)ror; cursor_cls)r r;rors r!r"zDBAPIProxyConnection.__init__s  $r#c:|j|j|jg|Ri|Sr)rr;ro)r rrs r!rzDBAPIProxyConnection.cursors*tt{DIGGGGGGGr#c8|jdSr)ror'rs r!r'zDBAPIProxyConnection.closes r#c,t|j|Sr)rrors r!rz DBAPIProxyConnection.__getattr__sty#&&&r#N)r}r~rrr"rr'rrr#r!rrs]%%% HHH'''''r#rr)NN)NNFF) r]rrrrrrrrr)NNTF) r]rrrrrrrrr)NNFFFF)) __future__rrrtypingrrrrArr[rutilr r r r r util.typingr TYPE_CHECKINGr;r engine.urlr ext.asynciorrrrrrrrrroverloadrrr rrr#r!r&s@#""""" !!!!!! *      ))))))CCCCCCCCL"!## ++ + && &   %% %    ( ( ( ( ( ( ( ( V((,# % (,! %    PPPPf####L)))))))):''''''''''r#