Skip to content

Commit 419a23b

Browse files
committed
test fix
1 parent 341e283 commit 419a23b

File tree

1 file changed

+10
-4
lines changed

1 file changed

+10
-4
lines changed

packages/service-library/tests/test_celery.py

Lines changed: 10 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55
# pylint: disable=protected-access
66
import pydantic
77
import pytest
8+
from common_library.json_serialization import json_dumps
89
from faker import Faker
910
from pydantic import StringConstraints
1011
from servicelib.celery.models import (
@@ -51,12 +52,17 @@ class _OwnerMetadata(OwnerMetadata):
5152
a: int | Wildcard
5253
b: str | Wildcard
5354

54-
keys = ["a", "b", "owner"]
55-
task_filter = _OwnerMetadata.model_validate(
55+
owner_metadata = _OwnerMetadata.model_validate(
5656
{"a": _faker.random_int(), "b": _faker.word(), "owner": _faker.word().lower()}
5757
)
58-
expected_key = ":".join([f"{k}={getattr(task_filter, k)}" for k in sorted(keys)])
59-
assert task_filter._build_task_id_prefix() == expected_key
58+
task_uuid = TaskUUID(_faker.uuid4())
59+
copy_owner_metadata = owner_metadata.model_dump()
60+
copy_owner_metadata.update({"task_uuid": f"{task_uuid}"})
61+
62+
expected_key = ":".join(
63+
[f"{k}={json_dumps(v)}" for k, v in sorted(copy_owner_metadata.items())]
64+
)
65+
assert owner_metadata.model_dump_task_id(task_uuid=task_uuid) == expected_key
6066

6167

6268
async def test_task_filter_task_uuid(

0 commit comments

Comments
 (0)