diff --git a/tests/utils/test_serializer.py b/tests/utils/test_serializer.py index 096dec3..cc47707 100644 --- a/tests/utils/test_serializer.py +++ b/tests/utils/test_serializer.py @@ -405,6 +405,9 @@ def setup_dict() -> dict[str, Any]: enum_attr = MyEnum.RUNNING attr_list = [0, 1, MySubclass()] + def my_task(self) -> None: + pass + return ServiceClass().serialize()["value"] @@ -439,6 +442,28 @@ def test_update_enum_attribute_to_float(setup_dict: dict[str, Any]) -> None: } +def test_update_task_state(setup_dict: dict[str, Any]) -> None: + assert setup_dict["my_task"] == { + "async": False, + "doc": None, + "frontend_render": False, + "readonly": True, + "signature": {"parameters": {}, "return_annotation": {}}, + "type": "method", + "value": None, + } + set_nested_value_by_path(setup_dict, "my_task", TaskStatus.RUNNING) + assert setup_dict["my_task"] == { + "async": False, + "doc": None, + "frontend_render": False, + "readonly": True, + "signature": {"parameters": {}, "return_annotation": {}}, + "type": "method", + "value": "RUNNING", + } + + def test_update_list_entry(setup_dict: dict[str, Any]) -> None: set_nested_value_by_path(setup_dict, "attr_list[1]", 20) assert setup_dict["attr_list"]["value"][1]["value"] == 20