fix test

skypilot-org · Apr 4, 2024 · a415639 · a415639
1 parent 6947d53
commit a415639
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/.github/workflows/pytest.yml b/.github/workflows/pytest.yml
@@ -27,7 +27,7 @@ jobs:
           - tests/test_optimizer_random_dag.py
           - tests/test_storage.py
           - tests/test_wheels.py
-          - tests/test_spot_serve.py
+          - tests/test_job_serve.py
           - tests/test_yaml_parser.py
     runs-on: ubuntu-latest
     steps:

diff --git a/tests/test_spot_serve.py → tests/test_job_serve.py b/tests/test_spot_serve.py → tests/test_job_serve.py
@@ -261,23 +261,23 @@ def test_cancel_on_spot_controller(self, _mock_cluster_state,
     @pytest.mark.timeout(60)
     def test_cancel(self, _mock_db_conn):
         cli_runner = cli_testing.CliRunner()
-        result = cli_runner.invoke(cli.spot_cancel, ['-a'])
+        result = cli_runner.invoke(cli.job_cancel, ['-a'])
         assert result.exit_code == 1
         assert controller_utils.Controllers.JOB_CONTROLLER.value.default_hint_if_non_existent in str(
             result.output), (result.exception, result.output, result.exc_info)
 
     @pytest.mark.timeout(60)
     def test_logs(self, _mock_db_conn):
         cli_runner = cli_testing.CliRunner()
-        result = cli_runner.invoke(cli.spot_logs, ['1'])
+        result = cli_runner.invoke(cli.job_logs, ['1'])
         assert result.exit_code == 1
         assert controller_utils.Controllers.JOB_CONTROLLER.value.default_hint_if_non_existent in str(
             result.output), (result.exception, result.output, result.exc_info)
 
     @pytest.mark.timeout(60)
     def test_queue(self, _mock_db_conn):
         cli_runner = cli_testing.CliRunner()
-        result = cli_runner.invoke(cli.spot_queue)
+        result = cli_runner.invoke(cli.job_queue)
         assert result.exit_code == 0
         assert controller_utils.Controllers.JOB_CONTROLLER.value.default_hint_if_non_existent in str(
             result.output), (result.exception, result.output, result.exc_info)