Add unit test

uweseimet · Dec 12, 2023 · 7be4eec · 7be4eec
1 parent a4f4303
commit 7be4eec
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/cpp/test/abstract_controller_test.cpp b/cpp/test/abstract_controller_test.cpp
@@ -11,6 +11,21 @@
 
 using namespace scsi_defs;
 
+TEST(AbstractControllerTest, ShutdownMode)
+{
+    MockAbstractController controller;
+
+    EXPECT_EQ(AbstractController::shutdown_mode::NONE, controller.GetShutdownMode());
+    controller.ScheduleShutdown(AbstractController::shutdown_mode::RESTART_PI);
+    EXPECT_EQ(AbstractController::shutdown_mode::RESTART_PI, controller.GetShutdownMode());
+    controller.ScheduleShutdown(AbstractController::shutdown_mode::STOP_S2P);
+    EXPECT_EQ(AbstractController::shutdown_mode::STOP_S2P, controller.GetShutdownMode());
+    controller.ScheduleShutdown(AbstractController::shutdown_mode::STOP_PI);
+    EXPECT_EQ(AbstractController::shutdown_mode::STOP_PI, controller.GetShutdownMode());
+    controller.ScheduleShutdown(AbstractController::shutdown_mode::RESTART_PI);
+    EXPECT_EQ(AbstractController::shutdown_mode::RESTART_PI, controller.GetShutdownMode());
+}
+
 TEST(AbstractControllerTest, AllocateCmd)
 {
     MockAbstractController controller;