adapt integration test

Matgenix · Mar 19, 2024 · 0a9dbdb · 0a9dbdb
1 parent 1f7947f
commit 0a9dbdb
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/tests/integration/test_slurm.py b/tests/integration/test_slurm.py
@@ -35,19 +35,15 @@ def test_paramiko_ssh_connection(job_controller, slurm_ssh_port):
 
 
 def test_project_check(job_controller, capsys):
-    from jobflow_remote.cli.project import check
+    from jobflow_remote.testing.cli import run_check_cli
 
-    check(print_errors=True)
-    captured = capsys.readouterr()
-    assert not captured.err
     expected = [
         "✓ Worker test_local_worker",
         "✓ Worker test_remote_worker",
         "✓ Jobstore",
         "✓ Queue store",
     ]
-    for line in expected:
-        assert line in captured.out
+    run_check_cli(["project", "check"], required_out=expected)
 
 
 @pytest.mark.parametrize(