|
1 | 1 | import sys
|
2 | 2 | import os
|
3 |
| -from sonic_platform_base import device_base |
4 | 3 | from unittest.mock import patch, MagicMock
|
| 4 | +sys.modules['sonic_platform'] = MagicMock() |
| 5 | +sys.modules['argparse'] = MagicMock() |
5 | 6 |
|
6 |
| -import pytest |
7 |
| -from click.testing import CliRunner |
8 |
| -from sonic_py_common import device_info |
| 7 | +import ssdutil.main as ssdutil # noqa: E402 |
9 | 8 |
|
10 | 9 | test_path = os.path.dirname(os.path.abspath(__file__))
|
11 | 10 | modules_path = os.path.dirname(test_path)
|
12 | 11 | sys.path.insert(0, modules_path)
|
13 | 12 |
|
14 |
| -sys.modules['sonic_platform'] = MagicMock() |
15 |
| -sys.modules['argparse'] = MagicMock() |
16 |
| - |
17 |
| -import ssdutil.main as ssdutil |
18 |
| -from .utils import get_result_and_return_code |
19 |
| - |
20 | 13 | ssdutil_output = """
|
21 | 14 | Device Model : InnoDisk Corp. - mSATA 3IE3
|
22 | 15 | Health : 99%
|
|
31 | 24 | Temperature : 30C
|
32 | 25 | """
|
33 | 26 |
|
| 27 | + |
34 | 28 | class Ssd():
|
| 29 | + |
35 | 30 | def get_model(self):
|
36 | 31 | return 'SkyNet'
|
| 32 | + |
37 | 33 | def get_firmware(self):
|
38 | 34 | return 'ABC'
|
| 35 | + |
39 | 36 | def get_serial(self):
|
40 | 37 | return 'T1000'
|
| 38 | + |
41 | 39 | def get_health(self):
|
42 | 40 | return 5
|
| 41 | + |
43 | 42 | def get_temperature(self):
|
44 | 43 | return 3000
|
| 44 | + |
45 | 45 | def get_vendor_output(self):
|
46 | 46 | return 'SONiC Test'
|
47 | 47 |
|
| 48 | + |
48 | 49 | class TestSsdutil:
|
49 | 50 |
|
50 |
| - @patch('sonic_py_common.device_info.get_paths_to_platform_and_hwsku_dirs', MagicMock(return_value = ("test_path", ""))) |
| 51 | + @patch('sonic_py_common.device_info.get_paths_to_platform_and_hwsku_dirs', |
| 52 | + MagicMock(return_value=("test_path", ""))) |
51 | 53 | @patch('os.geteuid', MagicMock(return_value=0))
|
52 |
| - @patch('sonic_platform_base.sonic_storage.ssd.SsdUtil', MagicMock(return_value=Ssd())) |
| 54 | + @patch('sonic_platform_base.sonic_storage.ssd.SsdUtil', |
| 55 | + MagicMock(return_value=Ssd())) |
53 | 56 | def test_happy_path(self):
|
54 |
| - ssdutil.ssdutil() |
55 |
| - |
56 | 57 |
|
| 58 | + ssdutil.ssdutil() |
0 commit comments