74 lines
2.0 KiB
C++
74 lines
2.0 KiB
C++
/**
|
|
* 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
|