From 0cca8ef08158d45d4a7080e90a50c786a0fe0df1 Mon Sep 17 00:00:00 2001 From: "N.Sehrawat" Date: Mon, 8 Jan 2024 17:33:55 +0100 Subject: [PATCH] Fixing results not completed --- tests/util/api/test_remote_backend.py | 15 +++++++++++---- 1 file changed, 11 insertions(+), 4 deletions(-) diff --git a/tests/util/api/test_remote_backend.py b/tests/util/api/test_remote_backend.py index 18e42df..6498fce 100644 --- a/tests/util/api/test_remote_backend.py +++ b/tests/util/api/test_remote_backend.py @@ -89,9 +89,16 @@ def test_get_results_not_completed( ) -> None: backend = RemoteBackend() jobs_api_instance = AsyncMock() - jobs = MagicMock() - jobs.status = JobStatus.RUNNING - jobs_api_instance.read_job_jobs_id_get.return_value = jobs + job = MagicMock() + job.id = 1 + job.status = JobStatus.RUNNING + jobs_api_instance.read_job_jobs_id_get.return_value = job + + results_api_instance = AsyncMock() + mocker.patch("quantuminspire.util.api.remote_backend.JobsApi", return_value=jobs_api_instance) - backend.get_results(1) + mocker.patch("quantuminspire.util.api.remote_backend.ResultsApi", return_value=results_api_instance) + backend.get_results(job.id) api_client.assert_has_calls([call().__aenter__(), call().__aexit__(None, None, None)]) + jobs_api_instance.read_job_jobs_id_get.assert_called_with(job.id) + results_api_instance.read_results_by_job_id_results_job_job_id_get.assert_not_called()