From 6bb3bc5e4a88c1227bbc2dc039b251c909a898db Mon Sep 17 00:00:00 2001
From: "Bradley A. Thornton" <bthornto@redhat.com>
Date: Tue, 1 Aug 2023 09:10:08 -0700
Subject: [PATCH] Fix for ansible 2.9

---
 src/pytest_ansible/units.py | 26 ++++++++++++++++++++++----
 1 file changed, 22 insertions(+), 4 deletions(-)

diff --git a/src/pytest_ansible/units.py b/src/pytest_ansible/units.py
index 1fec0f8f..85ea47ef 100644
--- a/src/pytest_ansible/units.py
+++ b/src/pytest_ansible/units.py
@@ -119,14 +119,17 @@ def inject(start_path: Path) -> None:
     # e.g. ansible-galaxy collection install etc
 
     # Set the environment variable as courtesy for integration tests
+
+    envvar_name = determine_envvar()
     env_paths = os.pathsep.join(paths)
-    logger.info("Setting ANSIBLE_COLLECTIONS_PATH to %s", env_paths)
-    os.environ["ANSIBLE_COLLECTIONS_PATH"] = env_paths
+    logger.info("Setting %s to %s", envvar_name, env_paths)
+    os.environ[envvar_name] = env_paths
 
 
 def inject_only() -> None:
-    """Inject the current ANSIBLE_COLLECTIONS_PATH."""
-    env_paths = os.environ.get("ANSIBLE_COLLECTIONS_PATH", "")
+    """Inject the current ANSIBLE_COLLECTIONS_PATH(S)."""
+    envvar_name = determine_envvar()
+    env_paths = os.environ.get(envvar_name, "")
     path_list = env_paths.split(os.pathsep)
     for path in path_list:
         if path:
@@ -148,3 +151,18 @@ def acf_inject(paths: list[str]) -> None:
         logger.debug("_ACF configured paths: %s", acf._n_configured_paths)
     else:
         logger.debug("_ACF not available")
+
+
+def determine_envvar() -> str:
+    """Use the existance of the AnsibleCollectioFinder to determine
+    the ansible version.
+
+        ansible 2.9 did not have AnsibleCollectionFinder and did not support ANSIBLE_COLLECTIONS_PATH
+    later versions do.
+
+    :returns: The appropriate environment variable to use
+    """
+    if not HAS_COLLECTION_FINDER:
+        return "ANSIBLE_COLLECTIONS_PATHS"
+    else:
+        return "ANSIBLE_COLLECTIONS_PATH"