1175143af1
- Why I did it As a LED indicator to help user to find switch location in the lab, UID LED is a useful LED in Mellanox switch. - How I did it I add a new member _led_uid in Mellanox/Chassis.py, and extend Mellanox/led.py to support blue color. Relevant platform-common PR sonic-net/sonic-platform-common#369 - How to verify it Add unit test cases in test.py, and do manual test including turn-on/off/show uid led. Signed-off-by: David Xia <daxia@nvidia.com>
182 lines
7.9 KiB
Python
182 lines
7.9 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 sys
|
|
if sys.version_info.major == 3:
|
|
from unittest import mock
|
|
else:
|
|
import mock
|
|
|
|
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 import utils
|
|
from sonic_platform.chassis import Chassis
|
|
from sonic_platform.fan import Fan
|
|
from sonic_platform.fan_drawer import RealDrawer, VirtualDrawer
|
|
from sonic_platform.led import Led
|
|
from sonic_platform.psu import FixedPsu, Psu
|
|
|
|
class TestLed:
|
|
@mock.patch('sonic_platform.led.Led._wait_files_ready', mock.MagicMock(return_value=True))
|
|
@mock.patch('sonic_platform.chassis.extract_RJ45_ports_index', mock.MagicMock(return_value=True))
|
|
def test_chassis_led(self):
|
|
chassis = Chassis()
|
|
assert chassis._led is None
|
|
assert chassis.set_status_led('red') is False
|
|
physical_led = chassis._led
|
|
assert physical_led is not None
|
|
self._verify_non_shared_led(physical_led, chassis)
|
|
|
|
def test_uid_led(self):
|
|
chassis = Chassis()
|
|
assert chassis.set_uid_led('blue') is False
|
|
physical_led = chassis._led_uid
|
|
assert physical_led is not None
|
|
self._verify_uid_led(physical_led, chassis)
|
|
|
|
def _verify_non_shared_led(self, physical_led, obj):
|
|
mock_file_content = self._mock_led_file_content(physical_led)
|
|
|
|
def mock_read_str_from_file(file_path, **kwargs):
|
|
return mock_file_content[file_path]
|
|
|
|
def mock_write_file(file_path, content, **kwargs):
|
|
mock_file_content[file_path] = content
|
|
|
|
utils.read_str_from_file = mock_read_str_from_file
|
|
utils.write_file = mock_write_file
|
|
|
|
assert obj.get_status_led() == Led.STATUS_LED_COLOR_GREEN
|
|
mock_file_content[physical_led.get_led_path('green')] = Led.LED_OFF
|
|
assert obj.set_status_led(Led.STATUS_LED_COLOR_RED) is True
|
|
assert obj.get_status_led() == Led.STATUS_LED_COLOR_RED
|
|
mock_file_content[physical_led.get_led_path('red')] = Led.LED_OFF
|
|
assert obj.set_status_led(Led.STATUS_LED_COLOR_GREEN) is True
|
|
assert obj.get_status_led() == Led.STATUS_LED_COLOR_GREEN
|
|
mock_file_content[physical_led.get_led_path('green')] = Led.LED_OFF
|
|
assert obj.set_status_led(Led.STATUS_LED_COLOR_ORANGE) is True
|
|
assert obj.get_status_led() == Led.STATUS_LED_COLOR_RED
|
|
mock_file_content[physical_led.get_led_path('orange')] = Led.LED_OFF
|
|
|
|
assert obj.set_status_led(Led.STATUS_LED_COLOR_RED_BLINK)
|
|
assert obj.get_status_led() == Led.STATUS_LED_COLOR_RED_BLINK
|
|
|
|
mock_file_content[physical_led.get_led_delay_off_path('red')] = Led.LED_OFF
|
|
mock_file_content[physical_led.get_led_delay_on_path('red')] = Led.LED_OFF
|
|
|
|
assert obj.set_status_led(Led.STATUS_LED_COLOR_GREEN_BLINK)
|
|
assert obj.get_status_led() == Led.STATUS_LED_COLOR_GREEN_BLINK
|
|
mock_file_content[physical_led.get_led_delay_off_path('green')] = Led.LED_OFF
|
|
mock_file_content[physical_led.get_led_delay_on_path('green')] = Led.LED_OFF
|
|
|
|
assert obj.set_status_led(Led.STATUS_LED_COLOR_ORANGE_BLINK)
|
|
assert obj.get_status_led() == Led.STATUS_LED_COLOR_RED_BLINK
|
|
mock_file_content[physical_led.get_led_delay_off_path('green')] = Led.LED_OFF
|
|
mock_file_content[physical_led.get_led_delay_on_path('green')] = Led.LED_OFF
|
|
|
|
def _verify_uid_led(self, physical_led, obj):
|
|
mock_file_content = self._mock_led_file_content(physical_led)
|
|
|
|
def mock_read_str_from_file(file_path, **kwargs):
|
|
return mock_file_content[file_path]
|
|
|
|
def mock_write_file(file_path, content, **kwargs):
|
|
mock_file_content[file_path] = content
|
|
|
|
utils.read_str_from_file = mock_read_str_from_file
|
|
utils.write_file = mock_write_file
|
|
|
|
assert obj.get_uid_led() == Led.STATUS_LED_COLOR_GREEN
|
|
mock_file_content[physical_led.get_led_path('green')] = Led.LED_OFF
|
|
assert obj.set_uid_led(Led.STATUS_LED_COLOR_BLUE) is True
|
|
assert obj.get_uid_led() == Led.STATUS_LED_COLOR_BLUE
|
|
mock_file_content[physical_led.get_led_path('blue')] = Led.LED_OFF
|
|
|
|
def _mock_led_file_content(self, led):
|
|
return {
|
|
led.get_led_path('green'): Led.LED_ON,
|
|
led.get_led_path('red'): Led.LED_OFF,
|
|
led.get_led_path('orange'): Led.LED_OFF,
|
|
led.get_led_path('blue'): Led.LED_OFF,
|
|
led.get_led_cap_path(): 'none green green_blink red red_blink orange blue',
|
|
led.get_led_delay_off_path('green'): Led.LED_OFF,
|
|
led.get_led_delay_on_path('green'): Led.LED_OFF,
|
|
led.get_led_delay_off_path('red'): Led.LED_OFF,
|
|
led.get_led_delay_on_path('red'): Led.LED_OFF,
|
|
led.get_led_delay_off_path('orange'): Led.LED_OFF,
|
|
led.get_led_delay_on_path('orange'): Led.LED_OFF,
|
|
led.get_led_delay_off_path('blue'): Led.LED_OFF,
|
|
led.get_led_delay_on_path('blue'): Led.LED_OFF,
|
|
}
|
|
|
|
@mock.patch('sonic_platform.led.Led._wait_files_ready', mock.MagicMock(return_value=True))
|
|
def test_fan_led(self):
|
|
fan_drawer = RealDrawer(0)
|
|
self._verify_fan_led(fan_drawer)
|
|
fan_drawer = VirtualDrawer(0)
|
|
self._verify_fan_led(fan_drawer)
|
|
|
|
def _verify_fan_led(self, fan_drawer):
|
|
fan1 = Fan(0, fan_drawer, 1)
|
|
fan2 = Fan(1, fan_drawer, 2)
|
|
physical_led = fan_drawer.get_led()._led
|
|
self._verify_shared_led(physical_led, fan1, fan2)
|
|
|
|
def _verify_shared_led(self, physical_led, obj1, obj2):
|
|
mock_file_content = self._mock_led_file_content(physical_led)
|
|
|
|
def mock_read_str_from_file(file_path, **kwargs):
|
|
return mock_file_content[file_path]
|
|
|
|
def mock_write_file(file_path, content, **kwargs):
|
|
mock_file_content[file_path] = content
|
|
|
|
utils.read_str_from_file = mock_read_str_from_file
|
|
utils.write_file = mock_write_file
|
|
assert obj1.set_status_led(Led.STATUS_LED_COLOR_GREEN)
|
|
assert obj2.get_status_led() == Led.STATUS_LED_COLOR_GREEN
|
|
mock_file_content[physical_led.get_led_path('green')] = Led.LED_OFF
|
|
assert obj2.set_status_led(Led.STATUS_LED_COLOR_RED)
|
|
assert obj2.get_status_led() == Led.STATUS_LED_COLOR_RED
|
|
assert obj1.set_status_led(Led.STATUS_LED_COLOR_RED)
|
|
assert obj2.get_status_led() == Led.STATUS_LED_COLOR_RED
|
|
|
|
mock_file_content[physical_led.get_led_path('red')] = Led.LED_OFF
|
|
assert obj1.set_status_led(Led.STATUS_LED_COLOR_GREEN)
|
|
assert obj1.get_status_led() == Led.STATUS_LED_COLOR_RED
|
|
assert obj2.get_status_led() == Led.STATUS_LED_COLOR_RED
|
|
mock_file_content[physical_led.get_led_path('red')] = Led.LED_OFF
|
|
assert obj2.set_status_led(Led.STATUS_LED_COLOR_GREEN)
|
|
assert obj1.get_status_led() == Led.STATUS_LED_COLOR_GREEN
|
|
assert obj2.get_status_led() == Led.STATUS_LED_COLOR_GREEN
|
|
|
|
@mock.patch('sonic_platform.led.Led._wait_files_ready', mock.MagicMock(return_value=True))
|
|
def test_psu_led(self):
|
|
psu1 = Psu(0)
|
|
psu2 = Psu(1)
|
|
physical_led = Psu.get_shared_led()._led
|
|
self._verify_shared_led(physical_led, psu1, psu2)
|
|
|
|
@mock.patch('sonic_platform.led.Led._wait_files_ready', mock.MagicMock(return_value=True))
|
|
def test_fixed_psu_led(self):
|
|
psu = FixedPsu(0)
|
|
physical_led = psu.led
|
|
self._verify_non_shared_led(physical_led, psu)
|