neL d Z ddlmZ ddlZddlZddlZej dk Zer ddlZej Z nddl
Zddl Z G d de
Z dadadadadadadadZdadadadada e a e a da d Z d Z d Z!d
Z"d Z#d Z$d
Z%d Z&d Z'd Z(d Z)d Z*d Z+g Z,g Z-d Z.d Z/e.d Z0e.d Z1e.d Z2e/d Z3e/d Z4e/d Z5d Z6e/d Z7e/d Z8e/d Z9e/d Z:d! Z;d" Zd% Z?d& Z@d' ZAd( ZBd) ZCd* ZDd2d,ZEd- ZFd. ZG G d/ d0e ZHdaId1 ZJdS )3zTesting extensions.
this module is designed to work as a testing-framework-agnostic library,
created so that multiple test frameworks can be supported at once
(mostly so that we can migrate to new ones). The current target
is pytest.
)absolute_importN) r c e Zd Zej ZdS )ABCN)__name__
__module__ __qualname__abcABCMeta
__metaclass__ y/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/sqlalchemy/testing/plugin/plugin_base.pyr r " s
r r c r | dddt d | dddt d | dd dd
d | d
dt d | dd ddd | dddd | dddd | ddd d | d dd | d!dd"d# | d$dd%d&' | d(dd)d*d+, | d-ddt d. | d/dd0d*d1, | d2dt dd34 | d5dt dd64 | d7dd8d*d9, | d:dd;d*d<, d S )=Nz
--log-infocallbackstringz,turn on info logging for (multiple OK))actiontyper helpz --log-debugz-turn on debug logging for (multiple OK)z--dbappenddbzBUse prefab database uri. Multiple OK, first one is run by default.)r r destr z--dbszList available prefab dbs)r zeroarg_callbackr z--dburidburiz8Database uri. Multiple OK, first one is run by default.z --dropfirst
store_true dropfirstz,Drop all tables in the target database first)r r r z--backend-only backend_onlyzUse a low number of distinct connections - i.e. for Oracle TNSz--write-idents write_identszCwrite out generated follower idents to , when -n is used)r r r z --reversetop
reversetopFzTUse a random-ordering set implementation in the ORM (helps reveal dependency issues))r r defaultr z--requirementsz3requirements class for testing, overrides setup.cfgz--with-cdecimalcdecimalzDMonkeypatch the cdecimal library into Python 'decimal' for all testsz
--include-tagz Include tests with tag )r r r r z
--exclude-tagz Exclude tests with tag z--write-profileswrite_profilesz$Write/update failing profiling data.z--force-write-profilesforce_write_profilesz,Unconditionally write/update profiling data.)_log _list_dbs_requirements_opt _include_tag _exclude_tag) make_options r
setup_optionsr, ; s K
; K
< K
' K"
( K
J K
; K
K K
/ K!
9 K
K
K
+ K
"
B K
K
+ K
+ K
3 K
#
; r c ddl m} | |_ dS )zConfigure required state for a follower.
This invokes in the parent process and typically includes
database creation.
r provisionN)sqlalchemy.testingr/ FOLLOWER_IDENT)follower_identr/ s r configure_followerr3 s$ -,,,,, -I r c * t t d| d< dS )aS Store important configuration we will need to send to a follower.
This invokes in the parent process after normal config is set up.
This is necessary as pytest seems to not be using forking, so we
start with nothing in memory, *but* it isn't running our argparse
callables, so we have to just copy all of that over.
include_tags exclude_tagsmemoized_configNr5 dict_s r !memoize_important_follower_configr; s" %$ E
r c t | d d t | d d dS )zfRestore important configuration needed by a follower.
This invokes in the follower process.
r8 r6 r7 N)r6 updater7 r9 s r !restore_important_follower_configr> sH /0 @AAA /0 @AAAAAr c d t j at ddg d S )Nz setup.cfgztest.cfg) configparser ConfigParser file_configreadr
r r read_configrD s/ +--Kk:./////r c J | a t D ]} |t t dS )z7things to set up early, before coverage might be setup.N)options
pre_configurerB )optfns r pre_beginrJ s7 G ! !
7K ! !r c | t _ d S N)rF has_coverage)values r set_coverage_flagrO s Gr c t D ]} | t t ddlma ddlmamama ddlm a m
a
m a ddlm a m
a
ddlma t dS )z9things to set up later, once we know coverage is running.r testing)fixturesengines
exclusions)
assertionswarnings profiling)configr/ )utilN)post_configurerF rB
sqlalchemyrR r0 rS rT rU rV rW rX rY r/ rZ
setup_filtersrI s r
post_beginr_ s ! !
7K
#"""""@@@@@@@@@@BBBBBBBBBB44444444 r c t s dd l a t | d r9t | t j d S | d r9t | t j d S d S )Nr z-infoz-debug)logging basicConfigendswith getLoggersetLevelINFODEBUGopt_strrN parsers r r& r& s 9% ))',77777 ( # # 9% ))'-888889 9r c t d t t d D ]0}t |ddt d| 1t j d d S )Nz0Available --db options (use --dburi to override)r z>20 r )printsortedrB rF getsysexit)argsmacros r r' r' sv
<=== ++D1122 B B
EEEE;??4#?#?#?@AAAAHQKKKKKr c $ t | d S rL )_setup_requirementsrh s r r( r( s r c b t | dd d S N-_)r7 addreplacerh s r r* r* * U]]3,,-----r c b t | dd d S rw )r6 rz r{ rh s r r) r) ! r| r c : t | | S rL )rG r r^ s r prer ) s
Ir c : t | | S rL )r[ r r^ s r postr . s "
Ir c
| a d S rL rF rH rB s r _setup_optionsr 3 s
GGGr c L | j r t d d S d S )Nmemory_intensive)r r7 rz r s r
_set_nomemoryr 9 s1
| -+,,,,,- -r c > | j rdd l }|t j d< d S d S )Nr decimal)r# rp modules)rF rB r# s r _monkeypatch_cdecimalr ? s2 *!) I * *r c 8 ddl m} t |_ d S Nr rY )r0 rY _fixture_fn_class_fixture_functions)rF rB rY s r
_init_symbolsr G s( )))))) 1 3 3F r c ddl m} ddlm} ddl m} | j rt
| j }ng }| j rt| j D ]l}t j d| D ]T}||
d vrt d|z | |
d| Um|s)| |
dd d |_ |D ]}| j rW|j rPt# | j d 5 } | |j d
z |z d z d d d n # 1 swxY w Y | || ||j }
|j s|
|
| d S ) Nr r rQ r. z[,\s]+r z:Unknown URI specifier '%s'. Specify --dbs for known uris.r" a
)r0 rY r\ rR r/ r listr resplitrF RuntimeErrorr ro _currentr r1 openwrite setup_configset_as_current) rF rB rY rR r/ db_urlsdb_tokenr db_urlfile_cfgs r _engine_urir N s0 ))))))"""""",,,,,,} w}%%z >
> >Hhy(33
>
>[ 0 0 6 666&8:<=
NN;??4#<#<====
> 9{tY77888FO - - LI$< Lg*C00
LE I 4s :V Cd JKKK
L
L
L
L
L
L
L
L
L
L
L
L
L
L
L $$ G[)*B
- sG , , , - -s $D55D9 <D9 c P | dd }t | d S )N sqla_testingrequirement_cls)ro ru )rF rB r s r
_requirementsr v s, "oon6GHHO(((((r c * ddl m} ddlm} |j d S | d \ }}t
| }| d dd D ]}t || }t || } | x|_ |_ d S )Nr r rQ :. ) r0 rY r\ rR requirementsr
__import__getattrrequires)argumentrY rR modnameclsnamemod componentreq_clss r ru ru } s ))))))""""""
& ~~c**GW W
C ]]3''+ & & c9%%c7##G-4WYY6F'***r c Z ddl m}m} ddlm} ddlm}m} | j r
|j
D ]}|j } || } | }
|
D ]P} |
| | | | Qn# t" $ r Y nw xY w|j j | r} | d }
|
D ]R} |
| | | | d Sn# t" $ r Y nw xY w| || |j j | r | || |j ||d rgddlm} | d D ]K}
|
| | |
d
|
d Ld S d S )
Nr )rY rZ )against)schemainspect test_schema)r
postgresql)r *namer )r r ) r0 rY rZ sqlalchemy.testing.exclusionsr r\ r r r Config all_configsr get_view_namesexecute _DropViewTableMetaDataNotImplementedErrorr schemasenabled_for_configdrop_all_tablesr sqlalchemy.dialectsr get_enums DropEnumTypeENUM)rF rB rY rZ r r r r e inspector
view_namesvnamer enums r _prep_testing_databaser s ////////555555******** 0=,,.. / / CA
I
& 5 5 7 7
( EII (( "LL0A0A B B '
"*==cBB
!*!9!9 !9!O!OJ ", " , , & $)$*OO$5$5+8 !- !" !" + D
I . . ."*==cBB
K$$Q #/$JJJwsL))
:::::: % / / 4 4 DII"// &OO%)&\$x. , U0 0/ / s$ B66
CC&E
E E c 8 | j rddlm} | d S d S )Nr )randomize_unitofwork)r! sqlalchemy.orm.utilr )rF rB r s r _reverse_topologicalr s> <<<<<< r c 8 ddl m} t |_ ||_ d S r )r0 rY rF rB )rH rB rY s r _post_setup_optionsr s) )))))) FN$Fr c n ddl m} | | dd |_ d S )Nr )rX r profile_file)r0 rX ProfileStatsFilero _profile_stats)rF rB rX s r _setup_profilingr sC ,,,,,, ( 9 977 I r c t |t j sdS | d rdS t j j r$t |dd st |dd sdS dS )NFry __backend____sparse_backend__T)
issubclassrS TestBase
startswithrY rF r r )r clss r
want_classr sy c8, - - u
u#]E22 1599
utr c |j d sdS |j dS t rt | d oz+_possible_configs_for_cls.. s FKFN6! r )key) setrY r r __unsupported_on__rU db_specr remover rZ to_listr intersection_updater r r r matching_config_reasonsextendr! r rz
differencedifference_updatereversedrn r r values)r reasonsr r spec
config_objr requirementcheck skip_reasons
non_preferred per_dialectr r s r r r X s fm//1122K
/!3#9: {++ / /JtJ
/"":...sM4(( /!4<<#@#@A {++ / /J4
##
/"":...s($// B'')?(@AAAsN## * {++ J " /
k:: $ < z&_do_skips... s< C r z)+N)r r joinrU _server_versiondriver)rB r5 s r rD z _do_skips.. s J M&&&HH #-#=#=jm#L#L M((( r __prefer_backends__)r r rY skip_testr rE r r r r' rU r) rZ r+ rH rz r/ r0 r
_setup_configpop)r r3 r cmsgr9 r4 r5 s r r r s G+C99KsM5)) m,, Aqss
,/LLL!**E 9 9 LLL II #)-";";"="=
IIg
+ , , 9
!4<<0G#H#HI% . .J4
##
. !!*--- ! !- 0 0 9 ) )- 8 8 8
k))koo''----- *)r c P t j | t d S rL )rY r pushrR )r5 ctxs r rJ rJ s
OW-----r c e Zd Zej d Zej d Zej d Zej d Zd Z dS )FixtureFunctionsc t rL r selfargkws r skip_test_exceptionz$FixtureFunctions.skip_test_exception !###r c t rL rT rV rr rX s r combinationsz FixtureFunctions.combinations rZ r c t rL rT r\ s r param_identz FixtureFunctions.param_ident rZ r c t rL rT rU s r fixturezFixtureFunctions.fixture rZ r c t rL rT )rV s r get_current_test_namez&FixtureFunctions.get_current_test_name s !###r N)
r r r r
abstractmethodrY r] r_ ra rc r
r r rR rR s $ $ $ $ $ $ $ $ $ $ $ $$ $ $ $ $r rR c
| a d S rL )r )fixture_fn_classs r set_fixture_functionsrg s (r )NF)K__doc__
__future__r r
r rp version_infopy3kr@ r rA collectionscollections_abcobjectrS rT rU rW rX r/ rV r rY rR rZ rB ra r' r6 r7 rF r, r3 r; r> rD rJ rO r_ r& r' r( r* r) rG r[ r r r r r r r r ru r r r r r r r r r r r
r r r r r rJ rR r rg r
r r ro s4 ' & & & & &
6! $
'CC''''))))$ $ $ $ $f $ $ $
suu suu
| | |~ . . .
B B B0 0 0 ! ! !! ! ! $
9
9
9 . . .. . .
- - -
* * * 4 4 4 $- $- $-N ) ) ) 7 7 7& 5 5 5p % % % 6 4
# # # 2 2 2 " " "- - -> > > >B'. '. '.T. . .$ $ $ $ $s $ $ $* ) ) ) ) )r