Wait behaviour bug fix

Author: Milian Lichere

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));
+        }
     }
 }

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;
         }