Add ParentThread to WfRunVariables in tests

littlehorse-enterprises · Nov 27, 2024 · 1d8fd25 · 1d8fd25
1 parent 42caf83
commit 1d8fd25
Showing 1 changed file with 11 additions and 11 deletions.
diff --git a/sdk-python/tests/test_workflow.py b/sdk-python/tests/test_workflow.py
@@ -286,15 +286,15 @@ class MyClass:
             def if_condition(self, thread: WorkflowThread) -> None:
                 thread.mutate(
                     WfRunVariable(
-                        variable_name="variable-1", variable_type=VariableType.INT, parent=None
+                        variable_name="variable-1", variable_type=VariableType.INT, parent=thread
                     ),
                     VariableMutationType.ASSIGN,
                     1,
                 )
                 thread.execute("task-a")
                 thread.mutate(
                     WfRunVariable(
-                        variable_name="variable-3", variable_type=VariableType.INT, parent=None
+                        variable_name="variable-3", variable_type=VariableType.INT, parent=thread
                     ),
                     VariableMutationType.ASSIGN,
                     3,
@@ -304,7 +304,7 @@ def if_condition(self, thread: WorkflowThread) -> None:
             def else_condition(self, thread: WorkflowThread) -> None:
                 thread.mutate(
                     WfRunVariable(
-                        variable_name="variable-2", variable_type=VariableType.INT, parent=None
+                        variable_name="variable-2", variable_type=VariableType.INT, parent=thread
                     ),
                     VariableMutationType.ASSIGN,
                     2,
@@ -313,7 +313,7 @@ def else_condition(self, thread: WorkflowThread) -> None:
                 thread.execute("task-d")
                 thread.mutate(
                     WfRunVariable(
-                        variable_name="variable-4", variable_type=VariableType.INT, parent=None
+                        variable_name="variable-4", variable_type=VariableType.INT, parent=thread
                     ),
                     VariableMutationType.ASSIGN,
                     4,
@@ -436,7 +436,7 @@ class MyClass:
             def if_condition(self, thread: WorkflowThread) -> None:
                 thread.mutate(
                     WfRunVariable(
-                        variable_name="variable-1", variable_type=VariableType.INT, parent=None
+                        variable_name="variable-1", variable_type=VariableType.INT, parent=thread
                     ),
                     VariableMutationType.ASSIGN,
                     1,
@@ -445,7 +445,7 @@ def if_condition(self, thread: WorkflowThread) -> None:
             def else_condition(self, thread: WorkflowThread) -> None:
                 thread.mutate(
                     WfRunVariable(
-                        variable_name="variable-2", variable_type=VariableType.INT, parent=None
+                        variable_name="variable-2", variable_type=VariableType.INT, parent=thread
                     ),
                     VariableMutationType.ASSIGN,
                     2,
@@ -530,7 +530,7 @@ class MyClass:
             def if_condition(self, thread: WorkflowThread) -> None:
                 thread.mutate(
                     WfRunVariable(
-                        variable_name="variable-2", variable_type=VariableType.INT, parent=None
+                        variable_name="variable-2", variable_type=VariableType.INT, parent=thread
                     ),
                     VariableMutationType.ASSIGN,
                     2,
@@ -539,7 +539,7 @@ def if_condition(self, thread: WorkflowThread) -> None:
             def my_entrypoint(self, thread: WorkflowThread) -> None:
                 thread.mutate(
                     WfRunVariable(
-                        variable_name="variable-1", variable_type=VariableType.INT, parent=None
+                        variable_name="variable-1", variable_type=VariableType.INT, parent=thread
                     ),
                     VariableMutationType.ASSIGN,
                     1,
@@ -549,7 +549,7 @@ def my_entrypoint(self, thread: WorkflowThread) -> None:
                 )
                 thread.mutate(
                     WfRunVariable(
-                        variable_name="variable-3", variable_type=VariableType.INT, parent=None
+                        variable_name="variable-3", variable_type=VariableType.INT, parent=thread
                     ),
                     VariableMutationType.ASSIGN,
                     3,
@@ -741,15 +741,15 @@ class MyClass:
             def my_condition(self, thread: WorkflowThread) -> None:
                 thread.mutate(
                     WfRunVariable(
-                        variable_name="variable-1", variable_type=VariableType.INT, parent=None
+                        variable_name="variable-1", variable_type=VariableType.INT, parent=thread
                     ),
                     VariableMutationType.ASSIGN,
                     1,
                 )
                 thread.execute("my-task")
                 thread.mutate(
                     WfRunVariable(
-                        variable_name="variable-2", variable_type=VariableType.INT, parent=None
+                        variable_name="variable-2", variable_type=VariableType.INT, parent=thread
                     ),
                     VariableMutationType.ASSIGN,
                     2,