mirror of
https://github.com/hatchet-dev/hatchet.git
synced 2026-01-01 22:29:54 -06:00
7679732b15
* fix: skipping bug * fix: move `waits` -> `conditions` * fix: refs * chore: ver * feat: add skipped task to test * feat: start implementing or groups in wait for * feat: test of or groups on durable context * fix: lint * chore: gen * fix: lint * fix: branching hell
46 lines
1.2 KiB
Python
46 lines
1.2 KiB
Python
import asyncio
|
|
|
|
import pytest
|
|
|
|
from examples.durable.worker import EVENT_KEY, SLEEP_TIME, durable_workflow
|
|
from hatchet_sdk import Hatchet
|
|
|
|
|
|
@pytest.mark.asyncio(loop_scope="session")
|
|
async def test_durable(hatchet: Hatchet) -> None:
|
|
ref = durable_workflow.run_no_wait()
|
|
|
|
await asyncio.sleep(SLEEP_TIME + 10)
|
|
|
|
hatchet.event.push(EVENT_KEY, {"test": "test"})
|
|
|
|
result = await ref.aio_result()
|
|
|
|
workers = await hatchet.workers.aio_list()
|
|
|
|
assert workers.rows
|
|
|
|
active_workers = [w for w in workers.rows if w.status == "ACTIVE"]
|
|
|
|
assert len(active_workers) == 2
|
|
assert any(
|
|
w.name == hatchet.config.apply_namespace("e2e-test-worker")
|
|
for w in active_workers
|
|
)
|
|
assert any(
|
|
w.name == hatchet.config.apply_namespace("e2e-test-worker_durable")
|
|
for w in active_workers
|
|
)
|
|
|
|
assert result["durable_task"]["status"] == "success"
|
|
|
|
wait_group_1 = result["wait_for_or_group_1"]
|
|
wait_group_2 = result["wait_for_or_group_2"]
|
|
|
|
assert abs(wait_group_1["runtime"] - SLEEP_TIME) < 3
|
|
|
|
assert wait_group_1["key"] == wait_group_2["key"]
|
|
assert wait_group_1["key"] == "CREATE"
|
|
assert "sleep" in wait_group_1["event_id"]
|
|
assert "event" in wait_group_2["event_id"]
|