Skip to content

Commit

Permalink
Fix tests
Browse files Browse the repository at this point in the history
  • Loading branch information
rathishcholarajan committed Apr 26, 2022
1 parent fce3781 commit 19cab8f
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 6 deletions.
4 changes: 2 additions & 2 deletions test/fake_account_client.py
Original file line number Diff line number Diff line change
Expand Up @@ -349,8 +349,8 @@ def __init__(
self._run_mode = run_mode
self._default_job_class = BaseFakeJob

def list_jobs_statuses(self, limit, skip, descending=True, extra_filter=None):
"""Return a list of statuses of jobs."""
def list_jobs(self, limit, skip, descending=True, extra_filter=None):
"""Return a list of jobs."""
# pylint: disable=unused-argument
extra_filter = extra_filter or {}
if all(fil in extra_filter for fil in ["creationDate", "id"]):
Expand Down
8 changes: 4 additions & 4 deletions test/integration/test_account_client.py
Original file line number Diff line number Diff line change
Expand Up @@ -204,12 +204,12 @@ def test_job_final_status_polling(self):
self.assertEqual(response.pop("status", None), ApiJobStatus.COMPLETED.value)
self.assertNotEqual(status_queue.qsize(), 0)

def test_list_jobs_statuses_skip(self):
"""Test listing job statuses with an offset."""
jobs_raw = self.client.list_jobs_statuses(
def test_list_jobs_skip(self):
"""Test listing jobs with an offset."""
jobs_raw = self.client.list_jobs(
limit=1,
skip=1,
extra_filter={"creationDate": {"lte": self.job["creation_date"]}},
extra_filter={"createdBefore": self.job["creation_date"]},
)

# Ensure our job is skipped
Expand Down

0 comments on commit 19cab8f

Please sign in to comment.