BUJΖ d dl Z d dlZddlmZ ddlmZ ddlmZ ddlm Z ej
e Z dZ
G d d e Zd
Zd Zd Zd
Zej d Zej d Zej d Zej d Zej d Zej d Zej d Zd Z ej d Z ej d Z dS ) N )config)engines )url)compatc 6 e Zd Zd Zed Zd Zd ZdS )registerc i | _ d S Nfns)selfs p/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/sqlalchemy/testing/provision.py__init__zregister.__init__ s
c V t d | S N*)r
for_db)clsfns r initz
register.init s$ %xzz %%b)))r c fd}|S )Nc | j < S r r
)r dbnamer s r decoratez!register.for_db..decorate s !DHV Kr )r r r s `` r r zregister.for_db s) r c < t |t j rt j | }n)t |t j r|}n |j j }| }|| j v r | j | |g|R S | j d |g|R S r )
isinstancer string_typessa_urlmake_urlURLdbr get_backend_namer )r cfgargr backends r __call__zregister.__call__ s c6. / / /#&&CC
VZ
(
( CC&*C&&(( dh $48G$S/3/// / 48C=+s+++ +r N)__name__
__module__ __qualname__r classmethodr r r* r r r r
r
s\ * * [* , , , , ,r r
c t D ]>}t d| |j j t ||j | ?d S )NzCREATE database %s, URI %r)_configs_for_db_operationloginfor% r create_dbfollower_identr' s r create_follower_dbr6 + sS (** / / -~svzJJJ#sv~..../ /r c t j | }| |rt | | } i }t | | t
j | | }t | || |
t j
|||| }|rt || |S r )r" r# get_dialectload_provisioningfollower_url_from_mainupdate_db_optsr testing_enginepost_configure_engineconnectcloser Configr
configure_follower)db_urloptions file_configr5 dialectdb_optsengr' s r setup_configrH 1 s of%%1133G @'??G67###
1
1C&#~666KKMM
-
gw
D
DC 03/// Jr c t D ]>}t d| |j j t ||j | ?d S )NzDROP database %s, URI %r)r0 r1 r2 r% r drop_dbr4 s r drop_follower_dbrK F sS (** - - +^SVZHHHSV^,,,,- -r c # K t } t j D ]}|j t j D ]T}|j j }| }||j |j |j
f}|| vr|V | | Ut j D ]}|j d S r ) setr r@ all_configsr% disposer r& usernamehostdatabaseadd)hostsr' r r) host_confs r r0 r0 L s EEE}((** }((** ! !fj&&(( clCHclC E ! !III IIi }((** r c 0 t d|j z )zDynamically create a database for testing.
Used when a test run will employ multiple processes, e.g., when run
via `tox` or `pytest -n4`.
z"no DB creation routine for cfg: %sNotImplementedErrorr r' rG idents r r3 r3 _ s BSW L
M
MMr c 0 t d|j z )z8Drop a database that we dynamically created for testing.z no DB drop routine for cfg: %srW rY s r rJ rJ i s > H
I
IIr c dS )zCSet database options (db_opts) for a test database that we created.Nr )rB rF s r r; r; o Dr c dS )zPerform extra steps after configuring an engine for testing.
(For the internal dialects, currently only used by sqlite.)
Nr )r enginer5 s r r= r= u s Dr c < t j | } || _ | S )zCreate a connection URL for a dynamically-created test database.
:param url: the connection URL specified when the test run was invoked
:param ident: the pytest-xdist "worker identifier" to be used as the
database name
)r" r# rR r rZ s r r: r: ~ s /#
CCL Jr c dS )z@Create dialect-specific config settings for a follower database.Nr )r' rZ s r rA rA r] r c dS )a[ Remove databases that were created during the test process, after the
process has ended.
This is an optional step that is invoked for certain backends that do not
reliably release locks on the database as long as a process is still in
use. For the internal dialects, this is currently only necessary for
mssql and oracle.
Nr ra s r run_reap_dbsrd s Dr c t d t j t }t j t }i }t | 5 }|D ]}| }| d \ }}t j | }||vr1|
||< || | |j
f} || | || | d d d n # 1 swxY w Y |D ]5} t || d }
|| } t! |
| 6d S )NzReaping databases... r )r1 r2 collections defaultdictrM openstripsplitr" r# r8 r9 r& rQ rS listrd ) idents_fileurlsidentsdialectsfile_linedb_namerB url_objurl_keyr rZ s r reap_dbsrv s HH
#$$$ "3 ' 'D
$S
)
)FH
k
)e ) )D::<D;c 0 t d|j z )aO Specify keyword arguments for creating a temporary Table.
Dialect-specific implementations of this method will return the
kwargs that are passed to the Table method when creating a temporary
table for testing, e.g., in the define_temp_tables method of the
ComponentReflectionTest class in suite/test_reflection.py
z.no temp table keyword args routine for cfg: %srW )r' rG s r temp_table_keyword_argsrx s 837B r c d S r r )r r% testclss r stop_test_classr{ s Dr ) rg logging r r r_ r r" utilr getLoggerr+ r1 FOLLOWER_IDENTobjectr
r6 rH rK r0 r r3 rJ r; r= r: rA rd rv rx r{ r r r r s " " " " " " g!!, , , , ,v , , ,:/ / / *- - - &
N N N
J J J
! ! !2
r