U
|
¸ý°d6 ã @ s¬ d dl mZ d dlZd dlZd dlZd dlmZ d dlmZ d dlmZ d dlZd dl Z ddl
|
m Z ddl m Z dd l mZ d
|
d l
|
mZ d
|
d l
|
mZ d
|
d l mZ d
|
dlmZ ejrÖd
|
dlmZ d
|
dlmZ d
|
dlmZ G dd dZe Z e dd Z e dd Z e dd Z e dd Z d6d d Z!G d d d Z"d7d d!Z#ej$d8d#d$d%d&d'd(d)d*Z%ej$d9d#d$d,d&d-d(d.d*Z%d:d/d*Z%d;d0d1Z&G d2d3 d3Z'G d4d5 d5Z(dS )<é )Ú annotationsN)ÚAny)ÚDict)ÚOptionalé )Úconfig)Ú decorator)Ú
|
gc_collecté )Úevent)Úpool)Ú
|
await_only)ÚLiteral)ÚEngine)ÚURL)Ú AsyncEnginec @ s e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dd Z dd Zdd Zdd Zd d Zd d Zd!d" Zd#S )$ÚConnectionKillerc C s" t ¡ | _t t¡| _t | _d S ©N)ÚweakrefÚWeakKeyDictionaryÚ
|
proxy_refsÚ collectionsÚ defaultdictÚsetÚtesting_enginesÚdbapi_connections©Úself© r úQd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\sqlalchemy/testing/engines.pyÚ__init__% s
|
zConnectionKiller.__init__c C sD t |d| j¡ t |d| j¡ t |d| j¡ t |d| j¡ d S )NÚcheckoutZcheckinÚcloseZclose_detached)r ÚlistenÚ _add_connÚ _remove_conn)r r r r r Úadd_pool* s zConnectionKiller.add_poolc C s | j |¡ d| j|< d S )NT)r Úaddr )r Z dbapi_conZ
|
con_recordZ con_proxyr r r r$ 2 s zConnectionKiller._add_connc G s | j |¡ d S r )r Údiscard)r Z
|
dbapi_connÚargr r r r% 6 s z ConnectionKiller._remove_connc C s, | |j¡ |dkst| j| |¡ d S )N)ÚclassÚglobalÚfunctionZfixture)r& r ÚAssertionErrorr r' )r ÚengineÚscoper r r Ú
|
add_engine9 s zConnectionKiller.add_enginec
|
C s@ z
|
| W n0 t k
|
r: } zt d| ¡ W 5 d }~X Y nX d S )Nz5testing_reaper couldn't rollback/close connection: %s©Ú ExceptionÚwarningsÚwarn©r ÚfnÚer r r Ú_safe? s
|
ÿzConnectionKiller._safec C s. t | jD ] }|d k r
|
|jr
|
| |j¡ q
|
d S r )Úlistr Úis_validr8 Úrollback©r Úrecr r r Ú rollback_allG s z ConnectionKiller.rollback_allc C s^ t | jD ],}|d k r
|
|jr
|
| j |j¡ | |j¡ q
|
| jD ]}| |j¡ q>| j ¡ d S r )
|
r9 r r: r r( Údbapi_connectionr8 Ú_checkinr; Úclear)r r= Úconr r r Ú checkin_allL s
|
z ConnectionKiller.checkin_allc C s | ¡ d S r )rC r r r r Ú close_all[ s zConnectionKiller.close_allc C s* t js
|
d S ddlm} | |jj|¡ d S ©Nr )Ú provision)r Úbootstrapped_as_sqlalchemyÚ rF Úprepare_for_drop_tablesr. Úurl)r Ú
|
connectionrF r r r rI ^ s z(ConnectionKiller.prepare_for_drop_tablesc C s | j | }t|D ]d}t| jD ]4}|d k r |jr |jd k r |j|jkr | |j¡ q t|drnt |
|
¡ q|
|
¡ q| ¡ d S )NÚ sync_engine) r r9 r r: Ú_poolr r8 r@ Úhasattrr ÚdisposerA )r r/ Zengr= Z proxy_refr r r Ú_drop_testing_enginesg s
|
ÿ
|
þ
|
|
z&ConnectionKiller._drop_testing_enginesc C s | d¡ d S )Nr, )rP r r r r Ú
|
after_testx s zConnectionKiller.after_testc C sT t js
|
d S |jjrd S | ¡ ddlm} t j ¡ }| |j
|
j |¡ W 5 Q R X d S rE ) r rG Ú __class__Z"__leave_connections_for_teardown__rC rH rF ÚdbÚconnectrI r. rJ )r ÚtestrF Úconnr r r Úafter_test_outside_fixtures{ s z,ConnectionKiller.after_test_outside_fixturesc C s | ¡ | d¡ | d¡ d S )Nr, r* )rC rP r r r r Ú stop_test_class_inside_fixtures s
|
z0ConnectionKiller.stop_test_class_inside_fixturesc C sB t jjr>t t jjr>tt jj}t jj ¡ ds>td| d S )NFz/%d connection recs not cleared after test suite)r ÚbaseZ _strong_ref_connection_recordsr ÚlenrA r- )r Úlnr r r Ú stop_test_class_outside_fixtures s ÿþz1ConnectionKiller.stop_test_class_outside_fixturesc C s" | ¡ | jD ]}| |¡ qd S r )rC r rP )r r/ r r r Ú final_cleanup s
|
z ConnectionKiller.final_cleanupc C s | j D ]}|jrdstqd S ©NF)r r: r- r< r r r Úassert_all_closed¤ s
|
z"ConnectionKiller.assert_all_closedN)Ú__name__Ú
|
__module__Ú __qualname__r r& r$ r% r0 r8 r> rC rD rI rP rQ rW rX r\ r] r_ r r r r r $ s" r c O s z| || W 5 t ¡ X d S r )Útesting_reaperr_ ©r6 ÚargsÚkwr r r Úassert_conns_closed s rg c O s z| || W 5 t ¡ X dS )zBDecorator that rolls back all open connections after fn execution.N)rc r> rd r r r Úrollback_open_connectionsµ s rh c O s t ¡ | || dS )z:Decorator that closes all connections before fn execution.N©rc rC rd r r r Ú close_first¿ s rj c O s z| || W 5 t ¡ X dS )z9Decorator that closes all connections after fn execution.Nri rd r r r Úclose_open_connectionsÇ s rk c c sX dd l m} |jD ]@}| r$|| kr$qt||d }|sHttd| j|}| ¡ V qd S )Nr zsqlalchemy.dialects.%s)Zsqlalchemy.dialectsZdialectsÚ__all__ÚgetattrÚ
|
__import__Údialect)ÚexcludeÚdÚnameÚmodr r r Ú all_dialectsÐ s
|
ÿrt c @ s> e Zd Zdd Zdd Zdd Zdd Zdd
|
d Zd d ZdS )ÚReconnectFixturec C s || _ g | _d| _d S r^ )ÚdbapiÚ connectionsÚ
|
is_stopped)r rv r r r r à s zReconnectFixture.__init__c C s t | j|S r )rm rv ©r Úkeyr r r Ú __getattr__å s z ReconnectFixture.__getattr__c O sT | j j||}| jr@| |j¡ | ¡ }| d¡ dsPtdn| j |¡ |S d S )Nzselect 1Fz%simulated connect failure didn't work)
|
rv rT rx r8 r" ÚcursorÚexecuter- rw Úappend)r re ÚkwargsrV Zcursr r r rT è s
|
zReconnectFixture.connectc
|
C s@ z
|
| W n0 t k
|
r: } zt d| ¡ W 5 d }~X Y nX d S )Nz.ReconnectFixture couldn't close connection: %sr1 r5 r r r r8 ö s
|
zReconnectFixture._safeFc C s, || _ t| jD ]}| |j¡ qg | _d S r )rx r9 rw r8 r" )r ÚstopÚcr r r Úshutdownü s zReconnectFixture.shutdownc C s
|
d| _ d S r^ )rx r r r r Úrestart s zReconnectFixture.restartN)F) r` ra rb r r{ rT r8 r r r r r r ru ß s
|
ru c sj | p
|
t jj} t jjj}|s i }t||d< t| |j fdd}jjj_ jjj
|
_ |_S )NÚmodulec s j j ¡ dj j_ d S r^ )ro rv r rx r ©Z_disposer. r r rO s
|
z$reconnecting_engine.<locals>.dispose) r rS rJ ro rv ru Útesting_enginerO r Z test_shutdownr Z test_restart)rJ Úoptionsrv rO r r
r Úreconnecting_engine s
|
|
r Fz Optional[URL]zOptional[Dict[str, Any]]zLiteral[False]Úboolr )rJ r ÚasyncioÚtransfer_staticpoolÚreturnc C s d S r r ©rJ r r r r r r r s r Tz Literal[True]r c C s d S r r r r r r r ) s c C sÊ |r|r t ddlm} n ddlm} ddlm} |sDd}d} d}
|
n$| dd¡}| d d¡} | d
|
d¡}
|
| prtj j
|
} || } |d kr¬tj d ks| j tj j
|
j kr¦tj }qÚi }n.tj d k rÚ| j tj j
|
j krÚtj ¡ } | |¡ || f|} |
|
r | jd kr t | d ¡d d } t | d¡dd }|rdddlm} tj d k rtttj j|rtd}| j tj j¡ n|rttj j| _| dkr|rd| j_nd| _t| jtjr´d| j_d| j_|rÆt | | ¡ | S )Nr )Úcreate_async_engine)Ú create_engine)Úmake_urlTr, FÚ
|
use_reaperr/ Úsqlite_savepointZsqliterT c S s
|
d | _ d S r )Zisolation_level)r? Zconnection_recordr r r Ú
|
do_connect] s z"testing_engine.<locals>.do_connectÚbeginc S s | d¡ d S )NZBEGIN)Zexec_driver_sql)rV r r r Údo_begina s z testing_engine.<locals>.do_begin)Ú
|
StaticPoolr+ ) r- Zsqlalchemy.ext.asyncior Ú
|
sqlalchemyr Zsqlalchemy.engine.urlr Úpopr rS rJ Z
|
drivernameZdb_optsÚcopyÚupdaterr r Z listens_forZsqlalchemy.poolr Ú
|
isinstancer Z_transfer_fromrL Z _has_eventsZ QueuePoolÚ_timeoutZ _max_overflowrc r0 )rJ r r r Z
|
share_poolZ_sqlite_savepointr r r r/ r Z default_optr. r r r r r r r 3 sZ
|
|
|
|
|
|
|
|
|
ÿ c sr ddl m} | stjj} g fdd} fdd}fdd}|| d |td
|
r\t _|_|_ 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_enginec s | ¡ d S r )r~ )ZsqlÚarf ©Úbufferr r Úexecutor s z mock_engine.<locals>.executorc s" dd D }|| ks t |d S )Nc S s g | ]}t d dt|¡qS )z[\n\t]rH )ÚreÚsubÚstr©Ú.0Úsr r r Ú
|
<listcomp> s z3mock_engine.<locals>.assert_sql.<locals>.<listcomp>)r- )ZstmtsÚrecvr r r Ú
|
assert_sql s z mock_engine.<locals>.assert_sqlc s j d fddjD ¡S )NÚ
|
c 3 s | ]}t |j d V qdS ))ro N)r¤ Úcompiler¥ ©rq r r Ú <genexpr> s z1mock_engine.<locals>.print_sql.<locals>.<genexpr>)ro ÚjoinÚmockr )r. r r Ú print_sql s z mock_engine.<locals>.print_sqlz://r° )
|
r r r rS rr rN r- r° rª r± )Z dialect_namer r¡ rª r± r )r r. r Ú mock_engine s r² c @ s: e Zd ZdZdd Zd ddZdd Zd d
|
Zd d ZdS )ÚDBAPIProxyCursorzuProxy a DBAPI cursor.
|
|
Tests can provide subclasses of this to intercept
|
DBAPI-level cursor operations.
|
|
c O s || _ || _|j||| _d S r )r. rK r| )r r. rV re r r r r r s zDBAPIProxyCursor.__init__Nc K s* |r| j j||f|S | j j|f|S d S r )r| r} )r ÚstmtÚ
|
parametersrf r r r r} ² s zDBAPIProxyCursor.executec K s | j j||f|S r )r| Ú executemany)r r´ Úparamsrf r r r r¶ ¸ s z DBAPIProxyCursor.executemanyc C s
|
t | jS r )Úiterr| r r r r Ú__iter__» s zDBAPIProxyCursor.__iter__c C s t | j|S r )rm r| ry r r r r{ ¾ s z DBAPIProxyCursor.__getattr__)N) r` ra rb Ú__doc__r r} r¶ r¹ r{ r r r r r³ ¥ s
|
r³ c @ s0 e Zd ZdZdd Zdd Zdd Zdd Zd
|
S ) ÚDBAPIProxyConnectionz}Proxy a DBAPI connection.
|
|
Tests can provide subclasses of this to intercept
|
DBAPI-level connection operations.
|
|
c C s || _ || _|| _d S r )rV r. Ú
|
cursor_cls)r r. rV r¼ r r r r Ê s z DBAPIProxyConnection.__init__c O s | j | j| jf||S r )r¼ r. rV )r re r r r r r| Ï s zDBAPIProxyConnection.cursorc C s | j ¡ d S r )rV r" r r r r r" Ò s zDBAPIProxyConnection.closec C s t | j|S r )rm rV ry r r r r{ Õ s z DBAPIProxyConnection.__getattr__N)r` ra rb rº r r| r" r{ r r r r r» Â s
|
r» )N)NN)NNFF)NNTF)NNFFFF)N))Ú
|
__future__r r r¢ Útypingr r r r3 r rH r Úutilr r r r r Z util.typingr Ú TYPE_CHECKINGr. r Z
|
engine.urlr Z ext.asyncior r rc rg rh rj rk rt ru r Úoverloadr r² r³ r» r r r r Ú<module>
|
sn
|
|
|
|
|
,
|
ü ü
|
ú
|
L
|
&
|
