pyinfra 3.3__py2.py3-none-any.whl → 3.4__py2.py3-none-any.whl

tests/test_api/test_api_facts.py

@@ -1,11 +1,13 @@
+from typing import cast
 from unittest.mock import MagicMock, patch
 
 from pyinfra.api import Config, State
-from pyinfra.api.arguments import CONNECTOR_ARGUMENT_KEYS, pop_global_arguments
+from pyinfra.api.arguments import CONNECTOR_ARGUMENT_KEYS, AllArguments, pop_global_arguments
 from pyinfra.api.connect import connect_all
 from pyinfra.api.exceptions import PyinfraError
 from pyinfra.api.facts import get_facts
 from pyinfra.connectors.util import CommandOutput, OutputLine
+from pyinfra.context import ctx_host, ctx_state
 from pyinfra.facts.server import Arch, Command
 
 from ..paramiko_util import PatchSSHTestCase
@@ -13,7 +15,9 @@ from ..util import make_inventory
 
 
 def _get_executor_defaults(state, host):
-    global_argument_defaults, _ = pop_global_arguments({}, state=state, host=host)
+    with ctx_state.use(state):
+        with ctx_host.use(host):
+            global_argument_defaults, _ = pop_global_arguments(state, host, {})
     return {
         key: value
         for key, value in global_argument_defaults.items()
@@ -52,13 +56,16 @@ class TestFactsApi(PatchSSHTestCase):
         anotherhost = inventory.get_host("anotherhost")
 
         connect_all(state)
-        anotherhost.current_op_global_arguments = {
-            "_sudo": True,
-            "_sudo_user": "someuser",
-            "_su_user": "someuser",
-            "_timeout": 10,
-            "_env": {"HELLO": "WORLD"},
-        }
+        anotherhost.current_op_global_arguments = cast(
+            AllArguments,
+            {
+                "_sudo": True,
+                "_sudo_user": "someuser",
+                "_su_user": "someuser",
+                "_timeout": 10,
+                "_env": {"HELLO": "WORLD"},
+            },
+        )
 
         with patch("pyinfra.connectors.ssh.SSHConnector.run_shell_command") as fake_run_command:
             fake_run_command.return_value = True, CommandOutput(
@@ -109,9 +116,12 @@ class TestFactsApi(PatchSSHTestCase):
 
         connect_all(state)
         anotherhost.in_op = True
-        anotherhost.current_op_global_arguments = {
-            "_ignore_errors": True,
-        }
+        anotherhost.current_op_global_arguments = cast(
+            AllArguments,
+            {
+                "_ignore_errors": True,
+            },
+        )
 
         with patch("pyinfra.connectors.ssh.SSHConnector.run_shell_command") as fake_run_command:
             fake_run_command.return_value = False, MagicMock()
@@ -188,9 +198,12 @@ class TestFactsApi(PatchSSHTestCase):
         anotherhost.data._sudo_user = "this-should-be-overridden"
         anotherhost.data._su_user = "this-should-be-overridden"
 
-        anotherhost.current_op_global_arguments = {
-            "_su_user": "override-su-user",
-        }
+        anotherhost.current_op_global_arguments = cast(
+            AllArguments,
+            {
+                "_su_user": "override-su-user",
+            },
+        )
 
         connect_all(state)
 

tests/test_api/test_api_operations.py

@@ -426,32 +426,31 @@ class TestNestedOperationsApi(PatchSSHTestCase):
 
         somehost = inventory.get_host("somehost")
 
-        ctx_state.set(state)
-        ctx_host.set(somehost)
-
         pyinfra.is_cli = True
 
-        try:
-            outer_result = server.shell(commands="echo outer")
-            assert outer_result._combined_output is None
+        with ctx_state.use(state):
+            with ctx_host.use(somehost):
+                try:
+                    outer_result = server.shell(commands="echo outer")
+                    assert outer_result._combined_output is None
 
-            def callback():
-                inner_result = server.shell(commands="echo inner")
-                assert inner_result._combined_output is not None
+                    def callback():
+                        inner_result = server.shell(commands="echo inner")
+                        assert inner_result._combined_output is not None
 
-            python.call(function=callback)
+                    python.call(function=callback)
 
-            assert len(state.get_op_order()) == 2
+                    assert len(state.get_op_order()) == 2
 
-            run_ops(state)
+                    run_ops(state)
 
-            assert len(state.get_op_order()) == 3
-            assert state.results[somehost].success_ops == 3
-            assert outer_result._combined_output is not None
+                    assert len(state.get_op_order()) == 3
+                    assert state.results[somehost].success_ops == 3
+                    assert outer_result._combined_output is not None
 
-            disconnect_all(state)
-        finally:
-            pyinfra.is_cli = False
+                    disconnect_all(state)
+                finally:
+                    pyinfra.is_cli = False
 
 
 class TestOperationFailures(PatchSSHTestCase):
@@ -519,40 +518,40 @@ class TestOperationOrdering(PatchSSHTestCase):
         state.current_deploy_filename = __file__
 
         pyinfra.is_cli = True
-        ctx_state.set(state)
 
-        # Add op to both hosts
-        for name in ("anotherhost", "somehost"):
-            ctx_host.set(inventory.get_host(name))
-            server.shell("echo hi")  # note this is called twice but on *the same line*
+        with ctx_state.use(state):
+            # Add op to both hosts
+            for name in ("anotherhost", "somehost"):
+                with ctx_host.use(inventory.get_host(name)):
+                    server.shell("echo hi")  # note this is called twice but on *the same line*
 
-        # Add op to just the second host - using the context modules such that
-        # it replicates a deploy file.
-        ctx_host.set(inventory.get_host("anotherhost"))
-        first_context_hash = server.user("anotherhost_user")._hash
+            # Add op to just the second host - using the context modules such that
+            # it replicates a deploy file.
+            ctx_host.set(inventory.get_host("anotherhost"))
+            first_context_hash = server.user("anotherhost_user")._hash
 
-        # Add op to just the first host - using the context modules such that
-        # it replicates a deploy file.
-        ctx_host.set(inventory.get_host("somehost"))
-        second_context_hash = server.user("somehost_user")._hash
+            # Add op to just the first host - using the context modules such that
+            # it replicates a deploy file.
+            ctx_host.set(inventory.get_host("somehost"))
+            second_context_hash = server.user("somehost_user")._hash
 
-        ctx_state.reset()
-        ctx_host.reset()
+            ctx_state.reset()
+            ctx_host.reset()
 
-        pyinfra.is_cli = False
+            pyinfra.is_cli = False
 
-        print(state.ops)
-        # Ensure there are two ops
-        op_order = state.get_op_order()
-        assert len(op_order) == 3
+            print(state.ops)
+            # Ensure there are two ops
+            op_order = state.get_op_order()
+            assert len(op_order) == 3
 
-        # And that the two ops above were called in the expected order
-        assert op_order[1] == first_context_hash
-        assert op_order[2] == second_context_hash
+            # And that the two ops above were called in the expected order
+            assert op_order[1] == first_context_hash
+            assert op_order[2] == second_context_hash
 
-        # Ensure somehost has two ops and anotherhost only has the one
-        assert len(state.ops[inventory.get_host("somehost")]) == 2
-        assert len(state.ops[inventory.get_host("anotherhost")]) == 2
+            # Ensure somehost has two ops and anotherhost only has the one
+            assert len(state.ops[inventory.get_host("somehost")]) == 2
+            assert len(state.ops[inventory.get_host("anotherhost")]) == 2
 
     # In API mode, pyinfra *overrides* the line numbers such that whenever an
     # operation or deploy is added it is simply appended. This makes sense as

tests/test_cli/test_cli.py

@@ -10,10 +10,10 @@ from .util import run_cli
 class TestCliEagerFlags(TestCase):
     def test_print_help(self):
         result = run_cli("--version")
-        assert result.exit_code == 0, result.stdout
+        assert result.exit_code == 0, result.stderr
 
         result = run_cli("--help")
-        assert result.exit_code == 0, result.stdout
+        assert result.exit_code == 0, result.stderr
 
 
 class TestOperationCli(PatchSSHTestCase):
@@ -22,7 +22,7 @@ class TestOperationCli(PatchSSHTestCase):
             path.join("tests", "test_cli", "deploy", "inventories", "inventory.py"),
             "not_a_module.shell",
         )
-        assert result.exit_code == 1, result.stdout
+        assert result.exit_code == 1, result.stderr
         assert "No such module: not_a_module"
 
     def test_invalid_operation_function(self):
@@ -30,7 +30,7 @@ class TestOperationCli(PatchSSHTestCase):
             path.join("tests", "test_cli", "deploy", "inventories", "inventory.py"),
             "server.not_an_operation",
         )
-        assert result.exit_code == 1, result.stdout
+        assert result.exit_code == 1, result.stderr
         assert "No such operation: server.not_an_operation"
 
     def test_deploy_operation(self):
@@ -40,7 +40,7 @@ class TestOperationCli(PatchSSHTestCase):
             "server.shell",
             "echo hi",
         )
-        assert result.exit_code == 0, result.stdout
+        assert result.exit_code == 0, result.stderr
 
     def test_deploy_operation_with_all(self):
         result = run_cli(
@@ -49,7 +49,7 @@ class TestOperationCli(PatchSSHTestCase):
             "server.shell",
             "echo hi",
         )
-        assert result.exit_code == 0, result.stdout
+        assert result.exit_code == 0, result.stderr
 
     def test_deploy_operation_json_args(self):
         result = run_cli(
@@ -58,7 +58,7 @@ class TestOperationCli(PatchSSHTestCase):
             "server.shell",
             '[["echo hi"], {}]',
         )
-        assert result.exit_code == 0, result.stdout
+        assert result.exit_code == 0, result.stderr
 
 
 class TestFactCli(PatchSSHTestCase):
@@ -68,8 +68,8 @@ class TestFactCli(PatchSSHTestCase):
             "fact",
             "server.Os",
         )
-        assert result.exit_code == 0, result.stdout
-        assert '"somehost": null' in result.stdout
+        assert result.exit_code == 0, result.stderr
+        assert '"somehost": null' in result.stderr
 
     def test_get_fact_with_kwargs(self):
         result = run_cli(
@@ -78,8 +78,8 @@ class TestFactCli(PatchSSHTestCase):
             "files.File",
             "path=.",
         )
-        assert result.exit_code == 0, result.stdout
-        assert '"somehost": null' in result.stdout
+        assert result.exit_code == 0, result.stderr
+        assert '"somehost": null' in result.stderr
 
 
 class TestExecCli(PatchSSHTestCase):
@@ -90,7 +90,7 @@ class TestExecCli(PatchSSHTestCase):
             "--",
             "echo hi",
         )
-        assert result.exit_code == 0, result.stdout
+        assert result.exit_code == 0, result.stderr
 
     def test_exec_command_with_options(self):
         result = run_cli(
@@ -108,7 +108,7 @@ class TestExecCli(PatchSSHTestCase):
             "--",
             "echo hi",
         )
-        assert result.exit_code == 0, result.stdout
+        assert result.exit_code == 0, result.stderr
 
     def test_exec_command_with_serial(self):
         result = run_cli(
@@ -118,7 +118,7 @@ class TestExecCli(PatchSSHTestCase):
             "--",
             "echo hi",
         )
-        assert result.exit_code == 0, result.stdout
+        assert result.exit_code == 0, result.stderr
 
     def test_exec_command_with_no_wait(self):
         result = run_cli(
@@ -128,7 +128,7 @@ class TestExecCli(PatchSSHTestCase):
             "--",
             "echo hi",
         )
-        assert result.exit_code == 0, result.stdout
+        assert result.exit_code == 0, result.stderr
 
     def test_exec_command_with_debug_operations(self):
         result = run_cli(
@@ -138,7 +138,7 @@ class TestExecCli(PatchSSHTestCase):
             "--",
             "echo hi",
         )
-        assert result.exit_code == 0, result.stdout
+        assert result.exit_code == 0, result.stderr
 
     def test_exec_command_with_debug_facts(self):
         result = run_cli(
@@ -148,7 +148,7 @@ class TestExecCli(PatchSSHTestCase):
             "--",
             "echo hi",
         )
-        assert result.exit_code == 0, result.stdout
+        assert result.exit_code == 0, result.stderr
 
 
 class TestDirectMainExecution(PatchSSHTestCase):

tests/test_cli/test_cli_inventory.py

@@ -80,7 +80,7 @@ class TestCliInventory(PatchSSHTestCase):
         assert result.exit_code == 0, result.stdout
         assert (
             'Ignoring variable "_hosts" in inventory file since it starts with a leading underscore'
-            in result.stdout
+            in result.stderr
         )
         assert inventory.hosts == {}
 
@@ -97,9 +97,9 @@ class TestCliInventory(PatchSSHTestCase):
         )
 
         assert result.exit_code == 0, result.stdout
-        assert 'Ignoring variable "dict_hosts" in inventory file' in result.stdout, result.stdout
+        assert 'Ignoring variable "dict_hosts" in inventory file' in result.stderr, result.stdout
         assert (
-            'Ignoring variable "generator_hosts" in inventory file' in result.stdout
+            'Ignoring variable "generator_hosts" in inventory file' in result.stderr
         ), result.stdout
         assert inventory.hosts == {}
 
@@ -115,5 +115,5 @@ class TestCliInventory(PatchSSHTestCase):
         )
 
         assert result.exit_code == 0, result.stdout
-        assert 'Ignoring host group "issue_662"' in result.stdout, result.stdout
+        assert 'Ignoring host group "issue_662"' in result.stderr, result.stdout
         assert inventory.hosts == {}

tests/test_cli/test_context_objects.py

@@ -5,8 +5,8 @@ from pyinfra.api import Host, Inventory
 from pyinfra.context import ctx_host, ctx_inventory
 
 
-def _create_host():
-    return Host(None, None, None, None)
+def _create_host(name: str = "host"):
+    return Host(name, Inventory(([name], {})), None, None)
 
 
 class TestHostContextObject(TestCase):
@@ -15,44 +15,44 @@ class TestHostContextObject(TestCase):
 
     def test_context_host_eq(self):
         host_obj = _create_host()
-        ctx_host.set(host_obj)
-        assert host == host_obj
+        with ctx_host.use(host_obj):
+            assert host == host_obj
 
     def test_context_host_repr(self):
-        host_obj = _create_host()
-        ctx_host.set(host_obj)
-        assert repr(host) == "ContextObject(Host):Host(None)"
+        host_obj = _create_host("somehost")
+        with ctx_host.use(host_obj):
+            assert repr(host) == "ContextObject(Host):Host(somehost)"
 
     def test_context_host_str(self):
         host_obj = _create_host()
-        ctx_host.set(host_obj)
-        assert str(host_obj) == "None"
+        with ctx_host.use(host_obj):
+            assert str(host_obj) == "host"
 
     def test_context_host_attr(self):
         host_obj = _create_host()
-        ctx_host.set(host_obj)
 
         with self.assertRaises(AttributeError):
             host_obj.hello
 
-        setattr(host, "hello", "world")
-        assert host_obj.hello == host.hello
+        with ctx_host.use(host_obj):
+            setattr(host, "hello", "world")
+            assert host_obj.hello == host.hello
 
     def test_context_host_class_attr(self):
         host_obj = _create_host()
-        ctx_host.set(host_obj)
-        assert ctx_host.isset() is True
 
-        with self.assertRaises(AttributeError):
-            host_obj.hello
+        with ctx_host.use(host_obj):
+            assert ctx_host.isset() is True
+
+            with self.assertRaises(AttributeError):
+                host_obj.hello
 
-        setattr(Host, "hello", "class_world")
-        setattr(host_obj, "hello", "instance_world")
+            setattr(Host, "hello", "class_world")
+            setattr(host_obj, "hello", "instance_world")
 
-        assert host.hello == host.hello
+            assert host.hello == host.hello
 
         # Reset and check fallback to class variable
-        ctx_host.reset()
         assert ctx_host.isset() is False
         assert host.hello == "class_world"
 
@@ -60,14 +60,14 @@ class TestHostContextObject(TestCase):
 class TestInventoryContextObject(TestCase):
     def test_context_inventory_len(self):
         inventory_obj = Inventory(("host", "anotherhost"))
-        ctx_inventory.set(inventory_obj)
-        assert ctx_inventory.isset() is True
 
-        assert len(inventory) == len(inventory_obj)
+        with ctx_inventory.use(inventory_obj):
+            assert ctx_inventory.isset() is True
+            assert len(inventory) == len(inventory_obj)
 
     def test_context_inventory_iter(self):
         inventory_obj = Inventory(("host", "anotherhost"))
-        ctx_inventory.set(inventory_obj)
-        assert ctx_inventory.isset() is True
 
-        assert list(iter(inventory)) == list(iter(inventory_obj))
+        with ctx_inventory.use(inventory_obj):
+            assert ctx_inventory.isset() is True
+            assert list(iter(inventory)) == list(iter(inventory_obj))