test SSHExecutor.from_machine

iterative · Oct 21, 2021 · d553e9e · d553e9e
1 parent ed3a172
commit d553e9e
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/tests/func/experiments/executor/test_ssh.py b/tests/func/experiments/executor/test_ssh.py
@@ -8,6 +8,7 @@
 from dvc.path_info import URLInfo
 from dvc.repo.experiments.base import EXEC_HEAD, EXEC_MERGE
 from dvc.repo.experiments.executor.ssh import SSHExecutor
+from tests.func.machine.conftest import *  # noqa, pylint: disable=wildcard-import
 from tests.remotes.ssh import TEST_SSH_KEY_PATH, TEST_SSH_USER
 
 
@@ -21,6 +22,12 @@ def _ssh_factory(cloud):
     )
 
 
+def test_from_machine(tmp_dir, scm, dvc, machine_instance, mocker):
+    mocker.patch.object(SSHExecutor, "_init_git")
+    executor = SSHExecutor.from_machine(dvc.machine, "foo", scm, "")
+    assert executor.host == machine_instance["instance_ip"]
+
+
 @pytest.mark.needs_internet
 @pytest.mark.parametrize("cloud", [pytest.lazy_fixture("git_ssh")])
 def test_init_git(tmp_dir, scm, cloud):