gNOI Warm Reboot - Added tests

Author: rkavitha-hcl

src/sonic-framework/rebootbackend/Makefile.am

@@ -10,7 +10,7 @@ DBGFLAGS = -g
 endif
 
 rebootbackend_SOURCES = rebootbackend.cpp rebootbe.cpp interfaces.cpp \
-                        reboot_thread.cpp redis_utils.cpp \
+                        reboot_thread.cpp \
                         reboot_common.cpp \
                         system/system.pb.cc types/types.pb.cc \
                         common/common.pb.cc 

src/sonic-framework/rebootbackend/redis_utils.cpp

@@ -89,6 +89,34 @@ TEST_F(RebootStatusTest, TestGetStatus) {
   EXPECT_EQ(0, response.when());
 }
 
+TEST_F(RebootStatusTest, TestGetWarmStatus) {
+  std::chrono::nanoseconds curr_ns = std::chrono::high_resolution_clock::now().time_since_epoch();
+
+  m_status.set_start_status(RebootMethod::WARM, "reboot because");
+
+  RebootStatusResponse response = m_status.get_response();
+  EXPECT_EQ(response.status().status(),
+            RebootStatus_Status::RebootStatus_Status_STATUS_UNKNOWN);
+
+  m_status.set_completed_status(
+      RebootStatus_Status::RebootStatus_Status_STATUS_SUCCESS, "anything");
+
+  response = m_status.get_response();
+
+  // message should be empty while reboot is active
+  EXPECT_THAT(response.status().message(), StrEq(""));
+
+  uint64_t reboot_ns = response.when();
+  EXPECT_TRUE(reboot_ns > (uint64_t)curr_ns.count());
+
+  m_status.set_inactive();
+  response = m_status.get_response();
+  EXPECT_THAT(response.status().message(), StrEq("anything"));
+  EXPECT_EQ(response.status().status(),
+            RebootStatus_Status::RebootStatus_Status_STATUS_SUCCESS);
+  EXPECT_EQ(0, response.when());
+}
+
 class RebootThreadTest : public ::testing::Test {
  protected:
   RebootThreadTest()
@@ -263,7 +291,7 @@ TEST_F(RebootThreadTest, TestUnsupportedRebootType) {
             "RebootThread: Start rx'd unsupported method");
 }
 
-TEST_F(RebootThreadTest, TestInvalidMethodfDoReboot) {
+TEST_F(RebootThreadTest, TestInvalidMethodDoReboot) {
   set_start_status(RebootMethod::POWERUP, "time to reboot");
   do_reboot();
   force_inactive();
@@ -273,4 +301,41 @@ TEST_F(RebootThreadTest, TestInvalidMethodfDoReboot) {
       IsStatus(RebootStatus_Status::RebootStatus_Status_STATUS_UNKNOWN, ""));
 }
 
+TEST_F(RebootThreadTest, TestNoWarmIfNonRetriableFailure) {
+  set_start_status(RebootMethod::WARM, "time to reboot");
+  set_completed_status(RebootStatus_Status::RebootStatus_Status_STATUS_FAILURE,
+                       "failed to warm reboot");
+  force_inactive();
+
+  RebootRequest request;
+  request.set_method(RebootMethod::WARM);
+
+  NotificationResponse response = m_reboot_thread.Start(request);
+  EXPECT_EQ(response.status, swss::StatusCode::SWSS_RC_FAILED_PRECONDITION);
+  EXPECT_EQ(response.json_string,
+            "RebootThread: last WARM reboot failed with non-retriable failure");
+}
+
+TEST_F(RebootThreadTest, TestSigTermStartofDoReboot) {
+  sigterm_requested = true;
+  set_start_status(RebootMethod::WARM, "time to reboot");
+  do_reboot();
+  force_inactive();
+  RebootStatusResponse response = m_reboot_thread.GetResponse();
+  EXPECT_THAT(
+      response,
+      IsStatus(RebootStatus_Status::RebootStatus_Status_STATUS_UNKNOWN, ""));
+}
+
+TEST_F(RebootThreadTest, TestWaitForRebootPositive) {
+  overwrite_reboot_timeout(1);
+  set_start_status(RebootMethod::WARM, "time to reboot");
+  swss::Select s;
+  swss::SelectableEvent m_stop;
+  s.addSelectable(&m_stop);
+  RebootThread::Progress progress = wait_for_platform_reboot(s);
+  EXPECT_EQ(progress, RebootThread::Progress::PROCEED);
+}
+
+
 }  // namespace rebootbackend