From ea49635728714696e178794d957cd67a24477dc0 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?C=C3=A9sar=20Francisco=20San=20Nicol=C3=A1s=20Mart=C3=ADne?=
 =?UTF-8?q?z?= <csannico@redhat.com>
Date: Thu, 30 Nov 2023 17:22:51 +0100
Subject: [PATCH] Testing with v2 coverage

---
 src/pytest_ansible/host_manager/v2.py      |  29 +++
 src/pytest_ansible/module_dispatcher/v2.py | 236 +++++++++++++++++++++
 2 files changed, 265 insertions(+)
 create mode 100644 src/pytest_ansible/host_manager/v2.py
 create mode 100644 src/pytest_ansible/module_dispatcher/v2.py

diff --git a/src/pytest_ansible/host_manager/v2.py b/src/pytest_ansible/host_manager/v2.py
new file mode 100644
index 00000000..cb549c4f
--- /dev/null
+++ b/src/pytest_ansible/host_manager/v2.py
@@ -0,0 +1,29 @@
+"""Fixme."""
+from ansible.inventory import Inventory
+from ansible.parsing.dataloader import DataLoader
+from ansible.vars import VariableManager
+
+from pytest_ansible.host_manager import BaseHostManager
+from pytest_ansible.module_dispatcher.v2 import ModuleDispatcherV2
+
+
+class HostManagerV2(BaseHostManager):
+    """Fixme."""
+
+    def __init__(self, *args, **kwargs) -> None:
+        """Fixme."""
+        super().__init__(*args, **kwargs)
+        self._dispatcher = ModuleDispatcherV2
+
+    def initialize_inventory(self):
+        """Fixme."""
+        self.options["loader"] = DataLoader()
+        self.options["variable_manager"] = VariableManager()
+        self.options["inventory_manager"] = Inventory(
+            loader=self.options["loader"],
+            variable_manager=self.options["variable_manager"],
+            host_list=self.options["inventory"],
+        )
+        self.options["variable_manager"].set_inventory(
+            self.options["inventory_manager"],
+        )
diff --git a/src/pytest_ansible/module_dispatcher/v2.py b/src/pytest_ansible/module_dispatcher/v2.py
new file mode 100644
index 00000000..e7cf93ae
--- /dev/null
+++ b/src/pytest_ansible/module_dispatcher/v2.py
@@ -0,0 +1,236 @@
+"""Fixme."""
+from __future__ import annotations
+
+import warnings
+
+from typing import TYPE_CHECKING
+
+import ansible.constants
+import ansible.errors
+import ansible.utils
+
+from ansible.cli import CLI
+from ansible.executor.task_queue_manager import TaskQueueManager
+from ansible.playbook.play import Play
+from ansible.plugins.callback import CallbackBase
+from ansible.plugins.loader import module_loader
+
+from pytest_ansible.errors import AnsibleConnectionFailure
+from pytest_ansible.has_version import has_ansible_v2
+from pytest_ansible.module_dispatcher import BaseModuleDispatcher
+from pytest_ansible.results import AdHocResult
+
+
+if not has_ansible_v2:
+    msg = "Only supported with ansible-2.* and newer"
+    raise ImportError(msg)
+
+
+class ResultAccumulator(CallbackBase):
+    """Fixme."""
+
+    def __init__(self, *args, **kwargs) -> None:
+        """Initialize object."""
+        super().__init__(*args, **kwargs)
+        self.contacted = {}
+        self.unreachable = {}
+
+    def v2_runner_on_failed(self, result, *args, **kwargs):
+        """Fixme."""
+        self.contacted[result._host.get_name()] = result._result
+
+    v2_runner_on_ok = v2_runner_on_failed
+
+    def v2_runner_on_unreachable(self, result):
+        """Fixme."""
+        self.unreachable[result._host.get_name()] = result._result
+
+    @property
+    def results(self):
+        """Fixme."""
+        return {"contacted": self.contacted, "unreachable": self.unreachable}
+
+
+class ModuleDispatcherV2(BaseModuleDispatcher):
+    """Pass."""
+
+    if TYPE_CHECKING:
+        from collections.abc import Sequence
+
+    required_kwargs: Sequence[str] = (
+        "inventory",
+        "inventory_manager",
+        "variable_manager",
+        "host_pattern",
+        "loader",
+    )
+
+    def has_module(self, name):
+        """Fixme."""
+        # Make sure we parse module_path and pass it to the loader,
+        # otherwise, only built-in modules will work.
+        if "module_path" in self.options:
+            paths = self.options["module_path"]
+            if isinstance(paths, (list, tuple, set)):
+                for path in paths:
+                    module_loader.add_directory(path)
+            else:
+                module_loader.add_directory(paths)
+
+        return module_loader.has_plugin(name)
+
+    def _run(self, *module_args, **complex_args):
+        """Execute an ansible adhoc command returning the result in a AdhocResult object."""
+        # Assemble module argument string
+        if module_args:
+            complex_args.update({"_raw_params": " ".join(module_args)})
+
+        # Assert hosts matching the provided pattern exist
+        hosts = self.options["inventory_manager"].list_hosts()
+        if self.options.get("extra_inventory_manager", None):
+            extra_hosts = self.options["extra_inventory_manager"].list_hosts()
+        else:
+            extra_hosts = []
+        no_hosts = False
+        if len(hosts + extra_hosts) == 0:
+            no_hosts = True
+            warnings.warn("provided hosts list is empty, only localhost is available")
+
+        self.options["inventory_manager"].subset(self.options.get("subset"))
+        hosts = self.options["inventory_manager"].list_hosts(
+            self.options["host_pattern"],
+        )
+
+        if self.options.get("extra_inventory_manager", None):
+            self.options["extra_inventory_manager"].subset(self.options.get("subset"))
+            extra_hosts = self.options["extra_inventory_manager"].list_hosts()
+        else:
+            extra_hosts = []
+        if len(hosts + extra_hosts) == 0 and not no_hosts:
+            raise ansible.errors.AnsibleError(
+                "Specified hosts and/or --limit does not match any hosts.",
+            )
+
+        # pylint: disable=no-member
+        parser = CLI.base_parser(
+            runas_opts=True,
+            inventory_opts=True,
+            async_opts=True,
+            output_opts=True,
+            connect_opts=True,
+            check_opts=True,
+            runtask_opts=True,
+            vault_opts=True,
+            fork_opts=True,
+            module_opts=True,
+        )
+        (options) = parser.parse_args([])
+
+        # Pass along cli options
+        options.verbosity = 5
+        options.connection = self.options.get("connection")
+        options.remote_user = self.options.get("user")
+        options.become = self.options.get("become")
+        options.become_method = self.options.get("become_method")
+        options.become_user = self.options.get("become_user")
+        options.module_path = self.options.get("module_path")
+
+        # Initialize callback to capture module JSON responses
+        callback = ResultAccumulator()
+
+        kwargs = {
+            "inventory": self.options["inventory_manager"],
+            "variable_manager": self.options["variable_manager"],
+            "loader": self.options["loader"],
+            "options": options,
+            "stdout_callback": callback,
+            "passwords": {"conn_pass": None, "become_pass": None},
+        }
+
+        kwargs_extra = {}
+        # If we have an extra inventory, do the same that we did for the inventory
+        if self.options.get("extra_inventory_manager", None):
+            callback_extra = ResultAccumulator()
+
+            kwargs_extra = {
+                "inventory": self.options["extra_inventory_manager"],
+                "variable_manager": self.options["extra_variable_manager"],
+                "loader": self.options["extra_loader"],
+                "stdout_callback": callback_extra,
+                "passwords": {"conn_pass": None, "become_pass": None},
+            }
+
+        # create a pseudo-play to execute the specified module via a single task
+        play_ds = {
+            "name": "pytest-ansible",
+            "hosts": self.options["host_pattern"],
+            "gather_facts": "no",
+            "tasks": [
+                {
+                    "action": {
+                        "module": self.options["module_name"],
+                        "args": complex_args,
+                    },
+                },
+            ],
+        }
+
+        play = Play().load(
+            play_ds,
+            variable_manager=self.options["variable_manager"],
+            loader=self.options["loader"],
+        )
+
+        play_extra = None
+        if self.options.get("extra_inventory_manager", None):
+            play_extra = Play().load(
+                play_ds,
+                variable_manager=self.options["extra_variable_manager"],
+                loader=self.options["extra_loader"],
+            )
+
+        # now create a task queue manager to execute the play
+        tqm = None
+        try:
+            tqm = TaskQueueManager(**kwargs)
+            tqm.run(play)
+        finally:
+            if tqm:
+                tqm.cleanup()
+
+        if self.options.get("extra_inventory_manager", None):
+            tqm_extra = None
+            try:
+                tqm_extra = TaskQueueManager(**kwargs_extra)
+                tqm_extra.run(play_extra)
+            finally:
+                if tqm_extra:
+                    tqm_extra.cleanup()
+
+        # Raise exception if host(s) unreachable
+        if callback.unreachable:
+            msg = "Host unreachable"
+            raise AnsibleConnectionFailure(
+                msg,
+                dark=callback.unreachable,
+                contacted=callback.contacted,
+            )
+
+        if (
+            self.options.get("extra_inventory_manager", None)
+            and callback_extra.unreachable
+        ):
+            raise AnsibleConnectionFailure(
+                "Host unreachable in the extra inventory",
+                dark=callback_extra.unreachable,
+                contacted=callback_extra.contacted,
+            )
+
+        # Success!
+        return AdHocResult(
+            contacted=(
+                {**callback.contacted, **callback_extra.contacted}
+                if self.options.get("extra_inventory_manager", None)
+                else callback.contacted
+            ),
+        )