f@L d dl Zn# e$ r ddlmZ Y nw xY wd dlZd dlZd dlmZ d dlZd dl Z d dl
Z
d dl Z d dl Z d dl
Z
d dlZ d dlZej rd dlmZ n # e$ r Y nw xY w d dlZdZn
# e$ r dZY nw xY wd Zd Zd
Zd Zer
d dlZd Zd
Zd Zd Z da d Z d Z d Z d Z!d Z"d Z#d Z$d Z%d Z& G d dej' Z(dS ) N ) plugin_base)update_wrapper)SequenceTFc | d fd}t j | t j d S )N
sqlalchemyc | dd r G fddt j |d< | dd r G fddt j |d< j | fi | d S )Ncallbackc e Zd Z d fd ZdS )=pytest_addoption..make_option..CallableActionNc " ||| d S N )selfparser namespacevalues
option_string callback_s z/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/sqlalchemy/testing/plugin/pytestplugin.py__call__zFpytest_addoption..make_option..CallableAction.__call__, s ImVV<<<<< r )__name__
__module__ __qualname__r )r s r CallableActionr + s6 CG= = = = = = = =r r actionzeroarg_callbackc 8 e Zd Z d fd Z dfd Z xZS )r FNc \ t | ||dd||| d S )Nr T)option_stringsdestnargsconstdefaultrequiredhelp)super__init__)r r! r" r% r& r' r __class__s r r) zFpytest_addoption..make_option..CallableAction.__init__7 sK .$//88'5 ! " '!) ! 9 r c " ||| d S r r )r r r r r r s r r zFpytest_addoption..make_option..CallableAction.__call__I s %$]FFCCCCCr )FFNr )r r r r) r
__classcell__)r* r r s @r r z=pytest_addoption..make_option..CallableAction6 s}
" "
& DHD D D D D D D D D Dr )popargparseAction addoption)namekwr r r groups @@@r make_optionz%pytest_addoption..make_option' s FF:t,, *
=
=
=
=
=
=
=
=
=
= *BxL 66"4d;; *
D
D
D
D
D
D
D
D
D
D
D0 *BxL#######r )getgroupr
setup_options read_config)r r4 r3 s @r pytest_addoptionr8 $ sV OOL ) )E)$ )$ )$ )$ )$V k*** r c 4 t j d t | d r9t j | j t j | j d nS| j j rGt j
| j j r t j | j j t j
| j t j t t! | j dd t j t$ d S )Nz sqlalchemy.testing.assertions workerinputfollower_ident
cov_sourceF)pytestregister_assert_rewritehasattrr !restore_important_follower_configr: configure_followeroption write_identsospathexistsremove pre_beginset_coverage_flagboolgetattrset_fixture_functionsPytestFixtureFunctions)configs r pytest_configurerO V s
"#BCCCv}%% 25f6HIII&v'9:J'KLLLL = % 2"'.. M &+
+
2
Ifm0 1 1 1&-(((! WV]L%
8
899 %&<=====r c , t j d S r )r
post_beginsessions r pytest_sessionstartrT k s r c , t j d S r )r final_process_cleanuprR s r pytest_sessionfinishrW o s %'''''r c t j | j dt j j dd z | j d< ddlm} | | j d d S )Nztest_%sr r; provision) r !memoize_important_follower_configr: uuiduuid4hexsqlalchemy.testingr[ create_follower_db)noder[ s r pytest_configure_noderc v sr 5d6FGGG-6 9I!B$9O-O)*000000$$T%56F%GHHHHHr c R ddl m} | | j d d S )Nr rZ r; )r` r[ drop_follower_dbr: )rb errorr[ s r pytest_testnodedownrg s7 000000""4#34D#EFFFFFr c t j d }d |D |d d < t d |D }|D ]}t j |j |j j D ]}||j ur||j }t t j
dt j
} ||j |j j D ]9} | D ]"}
||
j
|
#:g } |D ]V} | j j |v r1| || j j | j
A| | Wt! | d |d d < d S )Nc 4 t j t S r ) collections defaultdictlistr r r z/pytest_collection_modifyitems.. s '-- r c g | ]G}t |j t j !|j j j d E|HS )_)
isinstanceparentr= Instancer1
startswith.0items r
z1pytest_collection_modifyitems.. s\ dk6? 3 3 "'22377 r c 3 $ K | ] }|j V d S r )rq rt s r z0pytest_collection_modifyitems.. s$ 55tt{555555r from_parentr1 rq c \ | j j j j | j j j | j fS r )rq r1 rv s r rm z/pytest_collection_modifyitems.. s' K % * K # I
r )key)rj rk setr generate_sub_testsclsrq modulerK r= Classr collectr1 appendextendsorted)
rS rN items
rebuilt_items test_classes
test_classsub_cls per_cls_dictctorinsttnewitemsrv s
r pytest_collection_modifyitemsr s +-- M E!!!H 55u55555L"
7
7
"5 NJ - 4
7 7G jn,, ,Z^ < v|]FLII D )*2C2J '))7 7D "\\^^ 7 7$QV,33A66667 7 H " " ;?m + + OOM$+/ :49 E F F F F OOD ! ! ! !
E!!!HHHr c ~ t j | r\t j || rGt t
j dt
j fdt j | D S t j | r6t t
j r t j j
| rd S g S )Nrz c 4 g | ]} |j S )r{ )r )ru parametrize_cls collectorr s r rw z-pytest_pycollect_makeitem.. s;
Do.y A A A
r )inspectisclassr
want_classrK r= r _parametrize_clsr
isfunctionrp rr want_methodr )r r1 objr s ` @r pytest_pycollect_makeitemr s s 6tS A A v|]FLAA
#3I4Dc#J#J
3
y&/ 2 2
#IM3 7 7
t r c d|j vr|gS |j }g }t j d |D D ]}i }t d |D | D ]K\ }}|st d t
j d| }t ||j D ]
\ } }
|
|| < Ld d |D } |j
d| } t t | |f| }
t | | |
| |
|S )z6implement a class-based version of pytest parametrize._sa_parametrizec g | ]\ }}|S r r )ru argnameparamss r rw z$_parametrize_cls.. s 8 8 8_Wf& 8 8 8r c g | ]
}|d S )r r )ru _sa_params r rw z$_parametrize_cls.. s ; ; ;iYq\ ; ; ;r z*need argnames for class-based combinationsz,\s*ro c 3 K | ]D}|j d D ]'}t t j dd| V (EdS )-z\W N)idsplitstrresub)ru paramtokens r ry z#_parametrize_cls.. ss %
%
,, %
%
ub%(( ) )%
%
%
%
%
%
%
r )__dict__r itertoolsproductzip TypeErrorr r r joinr type__new__setattrr )r r r classesfull_param_set
cls_variablesr r
argname_splitargvalparametrized_namer1 newclss r r r sm ,,u )OG#+ 8 8 8 8 8
! ; ;? ; ; ;^
) )NGU
N LMMM HWg 6 6M
u| < <
)
)S%(
c""
) HH %
%
( %
%
%
,,,(9(9: dD3&-@@f%%%v Nr c t t j sd S t It j j j j a fd} j j | t d S )Nc > t j j d ad S r )class_teardownrq _current_classr} s r finalizez&pytest_runtest_setup..finalize s 4;- . . . !NNNr )rp r= Functionr class_setuprq addfinalizer
test_setup)rv r s ` r pytest_runtest_setupr s dFO , , DK&''' +
" " " " "
''111tr c $ t | d S r )
test_teardownr} s r pytest_runtest_teardownr s
$r c p t j | | j j j | j j | j d S r )r before_testrq r r r r1 r} s r r r ! s8 dk )4;?DI r c . t j | d S r )r
after_testr} s r r r ' s 4 r c 8 t j | j d S r )r start_test_classr r} s r r r + s *****r c 8 t j | j d S r )r stop_test_classr r} s r r r / s )))))r c | t j j dk rt j | S t j | S )N )sys version_infomajorr getfullargspec
getargspec)fns r r r 3 s5
""%b)))!"%%%r c > ddl m ddlm d d fd }|S )z:Port of langhelpers.decorator with pytest-specific tricks.r )format_argspec_plus)inspect_getfullargspecc 2 t | | || S r )exec)codeenvfn_names r _exec_code_in_envz/_pytest_fn_decorator.._exec_code_in_env@ s T37|r r c | }|r|j | t dd| j }| |d d|z } | | d| j }|s2t | d| j |_ | |_ t || S | j |_ | j |_ |S ) Ntargetr )r r r1 F)groupedzDdef %(name)s(%(args)s):
return %(target)s(%(fn)s, %(apply_kw)s)
)r r im_func)
argsr dictr updaterK __defaults__ __wrapped__r r )
r add_positional_parametersspecmetadatar decoratedr r r r s
r decoratez&_pytest_fn_decorator..decorateD s %%b)) $ 8 I 6 7 7 7xDr{CCC++D%@@@AAA
&% V2..
)
%,RB%?%?%LI "$&I !!)R00 0
$&=I !#I r )r )sqlalchemy.util.langhelpersr sqlalchemy.util.compatr )r r r r r s ` @@@r _pytest_fn_decoratorr : sp @?????====== < Or c > e Zd Zd Zd eed dZd Zd Zd Z d Z
d S )
rM c . t j j |i |S r )r= skip Exceptionr r r2 s r skip_test_exceptionz*PytestFixtureFunctions.skip_test_exceptionf s {$c0R000r c | S r r r s r rm z PytestFixtureFunctions.j s r c X t | d r| j nt | j S )Nr )r? r r r s r rm z PytestFixtureFunctions.m s+ 3
# #
#YY
r )irsnc J
ddl m t j j dk r?t | dk r+t
|d d rt |d }n>t | dk r+t
|d d rt |d }| dd fd }| d
d }g }d |r| j
t j dgd t | D R }
fd
t | D }|D ]vt t sf | \ }} || dd }
| rd| |
| d fd|D f wnG|D ]Dt t sf | \ }} | rd| || df Eg |D ]4\ }
} }r|
| fz
}
t# j |
d|i} | 5fd} | S )a Facade for pytest.mark.parametrize.
Automatically derives argument names from the callable which in our
case is always a method on a class with positional arguments.
ids for parameter sets are derived using an optional template.
r )
exclusionsr r __next__nextargnamesNc g }g }| D ]B}t |j r| | -| | C||fS r )rp compoundr )r resultgathered_exclusionsar s r _filter_exclusionsz?PytestFixtureFunctions.combinations.._filter_exclusions sj F"$
%
% a!455 %'..q1111MM!$$$$.. .r id_Fc g | ] \ }}|d v | S ))r r r r r )ru idxchars r rw z7PytestFixtureFunctions.combinations.. s2 !T 333 333r c V g | ]%\ }}|v t j | | f&S r )operator
itemgetter)ru r r _combination_id_fnss r rw z7PytestFixtureFunctions.combinations.. sI C... $S))+>t+DE...r Tr c 3 F K | ]\ }} | | V d S r r )ru gettercomb_fnr s r ry z6PytestFixtureFunctions.combinations.. sJ ! !5DVWGGFF3KK 0 0! ! ! ! ! !r r c t j | r?rt d d| j vrg | _ | j f | S t
| j dd }nt j d }r(|dgz
}t fd }d } || d } t j
| | S )
Nz5exclusions not supported for class level combinationsr r z, * _exclusionsc |d }|r' j | } || } | |dd i |S )Nr )r add)r r r2 r exlur s r check_exclusionszOPytestFixtureFunctions.combinations..decorate..check_exclusions sY &*2h & *#<:#6#6#8#8#.decorate..process_metadata s (( 77777r )r )r )r r NotImplementedErrorr r r r r r r r r= mark parametrize)r _argnamesr! r# r r has_exclusions
pytest_paramss r r z5PytestFixtureFunctions.combinations..decorate s* r"" #
M! -O %BK77)+B&"))8]*CDDD # *2 3ABB 7II " ! !I " - 0I)5 5 5 5 *)58 8 8 *)6F B {..y-HHLLLr )r` r r r r lenr? rl r- r r r enumeraterp tupler r r= r )r arg_setsr2 r r tobuild_pytest_params _arg_getterfns fn_paramsparam_exclusions
parametersr r r r r r r( r) s @@@@@@r combinationsz#PytestFixtureFunctions.combinationsr s; 211111 !Q & &8}}!!ghqk:&F&F! ,,8}}!!ghqk6&B&B! ,,66*d++ / / / / / ffUD!! " 6 "&":
#- %.s^^ K !*3 C
!#u-- ! &C.@.@.E.E+ + ([ 3 3ABB 7
# *%)N%,,"( ! ! ! !HK! ! !
.
!#u-- ! &C.@.@.E.E+ +# *%)N%,, 0$7
1F ( ( -J(#
2 / 11
L*555E ' ' ' '$ M $ M $ M $ M $ M $ M $ M $ ML r c D |d }t j |dd d|iS )Nr r r )r= r )r r3 idents r param_identz"PytestFixtureFunctions.param_ident s( 1
|Z^6666r c $ t j |i |S r )r= fixturer s r r9 z PytestFixtureFunctions.fixture s ~s)b)))r c @ t j d S )NPYTEST_CURRENT_TEST)rD environget)r s r get_current_test_namez,PytestFixtureFunctions.get_current_test_name s z~~ 3444r N) r r r r reprr r r4 r7 r9 r> r r r rM rM e s 1 1 1 _
I I IV7 7 7* * *5 5 5 5 5r rM ))sqla_plugin_baser ImportErrorr r. rj functoolsr r r r rD r r r= typing
TYPE_CHECKINGr xdist has_xdistr8 rO rT rW r] rc rg r r r r r r r r r r r r FixtureFunctionsrM r r r rH s ***** $ $ $ $ $ $
$MMM $###### D LLLII III/ / / d> > >* ( ( (
GKKK I I IG G G 6 6 6r 0 F 6 ! ! !+ + +* * *& & &( ( (V`5 `5 `5 `5 `5[ 9 `5 `5 `5 `5 `5s0 A A A !A( (A21A2