EI0 t d dl mZ d dlZd dlZd dlZddlmZ ddlmZ ddlm Z ddlm
Z
dd lm Z G d
d e Z
e
Zd Ze d
Ze d Ze d Ze d Zd dZ G d de Zd dZd dZd dZ G d de Z G d de ZeefdZdS ) )absolute_importN )config)uses_deprecated) decorator )event)poolc e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z d
Z e
d Z e
d Zd
ZdS )ConnectionKillerc t j | _ t j | _ t | _ d S N)weakrefWeakKeyDictionary
proxy_refstesting_enginessetconnsselfs n/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/sqlalchemy/testing/engines.py__init__zConnectionKiller.__init__ s2 !355 & 8 : : UU
c d| j |< d S NT)r )r engines r
add_enginezConnectionKiller.add_engine s '+ V$$$r c > | j ||f d S r )r add)r
dbapi_conn
con_records r connectzConnectionKiller.connect s!
J/00000r c d| j |< d S r )r )r dbapi_conr! con_proxys r checkoutzConnectionKiller.checkout! s %) """r c > | j ||f d S r )r discard)r r$ r! exceptions r
invalidatezConnectionKiller.invalidate$ s#
Iz233333r c z | d S # t $ r"}t j d|z Y d }~d S d }~ww xY w)Nz5testing_reaper couldn't rollback/close connection: %s Exceptionwarningswarnr fnes r _safezConnectionKiller._safe' so BDDDDD MJQN
:5:c z t | j D ]%}|!|j r| |j &d S r )listr is_validr3 rollbackr recs r rollback_allz ConnectionKiller.rollback_all/ sE (( ) )C3<
3<((( ) )r c z t | j D ]%}|!|j r| |j &d S r )r6 r r7 r3 _closer9 s r close_allzConnectionKiller.close_all4 sE (( ' 'C3<
3:&&& ' 'r c ~ t | j D ]'\ }}|j
| |j (d S r )r6 r
connectionr3 r8 r connr: s r _after_test_ctxz ConnectionKiller._after_test_ctx9 sO dj)) & &ID#~%
JJt} % % % % & &r c | t j j r| d S | d S r )r optionslow_connections_stop_test_ctx_minimal_stop_test_ctx_aggressiver s r _stop_test_ctxz ConnectionKiller._stop_test_ctxH s? > ) - ' ' ) ) ) ) ) * * , , , , ,r c | t | _ t | j D ]$}|t
j ur| %d S r )r> r r r6 r r dbdisposer9 s r rG z'ConnectionKiller._stop_test_ctx_minimalN s\ UU
,-- C&)##
r c | t | j D ]&\ }}| |j d |_ 't
| _ t | j D ]}| d S r ) r> r6 r r3 closer@ r r rL rA s r rH z*ConnectionKiller._stop_test_ctx_aggressiveX s dj)) " "ID# JJtz " " " !CNNUU
,-- C KKMMMM r c , | j D ] }|j rJ d S r )r r7 r9 s r assert_all_closedz"ConnectionKiller.assert_all_closedc s1 ? C|
u
r N)__name__
__module__ __qualname__r r r" r& r* r3 r; r> rC rI r rG rH rP r r r r s
, , ,1 1 1* * *4 4 4 ) ) )
' ' '
&
&
& - - - _ _ r r c p t t |d r| t j j j sMddlm } |
dd 5 | | d d d d S # 1 swxY w Y d S | | d S )NrN r )
assertionszCan't sort tablesF)assert_) testing_reaperr> hasattrrN r rK dialectsupports_alter rV expect_warningsdrop_all)metadatabindrV s r drop_all_tablesra l s tW
9 +
'
'(;U
'
K
K $ $ d # # # $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ s 1BBBc | |i | t d S # t w xY wr )rX rP r1 argskws r assert_conns_closedrf z sH +
DB((*****((**** % A c | |i | t dS # t w xY w)zBDecorator that rolls back all open connections after fn execution.N)rX r; rc s r rollback_open_connectionsri sH &
DB##%%%%%##%%%%rg c H t | |i | dS )z:Decorator that closes all connections before fn execution.NrX r> rc s r close_firstrl s+ BOOOOOOr c | |i | t dS # t w xY w)z9Decorator that closes all connections after fn execution.Nrk rc s r close_open_connectionsrn sH #
DB """"" """"rg c # K dd l m} |j D ]W}| r|| v r t ||d }|s%t t d|z j | }| V Xd S )Nr zsqlalchemy.databases.%s)sqlalchemy.databases databases__all__getattr
__import__rZ )excludednamemods r all_dialectsry s $$$$$$ tw at$$ 4t;<<F C kkmm r c 4 e Zd Zd Zd Zd Zd Zd dZd ZdS )
ReconnectFixturec 0 || _ g | _ d| _ d S NF)dbapi connections
is_stopped)r r~ s r r zReconnectFixture.__init__ s
r c , t | j | S r )rs r~ r keys r __getattr__z ReconnectFixture.__getattr__ s tz3'''r c | j j |i |}| j rM| |j | }| d J d | j | |S )Nzselect 1Fz%simulated connect failure didn't work) r~ r" r r3 rN cursorexecuter append)r rd kwargsrB curss r r" zReconnectFixture.connect s !tz!42622 ? JJtz " " ";;==D
LL $ $ $ AA A A5 # #D ) ) )Kr c z | d S # t $ r"}t j d|z Y d }~d S d }~ww xY w)Nz.ReconnectFixture couldn't close connection: %sr, r0 s r r3 zReconnectFixture._safe so BDDDDD MCaG
r4 Fc || _ t | j D ] }| |j g | _ d S r )r r6 r r3 rN )r stopcs r shutdownzReconnectFixture.shutdown sJ d&'' A JJqw r c d| _ d S r} )r r s r restartzReconnectFixture.restart s
r N)F) rQ rR rS r r r" r3 r r rT r r r{ r{ ss
( ( ( r r{ c ( | pt j j } t j j j }|si }t | |d< t
| | j fd}j j j _ j j j
_ |_ S )Nmodulec z j j dj j _ d S r} )rZ r~ r r )_disposer s r rL z$reconnecting_engine..dispose s6 %%'''*/ '
r ) r rK urlrZ r~ r{ testing_enginerL r
test_shutdownr test_restart)r rE r~ rL r r s @@r reconnecting_enginer s
C I #E (//GH
C
)
)F~H
">/8F ..6F FN Mr c j ddl m} ddlm} |sd}n| dd }| pt
j j } || } |;t
j | j t
j j j k r
t
j }nai }n^t
j R| j t
j j j k r3t
j
}| | || fi |}d|_ t |j t j rd|j _ d|j _ |rt% j |j dt( j t% j |j dt( j t% j |j d t( j t( | |S )
zBProduce an engine configured by --options with optional overrides.r
create_engine)make_urlT
use_reaperNr" r& r* )
sqlalchemyr sqlalchemy.engine.urlr popr rK r
drivernamedb_optscopyupdate _has_events
isinstancer
QueuePool_timeout
_max_overflowr listenrX r" r& r* r )r rE r r r default_optr s r r r s )(((((...... 5
[[ t44
C
(3--C 9 &)-2J J J nGGGG 3>VY]5M#M#M n))++ 7###
]3
*
*'
*
*F F&+t~.. & $% ! *
V[)^-CDDD
V[*n.EFFF
V[,0IJJJ!!&))) Mr c ddl m} | st j j } g fd}fd}fd} || dz d| t d rJ _ |_ |_ 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 r c 2 | d S r )r )sqlare buffers r executorz mock_engine..executor' s
cr c @ d D }|| k s
J | d S )Nc T g | ]%}t j d dt | &S )z[\n\t]r\ )resubstr).0ss r
z3mock_engine..assert_sql..+ s, >>>!y"c!ff-->>>r rT )stmtsrecvr s r
assert_sqlz mock_engine..assert_sql* s- >>v>>>u}}}d}}}}}r c ` j d fdj D S )N
c 3 ^ K | ]'}t | V (dS ))rZ N)r compile)r r rv s r z1mock_engine..print_sql..0 s9 HHqQYYqY 1 122HHHHHHr )rZ joinmock)rv r s @r print_sqlz mock_engine..print_sql. s4 NyyHHHHFKHHHHHHr z://r )strategyr ) r r r rK rw rY r r r ) dialect_namer r r r r r s @@r mock_enginer s )((((( & y~
F # # # # #I I I I I ]u v F vv&&&& &FK"F F Mr c 8 e Zd ZdZd Zd dZd Zd Zd Zd Z dS )
DBAPIProxyCursorzuProxy a DBAPI cursor.
Tests can provide subclasses of this to intercept
DBAPI-level cursor operations.
c F || _ || _ |j |i || _ d S r )r r@ r )r r rB rd r s r r zDBAPIProxyCursor.__init__D s, !dk42622 r Nc T |r | j j ||fi |S | j j |fi |S r )r r )r stmt
parametersre s r r zDBAPIProxyCursor.executeI sF 3&4;&tZ>>2>> >&4;&t22r22 2r c * | j j ||fi |S r )r executemany)r r paramsre s r r z DBAPIProxyCursor.executemanyO s! &t{&tV::r:::r c , t | j | S r )rs r r s r r z DBAPIProxyCursor.__getattr__R s t{C(((r c | S r rT r s r __enter__zDBAPIProxyCursor.__enter__U s r c d S r rT )r rd s r __exit__zDBAPIProxyCursor.__exit__X s r r )
rQ rR rS __doc__r r r r r r rT r r r r < s} 3 3 3
3 3 3 3 ; ; ;) ) )
r r c * e Zd ZdZd Zd Zd Zd ZdS )DBAPIProxyConnectionz}Proxy a DBAPI connection.
Tests can provide subclasses of this to intercept
DBAPI-level connection operations.
c l |j x| _ | _ || _ || _ d S r )r
_creatorrB _sqla_unwrapr
cursor_cls)r r r s r r z DBAPIProxyConnection.__init__d s1 (. (<(<(>(>> D% $r c : | j | j | j g|R i |S r )r r rB )r rd r s r r zDBAPIProxyConnection.cursori s* tt{DIGGGGGGGr c 8 | j d S r )rB rN r s r rN zDBAPIProxyConnection.closel s r c , t | j | S r )rs rB r s r r z DBAPIProxyConnection.__getattr__o s ty#&&&r N)rQ rR rS r r r rN r rT r r r r \ s] % % %
H H H ' ' ' ' 'r r c 6 fd}t d|i S )zIProduce an engine that provides proxy hooks for
common methods.
c 0 t j S r )r rK )conn_clsr s r mock_connz"proxying_engine..mock_conn{ s x :...r creator)rE )r )r r r s `` r proxying_enginer s s: / / / / / / 9i"8 9 9 99r r )NN)
__future__r r r. r r\ r r utilr r r
objectr rX ra rf ri rl rn ry r{ r r r r r r rT r r r sD ' & & & & & Q Q Q Q Q v Q Q Q h "!## + + + & & & # # # + + + + + v + + + \ (# # # #L% % % %P
v
@' ' ' ' '6 ' ' '0 ".> : : : : : :r