Skip to content

Commit 58b2a43

Browse files
committed
some initial tests
1 parent 1b0d899 commit 58b2a43

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

services/dask-sidecar/tests/unit/test_utils_dask.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,17 @@ def test_publish_event(
4646
progress=1,
4747
task_owner=task_owner,
4848
)
49+
50+
def handler_some_topic_event(event: tuple) -> None:
51+
print("received event", event)
52+
assert isinstance(event, tuple)
53+
received_task_log_event = TaskProgressEvent.model_validate_json(event[1])
54+
assert received_task_log_event == event_to_publish
55+
56+
dask_client.subscribe_topic("some_topic", handler_some_topic_event)
57+
time.sleep(1)
4958
dask_client.log_event("some_topic", event_to_publish.model_dump_json())
59+
time.sleep(1)
5060
# publish_event(dask_pub=dask_pub, event=event_to_publish)
5161

5262
# NOTE: this tests runs a sync dask client,

0 commit comments

Comments
 (0)