fix: add test for adding errors

chickensoft-games · Mar 20, 2024 · 6ddcfd8 · 6ddcfd8
1 parent 677e733
commit 6ddcfd8
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 0 deletions.
diff --git a/Chickensoft.LogicBlocks.Tests/test/fixtures/FakeLogicBlock.cs b/Chickensoft.LogicBlocks.Tests/test/fixtures/FakeLogicBlock.cs
@@ -92,6 +92,12 @@ public OnExitState(Action<State?> onExit) {
         this.OnExit(onExit);
       }
     }
+
+    public record AddErrorOnEnterState : State {
+      public AddErrorOnEnterState(Exception e) {
+        this.OnEnter(() => AddError(e));
+      }
+    }
   }
 
   public static class Output {

diff --git a/Chickensoft.LogicBlocks.Tests/test/fixtures/FakeLogicBlock.g.puml b/Chickensoft.LogicBlocks.Tests/test/fixtures/FakeLogicBlock.g.puml
@@ -8,6 +8,7 @@ state "FakeLogicBlock State" as Chickensoft_LogicBlocks_Tests_Fixtures_FakeLogic
   state "Custom" as Chickensoft_LogicBlocks_Tests_Fixtures_FakeLogicBlock_State_Custom
   state "OnEnterState" as Chickensoft_LogicBlocks_Tests_Fixtures_FakeLogicBlock_State_OnEnterState
   state "OnExitState" as Chickensoft_LogicBlocks_Tests_Fixtures_FakeLogicBlock_State_OnExitState
+  state "AddErrorOnEnterState" as Chickensoft_LogicBlocks_Tests_Fixtures_FakeLogicBlock_State_AddErrorOnEnterState
   Chickensoft_LogicBlocks_Tests_Fixtures_FakeLogicBlock_State : OnInputOne → OutputOne
   Chickensoft_LogicBlocks_Tests_Fixtures_FakeLogicBlock_State : OnInputTwo → OutputTwo
   Chickensoft_LogicBlocks_Tests_Fixtures_FakeLogicBlock_State : OnNoNewState → OutputOne

diff --git a/Chickensoft.LogicBlocks.Tests/test/src/LogicBlockTest.cs b/Chickensoft.LogicBlocks.Tests/test/src/LogicBlockTest.cs
@@ -511,4 +511,15 @@ public void ForceResetChangesStateAndProcessesInputs() {
 
     state.ShouldBeOfType<FakeLogicBlock.State.StateB>();
   }
+
+  [Fact]
+  public void AddsErrorToItself() {
+    var e = new InvalidOperationException();
+    var state = new FakeLogicBlock.State.AddErrorOnEnterState(e);
+
+    var context = state.CreateFakeContext();
+    state.Enter();
+
+    context.Errors.ShouldBe([e]);
+  }
 }