Fixed a bug where Wait action failed

Author: ashblue

Assets/FluidBehaviorTree/Scripts/Tasks/Actions/Editor/WaitTest.cs

@@ -1,4 +1,5 @@
-using NUnit.Framework;
+using Adnc.FluidBT.Trees;
+using NUnit.Framework;
 
 namespace Adnc.FluidBT.Tasks.Actions.Testing {
     public class WaitTest {
@@ -27,5 +28,15 @@ public void It_should_trigger_success_after_2_ticks () {
             Assert.AreEqual(TaskStatus.Continue, wait.Update());
             Assert.AreEqual(TaskStatus.Success, wait.Update());
         }
+
+        [Test]
+        public void It_should_trigger_continue_after_tree_restarts () {
+            var tree = new BehaviorTree(null);
+            tree.AddNode(tree.Root, new Wait());
+
+            Assert.AreEqual(TaskStatus.Continue, tree.Tick());
+            Assert.AreEqual(TaskStatus.Success, tree.Tick());
+            Assert.AreEqual(TaskStatus.Continue, tree.Tick());
+        }
     }
 }

Assets/FluidBehaviorTree/Scripts/Tasks/Actions/Wait.cs

@@ -3,7 +3,11 @@ public class Wait : ActionBase {
         public int turns = 1;
 
         private int _ticks;
-        
+
+        protected override void OnStart () {
+            _ticks = 0;
+        }
+
         protected override TaskStatus OnUpdate () {
             if (_ticks < turns) {
                 _ticks++;