tests: update tests to allow facts to call other facts

Author: Fizzadar

tests/facts/server.Mounts/mounts.json

@@ -5,6 +5,9 @@
         "33 29 8:1 / /boot rw,relatime - vfat /dev/sda1 rw,lazytime,fmask=0022,dmask=0022,codepage=437,iocharset=iso8859-1,shortname=mixed,utf8,errors=remount-ro",
         "408 35 0:352 / /mnt/overlay/overlay,dir\\040with\\040spaces\\040and\\040\\134backslashes ro,relatime shared:27 - overlay none ro,lowerdir=lower\\054dir,upperdir=upper\\054dir,workdir=work\\054dir"
     ],
+    "facts": {
+        "server.Kernel": "Linux"
+    },
     "fact": {
         "/": {
             "device": "/dev/sdb1",

tests/operations/server.mount/remount_options.json

@@ -4,6 +4,7 @@
         "options": ["rw", "noatime"]
     },
     "facts": {
+        "server.Kernel": "Linux",
         "server.Mounts": {
             "/data": {
                 "options": ["rw"]

tests/test_facts.py

@@ -6,9 +6,10 @@
 
 from pyinfra.api import StringCommand
 from pyinfra.api.facts import ShortFactBase
+from pyinfra.context import ctx_host, ctx_state
 from pyinfra_cli.util import json_encode
 
-from .util import YamlTest, get_command_string
+from .util import FakeState, YamlTest, create_host, get_command_string
 
 # show full diff on json
 TestCase.maxDiff = None
@@ -35,7 +36,18 @@ class TestTests(TestCase, metaclass=YamlTest):
         yaml_test_dir = path.join("tests", "facts", folder_name)
         yaml_test_prefix = "test_{0}_".format(fact.name)
 
+        @classmethod
+        def setUpClass(cls):
+            # Create a global fake state that attach to context state
+            cls.state = FakeState()
+
         def yaml_test_function(self, test_name, test_data, fact=fact):
+            host = create_host(self.state, facts=test_data.get("facts", {}))
+            with ctx_state.use(self.state):
+                with ctx_host.use(host):
+                    self._test_fn(test_name, test_data, fact)
+
+        def _test_fn(self, test_name, test_data, fact):
             short_fact = None
 
             if isinstance(fact, ShortFactBase):

tests/test_operations.py

@@ -111,7 +111,7 @@ def yaml_test_function(self, test_name, test_data):
             op_test_name = "{0}/{1}.json".format(arg, test_name)
 
             # Create a host with this tests facts and attach to context host
-            host = create_host(facts=test_data.get("facts", {}))
+            host = create_host(self.state, facts=test_data.get("facts", {}))
 
             allowed_exception = test_data.get("exception")
 

tests/util.py

@@ -171,7 +171,8 @@ class FakeHost:
     current_deploy_kwargs = None
     current_deploy_data = None
 
-    def __init__(self, name, facts, data):
+    def __init__(self, state, name, facts, data):
+        self.state = state
         self.name = name
         self.fact = FakeFacts(facts)
         self.data = data
@@ -368,7 +369,7 @@ def walk(self, dirname, topdown=True, onerror=None, followlinks=False):
                 yield recursive_return
 
 
-def create_host(name=None, facts=None, data=None):
+def create_host(state, name=None, facts=None, data=None):
     """
     Creates a FakeHost object with attached fact data.
     """
@@ -380,7 +381,7 @@ def create_host(name=None, facts=None, data=None):
     for name, fact_data in facts.items():
         real_facts[name] = fact_data
 
-    return FakeHost(name, facts=real_facts, data=data)
+    return FakeHost(state, name, facts=real_facts, data=data)
 
 
 class YamlTest(type):