o i @sdddlmZddlmZddlmZddlZddlZddlZddl m Z ddl Z ddl Z ddl ZddlmZddlZddlZddlmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZddl m!Z!ddl"m#Z#ddl$m%Z%ddl$m&Z&ddl'm(Z(ddl)m*Z*ddl)m+Z+ddl,m-Z-ddl.m/Z/ddl0m1Z1ddl2m3Z3ddl4m5Z5ddl6m7Z7ddl8m9Z9ddl8m:Z:ddl8m;Z;ddl8mZ>dd"l8m?Z?dd#l8m@Z@dd$lAmBZBdd%lAmCZCdd&lAmDZDdd'lEmFZFdd(lEmGZGdd)lEmHZHdd*lEmIZIdd+lEmJZJddlKZLdd,lMmNZNdd-lMmOZOejPr6dd.lQmRZRd/ZSd0ZTd1ZUzejVe5ZWWn eXyPd2ZWYnwd3ZYd4d5d6ZZe[e\Z]e^e_e`e_fZae^e_e_ejbe_e^e_ece_edffffZee^e_e`e_fZfdd;d<Zgdd>d?ZhddAdBZiddEdFZjGdGdHdHZkejPrece_ejlfZmnecZmGdIdJdJZnGdKdLdLenZoGdMdNdNZpGdOdPdPZqddTdUZrddYdZZsdd\d]ZtddbdcZuddfdgZvejwdhdhdiddkdlZxddmdnZyddpdqZzddrdsZ{ddwdxZ|dd}d~Z}dddZ~dddZdddZdddZdddZejddddZejddddZdS)) annotations) defaultdict)StringION)Path)runtestprotocol)clear_coverage_instance) stop_coverage)_is_pytest_cov_available)_is_pytest_cov_enabled)handle_coverage_report) get_source_lines_for_test_method) undecorated)CITag) SetupError)get_workspace_path)catch_and_log_exceptions) setup_logging)BddTestOptPlugin) BenchmarkData)get_benchmark_tags_and_metrics) TestOptHooks)print_test_report_links)item_to_test_ref) RetryHandler)SessionManager) TelemetryAPI)Test) TestModule)TestRef)TestRun) TestSession) TestStatus) TestSuite)TestTag)enable_all_ddtrace_integrations)install_global_trace_filter) trace_context)coverage_collection)get_coverage_percentage)install_coverage)install_coverage_percentage)uninstall_coverage_percentage) TestContext)asbool)TerminalReporterz!Flaky test is disabled by Datadogz*Skipped by Datadog Intelligent Test Runnerdatadog_itr_unskippablesession_manager_keypytestzno:rerunfailureszno:flaky) rerunfailuresflakyitem pytest.ItemreturnrcCsRzt|dd}|dur|jjWSt|jjjWSty(tYSw)Npath) getattrr7absoluteparentrmodule__file__ Exceptioncwd)r4 item_pathr@Z/home/ubuntu/.local/lib/python3.10/site-packages/ddtrace/testing/internal/pytest/plugin.py_get_module_path_from_item{s   rBworkspace_pathcCs,t|}z||WSty|YSw)z=Get module path from pytest item, converted to relative path.)rB relative_to ValueError)r4rCabs_pathr@r@rA#_get_relative_module_path_from_items   rG tuple[t.Optional[str], int, int]c Csz&tt|dr |jnt|dd}t||}t||\}}|||fWStt t fyMz| \}}}||pz(TestOptPlugin.__init__..)enable_ddtrace_trace_filterr-osenvirongetr$rreports_by_nodeidexcinfo_by_reportbenchmark_data_by_nodeidtests_by_nodeidis_xdist_workermanagersessionextra_failed_reports)selfrgr@r@rA__init__s  zTestOptPlugin.__init__rwpytest.SessioncCst|jdd}r|d}r|j|d|_|jdr"d|_|j|j t j t |j |j jtjdud|jrJt|j j|jrRtdSdS)N workerinput dd_session_idTddtrace-patch-all)test_frameworkhas_codeownersis_unsupported_ci)r8configrprwset_session_idru getoptionr$startrvrrecord_session_createdTEST_FRAMEWORKrenv_tagsr PROVIDER_NAMErmr%writer)ryrwxdist_worker_input session_idr@r@rApytest_sessionstarts$      z!TestOptPlugin.pytest_sessionstartcsj|jtjjkr tjntjj r"t |j dr"jj |j j d<jjjr7d fd d }t|j |ttd tt|j }|durO|jjtj<ttjtjtjjj t!j"duj#d j sxjj$%jjdS)N workeroutputtests_skipped_by_itrcoverage_report_bytesbytescoverage_formatrLr6boolcsjj||ddS)N)rrtags)rvupload_coverage_report)rrryr@rA upload_funcsz7TestOptPlugin.pytest_sessionfinish..upload_func)rris_pytest_cov_enabled_funcstop_coverage_func)rrrefd_abort_reason)rrrrLr6r)&rw set_status exitstatusr1ExitCode TESTS_FAILEDr!FAILPASSrurKrrrrvsettingscoverage_report_upload_enabledr r rr(metricsr#CODE_COVERAGE_LINES_PCTr+rfinishrrprecord_session_finishedrrrrr&get_early_flake_detection_abort_reasonrput_item)ryrwrcoverage_percentager@rrApytest_sessionfinish s8  z"TestOptPlugin.pytest_sessionfinishcCs0|jD] }t|}|jj|q|jdS)a& Discover modules, suites, and tests that have been selected by pytest. NOTE: Using pytest_collection_finish instead of pytest_collection_modifyitems allows us to capture only the tests that pytest has selection for run (eg: with the use of -k as an argument). N)itemsrrvcollected_testsaddfinish_collection)ryrwr4test_refr@r@rApytest_collection_finish>s z&TestOptPlugin.pytest_collection_finishr4r5rr"tuple[TestModule, TestSuite, Test]cs@dfdd }dd d }dfd d }jj|||dS)zn Return the module, suite and test objects for a given test item, creating them if necessary. r;rr6rhcs|jtjjddS)N) module_path) set_locationrGrvrC)r;)r4ryr@rA_on_new_modulePsz4TestOptPlugin._discover_test.._on_new_modulesuiter"cSsdSrjr@)rr@r@rA _on_new_suiteSsz3TestOptPlugin._discover_test.._on_new_suitetestrcstjj\}}}|j|pd|d|rt||jtj<t}r(| |j r6t r6| t }rC||dSdS)zAInitialize test with location, parameters, and custom attributes.rJ)r7rQN)rTrvrCrrLrr# SOURCE_END_get_test_parameters_jsonset_parametersis_skippable_test_is_test_unskippablemark_unskippable_get_test_custom_tagsset_tags)rrPrQrR parameters custom_tagsr4ryrr@rA _on_new_testVs   z2TestOptPlugin._discover_test.._on_new_test) on_new_module on_new_suite on_new_testN)r;rr6rh)rr"r6rh)rrr6rh)rv discover_test)ryr4rrrrr@rrA_discover_testKs zTestOptPlugin._discover_testr'Test'cCst|jjjjsdS|r|s|tjj t ddS| s(|r6|r8|tjj dddddSdSdS)uApply test management markers for the base plugin (used when an external rerun plugin drives execution). ATF retries are not supported in this mode — the external plugin controls the protocol and we cannot intercept individual test runs to capture real FAIL statuses. Disabled+ATF tests therefore get the same xfail treatment as quarantined tests: failures won't break the pipeline, but ATF retry semantics are silently unavailable. If ATF support is needed, disable the external rerun plugin (see the warning emitted at configure time). Overridden in TestOptPluginWithProtocol, which drives retries itself and needs real FAIL outcomes for ATF. NreasonFdd_quarantinedTstrictrrun) rvrtest_managementenabled is_disabledis_attempt_to_fix add_markerr1markskip"DISABLED_BY_TEST_MANAGEMENT_REASONis_quarantinedxfailryr4rr@r@rA_apply_test_management_markersws z,TestOptPlugin._apply_test_management_markersTpytest_runtest_protocol)tryfirst hookwrapperspecnamenextitemt.Optional[pytest.Item]t.Generator[None, None, None]c cst|}|r t|nd}|||\}}}|s&|tjtd|s6|tjtd|||j |j <| |||| ||t |j0}tjtddt } dVWdn1smwYtjtddWdn1swY|jstd|j |} | j|jd|| ||| | } | | || |jj| ||jj |j!| j"|jj#d|r|j$|j$kr||jj|tj%td|r|j$j&|j$j&kr ||jj|tj'tddSdS)N)rddtrace)rcoverage_libraryzrTest Optimization pytest_runtest_protocol did not run for %s; perhaps some plugin or conftest.py has overridden it)start_ns)rD)(rr is_startedrrrprecord_module_createdrrecord_suite_createdrtnodeid _handle_itrrr&rmrecord_coverage_startedr'record_coverage_finished test_runslogdebug make_test_runr_set_test_run_datar get_statusset_final_statusrrvrrcoverage_writer put_coverage last_test_runget_coverage_bitmapsrCrrecord_suite_finishedr;record_module_finished) ryr4rr next_test_ref test_module test_suitercontext coverage_datatest_run final_statusr@r@rApytest_runtest_protocol_wrappers^     z-TestOptPlugin.pytest_runtest_protocol_wrapperrr,tuple[TestRun, _ReportGroup]cCs||j|j}|}|tjt|dtt ||dd}| |||tj t||j j tj|j jd||fS)N)rrF)rr)rrci_provider_nameis_auto_injected)rtrrrrrprecord_test_createdr_make_reports_dictrrrecord_test_finishedrvrrrr)ryr4rrrrreportsr@r@rA_do_one_test_runs zTestOptPlugin._do_one_test_runcCs|j|j}||}t|j}||||\}}Wdn1s$wY|s>|r>||r>||||||nt |drI| ||| ||| | ||jrg|jd||jD] }|jj|qjdS)Ndd_disabled_attempt_to_fix)rtr _check_applicable_retry_handlersr&rmr is_skipped_by_itr should_retry _do_retries_get_user_property,_mark_attempt_to_fix_report_group_as_skipped_log_test_reportsrrrrrrvrr)ryr4rr retry_handlerrrrr@r@rA _do_test_runss"       zTestOptPlugin._do_test_runsrrcCsj||j\}}|||||||j|jd}r3||j||j | dSdSrj) _get_test_outcomerrr set_contextrspopr set_metricsrmark_benchmark)ryrr4rstatusrbenchmark_datar@r@rArs      z TestOptPlugin._set_test_run_datarrrr _ReportGroupcCspt}||||||tj|||tj|j}|||d}|rmt |j } | ||| \}}Wdn1sBwY| |}||||||||tjsg|||tj||s)| |} || |||| } ||| } r|j| t|dr||| |jj| d|tj} t|dr||| |jj| ddS)NTr report) RetryReports_mark_test_reports_as_retrylog_test_reportr[rbrcrset_tags_for_test_runrr&rmr rget_final_statusrmake_final_reportget_extra_failed_reportrxappendr&_mark_attempt_to_fix_report_as_skippedihookpytest_runtest_logreportrprd)ryr4rrrr retry_reportsrrrr final_reportextra_failed_reportteardown_reportr@r@rArs>              zTestOptPlugin._do_retriest.Optional[RetryHandler]cCs$|jjD] }||r|SqdSrj)rvretry_handlers should_apply)ryrhandlerr@r@rAr =s  z.TestOptPlugin._check_applicable_retry_handlerstuple[t.Any, t.Any]cCsDtjtjtjfD]}||}r|jr|jt|ddfSqdS)a< Extract the most relevant report `longrepr` for a report group. Errors that happened during the call phase have more useful information, so we try to use that if available. Also return the corresponding `wasxfail` attribute if present, for correct indication of xfail/xpass tests. wasxfailN)NN)r[rcrbrdrplongreprr8)ryrwhenrr@r@rA_extract_longreprDs zTestOptPlugin._extract_longreprcCs(|||tjs|||tjdSdSrj)_mark_test_report_as_retryr[rcrb)ryrrr@r@rAr Ssz)TestOptPlugin._mark_test_reports_as_retryrt.Optional[pytest.TestReport]cCsP|durdS|jtjkrd|_dS|jdpd}t|j|df}||_d|_dS)a? Modify a test report for an attempt-to-fix test to make it look like it was skipped. This is called *after* ``_get_test_outcome`` has already captured the real status (FAIL/PASS), so the backend sees the true result while the terminal/junitxml shows the test as skipped (quarantined). NpassedrWrzQuarantined (Attempt to Fix)skipped)r5r[rdoutcomelocationrLr7r4)ryr4r line_numberr4r@r@rAr'Ws   z4TestOptPlugin._mark_attempt_to_fix_report_as_skippedcCs`|tj}r|||d|tj_d|tj_dS|tj}|||d|tj_dS)zi Modify the test reports for an attempt-to-fix test to make it look like it was skipped. r9N)rpr[rcr'rbr;rd)ryr4r call_report setup_reportr@r@rAris    z:TestOptPlugin._mark_attempt_to_fix_report_group_as_skippedr5rLrcCs<||}r|jd|jfd|fg7_d|_dSdS)Ndd_retry_outcomedd_retry_reasonrerunTF)rpuser_propertiesr;get_pretty_name)ryrrr5r>r@r@rAr7vs z(TestOptPlugin._mark_test_report_as_retrycCs6tjtjtjfD]}||}r|jj|dqdS)Nr)r[rbrcrdrpr(r))ryr4rr5rr@r@rArs zTestOptPlugin._log_test_reports)rr]pytest.CallInfo[t.Any]-t.Generator[None, pluggy.Result[t.Any], None]ccs^dV}|}||j|j|j<|j|j|<|jtjkr+t|}r-||j |j<dSdSdS)zF Save report and exception information for later use. N) get_resultrqrr5excinforrr[rdrrs)ryr4r]r;rrr@r@rApytest_runtest_makereports   z'TestOptPlugin.pytest_runtest_makereportpytest.TestReportt.Optional[_ReportTestStatus]cCst|d}rt|d}ddd|d|dfSt|ddd kr+d d d d diffSt|drA|jtjkr?d d d d diffSdSt|drOdddddiffSdS)Nr@rArBRzRETRY z ()r3r quarantinedQ QUARANTINEDblueTr )rRrRdd_flakyr3KFLAKYyellow)rupperr8r5r[rd)ryr retry_outcome retry_reasonr@r@rApytest_report_teststatuss    z&TestOptPlugin.pytest_report_teststatusr!tuple[TestStatus, dict[str, str]]c Cstj}i}|j|i}tjtjtjfD]Y}||}|sqt |dd}r;|dkr;t ||t j <|j r6dnd|t j<|j|d}|jrTtj}|t|||fS|jrmtj}|rat |jnd} | |t j<||fSq||fS)z Return test status and tags with exception/skip information for a given executed test. This methods consumes the test reports and exception information for the specified test, and removes them from the dictionaries. r3NrxpassrzUnknown skip reason)r!rrqrr[rbrcrdrpr8rLr# XFAIL_REASONr9 TEST_RESULTrrfailedrupdate_get_exception_tagsr:SKIPvalue SKIP_REASON) ryrrr reports_dictphaserr3rHrr@r@rArs0  zTestOptPlugin._get_test_outcomecCsP|j|sdS|r|dS|rdS|tjjt d| dS)Nr) rvris_unskippablemark_forced_runrrr1rrSKIPPED_BY_ITR_REASONmark_skipped_by_itr)ryr4rrr@r@rArs  zTestOptPlugin._handle_itrrrterminalreporterr.rintr pytest.Configccs|jdd|jdg}|jD]}d|_dd|jD|_q||j|jd<dV||jd<|jds:|jd=t||jdS)a Modify terminal summary before letting pytest emit it. During the test session, all retry attempt reports are logged with a 'rerun' category. We remove this category here so it doesn't show up in the final stat counts. To make the extra failed reports collected during retries (see `get_extra_failed_report` for details) show up with the rest of the failure exception reports, we modify them to look like normal failures, and append them to the failed reports. After they have been shown by pytest, we undo the change so that the final count of failed tests is not affected. rBNr_cSs g|] \}}|dkr||fqS)r@r@).0kvr@r@rA s z9TestOptPlugin.pytest_terminal_summary..)statsrrprxr;rCrrv)ryrlrroriginal_failed_reportsrr@r@rApytest_terminal_summarys   z%TestOptPlugin.pytest_terminal_summaryN)rgrr6rh)rwr{r6rh)r4r5rrr6rr4r5rrr6rh)r4r5rrr6r)r4r5rrrr,r6r)r4r5rrr6rh)rrr4r5rr,r6rh) r4r5rrrrrrrrr6rh)rrr6r.)rrr6r2)rrrrr6rh)r4r5rr8r6rh)r4r5rrr6rh)rrrrr5rLr6r)r4r5r]rEr6rF)rrJr6rK)rrLr6r[)r4r5rrrrr6rh)rlr.rrmrrnr6r)r_r`ra__doc__rerzrrrrrr1hookimplrr rrrr r6r r'rr7rrIrZrrrur@r@r@rArfs:    2 ,  @   <         $ rfc@s*eZdZdZddd Zedd dZdS)TestOptPluginWithProtocolax TestOptPlugin subclass that registers pytest_runtest_protocol, taking ownership of test execution and retries. Used when no external rerun plugin (e.g. pytest-rerunfailures, flaky) is present. When such a plugin IS present, the base class is registered instead, so the external plugin drives retries while the wrapper hook still handles span bookkeeping. r4r5rrr6rhcCs|jjjjsdS|r|s|tjj t ddS| s(|rD|rF|r6|j dg7_ dS|tjj dddddSdSdS)uApply test management markers for the plugin that drives retries itself. ATF tests must NOT use xfail here: xfail converts failures into SKIP outcomes, which prevents AttemptToFixHandler.get_final_status from seeing real FAIL counts and computing correct tags (HAS_FAILED_ALL_RETRIES, ATTEMPT_TO_FIX_PASSED). Instead, ATF tests set a user property so _get_test_outcome captures the true status, and reports are mangled to "skipped" afterwards so failures still don't break the pipeline. The outer condition mirrors the original logic: only quarantined tests and disabled+ATF tests are treated as "run but don't fail the pipeline" — plain ATF (neither disabled nor quarantined) runs normally without any report suppression. Nr)r TFrTr)rvrrrrrrr1rrrrrCrrr@r@rAr s z8TestOptPluginWithProtocol._apply_test_management_markersrrrcCs8|jj|j|jd||||jj|j|jddS)N)rr<T)r(pytest_runtest_logstartrr<rpytest_runtest_logfinish)ryr4rr@r@rAr"s z1TestOptPluginWithProtocol.pytest_runtest_protocolNrv)r4r5rrr6r)r_r`rarwrrrr@r@r@rArys  ryc@s6eZdZdZddZdd d ZdddZdddZdS)rzF Collect and manage reports for the retries of a single test. cCstdd|_dS)NcSsgSrjr@r@r@r@rArk0rlz'RetryReports.__init__..)rreports_by_outcomerr@r@rArz/szRetryReports.__init__r4r5rrr5rLr6rcCsD||}r |jj|dt|dp|j}|j||dSdS)z Collect and log the test report for a given test phase, if it exists. Returns True if the report exists, and False if not. Tests that fail or skip during setup do not have the call phase report. rr@TF)rpr(r)rr;r|r&)ryr4rr5rr;r@r@rAr!2s zRetryReports.log_test_reportrrrr!rJc Cstjdtjdtjdi}||t|}z|j|d}|j}t|dd}Wnt y8t d|d}d}Ynwg} | rD| dg7} t j|j|jd d |jDtj|||j| d } |durft| d|| S) a Make the final report for the retries of a single test. The final status is provided by the retry handler. We assume that for a test to have a given outcome, at least one attempt must have had that outcome (e.g., for a test to have a 'failed' outcome, at least one retry must have failed). The attributes of the final report will be copied from the first retry report that had the same outcome. For instance, if the final report outcome is 'failed', it will have the same `longrepr` and `wasxfail` features as the first failed report. Additionally, if a test is marked as flaky (e.g., it both passed and failed during EFD), the final report will contain the `dd_flaky` user property. This allows us to identify the test as 'flaky' during logging, even though the final outcome is still one of 'passed', 'failed', or 'skipped'. r9r_r:rr3NzUTest %s has final outcome %r, but no retry had this outcome; this should never happen)rSTcSsi|]}|dqS)rWr@)rorpr@r@rA jsz2RetryReports.make_final_report..)rr<keywordsr5r4r;rC)r!rrrbrprLr|r4r8 IndexErrorrwarning is_flaky_runr1 TestReportrr<r~r[rcrCsetattr) ryrr4routcomesr; source_reportr4r3extra_user_propertiesr+r@r@rAr$As:    zRetryReports.make_final_reportr8cCsH|s|o | }|rdS|jdr"|tjkr"|jddSdS)aW Get an extra failed report to log at the end of the test session. If a test is retried and all retries failed, the final report will have a 'failed' status and contain the `longrepr` of one of the failures, and so the failure exception will be automatically logged at the end of the test session by pytest. In this case, this function returns None. But if some retries pass and others fail, the test will have a 'passed' final status, and no failure exception will be logged for the test. In this case, this function returns one of the failed reports, which is saved for logging at the end of the test session. This ensures we provide some failure log to the user. If the test is quarantined or disabled, and not attempt-to-fix, the failed report is not returned. If the test is attempt-to-fix, the failed report is returned, even if the test is quarantined or disabled: if the user is attempting to fix the test, and the attempt fails, we need to provide some feedback on the failure. Note that we only report _one_ failure per test (either the one embedded in the 'failed' final report, or the one returned by this function), even if the test failed multiple times. This is to avoid spamming the test output with multiple copies of the same error. Nr_r)rrrr|r!r)ryrrsuppress_errorsr@r@rAr%us z$RetryReports.get_extra_failed_reportN)r4r5rrr5rLr6r)rrr4r5rr!r6rJ)rrrr!r6r8)r_r`rarwrzr!r$r%r@r@r@rAr*s   4rc@s6eZdZdddZejdd d Zejdd d ZdS)XdistTestOptPlugin main_pluginrfr6rhcCs ||_dSrj)r)ryrr@r@rArzs zXdistTestOptPlugin.__init__nodet.AnycCs|jjj|jd<dS)zN Pass test session id from the main process to xdist workers. r}N)rrwitem_idr|)ryrr@r@rApytest_configure_nodesz(XdistTestOptPlugin.pytest_configure_nodeerrorcCs8t|dsdS|jd}r|jjj|7_dSdS)zs Collect count of tests skipped by ITR from a worker node and add it to the main process' session. rNr)rKrrprrwr)ryrrrr@r@rApytest_testnodedowns z&XdistTestOptPlugin.pytest_testnodedownN)rrfr6rh)rrr6rh)rrrrr6rh)r_r`rarzr1rxrrr@r@r@rArs   rrlist[pytest.TestReport]rcCsdd|DS)NcSsi|]}|j|qSr@)r5)rorr@r@rAr}sz&_make_reports_dict..r@)rr@r@rArsrparser pytest.ParserrhcCs|d}|jdddddd|jddddd d|jd dd dd d|jddd d|jddd d|jd d d dtjjjj|dS)zAdd ddtrace options.rz --ddtrace store_trueFz Enable Datadog Test Optimization)actiondestdefaulthelpz --no-ddtrace no-ddtracez7Disable Datadog Test Optimization (overrides --ddtrace)z--ddtrace-patch-allr~z$Enable all integrations with ddtracer)typez7Disable Datadog Test Optimization (overrides 'ddtrace')N) getgroup addoptionaddinirtestinginternal tracer_api pytest_hookspytest_addoption)rgroupr@r@rArs4 rrcCsttjdd S)NDD_CIVISIBILITY_ENABLEDtrue)r-rnrorpr@r@r@rA-_is_test_optimization_disabled_by_kill_switchsr early_configrnargs list[str]cCs&trdStd||r dStd||S)NFrr)r_is_option_true)rrr@r@rA_is_enabled_earlys   roptionrLcCs"||p||pd||vS)Nz--)rgetini)rrrr@r@rArs"rTrkrc cst||s dVdStttd}|jt|ttjdzt|d}Wnt yA}zt d|dVWYd}~dSd}~ww||j t <|jjrR|jjsRtdVdS)NrX) test_commandrtest_framework_version)rwz%s)rrr rset_attributes_get_test_commandr1 __version__rrrrstashSESSION_MANAGER_STASH_KEYrcoverage_enabledrsetup_coverage_collection)rrrrwrger@r@rApytest_load_initial_conftestss.     rcCst}t|dSrj)rr))rCr@r@rAr s rrc sddtr dSjtd}|stddSfddtD}|jj j p-|jj j }|rA|rA|D] }t d||q4t }n|r[|jjj rX|D] }t d|||t|qJt}nt }z||d}Wntystd YdSwj|jtjd rjt|jd rjt|tjjjj|jjrt sd d l!m"}t#}|t$|gdtdt%rt s|jjrt&dSdSdS)Nmarkersz+dd_tags(**kwargs): add tags to current spanz8Session manager not initialized (plugin was not enabled)csg|] }j|r|qSr@) pluginmanager hasplugin)rorXrr@rArrsz$pytest_configure..z%s is installed alongside Datadog Test Optimization. Datadog retries (ATR, EFD) take precedence over %s, which will be overridden for this test session.z%s is installed and Datadog Auto Test Retries and Early Flake Detection are disabled. Attempt to Fix retries will not work while %s drives test execution. To use Attempt to Fix, disable %s with: -p %s)rgz)Error setting up Test Optimization pluginxdistz pytest-bddr)start_coverage)sourcezIStarted coverage.py collection for report upload (pytest-cov not enabled))'addinivalue_linerrrprrr_EXTERNAL_RERUN_PLUGINSrauto_test_retriesrearly_flake_detectionrryrrfr= exceptionrregister add_hookspecsrrrrrrrrrpytest_configurerr 'ddtrace.contrib.internal.coverage.patchrrrLr r*) rrgdetected_rerun_pluginsdd_retries_enabled plugin_name plugin_classpluginrrCr@rrArsj             rcCsLd}t|dd}r|dd|j7}tjd}r$|d|7}|S)z@Extract and re-create pytest session command from pytest config.r1invocation_paramsNz {} PYTEST_ADDOPTS)r8formatjoinrrnrorp)rcommandraddoptsr@r@rAr^s rrH't.Optional[pytest.ExceptionInfo[t.Any]]dict[str, str]cCsb|duriSd}t}tj|j|j|j| |dtj|tj d|jj |jj ftj t |jiS)N)limitfilez%s.%s)r tracebackprint_exceptionrrctbr# ERROR_STACKgetvalue ERROR_TYPEr`r_ ERROR_MESSAGErL)rH max_entriesbufr@r@rArahs  rarrJ user_propertyt.Optional[t.Any]cCs.t|dg}|D] \}}||kr|SqdS)NrC)r8)rrrCkeyrcr@r@rArxs  rt.Optional[str]c Cst|dd}|dur dSiid}|jjD]%\}}z t||d|<Wqty<d|d|<tjd|ddYqwztj |ddWSt yWtjd |ddYdSw) Ncallspec) argumentsmetadatarzCould not encodezFailed to encode %rT)exc_info) sort_keysz*Failed to serialize parameters for test %s) r8rparamsr_encode_test_parameterr=rrjsondumps TypeError)r4rr param_name param_valr@r@rArs"     r parameterrcCst|}tdd|S)Nz at 0[xX][0-9a-fA-F]+rR)reprresub)r param_reprr@r@rArsrmarker pytest.MarkcCs2|jr |jd}|S|jr|jd}|Sd}|S)Nr conditionT)rkwargsrp)rrr@r@rA_get_skipif_conditions  rcCstdd|jddDS)Ncss,|]}t|duo|jdtkVqdS)FrN)rrrpITR_UNSKIPPABLE_REASON)rorr@r@rA s  z'_is_test_unskippable..skipifr)any iter_markers)r4r@r@rArs rcCs:i}|jddD]}|jD] \}}t|||<qq|S)Ndd_tagsr)rrrrL)r4rrrrcr@r@rArs rrw)scopecCsddl}|jS)z#Return the current tracer instance.rN)rtracer)rr@r@rAddtracersrfunctioncCs|durdS|S)zReturn the current root span.N)current_root_span)rr@r@rAddspansr)r4r5r6r)r4r5rCrr6r)r4r5rCrr6rH)r4r5r?rr6rU)rrr6r)rrr6rh)r6r)rrnrrr6r)rrLrrnrrr6r)rrnrrrrr6r)r6rh)rrnr6rh)rrnr6rL)rHrr6r)rrJrrLr6r)r4r5r6r)rrr6rL)rrr6r)r4r5r6r)r4r5r6r) __future__r collectionsriorrloggingrnpathlibrrrtypingt_pytest.runnerrpluggyr1rrr'ddtrace.contrib.internal.coverage.utilsr r r $ddtrace.internal.ci_visibility.utilsr !ddtrace.internal.utils.inspectionr ddtrace.testing.internal.cirddtrace.testing.internal.errorsrddtrace.testing.internal.gitr ddtrace.testing.internal.loggingrr#ddtrace.testing.internal.pytest.bddr)ddtrace.testing.internal.pytest.benchmarkrr)ddtrace.testing.internal.pytest.hookspecsr,ddtrace.testing.internal.pytest.report_linksr%ddtrace.testing.internal.pytest.utilsr'ddtrace.testing.internal.retry_handlersr(ddtrace.testing.internal.session_managerr"ddtrace.testing.internal.telemetryr"ddtrace.testing.internal.test_datarrrrr r!r"r#+ddtrace.testing.internal.tracer_api.contextr$r%r&,ddtrace.testing.internal.tracer_api.coverager'r(r)r*r+0ddtrace.testing.internal.tracer_api.pytest_hooksrddtrace.testing.internal.utilsr,r- TYPE_CHECKING_pytest.terminalr.rrirStashKeyrAttributeErrorrr getLoggerr_rtuplerLrm _LongreprUniondictr_ReportTestStatus _LocationrBrGrTrMr[rrrfryrrrrrrrrxrrrrrarrrrrrfixturerrr@r@r@rAs                                                 /)i   #    &  O