Update RobotController_test_hardware.cpp

leka · Feb 14, 2024 · 19aa686 · 19aa686
1 parent df8e38e
commit 19aa686
Showing 1 changed file with 5 additions and 10 deletions.
diff --git a/libs/RobotKit/tests/RobotController_test_hardware.cpp b/libs/RobotKit/tests/RobotController_test_hardware.cpp
@@ -3,6 +3,7 @@
 // SPDX-License-Identifier: Apache-2.0
 
 #include "./RobotController_test.h"
+#include "mocks/leka/DeepSleepEnabled.h"
 
 TEST_F(RobotControllerTest, stopActuators)
 {
@@ -32,18 +33,12 @@ TEST_F(RobotControllerTest, stopActuatorsAndLcd)
 
 TEST_F(RobotControllerTest, suspendHardwareForDeepSleep)
 {
-	EXPECT_CALL(mock_motor_left, enableDeepSleep);
-	EXPECT_CALL(mock_motor_right, enableDeepSleep);
-	EXPECT_CALL(mock_lcd, enableDeepSleep);
-	// TODO: Expect_call of RFID
-
-	auto components = std::to_array<interface::DeepSleepEnabled *>({
-		&mock_motor_left,
-		&mock_motor_right,
-		&mock_lcd,
-	});
+	mock::DeepSleepEnabled mock_deep_sleep_enabled {};
 
+	auto components = std::to_array<interface::DeepSleepEnabled *>({&mock_deep_sleep_enabled});
 	rc.registerDeepSleepEnabledComponents(components);
+
+	EXPECT_CALL(mock_deep_sleep_enabled, enableDeepSleep);
 	rc.suspendHardwareForDeepSleep();
 }