feat(sdk-dotnet): add unit test to handling errors in workflow thread

sdk-dotnet/LittleHorse.Sdk.Tests/Workflow/Spec/WorkflowThreadErrorsAndExceptionsTest.cs

@@ -0,0 +1,174 @@
+using System;
+using LittleHorse.Sdk.Common.Proto;
+using LittleHorse.Sdk.Workflow.Spec;
+using Moq;
+using Xunit;
+
+namespace LittleHorse.Sdk.Tests.Workflow.Spec;
+
+public class WorkflowThreadErrorsAndExceptionsTest
+{
+    private readonly Action<WorkflowThread> _action;
+    void ParentEntrypoint(WorkflowThread thread)
+    {
+    }
+
+    public WorkflowThreadErrorsAndExceptionsTest()
+    {
+        LHLoggerFactoryProvider.Initialize(null);
+        _action = ParentEntrypoint;
+    }
+
+    [Fact]
+    public void WfThread_WithoutSpecificError_ShouldCompileErrorHandling()
+    {
+        var numberOfExitNodes = 1;
+        var numberOfEntrypointNodes = 1;
+        var numberOfTasks = 2;
+        var workflowName = "TestWorkflow";
+        var mockParentWorkflow = new Mock<Sdk.Workflow.Spec.Workflow>(workflowName, _action);
+
+        void EntryPointAction(WorkflowThread wf)
+        {
+            NodeOutput node = wf.Execute("fail");
+            wf.HandleError(
+                node,
+                handler =>
+                {
+                    handler.Execute("my-task");
+                }
+            );
+            wf.Execute("my-task");
+        }
+        var workflowThread = new WorkflowThread(mockParentWorkflow.Object, EntryPointAction);
+
+        var compiledWfThread = workflowThread.Compile();
+
+        var expectedSpec = new ThreadSpec();
+        var entrypoint = new Node
+        {
+            Entrypoint = new EntrypointNode(),
+            OutgoingEdges =
+            {
+                new Edge { SinkNodeName = "1-fail-TASK" }
+            }
+        };
+
+        var failTask = new Node
+        {
+            Task = new TaskNode
+            {
+                TaskDefId = new TaskDefId { Name = "fail" }
+            },
+            OutgoingEdges = { new Edge { SinkNodeName = "2-my-task-TASK" } },
+            FailureHandlers =
+            {
+                new FailureHandlerDef
+                {
+                    HandlerSpecName = "exn-handler-1-fail-TASK-FAILURE_TYPE_ERROR",
+                    AnyFailureOfType = FailureHandlerDef.Types.LHFailureType.FailureTypeError
+                }
+            }
+        };
+
+        var myTask = new Node
+        {
+            Task = new TaskNode
+            {
+                TaskDefId = new TaskDefId { Name = "my-task" }
+            },
+            OutgoingEdges = { new Edge { SinkNodeName = "3-exit-EXIT" } }
+        };
+
+        var exitNode = new Node
+        {
+            Exit = new ExitNode()
+        };
+
+        expectedSpec.Nodes.Add("0-entrypoint-ENTRYPOINT", entrypoint);
+        expectedSpec.Nodes.Add("1-fail-TASK", failTask);
+        expectedSpec.Nodes.Add("2-my-task-TASK", myTask);
+        expectedSpec.Nodes.Add("3-exit-EXIT", exitNode);
+
+        var expectedNumberOfNodes = numberOfEntrypointNodes + numberOfExitNodes + numberOfTasks;
+        Assert.Equal(expectedNumberOfNodes, compiledWfThread.Nodes.Count);
+        Assert.Equal(expectedSpec, compiledWfThread);
+    }
+
+    [Fact]
+    public void WfThread_WithSpecificError_ShouldCompileErrorHandling()
+    {
+        var numberOfExitNodes = 1;
+        var numberOfEntrypointNodes = 1;
+        var numberOfTasks = 2;
+        var workflowName = "TestWorkflow";
+        var mockParentWorkflow = new Mock<Sdk.Workflow.Spec.Workflow>(workflowName, _action);
+
+        void EntryPointAction(WorkflowThread wf)
+        {
+            NodeOutput node = wf.Execute("fail");
+            wf.HandleError(
+                node,
+                LHErrorType.Timeout,
+                handler =>
+                {
+                    handler.Execute("my-task");
+                }
+            );
+            wf.Execute("my-task");
+        }
+        var workflowThread = new WorkflowThread(mockParentWorkflow.Object, EntryPointAction);
+
+        var compiledWfThread = workflowThread.Compile();
+
+        var expectedSpec = new ThreadSpec();
+        var entrypoint = new Node
+        {
+            Entrypoint = new EntrypointNode(),
+            OutgoingEdges =
+            {
+                new Edge { SinkNodeName = "1-fail-TASK" }
+            }
+        };
+
+        var failTask = new Node
+        {
+            Task = new TaskNode
+            {
+                TaskDefId = new TaskDefId { Name = "fail" }
+            },
+            OutgoingEdges = { new Edge { SinkNodeName = "2-my-task-TASK" } },
+            FailureHandlers =
+            {
+                new FailureHandlerDef
+                {
+                    HandlerSpecName = "exn-handler-1-fail-TASK-TIMEOUT",
+                    SpecificFailure = "TIMEOUT"
+                }
+            }
+        };
+
+        var myTask = new Node
+        {
+            Task = new TaskNode
+            {
+                TaskDefId = new TaskDefId { Name = "my-task" }
+            },
+            OutgoingEdges = { new Edge { SinkNodeName = "3-exit-EXIT" } }
+        };
+
+        var exitNode = new Node
+        {
+            Exit = new ExitNode()
+        };
+
+        expectedSpec.Nodes.Add("0-entrypoint-ENTRYPOINT", entrypoint);
+        expectedSpec.Nodes.Add("1-fail-TASK", failTask);
+        expectedSpec.Nodes.Add("2-my-task-TASK", myTask);
+        expectedSpec.Nodes.Add("3-exit-EXIT", exitNode);
+
+        var expectedNumberOfNodes = numberOfEntrypointNodes + numberOfExitNodes + numberOfTasks;
+        Assert.Equal(expectedNumberOfNodes, compiledWfThread.Nodes.Count);
+        Assert.Equal(expectedSpec, compiledWfThread);
+    }
+}