diff --git a/sonic-pcied/tests/mocked_libs/swsscommon/swsscommon.py b/sonic-pcied/tests/mocked_libs/swsscommon/swsscommon.py index 6947a8601..e5aa2a2ba 100644 --- a/sonic-pcied/tests/mocked_libs/swsscommon/swsscommon.py +++ b/sonic-pcied/tests/mocked_libs/swsscommon/swsscommon.py @@ -52,3 +52,12 @@ def __repr__(self): def __str__(self): return repr(self.fv_dict) + +class ConfigDBConnector: + pass + +class SonicDBConfig: + pass + +class SonicV2Connector: + pass diff --git a/sonic-pcied/tests/test_DaemonPcied.py b/sonic-pcied/tests/test_DaemonPcied.py index c8c17a276..2c3c953e7 100644 --- a/sonic-pcied/tests/test_DaemonPcied.py +++ b/sonic-pcied/tests/test_DaemonPcied.py @@ -11,14 +11,11 @@ else: import mock -from sonic_py_common import daemon_base - from .mock_platform import MockPcieUtil SYSLOG_IDENTIFIER = 'pcied_test' NOT_AVAILABLE = 'N/A' -daemon_base.db_connect = mock.MagicMock() tests_path = os.path.dirname(os.path.abspath(__file__)) @@ -26,6 +23,9 @@ mocked_libs_path = os.path.join(tests_path, "mocked_libs") sys.path.insert(0, mocked_libs_path) +from sonic_py_common import daemon_base +daemon_base.db_connect = mock.MagicMock() + # Add path to the file under test so that we can load it modules_path = os.path.dirname(tests_path) scripts_path = os.path.join(modules_path, "scripts") diff --git a/sonic-pcied/tests/test_pcied.py b/sonic-pcied/tests/test_pcied.py index a99331846..3850d38ed 100644 --- a/sonic-pcied/tests/test_pcied.py +++ b/sonic-pcied/tests/test_pcied.py @@ -10,7 +10,6 @@ else: from mock import MagicMock, patch, mock_open -from sonic_py_common import daemon_base, device_info from .mock_platform import MockPcieUtil tests_path = os.path.dirname(os.path.abspath(__file__)) @@ -18,6 +17,7 @@ # Add mocked_libs path so that the file under test can load mocked modules from there mocked_libs_path = os.path.join(tests_path, "mocked_libs") sys.path.insert(0, mocked_libs_path) +from sonic_py_common import daemon_base, device_info # Add path to the file under test so that we can load it modules_path = os.path.dirname(tests_path)