From eb433aadcee846ebb0ca4af8d82b0f210ced2ddd Mon Sep 17 00:00:00 2001 From: Qi Luo Date: Wed, 25 Aug 2021 02:59:59 +0000 Subject: [PATCH] Add mock support for swsscommon classes --- tests/mock_tables/dbconnector.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/mock_tables/dbconnector.py b/tests/mock_tables/dbconnector.py index b95c49511d..30a9f5a653 100644 --- 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