Ansible: Fix for missing group names in get_variables() (#724)

Author: marcandre-larochelle

test/test_backends.py

@@ -183,14 +183,14 @@ def get_vars(host):
             "c": "d",
             "x": "z",
             "inventory_hostname": "debian",
-            "group_names": ["g"],
+            "group_names": ["all", "g"],
             "groups": groups,
         }
         assert get_vars("rockylinux") == {
             "a": "a",
             "e": "f",
             "inventory_hostname": "rockylinux",
-            "group_names": ["ungrouped"],
+            "group_names": ["all", "ungrouped"],
             "groups": groups,
         }
 

test/test_modules.py

@@ -348,7 +348,7 @@ def test_ansible_module(host):
     assert variables["myhostvar"] == "bar"
     assert variables["mygroupvar"] == "qux"
     assert variables["inventory_hostname"] == "debian_bookworm"
-    assert variables["group_names"] == ["testgroup"]
+    assert variables["group_names"] == ["all", "testgroup"]
     assert variables["groups"] == {
         "all": ["debian_bookworm"],
         "testgroup": ["debian_bookworm"],

testinfra/utils/ansible_runner.py

@@ -311,12 +311,14 @@ def get_variables(self, host: str) -> dict[str, Any]:
         hostvars.setdefault("inventory_hostname", host)
         group_names = []
         groups = {}
+
         for group in sorted(inventory):
             if group == "_meta":
                 continue
             groups[group] = sorted(itergroup(inventory, group))
-            if group != "all" and host in inventory[group].get("hosts", []):
+            if host in groups[group]:
                 group_names.append(group)
+
         hostvars.setdefault("group_names", group_names)
         hostvars.setdefault("groups", groups)
         return hostvars