Wait behaviour bug fix

Eraclys · Aug 7, 2018 · 99158bc · 99158bc
1 parent 1d96c4e
commit 99158bc
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 0 deletions.
diff --git a/src/BehaviourTree.Tests/WaitTests.cs b/src/BehaviourTree.Tests/WaitTests.cs
@@ -35,5 +35,25 @@ public void WhenWaitTimeExpires_ReturnSuccessAndResetTimer()
 
             Assert.That(behaviourStatus, Is.EqualTo(BehaviourStatus.Running));
         }
+
+        [Test]
+        public void WhenResetIsCalled_ReturnResetTimer()
+        {
+            var sut = new Wait<MockContext>(1000);
+            var context = new MockContext();
+
+            context.AddMilliseconds(500);
+
+            var behaviourStatus = sut.Tick(context);
+
+            Assert.That(behaviourStatus, Is.EqualTo(BehaviourStatus.Running));
+
+            sut.Reset();
+            context.AddMilliseconds(600);
+
+            behaviourStatus = sut.Tick(context);
+
+            Assert.That(behaviourStatus, Is.EqualTo(BehaviourStatus.Running));
+        }
     }
 }
diff --git a/src/BehaviourTree/Behaviours/Wait.cs b/src/BehaviourTree/Behaviours/Wait.cs
@@ -37,6 +37,11 @@ protected override BehaviourStatus Update(TContext context)
         }
 
         protected override void OnTerminate(BehaviourStatus status)
+        {
+            DoReset(status);
+        }
+
+        protected override void DoReset(BehaviourStatus status)
         {
             _initialTimestamp = null;
         }