Refine getDbId() calling to fix build after swss-common change (sonic…

…-net#1245) * Refine getDbId() calling to fix build after swss-common change * Fix mock tests with multiDB support and config
raphaelt-nvidia · Aug 19, 2020 · 30ddd59 · 30ddd59
1 parent 1496233
commit 30ddd59
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 5 deletions.
diff --git a/orchagent/orch.h b/orchagent/orch.h
@@ -128,7 +128,7 @@ class Consumer : public Executor {
 
     int getDbId() const
     {
-        return getConsumerTable()->getDbId();
+        return getConsumerTable()->getDbConnector()->getDbId();
     }
 
     std::string dumpTuple(swss::KeyOpFieldsValuesTuple &tuple);

diff --git a/tests/mock_tests/mock_dbconnector.cpp b/tests/mock_tests/mock_dbconnector.cpp
@@ -39,6 +39,7 @@ namespace swss
     }
 
     DBConnector::DBConnector(const std::string& dbName, unsigned int timeout, bool isTcpConn)
+        : m_dbName(dbName)
     {
         if (swss::SonicDBConfig::isInit() == false)
             swss::SonicDBConfig::initialize("./database_config.json");

diff --git a/tests/mock_tests/mock_table.cpp b/tests/mock_tests/mock_table.cpp
@@ -23,7 +23,7 @@ namespace swss
 
     bool Table::get(const std::string &key, std::vector<FieldValueTuple> &ovalues)
     {
-        auto table = gDB[getDbId()][getTableName()];
+        auto table = gDB[m_pipe->getDbId()][getTableName()];
         if (table.find(key) == table.end())
         {
             return false;
@@ -35,7 +35,7 @@ namespace swss
 
     bool Table::hget(const std::string &key, const std::string &field, std::string &value)
     {
-        auto table = gDB[getDbId()][getTableName()];
+        auto table = gDB[m_pipe->getDbId()][getTableName()];
         if (table.find(key) == table.end())
         {
             return false;
@@ -58,14 +58,14 @@ namespace swss
                     const std::string &op,
                     const std::string &prefix)
     {
-        auto &table = gDB[getDbId()][getTableName()];
+        auto &table = gDB[m_pipe->getDbId()][getTableName()];
         table[key] = values;
     }
 
     void Table::getKeys(std::vector<std::string> &keys)
     {
         keys.clear();
-        auto table = gDB[getDbId()][getTableName()];
+        auto table = gDB[m_pipe->getDbId()][getTableName()];
         for (const auto &it : table)
         {
             keys.push_back(it.first);