Adding upstream version 2.14.3.upstream/2.14.3upstream

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>

1 files changed, 43 insertions, 0 deletions

diff --git a/test/lib/ansible_test/_internal/commands/coverage/erase.py b/test/lib/ansible_test/_internal/commands/coverage/erase.py
new file mode 100644
index 0000000..70b685c
--- /dev/null
+++ b/test/lib/ansible_test/_internal/commands/coverage/erase.py
@@ -0,0 +1,43 @@
+"""Erase code coverage files."""
+from __future__ import annotations
+
+import os
+
+from ...util_common import (
+    ResultType,
+)
+
+from ...executor import (
+    Delegate,
+)
+
+from ...provisioning import (
+    prepare_profiles,
+)
+
+from . import (
+    CoverageConfig,
+)
+
+
+def command_coverage_erase(args: CoverageEraseConfig) -> None:
+    """Erase code coverage data files collected during test runs."""
+    host_state = prepare_profiles(args)  # coverage erase
+
+    if args.delegate:
+        raise Delegate(host_state=host_state)
+
+    coverage_dir = ResultType.COVERAGE.path
+
+    for name in os.listdir(coverage_dir):
+        if not name.startswith('coverage') and '=coverage.' not in name:
+            continue
+
+        path = os.path.join(coverage_dir, name)
+
+        if not args.explain:
+            os.remove(path)
+
+
+class CoverageEraseConfig(CoverageConfig):
+    """Configuration for the coverage erase command."""