Add test for CLI install

aiida_hyperqueue/cli/install.py

@@ -21,12 +21,16 @@
     default=Path("$HOME/bin/"),
     help="remote bin path hq will stored.",
 )
+@click.option(
+    "--write-bashrc/--no-write-bashrc",
+    default=True,
+    help="write the bin path to bashrc.",
+)
 @click.option(
     "--hq-version", type=str, default="0.19.0", help="the hq version will be installed."
 )
-# TODO: separate the bashrc write and make it optional.
 # TODO: should also support different arch binary??
-def cmd_install(computer: orm.Computer, remote_bin_dir: Path, hq_version: str):
+def cmd_install(computer: orm.Computer, remote_bin_dir: Path, hq_version: str, write_bashrc: bool):
     """Install the hq binary to the computer through the transport"""
 
     # The minimal hq version we support is 0.13.0, check the minor version
@@ -95,15 +99,16 @@ def cmd_install(computer: orm.Computer, remote_bin_dir: Path, hq_version: str):
             transport.exec_command_wait(f"chmod +x {str(remote_bin_dir / 'hq')}")
 
             # write to bashrc
-            identity_str = "by aiida-hq"
-            retval, _, stderr = transport.exec_command_wait(
-                f"grep -q '# {identity_str}' ~/.bashrc || echo '# {identity_str}\nexport PATH=$HOME/bin:$PATH' >> ~/.bashrc"
-            )
-
-            if retval != 0:
-                echo.echo_critical(
-                    f"Not able to set set the path $HOME/bin to your remote bashrc, try to do it manually.\n"
-                    f"Info: {stderr}"
+            if write_bashrc:
+                identity_str = "by aiida-hq"
+                retval, _, stderr = transport.exec_command_wait(
+                    f"grep -q '# {identity_str}' ~/.bashrc || echo '# {identity_str}\nexport PATH=$HOME/bin:$PATH' >> ~/.bashrc"
                 )
 
-    echo.echo_success("The hq binary installed in remote")
+                if retval != 0:
+                    echo.echo_critical(
+                        f"Not able to set set the path $HOME/bin to your remote bashrc, try to do it manually.\n"
+                        f"Info: {stderr}"
+                    )
+
+    echo.echo_success(f"The hq binary installed into remote {remote_bin_dir}")

tests/test_cli_alloc.py

@@ -0,0 +1,13 @@
+import pytest
+from click.testing import CliRunner
+
+from aiida.transports.transport import Transport as TransportClass
+
+@pytest.fixture
+def runner():
+    return CliRunner()
+
+original_exec_command_wait = TransportClass.exec_command_wait
+
+# TODO: Not yet implemented, seems hard to just use hq_env
+# If using real command, I need to handle the grace for teardown of tests

tests/test_cli_install.py

@@ -0,0 +1,21 @@
+import pytest
+from click.testing import CliRunner
+
+from aiida_hyperqueue.cli import cmd_install
+
+@pytest.fixture
+def runner():
+    return CliRunner()
+
+def test_install(runner, tmp_path, aiida_computer_ssh):
+    aiida_computer_ssh(label="localhost-hq")
+
+    version = "0.19.0"
+    result = runner.invoke(cmd_install, ["-p", f"{str(tmp_path.resolve())}", "--hq-version", version, "--no-write-bashrc", "localhost-hq"])
+
+    assert result.exit_code == 0
+    assert f"hq version {version}" in result.output
+
+
+
+

tests/test_cli.py → tests/test_cli_server.py

@@ -139,3 +139,4 @@ def test_server_start_info_stop_circle(runner: CliRunner, aiida_computer_ssh, mo
 
     assert result.exit_code == 1
     assert "Critical: cannot obtain HyperQueue server information" in result.output
+