From 45b5685037f27a642161e48bb81b08bccdee277e Mon Sep 17 00:00:00 2001
From: Matt Clay <matt@mystile.com>
Date: Fri, 21 Sep 2018 10:33:35 -0700
Subject: [PATCH] Add python.py coverage injector for ansible-test.

This can be used to run Python scripts from the repository with the
correct interpreter and allow collection of code coverage.

Useful for testing contrib inventory scripts.
---
 test/runner/injector/injector.py | 16 ++++++++++++++++
 test/runner/injector/python.py   |  1 +
 2 files changed, 17 insertions(+)
 create mode 120000 test/runner/injector/python.py

diff --git a/test/runner/injector/injector.py b/test/runner/injector/injector.py
index 3bb9b9c22f3..a7f107b0645 100755
--- a/test/runner/injector/injector.py
+++ b/test/runner/injector/injector.py
@@ -105,6 +105,8 @@ def main():
 
         if os.path.basename(__file__) == 'injector.py':
             args, env = runner()  # code coverage collection is baked into the AnsiballZ wrapper when needed
+        elif os.path.basename(__file__) == 'python.py':
+            args, env = python()  # run arbitrary python commands using the correct python and with optional code coverage
         else:
             args, env = injector()
 
@@ -119,6 +121,20 @@ def main():
         raise
 
 
+def python():
+    """
+    :rtype: list[str], dict[str, str]
+    """
+    if config.coverage_file:
+        args, env = coverage_command()
+    else:
+        args, env = [config.python_interpreter], os.environ.copy()
+
+    args += config.arguments[1:]
+
+    return args, env
+
+
 def injector():
     """
     :rtype: list[str], dict[str, str]
diff --git a/test/runner/injector/python.py b/test/runner/injector/python.py
new file mode 120000
index 00000000000..1f9d09cbf2a
--- /dev/null
+++ b/test/runner/injector/python.py
@@ -0,0 +1 @@
+injector.py
\ No newline at end of file