mirror of
https://github.com/hatchet-dev/hatchet.git
synced 2026-01-05 16:19:43 -06:00
993817b049
* fix: register durable steps and workflows separately * chore: initial copy of pooled listener * feat: initial generic impl * feat: use pooled listener for wf run listener * refactor: move listeners to subdir * feat: refactor durable event listener * fix: bug * feat: share single pooled workflow listener and event listener everywhere * cruft: rm hatchet fixture * fix: rebase issue * feat: remove asyncio api client in favor of sync one * chore: minor version * proposal: crazy hack idea to make the workflow run listener work * fix: sleeps and error handling * Revert "cruft: rm hatchet fixture" This reverts commit b75f625e6ccec095e8c4e294d6727db166796411. * fix: set timeout * fix: rm pytest-timeout * fix: rm retry * fix: use v1 by default * fix: try removing retry state * fix: try using async client? * fix: try running sequentially * debug: loop * debug: maybe it's this? * fix: lint * fix: re-remove unused fixtures * fix: lazily create clients in admin client * fix: default * fix: lazily initialize dispatcher client * fix: hint * fix: no. way. * feat: add back retries in ci * fix: clients + imports * fix: loop scope * debug: try running skipped tests in ci again * Revert "debug: try running skipped tests in ci again" This reverts commit 8d9e18150e5207ee6051d8df8a6fe2a7504c722e. * fix: rm duped code * refactor: rename everything as `to_proto` * refactor: removals of `namespace` being passed around * fix: task output stupidity * feat: add deprecation warning * fix: remove more unused code * feat: mix sync and async in dag example * fix: autouse * fix: more input types * feat: remove ability to pass in loop * fix: overload key gen
30 lines
841 B
Python
30 lines
841 B
Python
import asyncio
|
|
|
|
import pytest
|
|
|
|
from examples.cancellation.worker import cancellation_workflow
|
|
from hatchet_sdk import Hatchet
|
|
from hatchet_sdk.clients.rest.models.v1_task_status import V1TaskStatus
|
|
|
|
|
|
@pytest.mark.asyncio(loop_scope="session")
|
|
async def test_cancellation(hatchet: Hatchet) -> None:
|
|
ref = await cancellation_workflow.aio_run_no_wait()
|
|
|
|
"""Sleep for a long time since we only need cancellation to happen _eventually_"""
|
|
await asyncio.sleep(10)
|
|
|
|
for i in range(30):
|
|
run = await hatchet.runs.aio_get(ref.workflow_run_id)
|
|
|
|
if run.run.status == V1TaskStatus.RUNNING:
|
|
await asyncio.sleep(1)
|
|
continue
|
|
|
|
assert run.run.status == V1TaskStatus.CANCELLED
|
|
assert not run.run.output
|
|
|
|
break
|
|
else:
|
|
assert False, "Workflow run did not cancel in time"
|