Update BehaviorKit_test.cpp

leka · Dec 14, 2023 · 87c4d6e · 87c4d6e
1 parent 55e4556
commit 87c4d6e
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/libs/BehaviorKit/tests/BehaviorKit_test.cpp b/libs/BehaviorKit/tests/BehaviorKit_test.cpp
@@ -76,8 +76,11 @@ TEST_F(BehaviorKitTest, startBehaviorIDNotRegistered)
 	});
 	behaviorkit.registerBehaviors(behaviors);
 
-	EXPECT_CALL(mock_behavior_a, id);
-	EXPECT_CALL(mock_behavior_b, id);
+	auto behavior_a_id = BehaviorID {1};
+	auto behavior_b_id = BehaviorID {2};
+
+	EXPECT_CALL(mock_behavior_a, id).WillOnce(Return(behavior_a_id));
+	EXPECT_CALL(mock_behavior_b, id).WillOnce(Return(behavior_b_id));
 
 	behaviorkit.start(3);
 }