Test more result types

Author: tpoliaw

tests/unit_tests/worker/test_task_worker.py

@@ -895,15 +895,21 @@ def test_plan_module_with_composite_devices_can_be_loaded_before_device_module(
     assert params["composite"].second_fake_device == second_fake_device
 
 
-class NotSerializable:
-    pass
-
-
 @pytest.mark.parametrize(
     "plan_result,task_result,type_name",
     (
-        (NotSerializable(), None, "NotSerializable"),
-        ((NotSerializable(), NotSerializable()), None, "tuple"),
+        (Unreturnable(foo=1, bar=[]), None, "Unreturnable"),
+        ((Unreturnable(foo=2, bar=[]), Unreturnable(foo=3, bar=[])), None, "tuple"),
+        (
+            ComplexReturn(foo=4, bar=["one"]),
+            {"foo": 4, "bar": ["one"]},
+            "ComplexReturn",
+        ),
+        (
+            ModelReturn(foo=5, bar=["two"]),
+            {"foo": 5, "bar": ["two"]},
+            "ModelReturn",
+        ),
         (42, 42, "int"),
         ((1, 2), [1, 2], "tuple"),
     ),