o i* @s8ddlmZddlZddlZddlmZddlmZddlmZddl m Z ddl m Z ddl mZdd lmZe eZd efd d Zd efd dZd efddZddZdeed efddZd efddZddZddZddZddZd d!Z d"d#Z!d$d%Z" d+d&ee#egefd'ed(eed dfd)d*Z$dS),)PathN)Callable)Optional)_original_sys_argv_command)is_coverage_running) get_logger) _get_config)asboolreturncCs dtjvS)Ncoverage)sysmodulesrr[/home/ubuntu/.local/lib/python3.10/site-packages/ddtrace/contrib/internal/coverage/utils.pyis_coverage_loadeds rcCs|jdduS)z4Check if pytest-cov plugin is available (installed). pytest_covN) pluginmanager get_plugin)configrrr_is_pytest_cov_availablesrcCsTt|sdS|jddd}|jddd}|durdSt|tr(|dgkr(|s(dS|S)zRCheck if pytest-cov plugin is both available and enabled via command-line options.Fz--cov)defaultz--no-covT)r getoption isinstancelist)r cov_option nocov_optionrrr_is_pytest_cov_enabledsrcCstsdSttjdddS)NFr _datadog_patch)rgetattrr r rrrr_is_coverage_patched'srsys_argv_commandcCsdd|vS)Nzcoverage run -m )join)r rrr_command_invokes_coverage_run.sr#cCstddtrdSttS)N COVERAGE_RUNFT)rr r#rrrrr$_is_coverage_invoked_by_coverage_run2s r%cCsL|jD]}|\}}t|dr#|jr#t|jdr#|jjr#|jjSqdS)zTFind and return pytest-cov coverage instance if available from config.pluginmanager.cov_controllercovN)rlist_name_pluginhasattrr&r')rplugin_plugin_instancerrr_find_pytest_cov_instance8s r-cCsBddlm}t|}|r||tdt|dStddS)z7Register pytest-cov instance with ddtrace if available.r)set_coverage_instancez8Registered pytest-cov coverage instance with ddtrace: %sTz0No pytest-cov controller found in plugin managerF)'ddtrace.contrib.internal.coverage.patchr.r-logdebugtype)rr. cov_instancerrr_register_pytest_cov_instanceBs  r4c Cst|}|sdSz0t|dr|tdt|dr0|jr3t|dr6|tdWdSWdSWdSWdStyQ}z td|WYd}~dSd}~ww)z.Save pytest-cov data before report generation.Nsavez7Saved pytest-cov coverage data before report generation_startedstopz&Stopped pytest-cov coverage collectionz"Could not save pytest-cov data: %s)r-r)r5r0r1r6r7 Exception)rr3 save_errorrrr_save_pytest_cov_dataPs"    r:c Csddlm}td|||rt|}d}|rtdd}z|}|r*|nt}t|}td|t |Wnt yS}z td|WYd}~nd}~ww|sz@d} t |d rit |j d ri|j j } n |rtt |d rt|} | sxd } td | ddlm} | | d} | | }td| Wnt y} z td| WYd} ~ nd} ~ wwz|jt|d} td| WSt y}z td|WYd}~nd}~ww|t|dS)z8Generate LCOV report using either pytest-cov or ddtrace.r)generate_lcov_reportz"Generating LCOV report to file: %sNz:Using pytest-cov instance directly to generate LCOV reportFz5pytest-cov has data in memory: %s (measured %d files)z#Could not check pytest-cov data: %sr data_file data_filenamez .coveragez8No data in memory, attempting to load from data file: %s)Coverage)r<zLoaded data from file: %sz%Could not load coverage data file: %s)outfilez7Generated LCOV report directly from pytest-cov instancez.Direct pytest-cov report generation failed: %s)r/r;r0r1r-get_datameasured_filessetboollenr8r)rr<r=r r>load lcov_reportstr)rtmp_pathis_pytest_cov_enabled_funcr;pytest_cov_instancedatahas_datarAer<r> fresh_cov load_error pct_covered direct_errorrrr_generate_lcov_reportfsZ        rRcCsP|s td|dS|}|std|t|dStdt||S)z2Validate report file exists and read its contents.z$LCOV report file was not created: %sNzLCOV report file is empty: %szRead coverage report: %d bytes)existsr0warning read_bytes_cleanup_temp_filer1rD)rHcoverage_report_bytesrrr_validate_and_read_reports  rXc CsBz|WdSty }z td|WYd}~dSd}~ww)z(Clean up temporary coverage report file.z5Failed to clean up temporary coverage report file: %sN)unlinkr8r0r1)rHrMrrrrVs rVcCs,|r||std|dddSdSdS)zEStop coverage collection if we started it ourselves (not pytest-cov).zStopping coverage.py collectionTr5N)r0r1)stop_coverage_funcrrIrrr_stop_coverage_if_neededs  r\ upload_funcrIr[c CsVztd||rtstdt|ts!tdWdStd||r.t|d}tjddd d  }t|j}Wdn1sHwYzt |||}|dur`td ||ntd |Wn%t y}zd t |vrtd|t |WYd}~WdSd}~wwzBt |} | sWt |t|||WdS|| |} | rtdntdWt |t|||WdSWt |t|||WdSt |t|||wt y*} zCtd| |r ||sz|ddWnt y tjdddYnwWYd} ~ dSWYd} ~ dSWYd} ~ dSWYd} ~ dSd} ~ ww)a Shared coverage report upload handling for pytest plugins. Args: config: pytest config object (to access pluginmanager for pytest-cov) upload_func: Function to call for uploading (signature: upload_func(bytes, format) -> bool) is_pytest_cov_enabled_func: Function to check if pytest-cov is enabled stop_coverage_func: Optional function to stop coverage collection z=Coverage report upload is enabled, checking for coverage datazVpytest-cov is enabled but coverage not running, trying to register pytest-cov instancez8Coverage is not running, skipping coverage report uploadNz;Coverage is running, attempting to generate coverage reportlcovwbz.lcovF)modesuffixdeletez4Generated LCOV coverage report: %s (%.1f%% coverage)zDGenerated LCOV coverage report: %s (coverage percentage unavailable)zNo data to reportz6No coverage data available to generate LCOV report: %sz%Successfully uploaded coverage reportz Failed to upload coverage reportz,Error in coverage report upload handling: %sTrZz#Could not stop coverage after error)exc_info)r0r1rr4r:tempfileNamedTemporaryFilernamerRr8rGrVrXr\) rr]rIr[coverage_formattmp_filerHrP report_errorrWupload_successrMrrrhandle_coverage_reportsv               rk)N)%pathlibrr rdtypingrr&ddtrace.contrib.internal.coverage.datarr/rddtrace.internal.loggerr!ddtrace.internal.settings._configrddtrace.internal.utils.formatsr __name__r0rCrrrrrrGr#r%r-r4r:rRrXrVr\bytesrkrrrrsD          <