U
|
¸ý°d¾m ã @ s d dl mZ d dlZd dlZd dlmZ d dlZd dlZd dlZd dl Z d dl
|
Z
|
d dl Z d dl Z d dl Z z d dlZW n ek
|
r ddlmZ Y nX dd Zdd d
|
d Zd ae jd ddd Zdd Zdd Zdd Zdd ZG dd dZdd Z d d Z d d Z d9d!d"Z d#d$ Z da!d%d& Z"e j#d d'd(d) Z$d*d+ Z%da&d,d- Z'e jd.d/d0d1 Z(e jd2d/d3d4 Z)d5d6 Z*G d7d8 d8ej+Z,dS ):é )Ú annotationsN)Úupdate_wrapperé )Ú plugin_basec s$ | d¡ fdd}t |¡ d S )NÚ
|
sqlalchemyc sn | dd ¡ r.G fdddtj}||d< | dd ¡r\G fdddtj}||d< j| f| d S )NÚcallbackc s e Zd Zd fdd ZdS )ú=pytest_addoption.<locals>.make_option.<locals>.CallableActionNc s ||| d S ©N© ©ÚselfÚparserÚ namespaceÚvaluesZ option_string©Ú callback_r
|
ú]d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\sqlalchemy/testing/plugin/pytestplugin.pyÚ__call__" s úFpytest_addoption.<locals>.make_option.<locals>.CallableAction.__call__)N)Ú__name__Ú
|
__module__Ú __qualname__r r
|
r r
|
r ÚCallableAction! s ÿr ÚactionÚzeroarg_callbackc s, e Zd Zd fdd Zdfdd Z ZS ) r FNc s t j||dd|||d d S )Nr T)Úoption_stringsÚdestÚnargsÚconstÚdefaultÚrequiredÚhelp)ÚsuperÚ__init__)r r r r r r! ©Ú __class__r
|
r r# - s ùzFpytest_addoption.<locals>.make_option.<locals>.CallableAction.__init__c s ||| d S r r
|
r ©r r
|
r r ? s r )FFN)N)r r r r# r Ú __classcell__r
|
r& r$ r r , s ú ÿ)ÚpopÚargparseZActionZ addoption)ÚnameÚkwr ©Úgroup)r r r Ú make_option s z%pytest_addoption.<locals>.make_option)Zgetgroupr Z setup_options)r r. r
|
r, r Úpytest_addoption s
|
+r/ z pytest.Config©Úconfigc C s t | j¡ t jst jr`d tt jdd t jD ¡}| jjrX| j jd| 7 _n|| j_| j
|
d¡rz| j t ¡ t | dr¢t | j¡ t | jd ¡ n&| jjrÈtj | jj¡rÈt | jj¡ t | j¡ t tt| jdd¡ t t¡ | jj rd a d S )
|
Nz and c S s g | ]}d | qS )znot r
|
)Ú.0Útagr
|
r
|
r Ú
|
<listcomp>P s z$pytest_configure.<locals>.<listcomp>ZxdistÚ workerinputÚfollower_identZ
|
cov_sourceFT) r Z read_configZrootpathÚ exclude_tagsZ include_tagsÚjoinÚlistÚoptionZmarkexprÚ pluginmanagerZ haspluginÚregisterÚ
|
XDistHooksÚhasattrZ!restore_important_follower_configr5 Zconfigure_followerZ write_identsÚosÚpathÚexistsÚremoveZ pre_beginZset_coverage_flagÚboolÚgetattrZset_fixture_functionsÚPytestFixtureFunctionsÚdump_pyannotateÚDUMP_PYANNOTATE)r1 Znew_exprr
|
r
|
r Úpytest_configureK s4 ÿÿ
|
ÿ ÿ
|
|
rH FT)Zautousec c s. t rddlm} | ¡ d V t r*| ¡ d S )Nr ©Ú collect_types)rG Úpyannotate_runtimerJ ÚstartÚstoprI r
|
r
|
r Úcollect_types_fixturet s rN c C sì dd l }ddl m} ddlm} ddlm} d}tjjr<dnd}d| d | d
|
d |j g}| rdd l
|
m } | d |j d¡ n| d| ¡ | j j d¡} | rÊ| d|¡ |D ]}
|
| |
|
¡ q¸n td|d |D ] }
|
t|
|
qÚd S )Nr )Ú __version__)Úhas_compiled_ext)Ú_CYEXTENSION_MSGzsqlalchemy installationz no user sitezuser site loadedz SQLAlchemy z (ú)zPath: )Úutilz!compiled extension enabled, e.g. ú z compiled extension not enabled; Zterminalreporterú=z=========================)r rO Zsqlalchemy.utilrP Zsqlalchemy.util._has_cyrQ ÚsysÚflagsÚ no_user_siteÚ__file__Zsqlalchemy.cyextensionrS Úappendr1 r; Z
|
get_pluginZ write_sepZ
|
write_lineÚprint) Úsessionr rO rP rQ ZgreetÚsiteZmsgsrS ZpmÚmr
|
r
|
r Ú_log_sqlalchemy_info s*
|
þ r_ c C s$ ddl m} t| | tj¡ d S ©Nr ©Úasyncio)Úsqlalchemy.testingrb r_ Z _assume_asyncr Z
|
post_begin)r\ rb r
|
r
|
r Úpytest_sessionstart s rd c C sB ddl m} | tj¡ | jjjr>ddlm } |
|
| jjj¡ d S )Nr ra rI ) rc rb Ú_maybe_async_provisioningr Zfinal_process_cleanupr1 r: rF rK rJ Z
|
dump_stats)r\ rb rJ r
|
r
|
r Úpytest_sessionfinish¦ s
|
|
rf c s> | j jjr:ddlm} tj d¡ fdd}|j|d d S )Nr rI úlib/sqlalchemyc s< t j t j | ¡¡} dt j | g¡kr,d S d| kr8d S | S )Nrg Ztesting)r? r@ ÚnormpathÚabspathÚ
|
commonpath)Úfilename©Zlib_sqlalchemyr
|
r Ú_filter· s ÿz)pytest_collection_finish.<locals>._filter)Zfilter_filename) r1 r: rF rK rJ r? r@ ri Zinit_types_collection)r\ rJ rm r
|
rl r Úpytest_collection_finish± s
|
|
rn c @ s e Zd Zdd Zdd ZdS )r= c C sX ddl m} ddl m} t |j¡ dt ¡ jdd
|jd< | |j
|
|jd ¡ d S )Nr ©Ú provisionra ztest_%sé r6 ) rc rp rb r Z!memoize_important_follower_configr5 ÚuuidÚuuid4Úhexre Zcreate_follower_db)r Únoderp rb r
|
r
|
r Úpytest_configure_nodeÆ s ÿz XDistHooks.pytest_configure_nodec C s0 ddl m} ddl m} | |j|jd ¡ d S )Nr ro ra r6 )rc rp rb re Zdrop_follower_dbr5 )r ru Úerrorrp rb r
|
r
|
r Úpytest_testnodedownÕ s ÿz XDistHooks.pytest_testnodedownN)r r r rv rx r
|
r
|
r
|
r r= Å s r= c s¼ ddl m} t dd ¡dd |D |d d
< dd |D fd d
|
fd d }| |¡ g }|D ]2}|j}|kr| | |j ¡ ql| |¡ qlt |d d d|d d
< d S )Nr ra c S s
|
t t¡S r )Ú collectionsÚ defaultdictr9 r
|
r
|
r
|
r Ú<lambda>í ó z/pytest_collection_modifyitems.<locals>.<lambda>c S s4 g | ],}| tj¡d k r| tj¡j d¡s|qS )NÚ_)Ú getparentÚpytestÚClassr* Ú
|
startswith©r2 Úitemr
|
r
|
r r4 ð s ýz1pytest_collection_modifyitems.<locals>.<listcomp>c S s h | ]}| tj¡qS r
|
)r~ r r r r
|
r
|
r Ú <setcomp>÷ s z0pytest_collection_modifyitems.<locals>.<setcomp>c 3 s4 | ¡ D ]&}t|tjr( |E d H q|V qd S r )ÚcollectÚ
|
isinstancer Z Collector)ÚelementZ
|
inst_or_fn)r
r
|
r r
ù s z.pytest_collection_modifyitems.<locals>.collectc
|
s
|
D ]þ} t }t| jdds,t| jddr4dh}n t| jddrJdh}nt }dd | ¡ D }|| }| |¡}|D ]}| |¡ qxt | j| j |¡D ]h}|| jk r| j }|
|
t j ¡}t j j|j|d }|D ]}| |¡ qÔ |D ]} || j | ¡ qìqqd S )
|
NZ __backend__FZ __only_on__ÚbackendZ__sparse_backend__Zsparse_backendc S s h | ]
|
}|j qS r
|
)r* )r2 Úmarkr
|
r
|
r r s zLpytest_collection_modifyitems.<locals>.setup_test_classes.<locals>.<setcomp>©r* Úparent)ÚsetrD ÚclsÚ frozensetZ iter_markersÚunionZ
|
add_markerr Zgenerate_sub_testsÚmoduler~ r ÚModuler Ú from_parentr r* rZ )
|
Z
|
test_classZ add_markersZexisting_markersZ all_markersÚmarkerZsub_clsZ per_cls_dictr Znew_clsÚfn©r
Z rebuilt_itemsZ test_classesr
|
r Úsetup_test_classes sD
|
ÿÿ
|
ÿ
|
|
ÿ z9pytest_collection_modifyitems.<locals>.setup_test_classesc S s | tj¡j| tj¡j| jfS r )r~ r r r* r )r r
|
r
|
r r{ 9 s ý)Úkey)
|
rc rb ry rz re r Úextendr* rZ Úsorted)r\ r1 Úitemsrb r Znewitemsr Zcls_r
|
r r Ú pytest_collection_modifyitemsÞ s* ÿþ +
|
þr c sx t |¡rJt ||¡rJddlm} |jr0t|} fddt j |D S t
|
|¡rp j d k rpt j |¡rpd S g S d S )Nr r0 c s g | ]}t jj|j d qS )r )r r r r )r2 Zparametrize_cls©Ú collectorr
|
r r4 H s
|
ý ÿz-pytest_pycollect_makeitem.<locals>.<listcomp>) ÚinspectÚisclassr Z
|
want_classrc r1 Ú any_asyncÚ_apply_maybe_asyncÚ_parametrize_clsr Ú
|
isfunctionr Z want_method)r r* Úobjr1 r
|
r r Úpytest_pycollect_makeitemA s
|
|
üÿþ ýr¥ c C s t | dr| j} q t | ¡S )NÚ __wrapped__)r> r¦ r Úiscoroutinefunction©r r
|
r
|
r Ú _is_wrapped_coroutine_function[ s
|
r© c sÐ ddl m t| ¡ D ]\}}t|s2t|trt|dds| d¡rt |sd}t|trh|j
|
}d}t fdd}||}|rt|}d|_ t | || q|rÌ| ¡ d d
D ]}|tkr´t|d q´| S )
|
Nr ra Ú_maybe_async_appliedFZtest_Tc s j | f||S r )Ú _maybe_async©r ÚargsÚkwargsra r
|
r Ú
|
make_asyncq s z&_apply_maybe_async.<locals>.make_asyncr )rc rb Úvarsr Úcallabler Ú classmethodrD r r© Ú__func__Ú_pytest_fn_decoratorrª ÚsetattrÚmroÚobjectr¡ )r¤ Zrecurser* ÚvalueZis_classmethodr¯ Zdo_asyncr r
|
ra r r¡ b s8 ÿÿ
|
þýü
|
r¡ c C sÚ d|j kr|gS |j}g }tjdd |D D ]¦}i }tdd |D |D ]>\}}|s^tdt d|¡}t||jD ]\} }
|
|
|
|| < qvqJd dd |D ¡} d
|
|j
|
| f } t t | |f|¡} t | | | | | ¡ q.|S ) z6implement a class-based version of pytest parametrize.Ú_sa_parametrizec S s g | ] \}}|qS r
|
r
|
)r2 ÚargnameÚparamsr
|
r
|
r r4 s z$_parametrize_cls.<locals>.<listcomp>c S s g | ] }|d qS )r r
|
)r2 Z _sa_paramr
|
r
|
r r4 s z*need argnames for class-based combinationsz,\s*r} c s s. | ]&}|j d ¡D ]}t dd|¡V qqdS )ú-z\WÚ N)ÚidÚsplitÚreÚsub)r2 ÚparamÚtokenr
|
r
|
r Ú <genexpr> s þz#_parametrize_cls.<locals>.<genexpr>z%s_%s)Ú__dict__r¹ Ú itertoolsÚproductÚzipÚ TypeErrorrÀ r¿ r r8 r ÚtypeÚ__new__rµ rZ )r r r¹ ÚclassesZfull_param_setZ cls_variablesrº r Z argname_splitÚargÚvalZparametrized_namer* Znewclsr
|
r
|
r r¢ s2
|
ÿ ÿ
|
þ r¢ c C s@ ddl m} t| tjr<td kr<| tj| j ¡ |
|
tj ¡ad S r` ) rc rb r r ZFunctionÚ_current_classre r Z!start_test_class_outside_fixturesr r~ r ©r rb r
|
r
|
r Úpytest_runtest_setup¦ s
|
þrÑ )Z hookwrapperc
|
c s ddl m} | tj| ¡ d V td k r|d ksD| tj¡tk rd az´z|
|
tj | j ¡ W n t k
|
rø } z|t jræ|jsdt j f|_nZ|jd rÒt|jd trÒt|j}|d dt j 7 < t||_n| jdt jf7 _ W 5 d }~X Y nX W 5 d a X d S )Nr ra z__Original test failure__:
|
éÿÿÿÿz
|
__Original test failure__:
|
z__Original test failure__)rc rb r« r Z
|
after_testrÏ r~ r r Ú_current_reportre Z stop_test_class_outside_fixturesr Ú ExceptionÚfailedr Z longreprtextr Ústrr9 Útuple)r Únextitemrb Úer r
|
r
|
r Úpytest_runtest_teardown¹ sF
|
þý ÿÿÿ
|
ÿÿ þrÚ c C s, ddl m} | tj| | jj| j| j¡ d S r` ) rc rb r« r Z before_testr r r r* rÐ r
|
r
|
r Úpytest_runtest_callï s ûrÛ c C s | j dkr| ad S )NÚcall)ÚwhenrÓ )Úreportr
|
r
|
r Úpytest_runtest_logreport s
|
rß Úclass)Zscopec c sV ddl m} | j}t|dr(| |j¡ d V t|drD| |j¡ | tj|¡ d S )Nr ra Úsetup_test_classÚteardown_test_class) rc rb r r> r« rá râ r Zstop_test_class)Úrequestrb r r
|
r
|
r Úsetup_class_methods s
|
|
rä Úfunctionc c s ddl m} | j}t|dr(| |j¡ t|dr>| |j¡ d V | tj|¡ t|drh| |j ¡ t|dr~| |j
|
¡ d S )Nr ra Ú
|
setup_testÚsetUpÚtearDownÚ teardown_test) rc rb Úinstancer> r« ræ rç r Zafter_test_fixturesrè ré )rã rb r r
|
r
|
r Úsetup_test_methods s
|
|
|
|
|
|
rë c s8 ddl m ddlm dd d fdd }|S )
|
z:Port of langhelpers.decorator with pytest-specific tricks.r )Úformat_argspec_plus)Úinspect_getfullargspecc S s t | | || S r )Úexec)ÚcodeÚenvÚfn_namer
|
r
|
r Ú_exec_code_in_env] s
|
z/_pytest_fn_decorator.<locals>._exec_code_in_envr
|
c s¤ | }|r|j |¡ tdd| jd}| |dd¡ d| } || d| j}|szt| d| j|_| |_t|| S | j |_ | j|_t
|
| d r| j |_ |S d S )
|
NÚ __target_fnÚ __orig_fn)ró rô r* F)ZgroupedzVdef %(name)s%(grouped_args)s:
|
return %(__target_fn)s(%(__orig_fn)s, %(apply_kw)s)
|
)ró rô r³ Ú
|
pytestmark) r r Údictr ÚupdaterD Ú __defaults__r¦ r r r> rõ )r Úadd_positional_parametersÚspecÚmetadatarï Z decorated©rò rì rí Útargetr
|
r Údecoratee s6 ÿüÿ ÿ
|
|
z&_pytest_fn_decorator.<locals>.decorate)r
|
)Zsqlalchemy.util.langhelpersrì Zsqlalchemy.util.compatrí )rý rþ r
|
rü r r´ W s
|
"r´ c @ sf e Zd Zdd Zedd Zdd Zdd eed d d
|
Z d d Z
|
d d Z dd Z dd Z dd ZdS )rE c O s t jj||S r )r ÚskiprÔ )r rÍ r+ r
|
r
|
r Úskip_test_exception s z*PytestFixtureFunctions.skip_test_exceptionc C s t jS r )r r ©r r
|
r
|
r Ú add_to_marker s z$PytestFixtureFunctions.add_to_markerc C s t j dd¡S )Nrä rë )r r Z usefixturesr r
|
r
|
r Úmark_base_test_class s ÿz+PytestFixtureFunctions.mark_base_test_classc C s | S r r
|
©r¤ r
|
r
|
r r{ r| z PytestFixtureFunctions.<lambda>c C s t | dr| jS t| jS )Nr )r> r rÊ r r
|
r
|
r r{ s ÿ)ÚiÚrÚsÚnc
|
s ddl m t|dkr2t|d dr2t|d }| dd¡fdd}| d d¡}g }d
|
|r | j tjdd d t |D } fd d t |D }|D ]\t
|
t s´f|\}} ||dd
}
|
| rØd| |
|
| d fdd|D ¡f¡ q nD|D ]>t
|
t sf|\}} | r0d| || df¡ qg |D ]6\}
|
} }rf|
|
| f7 }
|
tj|
|
d|i} | ¡ qLfdd} | 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 Ú__next__ÚargnamesNc s< g }g }| D ]&}t | jr(| |¡ q | |¡ q ||fS r )r ÚcompoundrZ )r ÚresultZgathered_exclusionsÚar r
|
r Ú_filter_exclusions° s z?PytestFixtureFunctions.combinations.<locals>._filter_exclusionsÚid_Fc S s g | ]\}}|d kr|qS ))r r r r r
|
©r2 ÚidxÚcharr
|
r
|
r r4 Ç s þz7PytestFixtureFunctions.combinations.<locals>.<listcomp>c s* g | ]"\}}| krt |¡ | fqS r
|
)ÚoperatorÚ
|
itemgetterr )Ú_combination_id_fnsr
|
r r4 Í s þTr¼ c 3 s | ]\}}|| V qd S r r
|
)r2 ÚgetterZcomb_fn)rÍ r
|
r rÄ â s z6PytestFixtureFunctions.combinations.<locals>.<genexpr>r¾ c sÆ t | ¡r:rtdd| jkr&g | _| j f¡ | S t | ¡jdd
} d kr\|}n t d ¡}r°t
|
dd |D }d| }||g7 }t fdd }|| |fd
|
} t j |¡| S d S ) Nz5exclusions not supported for class level combinationsr¹ r z, *c s s | ]}| d ¡rdV qdS )Ú _exclusionsr N)r )r2 r r
|
r
|
r rÄ s
|
zHPytestFixtureFunctions.combinations.<locals>.decorate.<locals>.<genexpr>Z _exclusions_c s4 |d }|r" ¡ j| }|| } | |d d
|S )NrÒ )r Úadd)r r r+ r Zexlur r
|
r Úcheck_exclusions s
|
zOPytestFixtureFunctions.combinations.<locals>.decorate.<locals>.check_exclusions)rù )r r ÚNotImplementedErrorrÅ r¹ rZ Úgetfullargspecr rÀ r¿ Úsumr´ r r Z parametrize)r Z _fn_argnamesZ _argnamesZ existing_exlZcurrent_exclusion_namer )r r
|
Úhas_exclusionsÚ pytest_paramsr
|
r rþ ÿ s4
|
ÿ
|
ÿ
|
|
ÿz5PytestFixtureFunctions.combinations.<locals>.decorate)r )rc r
|
Úlenr> r9 r( r r r Ú enumerater r× rZ r8 r r ) r Zarg_setsr+ r r Ztobuild_pytest_paramsZ _arg_getterÚfnsZ fn_paramsZparam_exclusionsÚ
|
parametersr rþ r
|
)r rÍ r r
|
r r r Ú combinations sh ÿþþ
|
þ
|
ÿýÿ ÿ
|
&z#PytestFixtureFunctions.combinationsc G s |d }t j|dd
d|iS )Nr r r¾ )r r )r r$ Úidentr
|
r
|
r Ú param_ident' s z"PytestFixtureFunctions.param_identc sz ddl m ddl m t|dkrFt|d rF|d }|dd
}nd }tj|| fdd}|rr||S |S d S )Nr r0 ra r c s j r | ¡} | } | S r )r Z_maybe_async_wrapperr¨ ©rb r1 Úfixturer
|
r ÚwrapA s
|
z,PytestFixtureFunctions.fixture.<locals>.wrap)rc r1 rb r! r± r r) )r rÍ r+ r r* r
|
r( r r) + s
|
z PytestFixtureFunctions.fixturec C s t j d¡S )NZPYTEST_CURRENT_TEST)r? ÚenvironÚgetr r
|
r
|
r Úget_current_test_nameP s z,PytestFixtureFunctions.get_current_test_namec s$ ddl m t fdd}||S )Nr ra c s j | f|| d S r )Z_run_coroutine_functionr¬ ra r
|
r rþ V s z3PytestFixtureFunctions.async_test.<locals>.decorate)rc rb r´ )r r rþ r
|
ra r Ú
|
async_testS s z!PytestFixtureFunctions.async_testN)r r r r Úpropertyr r ÚreprrÖ r r% r' r) r- r. r
|
r
|
r
|
r rE s
|
ü %rE )T)-Ú
|
__future__r r) ry Ú functoolsr r rÆ r r? rÀ rV rr r Zsqla_plugin_baser Ú ImportErrorr½ r/ rH rG r) rN r_ rd rf rn r= r r¥ r© r¡ r¢ rÏ rÑ ZhookimplrÚ rÛ rÓ rß rä rë r´ ZFixtureFunctionsrE r
|
r
|
r
|
r Ú<module> sV 1&
|
|
|
c
|
!
|
|
5
|
|
|
|
>3
|
