o i@s"dZddlZddlZddlmZddlZddlmZ ddlm Z ddlm Z ddlm ZddlmZdd lmZdd lmZddlZdd lmZdd lmZeeZeeZd eddfddZGdddZejdej eddffddZ!ddZ"ddZ#ede$dej%e&fddZdS)z API for code coverage collection for use by ddtrace.testing. The rest of ddtrace.testing should only use the interface exposed in this file to set up code coverage and get coverage data. N)Path)get_coverage_percentage)patch)run_coverage_report)unpatch)$_is_coverage_invoked_by_coverage_run)_is_coverage_patched)ModuleCodeCollector) CoverageLines)catch_and_log_exceptionsworkspace_pathreturncCs"tjjjj|gddtdS)NT) include_pathscollect_import_time_coverage)ddtraceinternalcoverage installerinstallr start_coverage)r r`/home/ubuntu/.local/lib/python3.10/site-packages/ddtrace/testing/internal/tracer_api/coverage.pyinstall_coverages rc@s6eZdZdddZdedejeee ffddZ dS) CoverageDatar NcCs d|_dSN)_covered_lines)selfrrr__init__%s zCoverageData.__init__ relative_toc csb|jsdS|jD]#\}}z t||}Wn ty!Yq wd|}||fVq dS)N/)ritemsrr ValueErrorto_bytes)rr absolute_path covered_lines relative_pathpath_strrrrget_coverage_bitmaps(s  z!CoverageData.get_coverage_bitmaps)r N) __name__ __module__ __qualname__rrtIterabletuplestrbytesr'rrrrr$s $rccsFt}t}|V||_WddS1swYdSr)r CollectInContextrget_covered_linesr)coverage_collector coverage_datarrrcoverage_collection6s   "r4cC tdS)zJ Patch coverage.py to obtain coverage percentage from pytest-cov. N)patch_coveragerrrrinstall_coverage_percentage> r7cCr5)z' Undo patching of coverage.py. N)unpatch_coveragerrrruninstall_coverage_percentageEr8r:pytest_cov_statuscCst|s tr tr tt}|durtddSt|ttfs0t dt |j t |j |dStd|t|S)a Retrieve coverage percentage collected during a pytest-cov test session, if available. This retrieves the coverage percentage from coverage.py patching, coverage report upload, or generates a report if using 'coverage run' without pytest-cov. Nz5Unable to retrieve coverage data for the session spanzDUnexpected format for total covered percentage: type=%s.%s, value=%rzCode coverage: %s%%) rrr_get_coverage_percentagelogdebug isinstancefloatintwarningtyper)r()r;lines_pct_valuerrrrLs    r)'__doc__ contextlibloggingpathlibrtypingr+'ddtrace.contrib.internal.coverage.patchrr<rr6rrr9'ddtrace.contrib.internal.coverage.utilsrrddtrace.internal.coverage.coder #ddtrace.internal.coverage.installerr/ddtrace.internal.test_visibility.coverage_linesr ddtrace.testing.internal.loggingr getLoggerr(r=rrcontextmanager Generatorr4r7r:boolOptionalr@rrrrs2