e8b4c2a1f4
- Why I did it * To support systems with dynamic port configuration * Apply lazy initialization to faster the speed of loading platform API - How I did it * Add module.py to implement dynamic port configuration (aka line card model) * Adjust chassis.py, platform.py, thermal.py, sfp.py to support dynamic port configuration * Optimize existing code - How to verify it Platform regression on MSN4700, MSN3800 and MSN2700, 100% pass Unit test covers all new changes.
109 lines
4.7 KiB
Python
109 lines
4.7 KiB
Python
#
|
|
# Copyright (c) 2021 NVIDIA CORPORATION & AFFILIATES.
|
|
# Apache-2.0
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
#
|
|
|
|
import os
|
|
import pytest
|
|
import sys
|
|
if sys.version_info.major == 3:
|
|
from unittest.mock import MagicMock, patch
|
|
else:
|
|
from mock import MagicMock, patch
|
|
|
|
test_path = os.path.dirname(os.path.abspath(__file__))
|
|
modules_path = os.path.dirname(test_path)
|
|
sys.path.insert(0, modules_path)
|
|
|
|
from sonic_platform.chassis import Chassis
|
|
from sonic_platform.eeprom import Eeprom, EepromContentVisitor
|
|
|
|
|
|
class TestEeprom:
|
|
@patch('os.path.exists', MagicMock(return_value=True))
|
|
@patch('os.path.islink', MagicMock(return_value=True))
|
|
@patch('sonic_platform.eeprom.Eeprom.get_system_eeprom_info')
|
|
def test_chassis_eeprom(self, mock_eeprom_info):
|
|
mock_eeprom_info.return_value = {
|
|
hex(Eeprom._TLV_CODE_PRODUCT_NAME): 'MSN3420',
|
|
hex(Eeprom._TLV_CODE_PART_NUMBER): 'MSN3420-CB2FO',
|
|
hex(Eeprom._TLV_CODE_MAC_BASE): '1C:34:DA:1C:9F:00',
|
|
hex(Eeprom._TLV_CODE_SERIAL_NUMBER): 'MT2019X13878'
|
|
}
|
|
chassis = Chassis()
|
|
assert chassis.get_name() == 'MSN3420'
|
|
assert chassis.get_model() == 'MSN3420-CB2FO'
|
|
assert chassis.get_base_mac() == '1C:34:DA:1C:9F:00'
|
|
assert chassis.get_serial() == 'MT2019X13878'
|
|
assert chassis.get_system_eeprom_info() == mock_eeprom_info.return_value
|
|
|
|
def test_eeprom_init(self):
|
|
# Test symlink not exist, there is an exception
|
|
with pytest.raises(RuntimeError):
|
|
Eeprom()
|
|
|
|
@patch('os.path.exists', MagicMock(return_value=True))
|
|
@patch('os.path.islink', MagicMock(return_value=True))
|
|
def test_get_system_eeprom_info_from_db(self):
|
|
return_values = {
|
|
('EEPROM_INFO|State', 'Initialized'): '1',
|
|
('EEPROM_INFO|{}'.format(hex(Eeprom._TLV_CODE_PRODUCT_NAME)), 'Value'): 'MSN3420',
|
|
('EEPROM_INFO|{}'.format(hex(Eeprom._TLV_CODE_PART_NUMBER)), 'Value'): 'MSN3420-CB2FO',
|
|
('EEPROM_INFO|{}'.format(hex(Eeprom._TLV_CODE_MAC_BASE)), 'Value'): '1C:34:DA:1C:9F:00',
|
|
('EEPROM_INFO|{}'.format(hex(Eeprom._TLV_CODE_SERIAL_NUMBER)), 'Value'): 'MT2019X13878',
|
|
('EEPROM_INFO|{}'.format(hex(Eeprom._TLV_CODE_VENDOR_EXT)), 'Num_vendor_ext'): '2',
|
|
('EEPROM_INFO|{}'.format(hex(Eeprom._TLV_CODE_VENDOR_EXT)), 'Value_0'): 'ext1',
|
|
('EEPROM_INFO|{}'.format(hex(Eeprom._TLV_CODE_VENDOR_EXT)), 'Value_1'): 'ext2',
|
|
('EEPROM_INFO|{}'.format(hex(Eeprom._TLV_CODE_CRC_32)), 'Value'): 'CRC_VALUE',
|
|
}
|
|
def side_effect(key, field):
|
|
return return_values.get((key, field))
|
|
eeprom = Eeprom()
|
|
eeprom._redis_hget = MagicMock(side_effect = side_effect)
|
|
|
|
info = eeprom.get_system_eeprom_info()
|
|
assert eeprom.get_product_name() == 'MSN3420'
|
|
assert eeprom.get_part_number() == 'MSN3420-CB2FO'
|
|
assert eeprom.get_base_mac() == '1C:34:DA:1C:9F:00'
|
|
assert eeprom.get_serial_number() == 'MT2019X13878'
|
|
assert info[hex(Eeprom._TLV_CODE_VENDOR_EXT)] == ['ext1', 'ext2']
|
|
assert info[hex(Eeprom._TLV_CODE_CRC_32)] == 'CRC_VALUE'
|
|
|
|
@patch('os.path.exists', MagicMock(return_value=True))
|
|
@patch('os.path.islink', MagicMock(return_value=True))
|
|
def test_get_system_eeprom_info_from_hardware(self):
|
|
eeprom = Eeprom()
|
|
eeprom.p = os.path.join(test_path, 'mock_eeprom_data')
|
|
eeprom._redis_hget = MagicMock()
|
|
info = eeprom.get_system_eeprom_info()
|
|
assert eeprom.get_product_name() == 'MSN3800'
|
|
assert eeprom.get_part_number() == 'MSN3800-CS2FO'
|
|
assert eeprom.get_base_mac() == 'B8:59:9F:A9:34:00'
|
|
assert eeprom.get_serial_number() == 'MT1937X00537'
|
|
assert info[hex(Eeprom._TLV_CODE_CRC_32)] == '0x9EFF0119'
|
|
|
|
def test_eeprom_content_visitor(self):
|
|
content = {}
|
|
v = EepromContentVisitor(content)
|
|
v.visit_tlv('tlv1', Eeprom._TLV_CODE_PRODUCT_NAME, 7, 'MSN3420')
|
|
v.visit_tlv('tlv2', Eeprom._TLV_CODE_VENDOR_EXT, 4, 'ext1')
|
|
v.visit_tlv('tlv3', Eeprom._TLV_CODE_VENDOR_EXT, 4, 'ext2')
|
|
assert content[hex(Eeprom._TLV_CODE_PRODUCT_NAME)] == 'MSN3420'
|
|
assert content[hex(Eeprom._TLV_CODE_VENDOR_EXT)] == ['ext1', 'ext2']
|
|
|
|
|
|
|
|
|