util/managedfile: replace ssh usage for local case

In case we operate on the local machine, there is no need to force the
user to have a setup where they can connect via SSH to localhost.
Instead implement the symlink functionality via pure python os module
functionality.

Signed-off-by: Rouven Czerwinski <[email protected]>

Author: Emantor

labgrid/util/managedfile.py

@@ -68,18 +68,23 @@ def sync_to_resource(self, symlink=None):
                     f"{self.rpath}{os.path.basename(self.local_path)}"
                 )
         else:
-            conn = sshmanager.open("localhost")
             self.rpath = os.path.dirname(self.local_path) + "/"
 
         if symlink is not None:
             self.logger.info("Linking")
-            try:
-                conn.run_check(f"test ! -e {symlink} -o -L {symlink}")
-            except ExecutionError:
-                raise ManagedFileError(f"Path {symlink} exists but is not a symlink.")
-            # use short options to be compatible with busybox
-            # --symbolic --force --no-dereference
-            conn.run_check(f"ln -sfn {self.rpath}{os.path.basename(self.local_path)} {symlink}")
+            if isinstance(self.resource, NetworkResource):
+                try:
+                    conn.run_check(f"test ! -e {symlink} -o -L {symlink}")
+                except ExecutionError:
+                    raise ManagedFileError(f"Path {symlink} exists but is not a symlink.")
+                # use short options to be compatible with busybox
+                # --symbolic --force --no-dereference
+                conn.run_check(f"ln -sfn {self.rpath}{os.path.basename(self.local_path)} {symlink}")
+            else:
+                if os.path.exists(symlink) and not os.path.islink(symlink):
+                    raise ManagedFileError(f"Path {symlink} exists but is not a symlink.")
+                os.symlink(f"{self.rpath}{os.path.basename(self.local_path)}", symlink)
+
 
     def _on_nfs(self, conn):
         if self._on_nfs_cached is not None:

tests/test_util.py

@@ -354,9 +354,7 @@ def test_local_managedfile(target, tmpdir):
     assert str(t) == mf.get_remote_path()
 
 def test_network_managedfile_no_sync(target, tmpdir):
-    import hashlib
-
-    res = NetworkResource(target, "localhost", "test")
+    res = NetworkResource(target, "test", "localhost")
     t = tmpdir.join("test")
     t.write(
 """
@@ -369,6 +367,33 @@ def test_network_managedfile_no_sync(target, tmpdir):
     with pytest.raises(ManagedFileError, match=expected) as e:
         mf.get_remote_path()
 
+def test_local_managedfile_symlink(target, tmpdir):
+    res = Resource(target, "test")
+    t = tmpdir.join("test")
+    t.write(
+"""
+Test
+"""
+    )
+    mf = ManagedFile(t, res, detect_nfs=False)
+    mf.sync_to_resource(symlink=tmpdir.join("link"))
+
+    assert str(t) == mf.get_remote_path()
+    assert os.path.islink(tmpdir.join("link"))
+
+@pytest.mark.localsshmanager
+def test_remote_managedfile_symlink(target, tmpdir):
+    res = NetworkResource(target, "test", "localhost")
+    t = tmpdir.join("test")
+    t.write(
+"""
+Test
+"""
+    )
+    mf = ManagedFile(t, res, detect_nfs=False)
+    mf.sync_to_resource(symlink=tmpdir.join("link"))
+
+    assert os.path.islink(tmpdir.join("link"))
 
 def test_find_dict():
     dict_a = {"a": {"a.a": {"a.a.a": "a.a.a_val"}}, "b": "b_val"}