B ó±ô`0 ã@sªdZddlmZmZmZeZddlZddlZddl Z ddl m Z m Z m Z ddlmZddlmZmZddlmZejd d „ƒZd d „Zd d„Zdd„Zdd„Zdd„ZdS)zMUtility code for facilitating collection of code coverage when running tests.é)Úabsolute_importÚdivisionÚprint_functionNé)ÚIntegrationConfigÚ SanityConfigÚ TestConfig)Úwrite_text_file)ÚCOVERAGE_CONFIG_NAMEÚ remove_tree)Ú data_contextccs"t|ƒz dVWdt|ƒXdS)zEContent to set up and clean up code coverage configuration for tests.N)Úcoverage_setupÚcoverage_cleanup)Úargs©rúS/home/dcms/DCMS/lib/python3.7/site-packages/ansible_test/_internal/coverage_util.pyÚcoverage_contexts rcCsD|js dSt|ƒ}|jr d|_n t ¡|_ttj  |jt ¡|ƒdS)z8Set up code coverage configuration before running tests.Nz/tmp/coverage-temp-dir) ZcoverageÚgenerate_coverage_configÚexplainÚcoverage_config_base_pathÚtempfileÚmkdtempr ÚosÚpathÚjoinr )rÚcoverage_configrrrr (s r cCs |jr|jst|jƒd|_dS)z?Clean up code coverage configuration after tests have finished.N)rrr )rrrrr7s  rcCstƒjjrt|ƒ}ntƒ}|S)z/Generate code coverage configuration for tests.)r ÚcontentÚ collectionÚ#generate_collection_coverage_configÚ generate_ansible_coverage_config)rrrrrr>s  rcCsd}|S)z7Generate code coverage configuration for Ansible tests.zé [run] branch = True concurrency = multiprocessing parallel = True omit = */python*/dist-packages/* */python*/site-packages/* */python*/distutils/* */pyshared/* */pytest */AnsiballZ_*.py */test/results/* r)rrrrrHsrcCs|d}t|tƒr.|dtƒjjtƒjjjf7}nJt|tƒrf|dtƒjjtj   tƒjjtƒjj ¡f7}n|dtƒjj7}|S)zBGenerate code coverage configuration for Ansible Collection tests.zl [run] branch = True concurrency = multiprocessing parallel = True disable_warnings = no-data-collected z include = %s/* */%s/* z% include = %s/* omit = %s/* z include = %s/* ) Ú isinstancerr rÚrootrÚ directoryrrrrZ results_path)rrrrrr]s   ,r)Ú__doc__Ú __future__rrrÚtypeÚ __metaclass__Ú contextlibrrÚconfigrrrÚior Úutilr r Údatar Úcontextmanagerrr rrrrrrrrÚs