add unit tests

Author: zjswhhh

src/DbInterface.cpp

@@ -1169,7 +1169,7 @@ void DbInterface::getWarmRestartFlag(std::shared_ptr<swss::DBConnector> stateDbC
 //
 // check if warm restart is on when port completes reconciliation
 //
-void DbInterface::checkWarmRestart(const std::string portName)
+void DbInterface::checkWarmRestart(const std::string &portName)
 {
     MUXLOGDEBUG(portName);
 
@@ -1213,7 +1213,7 @@ void DbInterface::handleSetMuxReconciled()
 //
 // set config db mux mode 
 // 
-void DbInterface::setMuxMode(const std::string portName, const std::string state)
+void DbInterface::setMuxMode(const std::string &portName, const std::string state)
 {
     MUXLOGDEBUG(portName);
 

src/DbInterface.h

@@ -271,7 +271,7 @@ class DbInterface
      *  
      * @return none
      */
-    void setMuxMode(const std::string portName, const std::string state);
+    virtual void setMuxMode(const std::string &portName, const std::string state);
 
     /**
      * @method setMuxReconciled
@@ -280,7 +280,7 @@ class DbInterface
      * 
      * @return none
      */
-    void setMuxReconciled();
+    virtual void setMuxReconciled();
 
     /**
      * @method checkWarmRestart
@@ -291,7 +291,7 @@ class DbInterface
      * 
      * @return none
      */
-    void checkWarmRestart(const std::string portName);
+    virtual void checkWarmRestart(const std::string &portName);
 
 private:
     friend class test::MuxManagerTest;

src/MuxManager.cpp

@@ -476,7 +476,7 @@ void MuxManager::generateServerMac(uint16_t serverId, std::array<uint8_t, ETHER_
 //
 // process warm restart flag
 //
-void MuxManager::processWarmRestartFlag(const std::string flag)
+void MuxManager::processWarmRestartFlag(const std::string &flag)
 {
     mWarmRestartFlag = flag;
 

src/MuxManager.h

@@ -394,7 +394,7 @@ class MuxManager
      * 
      * @return 
      */
-    void processWarmRestartFlag(const std::string flag);
+    void processWarmRestartFlag(const std::string &flag);
 
     /**
      * @method getWarmRestartFlag 

test/FakeDbInterface.cpp

@@ -92,4 +92,14 @@ void FakeDbInterface::postPckLossRatio(
     mExpectedPacketCount = expectedPacketCount;
 } 
 
+void FakeDbInterface::setMuxReconciled()
+{
+    mSetMuxReconciledInvokeCount += 1;
+}
+
+void FakeDbInterface::checkWarmRestart(const std::string &portName)
+{
+    mCheckWarmRestartInvokeCount += 1;
+}
+
 } /* namespace test */

test/FakeDbInterface.h

@@ -62,6 +62,8 @@ class FakeDbInterface: public mux::DbInterface
 
     void setNextMuxState(mux_state::MuxState::Label label) {mNextMuxState = label;};
 
+    virtual void setMuxReconciled() override;
+    virtual void checkWarmRestart(const std::string &portName) override;
 
 public:
     mux_state::MuxState::Label mNextMuxState;
@@ -80,6 +82,9 @@ class FakeDbInterface: public mux::DbInterface
     uint32_t mPostLinkProberMetricsInvokeCount = 0;
     uint64_t mUnknownEventCount = 0;
     uint64_t mExpectedPacketCount = 0;
+
+    uint32_t mSetMuxReconciledInvokeCount = 0;
+    uint32_t mCheckWarmRestartInvokeCount = 0;
 };
 
 } /* namespace test */

test/LinkManagerStateMachineTest.cpp

@@ -188,6 +188,7 @@ void LinkManagerStateMachineTest::activateStateMachine()
 {
     mFakeMuxPort.activateStateMachine();
     mFakeMuxPort.handleDefaultRouteState("ok");
+    mFakeMuxPort.checkWarmRestart();
 }
 
 void LinkManagerStateMachineTest::setMuxActive()
@@ -1164,4 +1165,11 @@ TEST_F(LinkManagerStateMachineTest, EnableDecreaseLinkProberIntervalFeature)
     EXPECT_EQ(mFakeMuxPort.mFakeLinkProber->mDecreaseIntervalCallCount, 1);
 }
 
+TEST_F(LinkManagerStateMachineTest, CheckWarmRestart)
+{
+    setMuxActive();
+
+    EXPECT_EQ(mDbInterfacePtr->mCheckWarmRestartInvokeCount, 1);
+}
+
 } /* namespace test */

test/LinkManagerStateMachineTest.h

@@ -55,6 +55,7 @@ class LinkManagerStateMachineTest: public ::testing::Test
     void postPeerLinkStateEvent(std::string linkState, uint32_t count = 0);
     void postPckLossRatioUpdateEvent(uint64_t unknownCount, uint64_t totalCount);
     void postPckLossCountsResetEvent();
+    void checkWarmRestart();
 
 public:
     boost::asio::io_service mIoService;

test/MuxManagerTest.cpp

@@ -200,6 +200,11 @@ void MuxManagerTest::updatePortCableType(const std::string &port, const std::str
     mMuxManagerPtr->updatePortCableType(port, cableType);
 }
 
+void MuxManagerTest::processWarmRestartFlag(const std::string &flag)
+{
+    mMuxManagerPtr->processWarmRestartFlag(flag);
+}
+
 void MuxManagerTest::initLinkProberActiveActive(std::shared_ptr<link_manager::ActiveActiveStateMachine> linkManagerStateMachineActiveActive)
 {
     mFakeLinkProber = std::make_shared<FakeLinkProber> (linkManagerStateMachineActiveActive->getLinkProberStateMachinePtr().get());
@@ -631,4 +636,21 @@ INSTANTIATE_TEST_CASE_P(
     )
 );
 
+TEST_F(MuxManagerTest, WarmRestart)
+{
+    std::string port = "Ethernet0";
+
+    createPort(port);
+    EXPECT_FALSE(mMuxManagerPtr->getWarmRestartFlag());
+
+    std::string flag = "true";
+    processWarmRestartFlag(flag);
+    EXPECT_TRUE(mMuxManagerPtr->getWarmRestartFlag());
+
+    EXPECT_EQ(mDbInterfacePtr->mSetMuxReconciledInvokeCount, 0);
+    runIoService();
+    EXPECT_EQ(mDbInterfacePtr->mSetMuxReconciledInvokeCount, 1);
+}
+
 } /* namespace test */
+

test/MuxManagerTest.h

@@ -75,6 +75,7 @@ class MuxManagerTest: public testing::Test
     void initLinkProberActiveActive(std::shared_ptr<link_manager::ActiveActiveStateMachine> linkManagerStateMachine);
     void initLinkProberActiveStandby(std::shared_ptr<link_manager::ActiveStandbyStateMachine> linkManagerStateMachine);
     void generateServerMac(const std::string &portName, std::array<uint8_t, ETHER_ADDR_LEN> &address);
+    void processWarmRestartFlag(const std::string &flag);
     void createPort(std::string port, common::MuxPortConfig::PortCableType portCableType = common::MuxPortConfig::PortCableType::ActiveStandby);
 
 public: