Skip to content

Commit e17a597

Browse files
committed
Cover async export job facade
Co-authored-by: Copilot App <223556219+Copilot@users.noreply.github.com> Copilot-Session: 76f2163a-0773-4889-b627-1bd4e5c7bea0
1 parent f7f724b commit e17a597

1 file changed

Lines changed: 11 additions & 5 deletions

File tree

tests/durabletask/extensions/history_export/test_client.py

Lines changed: 11 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@
2121
from durabletask import client, task, worker
2222
from durabletask.extensions.history_export import (
2323
AsyncExportHistoryClient,
24+
AsyncExportHistoryJobClient,
2425
ExportDestination,
2526
ExportFormat,
2627
ExportFormatKind,
@@ -164,16 +165,21 @@ async def test_async_client_create_list_wait_and_delete(
164165
format=ExportFormat(kind=ExportFormatKind.JSON),
165166
))
166167

167-
final = await export_client.wait_for_job(
168-
desc.job_id, timeout=30, poll_interval=0.1)
168+
job_client = export_client.get_job_client(desc.job_id)
169+
assert isinstance(job_client, AsyncExportHistoryJobClient)
170+
assert job_client.job_id == desc.job_id
171+
assert job_client.orchestrator_instance_id == orchestrator_instance_id_for(
172+
desc.job_id)
173+
174+
final = await job_client.wait(timeout=30, poll_interval=0.1)
169175
assert final.status == ExportJobStatus.COMPLETED
170-
assert (await export_client.get_job(desc.job_id)) is not None
176+
assert (await job_client.describe()) is not None
171177
assert desc.job_id in {
172178
job.job_id async for job in export_client.list_jobs()
173179
}
174180

175-
await export_client.delete_job(desc.job_id)
176-
assert await export_client.get_job(desc.job_id) is None
181+
await job_client.delete()
182+
assert await job_client.describe() is None
177183

178184

179185
async def test_async_client_get_job_returns_none_for_empty_state(writer):

0 commit comments

Comments
 (0)