o gl@sddlmZddlZddlZddlmZddlZddlZddlZddl Z ddl Z ddl Z ddl m Z ddlZddlZze sAddlZWneyQddlmZYnwddZd) annotationsN)update_wrapper) TYPE_CHECKING) plugin_basecs$|dfdd}t|dS)N sqlalchemycsr|ddrGfdddtj}||d<|ddr.Gfdddtj}||d<j|fi|dS)NcallbackcseZdZ dfdd ZdS)=pytest_addoption..make_option..CallableActionNc|||dSNselfparser namespacevalues option_string callback_r b/var/www/html/flaskapp/venv/lib/python3.10/site-packages/sqlalchemy/testing/plugin/pytestplugin.py__call__*Fpytest_addoption..make_option..CallableAction.__call__r )__name__ __module__ __qualname__rr rr rCallableAction)sractionzeroarg_callbackcs4eZdZ   dfdd Z dfdd ZZS) r FNc stj||dd|||ddS)NrT)option_stringsdestnargsconstdefaultrequiredhelp)super__init__)rrr r#r$r% __class__r rr'5s zFpytest_addoption..make_option..CallableAction.__init__cr r r r rr rrGrr)FFNr )rrrr'r __classcell__r r*r(rr4s)popargparseAction addoption)namekwrgroup)rrr make_option%s  z%pytest_addoption..make_option)getgroupr setup_options)rr4r r2rpytest_addoption"s  +r7config pytest.ConfigcCs t|jtjs tjr0dttjddtjD}|jjr,|jjd|7_n||j_|j dr=|j t t |drQt|jt|jdn|jjrdtj|jjrdt|jjt|jttt|jddtt|jjrd adSdS) Nz and cSsg|]}d|qS)znot r ).0tagr r r Xsz$pytest_configure..xdist workerinputfollower_ident cov_sourceFT)r read_configrootpath exclude_tags include_tagsjoinlistoptionmarkexpr pluginmanager haspluginregister XDistHookshasattr!restore_important_follower_configr>configure_follower write_identsospathexistsremove pre_beginset_coverage_flagboolgetattrset_fixture_functionsPytestFixtureFunctionsdump_pyannotateDUMP_PYANNOTATE)r8new_exprr r rpytest_configureSs6       r^FT)autouseccs4tr ddlm}|dVtr|dSdS)Nr collect_types)r\pyannotate_runtimerastartstopr`r r rcollect_types_fixture|s  rec Csddl}ddlm}ddlm}ddlm}d}tjjrdnd}d|d |d d |j g}|rCdd l m }| d |j dn| d||j jd} | rf| d||D]} | | q\dStd|d|D]} t| qndS)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; terminalreporter=z=========================)rrfsqlalchemy.utilrgsqlalchemy.util._has_cyrhsysflags no_user_site__file__sqlalchemy.cyextensionrjappendr8rI get_plugin write_sep write_lineprint) sessionrrfrgrhgreetsitemsgsrjpmmr r r_log_sqlalchemy_infos.         rcCs$ddlm}t||tjdSNrasyncio)sqlalchemy.testingrr _assume_asyncr post_begin)rzrr r rpytest_sessionstarts rcCsFddlm}|tj|jjjr!ddlm }| |jjjdSdS)Nrrr`) rr_maybe_async_provisioningrfinal_process_cleanupr8rGr[rbra dump_stats)rzrrar r rpytest_sessionfinishs    rcCsddlm}|dSr)rr _shutdown)r8rr r rpytest_unconfigures  rcsB|jjjrddlm}tjdfdd}|j|ddSdS)Nrr`lib/sqlalchemycs<tjtj|}dtj|gvrdSd|vrdS|S)Nrtesting)rQrRnormpathabspath commonpath)filenamelib_sqlalchemyr r_filtersz)pytest_collection_finish.._filter)filter_filename) r8rGr[rbrarQrRrinit_types_collection)rzrarr rrpytest_collection_finishs     rc@seZdZddZddZdS)rLcCsXddlm}ddlm}t|jdtjdd|jd<| |j |jddS)Nr provisionrztest_%s r?) rrrr!memoize_important_follower_configr>uuiduuid4hexrcreate_follower_db)rnoderrr r rpytest_configure_nodes    z XDistHooks.pytest_configure_nodecCs0ddlm}ddlm}||j|jddS)Nrrrr?)rrrrdrop_follower_dbr>)rrerrorrrr r rpytest_testnodedowns   zXDistHooks.pytest_testnodedownN)rrrrrr r r rrLs rLcsddlm}tdddd|D|dd<dd|Dfd d fd d }||g}|D]}|j}|vrJ|||jq6||q6t |d dd|dd<dS)NrrcSs ttSr ) collections defaultdictrFr r r rs z/pytest_collection_modifyitems..cSs4g|]}|tjdur|tjjds|qS)N_) getparentpytestClassr0 startswithr:itemr r rr<sz1pytest_collection_modifyitems..cSsh|]}|tjqSr )rrrrr r r sz0pytest_collection_modifyitems..c3s6|D]}t|tjr|EdHq|VqdSr )collect isinstancer Collector)element inst_or_fn)rr rrs   z.pytest_collection_modifyitems..collectc sD]}t}t|jddst|jddrdh}nt|jddr$dh}nt}dd|D}||}||}|D]}||q;t|j|j |D]4}||jur|j}| t j }t j j|j|d }|D]}||qi|D] } || j| quqLqdS) N __backend__F __only_on__backend__sparse_backend__sparse_backendcSsh|]}|jqSr )r0)r:markr r rrszLpytest_collection_modifyitems..setup_test_classes..r0parent)setrXcls frozenset iter_markersunion add_markerrgenerate_sub_testsmodulerrModuler from_parentrr0ru) test_class add_markersexisting_markers all_markersmarkersub_cls per_cls_dictrnew_clsfnr rebuilt_items test_classesr rsetup_test_classes s@        z9pytest_collection_modifyitems..setup_test_classescSs |tjj|tjj|jfSr )rrrr0r)rr r rrEs  )key) rrrrrrextendr0rusorted)rzr8itemsrrnewitemsrcls_r rrpytest_collection_modifyitemss*   * rcstt|r%t||r%ddlm}|jrt|}fddtj |DSt |r8j dur8t j |r8dSgS)Nrr8csg|] }tjj|jdqS)r)rrrr)r:parametrize_cls collectorr rr<Ts z-pytest_pycollect_makeitem..) inspectisclassr want_classrr8 any_async_apply_maybe_async_parametrize_clsr isfunctionr want_method)rr0objr8r rrpytest_pycollect_makeitemMs     rcCs$t|dr |j}t|dst|S)N __wrapped__)rMrriscoroutinefunctionrr r r_is_wrapped_coroutine_functiongs  rcsddlmt|D]C\}}t|st|trOt|ddsO|drOt |sOd}t|tr4|j }d}t fdd}||}|rFt|}d|_ t |||q |rf|d dD] }|tkret|dqZ|S) Nrr_maybe_async_appliedFtest_Tcsj|g|Ri|Sr ) _maybe_asyncrargskwargsrr r make_async}sz&_apply_maybe_async..make_asyncr)rrvarsrcallabler classmethodrXrr__func___pytest_fn_decoratorrsetattrmroobjectr)rrecurser0valueis_classmethodrdo_asyncrr rrrns<     rcCsd|jvr|gS|j}g}tjdd|DD]S}i}tdd|D|D]\}}|s/tdtd|}t||jD]\} } | || <q;q%d dd |D} d |j | f} t t | |f|} t || | || q|S) z6implement a class-based version of pytest parametrize._sa_parametrizecSsg|]\}}|qSr r )r:argnameparamsr r rr<z$_parametrize_cls..cSsg|]}|dqS)rr )r: _sa_paramr r rr<rz*need argnames for class-based combinationsz,\s*rcss0|]}|jdD] }tdd|Vq qdS)-z\WN)idsplitresub)r:paramtokenr r r s  z#_parametrize_cls..z%s_%s)__dict__r itertoolsproductzip TypeErrorr r rrErtype__new__rru)rrrclassesfull_param_set cls_variablesrr  argname_splitargvalparametrized_namer0newclsr r rrs2        rcCsHddlm}t|tjr tdur"|tj|j | tj adSdSdSr) rrrrFunction_current_classrr!start_test_class_outside_fixturesrrrrrr r rpytest_runtest_setups  r#) hookwrapperc csddlm}|tj|dVtdur|dus!|tjturdazXz | tj |j WnHt yv}z._exec_code_in_envr cs|}|r |j|tdd|jd}||ddd|}||d|j}|s=t|d|j|_||_t||S|j |_ |j|_t |d rN|j |_ |S) N __target_fn __orig_fn)rMrNr0F)groupedzVdef %(name)s%(grouped_args)s: return %(__target_fn)s(%(__orig_fn)s, %(apply_kw)s) )rMrNr pytestmark) rrdictrupdaterX __defaults__rrrrMrP)radd_positional_parametersspecmetadatarI decoratedrLrFrGtargetr rdecorateqs.   z&_pytest_fn_decorator..decorateN)r )sqlalchemy.util.langhelpersrFsqlalchemy.util.compatrG)rYrZr rXrrcs  !rc@sfeZdZddZeddZddZddeed dd Z d d Z d dZ ddZ ddZ ddZdS)rZcOstjj|i|Sr )rskipr()rrr1r r rskip_test_exceptionsz*PytestFixtureFunctions.skip_test_exceptioncCstjSr )rrrr r r add_to_markersz$PytestFixtureFunctions.add_to_markercCstjddS)Nr=rE)rr usefixturesr_r r rmark_base_test_classsz+PytestFixtureFunctions.mark_base_test_classcCs|Sr r rr r rrszPytestFixtureFunctions.cCst|dr|jSt|jS)Nr)rMrrrcr r rrs)irsnc sddlmt|dkrt|ddrt|d}|ddfdd}|d d}g}d |r|jtjdgd d t |DR}fd d t |D}|D].t t s[f|\}} ||dd} | rmd| | | d fdd|DfqQn|D]t t sf|\}} | rd| || dfqg|D]\} } }r| | f7} tj| d|i}  | qfdd} | S)aFacade 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__argnamesNcs<g}g}|D]}t|jr||q||q||fSr )rcompoundru)rresultgathered_exclusionsarhr r_filter_exclusionss   z?PytestFixtureFunctions.combinations.._filter_exclusionsid_FcSsg|] \}}|dvr|qS))rgrerfror r:idxcharr r rr<s z7PytestFixtureFunctions.combinations..cs*g|]\}}|vrt||fqSr )operator itemgetterrr)_combination_id_fnsr rr<s Trc3s |] \}}||VqdSr r )r:gettercomb_fn)rr rrs z6PytestFixtureFunctions.combinations..r cst|rr tdd|jvrg|_|jf|St|jdd}dur.|}nt d}rXt dd|D}d|}||g7}t fdd }|||fd }t j ||S) Nz5exclusions not supported for class level combinationsrrz, *css|] }|drdVqdS) _exclusionsrN)r)r:rgr r rrs   zHPytestFixtureFunctions.combinations..decorate.. _exclusions_cs8|d}|rj|}||}||ddi|S)Nr%)rladd)rrr1rzexlurhr rcheck_exclusions!s zOPytestFixtureFunctions.combinations..decorate..check_exclusions)rT)rrNotImplementedErrorrrrugetfullargspecrr r sumrrr parametrize)r _fn_argnames _argnames existing_exlcurrent_exclusion_namer~)rkrihas_exclusions pytest_paramsr rrZ s2     z5PytestFixtureFunctions.combinations..decorate)rrilenrMrFr,rwrurv enumeraterr-rurErr ) rarg_setsr1rprqtobuild_pytest_params _arg_getterfns fn_paramsparam_exclusions parametersr rZr )rwrrkrirrr combinationssj            &z#PytestFixtureFunctions.combinationscGs |d}tj|ddd|iS)Nrrr )rr )rridentr r r param_ident1sz"PytestFixtureFunctions.param_identcszddlmddlmt|dkr#t|dr#|d}|dd}nd}tj|i|fdd}|r;||S|S)Nrrrrcsjr|}|}|Sr )r_maybe_async_wrapperrrr8fixturer rwrapKs z,PytestFixtureFunctions.fixture..wrap)rr8rrrrr)rrr1rrr rrr5s   zPytestFixtureFunctions.fixturecCs tjdS)NPYTEST_CURRENT_TEST)rQenvirongetr_r r rget_current_test_nameZs z,PytestFixtureFunctions.get_current_test_namecs$ddlmtfdd}||S)Nrrcsj|g|Ri|dSr )_run_coroutine_functionrrr rrZ`sz3PytestFixtureFunctions.async_test..decorate)rrr)rrrZr rr async_test]s z!PytestFixtureFunctions.async_testN)rrrr^propertyr`rbreprr,rwrrrrrr r r rrZs   % rZ)r8r9)T)0 __future__rr-r functoolsrrrrurQr rptypingrrrsqla_plugin_baser ImportErrorrr7r^r\rrerrrrrrLrrrrrr r#hookimplr0r2r)r6r=rErFixtureFunctionsrZr r r rsb     1&     a  !  5    >2