Skip to content

Commit fc3a8fc

Browse files
04diiguyiYi Gu
andauthored
fix live test (microsoft#665)
Co-authored-by: Yi Gu <guyi@microsoft.com>
1 parent 5228f10 commit fc3a8fc

File tree

6 files changed

+1470
-198
lines changed

6 files changed

+1470
-198
lines changed

azure-quantum/tests/unit/recordings/test_list_session_jobs.yaml

Lines changed: 281 additions & 27 deletions
Large diffs are not rendered by default.

azure-quantum/tests/unit/recordings/test_list_session_jobs_filtered_by_name.yaml

Lines changed: 281 additions & 27 deletions
Large diffs are not rendered by default.

azure-quantum/tests/unit/recordings/test_list_session_jobs_filtered_by_status.yaml

Lines changed: 281 additions & 27 deletions
Large diffs are not rendered by default.

azure-quantum/tests/unit/recordings/test_list_session_jobs_orderby_asc.yaml

Lines changed: 281 additions & 27 deletions
Large diffs are not rendered by default.

azure-quantum/tests/unit/recordings/test_list_session_jobs_orderby_desc.yaml

Lines changed: 281 additions & 27 deletions
Large diffs are not rendered by default.

azure-quantum/tests/unit/test_pagination.py

Lines changed: 65 additions & 63 deletions
Original file line numberDiff line numberDiff line change
@@ -312,96 +312,98 @@ def test_list_sessions_orderby_desc_filtered_by_created_before(self):
312312

313313
@pytest.mark.live_test
314314
def test_list_session_jobs(self):
315-
# please change to valid session id for recording
316-
#session_id = "cd61afdb-eb34-11ef-a0c7-5414f37777d8"
317-
session_id = "00000000-0000-0000-0000-000000000001"
318-
319315
ws = self.create_workspace()
320-
jobs = ws.list_session_jobs(session_id=session_id)
321316

322-
for job in jobs:
323-
self.assertEqual(job.item_type, "Job")
324-
self.assertEqual(job._details.session_id, session_id)
317+
sessions = ws.list_sessions()
318+
319+
if len(sessions) > 0:
320+
session = sessions[0]
321+
jobs = ws.list_session_jobs(session_id=session.id)
322+
323+
for job in jobs:
324+
self.assertEqual(job.item_type, "Job")
325+
self.assertEqual(job._details.session_id, session.id)
325326

326327
@pytest.mark.live_test
327328
def test_list_session_jobs_filtered_by_name(self):
328-
# please change to valid session id for recording
329-
#session_id = "cd61afdb-eb34-11ef-a0c7-5414f37777d8"
330-
session_id = "00000000-0000-0000-0000-000000000001"
331-
332329
ws = self.create_workspace()
333-
jobs = ws.list_session_jobs(session_id=session_id, name_match="Job")
334330

335-
for job in jobs:
336-
self.assertEqual(job.item_type, "Job")
337-
self.assertEqual(job._details.session_id, session_id)
338-
339-
check_job_name = job.details.name.startswith("Job")
340-
self.assertTrue( check_job_name, job.details.name)
331+
sessions = ws.list_sessions()
332+
333+
if len(sessions) > 0:
334+
session = sessions[0]
335+
jobs = ws.list_session_jobs(session_id=session.id, name_match="Job")
336+
337+
for job in jobs:
338+
self.assertEqual(job.item_type, "Job")
339+
self.assertEqual(job._details.session_id, session.id)
340+
341+
check_job_name = job.details.name.startswith("Job")
342+
self.assertTrue( check_job_name, job.details.name)
341343

342344
@pytest.mark.live_test
343345
def test_list_session_jobs_filtered_by_status(self):
344-
# please change to valid session id for recording
345-
#session_id = "cd61afdb-eb34-11ef-a0c7-5414f37777d8"
346-
session_id = "00000000-0000-0000-0000-000000000001"
347-
348346
ws = self.create_workspace()
349-
jobs = ws.list_session_jobs(session_id=session_id, status=["Succeeded"])
350347

351-
for job in jobs:
352-
self.assertEqual(job.item_type, "Job")
353-
self.assertEqual(job._details.session_id, session_id)
348+
sessions = ws.list_sessions()
349+
if len(sessions) > 0:
350+
session = sessions[0]
351+
jobs = ws.list_session_jobs(session_id=session.id, status=["Succeeded"])
354352

355-
check_job_status = job.details.status == "Succeeded"
356-
self.assertTrue( check_job_status, job.details.status)
353+
for job in jobs:
354+
self.assertEqual(job.item_type, "Job")
355+
self.assertEqual(job._details.session_id, session.id)
356+
357+
check_job_status = job.details.status == "Succeeded"
358+
self.assertTrue( check_job_status, job.details.status)
357359

358360
@pytest.mark.live_test
359361
def test_list_session_jobs_orderby_asc(self):
360-
# please change to valid session id for recording
361-
#session_id = "cd61afdb-eb34-11ef-a0c7-5414f37777d8"
362-
session_id = "00000000-0000-0000-0000-000000000001"
363-
364362
ws = self.create_workspace()
365-
jobs = ws.list_session_jobs(session_id=session_id, status=["Succeeded"], orderby_property="CreationTime", is_asc=True)
363+
sessions = ws.list_sessions()
364+
365+
if len(sessions) > 0:
366+
session = sessions[0]
367+
jobs = ws.list_session_jobs(session_id=session.id, status=["Succeeded"], orderby_property="CreationTime", is_asc=True)
366368

367-
creation_time = None
368-
for job in jobs:
369-
self.assertEqual(job.item_type, "Job")
370-
self.assertEqual(job._details.session_id, session_id)
369+
creation_time = None
370+
for job in jobs:
371+
self.assertEqual(job.item_type, "Job")
372+
self.assertEqual(job._details.session_id, session.id)
371373

372-
check_job_status = job.details.status == "Succeeded"
373-
self.assertTrue( check_job_status, job.details.status)
374+
check_job_status = job.details.status == "Succeeded"
375+
self.assertTrue( check_job_status, job.details.status)
374376

375-
if creation_time is None:
376-
creation_time = job.details.creation_time
377-
else:
378-
check_item_created_before_order = job.details.creation_time >= creation_time
379-
self.assertTrue( check_item_created_before_order, job.details.creation_time)
380-
creation_time = job.details.creation_time
377+
if creation_time is None:
378+
creation_time = job.details.creation_time
379+
else:
380+
check_item_created_before_order = job.details.creation_time >= creation_time
381+
self.assertTrue( check_item_created_before_order, job.details.creation_time)
382+
creation_time = job.details.creation_time
381383

382384
@pytest.mark.live_test
383385
def test_list_session_jobs_orderby_desc(self):
384-
# please change to valid session id for recording
385-
#session_id = "cd61afdb-eb34-11ef-a0c7-5414f37777d8"
386-
session_id = "00000000-0000-0000-0000-000000000001"
387-
388386
ws = self.create_workspace()
389-
jobs = ws.list_session_jobs(session_id=session_id, status=["Succeeded"], orderby_property="CreationTime", is_asc=False)
387+
sessions = ws.list_sessions()
390388

391-
creation_time = None
392-
for job in jobs:
393-
self.assertEqual(job.item_type, "Job")
394-
self.assertEqual(job._details.session_id, session_id)
389+
if len(sessions) > 0:
390+
session = sessions[0]
391+
jobs = ws.list_session_jobs(session_id=session.id, status=["Succeeded"], orderby_property="CreationTime", is_asc=False)
395392

396-
check_job_status = job.details.status == "Succeeded"
397-
self.assertTrue( check_job_status, job.details.status)
393+
creation_time = None
394+
for job in jobs:
395+
self.assertEqual(job.item_type, "Job")
396+
self.assertEqual(job._details.session_id, session.id)
398397

399-
if creation_time is None:
400-
creation_time = job.details.creation_time
401-
else:
402-
check_item_created_before_order = job.details.creation_time <= creation_time
403-
self.assertTrue( check_item_created_before_order, job.details.creation_time)
404-
creation_time = job.details.creation_time
398+
check_job_status = job.details.status == "Succeeded"
399+
self.assertTrue( check_job_status, job.details.status)
400+
401+
if creation_time is None:
402+
creation_time = job.details.creation_time
403+
else:
404+
check_item_created_before_order = job.details.creation_time <= creation_time
405+
self.assertTrue( check_item_created_before_order, job.details.creation_time)
406+
creation_time = job.details.creation_time
405407

406408
@pytest.mark.live_test
407409
def test_list_top_level_items(self):

0 commit comments

Comments
 (0)