From 8965cb5ce1e5ae140314611fe70978c39c90f7af Mon Sep 17 00:00:00 2001
From: Serhii Snitsaruk <limbonaut@pm.me>
Date: Fri, 1 Sep 2023 13:11:00 +0200
Subject: [PATCH] Add tests for BTSubtree

---
 tests/test_subtree.h | 73 ++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 73 insertions(+)
 create mode 100644 tests/test_subtree.h

diff --git a/tests/test_subtree.h b/tests/test_subtree.h
new file mode 100644
index 0000000..064ad1d
--- /dev/null
+++ b/tests/test_subtree.h
@@ -0,0 +1,73 @@
+/**
+ * test_subtree.h
+ * =============================================================================
+ * Copyright 2021-2023 Serhii Snitsaruk
+ *
+ * Use of this source code is governed by an MIT-style
+ * license that can be found in the LICENSE file or at
+ * https://opensource.org/licenses/MIT.
+ * =============================================================================
+ */
+
+#ifndef TEST_SUBTREE_H
+#define TEST_SUBTREE_H
+
+#include "limbo_test.h"
+
+#include "modules/limboai/bt/behavior_tree.h"
+#include "modules/limboai/bt/tasks/bt_task.h"
+#include "modules/limboai/bt/tasks/decorators/bt_subtree.h"
+
+namespace TestSubtree {
+
+TEST_CASE("[Modules][LimboAI] BTSubtree") {
+	ClassDB::register_class<BTTestAction>();
+
+	Ref<BTSubtree> st = memnew(BTSubtree);
+	Ref<Blackboard> bb = memnew(Blackboard);
+	Node *dummy = memnew(Node);
+
+	SUBCASE("When empty") {
+		ERR_PRINT_OFF;
+		st->initialize(dummy, bb);
+		CHECK(st->execute(0.01666) == BTTask::FAILURE);
+		ERR_PRINT_ON;
+	}
+
+	SUBCASE("With a subtree assigned") {
+		Ref<BehaviorTree> bt = memnew(BehaviorTree);
+		Ref<BTTestAction> task = memnew(BTTestAction(BTTask::SUCCESS));
+		bt->set_root_task(task);
+		st->set_subtree(bt);
+
+		CHECK(st->get_child_count() == 0);
+		st->initialize(dummy, bb);
+		CHECK(st->get_child_count() == 1);
+		CHECK(st->get_child(0) != task);
+
+        Ref<BTTestAction> ta = st->get_child(0);
+		REQUIRE(ta.is_valid());
+
+		SUBCASE("When child succeeds") {
+			ta->ret_status = BTTask::SUCCESS;
+			CHECK(st->execute(0.01666) == BTTask::SUCCESS);
+			CHECK_STATUS_ENTRIES_TICKS_EXITS(ta, BTTask::SUCCESS, 1, 1, 1);
+		}
+		SUBCASE("When child fails") {
+			ta->ret_status = BTTask::FAILURE;
+			CHECK(st->execute(0.01666) == BTTask::FAILURE);
+			CHECK_STATUS_ENTRIES_TICKS_EXITS(ta, BTTask::FAILURE, 1, 1, 1);
+		}
+		SUBCASE("When child is running") {
+			ta->ret_status = BTTask::RUNNING;
+			CHECK(st->execute(0.01666) == BTTask::RUNNING);
+			CHECK_STATUS_ENTRIES_TICKS_EXITS(ta, BTTask::RUNNING, 1, 1, 0);
+		}
+	}
+
+	memdelete(dummy);
+}
+
+} //namespace TestSubtree
+
+#endif // TEST_SUBTREE_H