mirror of
https://github.com/hatchet-dev/hatchet.git
synced 2026-01-03 15:19:44 -06:00
8e80faf2d6
* api changes * doc changes * move docs * generated * generate * pkg * backmerge main * revert to main * revert main * race? * remove go tests
50 lines
1.3 KiB
Python
50 lines
1.3 KiB
Python
import asyncio
|
|
import os
|
|
|
|
import pytest
|
|
|
|
from examples.waits.worker import task_condition_workflow
|
|
from hatchet_sdk import Hatchet
|
|
|
|
|
|
@pytest.mark.skipif(
|
|
os.getenv("CI", "false").lower() == "true",
|
|
reason="Skipped in CI because of unreliability",
|
|
)
|
|
@pytest.mark.asyncio(loop_scope="session")
|
|
async def test_waits(hatchet: Hatchet) -> None:
|
|
|
|
ref = task_condition_workflow.run_no_wait()
|
|
|
|
await asyncio.sleep(15)
|
|
|
|
hatchet.event.push("skip_on_event:skip", {})
|
|
hatchet.event.push("wait_for_event:start", {})
|
|
|
|
result = await ref.aio_result()
|
|
|
|
assert result["skip_on_event"] == {"skipped": True}
|
|
|
|
first_random_number = result["start"]["random_number"]
|
|
wait_for_event_random_number = result["wait_for_event"]["random_number"]
|
|
wait_for_sleep_random_number = result["wait_for_sleep"]["random_number"]
|
|
|
|
left_branch = result["left_branch"]
|
|
right_branch = result["right_branch"]
|
|
|
|
assert left_branch.get("skipped") is True or right_branch.get("skipped") is True
|
|
|
|
branch_random_number = left_branch.get("random_number") or right_branch.get(
|
|
"random_number"
|
|
)
|
|
|
|
result_sum = result["sum"]["sum"]
|
|
|
|
assert (
|
|
result_sum
|
|
== first_random_number
|
|
+ wait_for_event_random_number
|
|
+ wait_for_sleep_random_number
|
|
+ branch_random_number
|
|
)
|