Skip to content

Commit

Permalink
Filled out init files in mocked libs to prevent unintentional module …
Browse files Browse the repository at this point in the history
…hiding
  • Loading branch information
assrinivasan committed May 31, 2024
1 parent 36ab2f8 commit 28db41d
Show file tree
Hide file tree
Showing 3 changed files with 15 additions and 1 deletion.
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
from . import ssd_generic # noqa F401
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
from . import ssd # noqa F401
14 changes: 13 additions & 1 deletion tests/ssdutil_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,7 @@
from unittest.mock import patch, MagicMock
sys.modules['sonic_platform'] = MagicMock()
sys.modules['argparse'] = MagicMock()
sys.modules['subprocess'] = MagicMock()

import ssdutil.main as ssdutil # noqa: E402

Expand Down Expand Up @@ -58,6 +59,17 @@ class TestSsdutil:

@patch('sonic_py_common.device_info.get_paths_to_platform_and_hwsku_dirs', MagicMock(return_value=("test_path", ""))) # noqa: E501
@patch('os.geteuid', MagicMock(return_value=0))
def test_happy_path(self):
@patch('builtins.print')
def test_sonic_storage_path(self, mock_print):

ssdutil.ssdutil()
assert mock_print.call_count == 6

@patch('sonic_py_common.device_info.get_paths_to_platform_and_hwsku_dirs', MagicMock(return_value=("test_path", ""))) # noqa: E501
@patch('sonic_platform_base.sonic_storage.ssd', MagicMock(side_effect=ImportError())) # noqa: E501
@patch('os.geteuid', MagicMock(return_value=0))
@patch('builtins.print')
def test_sonic_ssd_path(self, mock_print):

ssdutil.ssdutil()
assert mock_print.call_count == 6

0 comments on commit 28db41d

Please sign in to comment.