From 9a38adb67ea1234cbccccf19d5b4ca0c01250490 Mon Sep 17 00:00:00 2001 From: "changhao@mit.edu" Date: Thu, 13 Feb 2025 11:23:13 -0500 Subject: [PATCH] remove mocked JobManager --- tests/test_cli.py | 13 +++++-------- 1 file changed, 5 insertions(+), 8 deletions(-) diff --git a/tests/test_cli.py b/tests/test_cli.py index f632168..3b41af9 100644 --- a/tests/test_cli.py +++ b/tests/test_cli.py @@ -12,11 +12,10 @@ def mock_job_manager(): """Fixture to provide a mocked JobManager.""" return MagicMock(spec=JobManager) - -def test_list_jobs_all(mock_job_manager, capsys): +def test_list_jobs_all(capsys): """Test listing all jobs without filters.""" # Mock jobs - mock_job_manager.get_jobs.return_value = [ + mock_jobs = [ MetriqGymJob( id="1234", device_name="ibmq_qasm_simulator", @@ -37,7 +36,6 @@ def test_list_jobs_all(mock_job_manager, capsys): ), ] - mock_jobs: list[MetriqGymJob] = mock_job_manager.get_jobs() list_jobs(mock_jobs, show_index=False) # Capture the output @@ -53,12 +51,11 @@ def test_list_jobs_all(mock_job_manager, capsys): assert captured.out == expected_output -def test_list_jobs_no_jobs(mock_job_manager, capsys): +def test_list_jobs_no_jobs(capsys): """Test listing jobs when no jobs are recorded.""" # Mock no jobs - mock_job_manager.get_jobs.return_value = [] - - mock_jobs: list[MetriqGymJob] = mock_job_manager.get_jobs() + mock_jobs = [] + list_jobs(mock_jobs, show_index=False) # Capture the output