2c3dc47b2c
#### Why I did it These methods were added to make some convenient platform and chassis information methods accessible through sonic-py-common. These methods were refactored from sonic-utilities and are used in the `show platform summary` and `show version` commands. #### How I did it There are two methods, one is `get_platform_info()` which simply calls local methods to collect useful platform information into a dictionary format, this came directly from sonic-utilities.
89 lines
3.0 KiB
Python
89 lines
3.0 KiB
Python
import os
|
|
import sys
|
|
|
|
# TODO: Remove this if/else block once we no longer support Python 2
|
|
if sys.version_info.major == 3:
|
|
from unittest import mock
|
|
else:
|
|
# Expect the 'mock' package for python 2
|
|
# https://pypi.python.org/pypi/mock
|
|
import mock
|
|
|
|
import pytest
|
|
|
|
from sonic_py_common import device_info
|
|
|
|
from .mock_swsssdk import SonicV2Connector
|
|
|
|
# TODO: Remove this if/else block once we no longer support Python 2
|
|
if sys.version_info.major == 3:
|
|
BUILTINS = "builtins"
|
|
else:
|
|
BUILTINS = "__builtin__"
|
|
|
|
MACHINE_CONF_CONTENTS = """\
|
|
onie_version=2016.11-5.1.0008-9600
|
|
onie_vendor_id=33049
|
|
onie_machine_rev=0
|
|
onie_arch=x86_64
|
|
onie_config_version=1
|
|
onie_build_date="2017-04-26T11:01+0300"
|
|
onie_partition_type=gpt
|
|
onie_kernel_version=4.10.11
|
|
onie_firmware=auto
|
|
onie_switch_asic=mlnx
|
|
onie_skip_ethmgmt_macs=yes
|
|
onie_machine=mlnx_msn2700
|
|
onie_platform=x86_64-mlnx_msn2700-r0"""
|
|
|
|
EXPECTED_GET_MACHINE_INFO_RESULT = {
|
|
'onie_arch': 'x86_64',
|
|
'onie_skip_ethmgmt_macs': 'yes',
|
|
'onie_platform': 'x86_64-mlnx_msn2700-r0',
|
|
'onie_machine_rev': '0',
|
|
'onie_version': '2016.11-5.1.0008-9600',
|
|
'onie_machine': 'mlnx_msn2700',
|
|
'onie_config_version': '1',
|
|
'onie_partition_type': 'gpt',
|
|
'onie_build_date': '"2017-04-26T11:01+0300"',
|
|
'onie_switch_asic': 'mlnx',
|
|
'onie_vendor_id': '33049',
|
|
'onie_firmware': 'auto',
|
|
'onie_kernel_version': '4.10.11'
|
|
}
|
|
|
|
class TestDeviceInfo(object):
|
|
@pytest.fixture(scope="class", autouse=True)
|
|
def sanitize_environment(self):
|
|
# Clear environment variables, in case a variable is set in the test
|
|
# environment (e.g., PLATFORM) which could modify the behavior of sonic-py-common
|
|
with mock.patch.dict(os.environ, {}, clear=True):
|
|
yield
|
|
|
|
def test_get_machine_info(self):
|
|
with mock.patch("os.path.isfile") as mock_isfile:
|
|
mock_isfile.return_value = True
|
|
open_mocked = mock.mock_open(read_data=MACHINE_CONF_CONTENTS)
|
|
with mock.patch("{}.open".format(BUILTINS), open_mocked):
|
|
result = device_info.get_machine_info()
|
|
assert result == EXPECTED_GET_MACHINE_INFO_RESULT
|
|
open_mocked.assert_called_once_with("/host/machine.conf")
|
|
|
|
def test_get_platform(self):
|
|
with mock.patch("sonic_py_common.device_info.get_machine_info") as get_machine_info_mocked:
|
|
get_machine_info_mocked.return_value = EXPECTED_GET_MACHINE_INFO_RESULT
|
|
result = device_info.get_platform()
|
|
assert result == "x86_64-mlnx_msn2700-r0"
|
|
|
|
def test_get_chassis_info(self):
|
|
with mock.patch("sonic_py_common.device_info.SonicV2Connector", new=SonicV2Connector):
|
|
result = device_info.get_chassis_info()
|
|
truth = {"serial": SonicV2Connector.TEST_SERIAL,
|
|
"model": SonicV2Connector.TEST_MODEL,
|
|
"revision": SonicV2Connector.TEST_REV}
|
|
assert result == truth
|
|
|
|
@classmethod
|
|
def teardown_class(cls):
|
|
print("TEARDOWN")
|