fix UT failures

Signed-off-by: Prabhat Aravind <paravind@microsoft.com>
prabhataravind · Oct 26, 2024 · e71ac5c · e71ac5c
1 parent c19b918
commit e71ac5c
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/tests/mock_tests/database_config.json b/tests/mock_tests/database_config.json
@@ -76,6 +76,16 @@
             "id" : 14,
             "separator": ":",
             "instance" : "redis"
+        },
+        "DPU_APPL_DB" : {
+            "id" : 15,
+            "separator": ":",
+            "instance" : "redis"
+        },
+        "DPU_APPL_STATE_DB" : {
+            "id" : 16,
+            "separator": "|",
+            "instance" : "redis"
         }
     },
     "VERSION" : "1.0"

diff --git a/tests/mock_tests/mock_orch_test.cpp b/tests/mock_tests/mock_orch_test.cpp
@@ -65,6 +65,8 @@ void MockOrchTest::SetUp()
     m_app_db = make_shared<swss::DBConnector>("APPL_DB", 0);
     m_config_db = make_shared<swss::DBConnector>("CONFIG_DB", 0);
     m_state_db = make_shared<swss::DBConnector>("STATE_DB", 0);
+    m_dpu_app_db = make_shared<swss::DBConnector>("DPU_APPL_DB", 0);
+    m_dpu_app_state_db = make_shared<swss::DBConnector>("DPU_APPL_STATE_DB", 0);
     m_chassis_app_db = make_shared<swss::DBConnector>("CHASSIS_APP_DB", 0);
 
     PrepareSai();
@@ -276,4 +278,4 @@ void MockOrchTest::TearDown()
 
     ut_helper::uninitSaiApi();
 }
-}
+}