Add SwitchContainer tests

unittest/vslib/TestSwitchContainer.cpp

@@ -0,0 +1,74 @@
+#include "SwitchContainer.h"
+
+#include <gtest/gtest.h>
+
+using namespace saivs;
+
+TEST(SwitchContainer, insert)
+{
+    SwitchContainer sc;
+
+    EXPECT_THROW(sc.insert(nullptr), std::runtime_error);
+
+    auto s = std::make_shared<Switch>(0x2100000000);
+
+    sc.insert(s);
+
+    EXPECT_THROW(sc.insert(s), std::runtime_error);
+}
+
+TEST(SwitchContainer, removeSwitch)
+{
+    SwitchContainer sc;
+
+    EXPECT_THROW(sc.removeSwitch(0), std::runtime_error);
+
+    auto s = std::make_shared<Switch>(0x2100000000);
+
+    sc.insert(s);
+
+    sc.removeSwitch(0x2100000000);
+
+    s = std::make_shared<Switch>(0x2100000000);
+
+    sc.insert(s);
+
+    EXPECT_THROW(sc.removeSwitch(nullptr), std::runtime_error);
+
+    EXPECT_THROW(sc.removeSwitch(1), std::runtime_error);
+
+    sc.removeSwitch(s);
+}
+
+TEST(SwitchContainer, clear)
+{
+    SwitchContainer sc;
+
+    sc.clear();
+}
+
+TEST(SwitchContainer, contains)
+{
+    SwitchContainer sc;
+
+    auto s = std::make_shared<Switch>(0x2100000000);
+
+    sc.insert(s);
+
+    EXPECT_TRUE(sc.contains(0x2100000000));
+
+    EXPECT_FALSE(sc.contains(1));
+}
+
+TEST(SwitchContainer, getSwitch)
+{
+    SwitchContainer sc;
+
+    auto s = std::make_shared<Switch>(0x2100000000);
+
+    sc.insert(s);
+
+    EXPECT_NE(sc.getSwitch(0x2100000000), nullptr);
+
+    EXPECT_EQ(sc.getSwitch(1), nullptr);
+}

vslib/SwitchContainer.cpp

@@ -10,6 +10,11 @@ void SwitchContainer::insert(
 {
     SWSS_LOG_ENTER();
 
+    if (sw == nullptr)
+    {
+        SWSS_LOG_THROW("switch pointer can't be nullptr");
+    }
+
     auto switchId = sw->getSwitchId();
 
     if (m_switchMap.find(switchId) != m_switchMap.end())
@@ -42,6 +47,11 @@ void SwitchContainer::removeSwitch(
 {
     SWSS_LOG_ENTER();
 
+    if (sw == nullptr)
+    {
+        SWSS_LOG_THROW("switch pointer can't be nullptr");
+    }
+
     removeSwitch(sw->getSwitchId());
 }