Use mock folder for ssh config testing (#523)

aiidalab_widgets_base/computational_resources.py

@@ -12,7 +12,7 @@
 import traitlets
 from aiida import common, orm, plugins
 from aiida.orm.utils.builders.computer import ComputerBuilder
-from aiida.transports.plugins.ssh import parse_sshconfig
+from aiida.transports.plugins import ssh as aiida_ssh_plugin
 from humanfriendly import InvalidSize, parse_size
 from IPython.display import clear_output, display
 
@@ -425,7 +425,7 @@ def _is_in_config(self):
         fpath = Path.home() / ".ssh" / "config"
         if not fpath.exists():
             return False
-        sshcfg = parse_sshconfig(self.hostname.value)
+        sshcfg = aiida_ssh_plugin.parse_sshconfig(self.hostname.value)
         # NOTE: parse_sshconfig returns a dict with a hostname
         # even if it is not in the config file.
         # We require at least the user to be specified.
@@ -850,7 +850,7 @@ def _configure_computer(self, computer: orm.Computer, transport: str):
 
     def _configure_computer_ssh(self, computer: orm.Computer, user: orm.User):
         """Configure the computer with SSH transport"""
-        sshcfg = parse_sshconfig(self.hostname.value)
+        sshcfg = aiida_ssh_plugin.parse_sshconfig(self.hostname.value)
         authparams = {
             "port": int(sshcfg.get("port", 22)),
             "look_for_keys": True,

tests/test_computational_resources.py

@@ -17,8 +17,10 @@ def test_computational_resources_widget(aiida_local_code_bash):
 
 
 @pytest.mark.usefixtures("aiida_profile_clean")
-def test_ssh_computer_setup_widget():
+def test_ssh_computer_setup_widget(monkeypatch, tmp_path):
     """Test the SshComputerSetup."""
+    # mock home directory for ssh config file
+    monkeypatch.setenv("HOME", str(tmp_path))
     widget = computational_resources.SshComputerSetup()
 
     ssh_config = {
@@ -63,8 +65,13 @@ def test_ssh_computer_setup_widget():
 
 
 @pytest.mark.usefixtures("aiida_profile_clean")
-def test_aiida_computer_setup_widget():
+def test_aiida_computer_setup_widget(monkeypatch):
     """Test the AiidaComputerSetup."""
+    # monkeypatch the parse_sshconfig
+    monkeypatch.setattr(
+        "aiida.transports.plugins.ssh.parse_sshconfig",
+        lambda _: {"hostname": "daint.cscs.ch", "user": "aiida"},
+    )
     widget = computational_resources.AiidaComputerSetup()
 
     # At the beginning, the computer_name should be an empty string.