Fix coverage

Signed-off-by: Vivek Reddy <[email protected]>
sonic-net · Dec 8, 2023 · 504ae4a · 504ae4a
1 parent 3456f5c
commit 504ae4a
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/tests/mock_tests/warmrestarthelper_ut.cpp b/tests/mock_tests/warmrestarthelper_ut.cpp
@@ -47,6 +47,13 @@ namespace wrhelper_test
                             {"nexthop", "2.1.0.0"},
                             {"weight", "1"},
                         });
+        m_routeTable->set("1.2.0.0/24",
+                        {
+                            {"ifname", "eth2"},
+                            {"nexthop", "2.2.0.0"},
+                            {"weight", "1"},
+                            {"random_attrib", "random_val"},
+                        });
         wrHelper->runRestoration();
         ASSERT_EQ(wrHelper->getState(), WarmStart::RESTORED);
 
@@ -70,6 +77,16 @@ namespace wrhelper_test
                                         {"protocol", "kernel"}
                                     }
                                 });
+        wrHelper->insertRefreshMap({
+                                    "1.2.0.0/24",
+                                    "SET",
+                                    {
+                                        {"ifname", "eth2"},
+                                        {"nexthop", "2.2.0.0"},
+                                        {"weight", "1"},
+                                        {"protocol", "kernel"}
+                                    }
+                                });
         wrHelper->reconcile();
         ASSERT_EQ(wrHelper->getState(), WarmStart::RECONCILED);
 
@@ -82,5 +99,8 @@ namespace wrhelper_test
 
         m_routeTable->hget("1.1.0.0/24", "weight", val);
         ASSERT_EQ(val, "4");
+
+        m_routeTable->hget("1.2.0.0/24", "protocol", val);
+        ASSERT_EQ(val, "kernel");
     }
 }