pyinfra 3.0.2__py2.py3-none-any.whl → 3.1.1__py2.py3-none-any.whl

tests/test_api/test_api_inventory.py

@@ -51,3 +51,24 @@ class TestInventoryApi(TestCase):
 
         assert inventory.get_host("somehost").data.override_data == "override_data"
         assert inventory.get_host("anotherhost").data.override_data == "override_data"
+
+    def test_inventory_group_data_not_shared(self):
+        group_data = {"test": {}}
+        hosts = ["hosthost", "anotherhost"]
+
+        inventory = Inventory(
+            (hosts, {}),
+            group=(hosts, group_data),
+        )
+
+        hosthost = inventory.get_host("hosthost")
+
+        # Test that modifying host.data.<X> *does not* stick (both on the same
+        # host and also other hosts).
+        hosthost.data.test["hi"] = "no"
+        assert hosthost.data.test == {}
+        assert inventory.get_host("anotherhost").data.test == {}
+
+        # Test that setting host.data.<X> *does* persist
+        hosthost.data.somethingelse = {"hello": "world"}
+        assert hosthost.data.somethingelse == {"hello": "world"}

tests/test_cli/test_cli_exceptions.py

@@ -38,13 +38,13 @@ class TestCliExceptions(TestCase):
     def test_no_fact_module(self):
         self.assert_cli_exception(
             ["my-server.net", "fact", "not_a_module.SomeFact"],
-            "No such module: pyinfra.facts.not_a_module",
+            "No such module: not_a_module",
         )
 
     def test_no_fact_cls(self):
         self.assert_cli_exception(
             ["my-server.net", "fact", "server.NotAFact"],
-            "No such attribute in module pyinfra.facts.server: NotAFact",
+            "No such attribute in module server: NotAFact",
         )
 
 

tests/test_cli/test_cli_util.py

@@ -30,15 +30,13 @@ class TestCliUtil(TestCase):
     def test_setup_no_module(self):
         with self.assertRaises(CliError) as context:
             get_func_and_args(("no.op",))
-        assert context.exception.message == "No such module: pyinfra.operations.no"
+        assert context.exception.message == "No such module: no"
 
     def test_setup_no_op(self):
         with self.assertRaises(CliError) as context:
             get_func_and_args(("server.no",))
 
-        assert (
-            context.exception.message == "No such attribute in module pyinfra.operations.server: no"
-        )
+        assert context.exception.message == "No such attribute in module server: no"
 
     def test_setup_op_and_args(self):
         commands = ("pyinfra.operations.server.user", "one", "two", "hello=world")

tests/test_connectors/test_terraform.py

@@ -20,9 +20,9 @@ class TestTerraformConnector(TestCase):
         with self.assertRaises(InventoryError) as context:
             list(TerraformInventoryConnector.make_names_data("output_key"))
 
-        assert (
-            context.exception.args[0]
-            == "No Terraform output with key: `output_key`, valid keys:\n   - hello.world"
+        assert context.exception.args[0] == (
+            "No Terraform output with key: `output_key`, "
+            "valid keys:\n   - hello\n   - hello.world"
         )
 
     @patch("pyinfra.connectors.terraform.local.shell")
@@ -58,7 +58,7 @@ class TestTerraformConnector(TestCase):
             (
                 "@terraform/somehost",
                 {"ssh_hostname": "somehost"},
-                ["@terraform"],
+                ["@terraform", "all"],
             ),
         ]
 
@@ -71,7 +71,7 @@ class TestTerraformConnector(TestCase):
             (
                 "@terraform/somehost",
                 {"ssh_hostname": "somehost"},
-                ["@terraform"],
+                ["@terraform", "all"],
             ),
         ]
 
@@ -88,7 +88,7 @@ class TestTerraformConnector(TestCase):
             (
                 "@terraform/a name",
                 {"ssh_hostname": "hostname"},
-                ["@terraform"],
+                ["@terraform", "all"],
             ),
         ]