Add mock support for swsscommon classes (#1780)

This is a partial backport of #1392 Only add new mock support in unit test.
sonic-net · Aug 25, 2021 · 8aa2226 · 8aa2226
1 parent d939cfb
commit 8aa2226
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/mock_tables/dbconnector.py b/tests/mock_tables/dbconnector.py
@@ -8,7 +8,7 @@
 import redis
 import swsssdk
 from sonic_py_common import multi_asic
-from swsssdk import SonicDBConfig, SonicV2Connector
+from swsssdk import SonicDBConfig, SonicV2Connector, ConfigDBConnector, ConfigDBPipeConnector
 from swsscommon import swsscommon
 
 
@@ -151,3 +151,5 @@ def keys(self, pattern='*'):
 redis.StrictRedis = SwssSyncClient
 SonicV2Connector.connect = connect_SonicV2Connector
 swsscommon.SonicV2Connector = SonicV2Connector
+swsscommon.ConfigDBConnector = ConfigDBConnector
+swsscommon.ConfigDBPipeConnector = ConfigDBPipeConnector