83 lines
3.2 KiB
Python
83 lines
3.2 KiB
Python
|
import os
|
||
|
|
import sys
|
||
|
|
import pytest
|
||
|
|
from mock import MagicMock
|
||
|
|
from .mock_platform import MockFan
|
||
|
|
|
||
|
|
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.component import Component, ComponentSSD
|
||
|
|
|
||
|
|
from sonic_platform_base.component_base import ComponentBase, \
|
||
|
|
FW_AUTO_INSTALLED, \
|
||
|
|
FW_AUTO_ERR_BOOT_TYPE, \
|
||
|
|
FW_AUTO_ERR_IMAGE, \
|
||
|
|
FW_AUTO_ERR_UKNOWN
|
||
|
|
|
||
|
|
def mock_install_firmware_success(image_path):
|
||
|
|
return True
|
||
|
|
|
||
|
|
def mock_install_firmware_fail(image_path):
|
||
|
|
return False
|
||
|
|
|
||
|
|
def mock_update_notification_cold_boot(image_path):
|
||
|
|
return "Immediate power cycle is required to complete NAME firmware update"
|
||
|
|
|
||
|
|
def mock_update_notification_warm_boot(image_path):
|
||
|
|
return None
|
||
|
|
|
||
|
|
def mock_update_notification_error(image_path):
|
||
|
|
raise RuntimeError("Failed to parse NAME firmware upgrade status")
|
||
|
|
|
||
|
|
test_data_default = [
|
||
|
|
(None, False, None, FW_AUTO_ERR_IMAGE),
|
||
|
|
(None, True, 'warm', FW_AUTO_ERR_BOOT_TYPE),
|
||
|
|
(mock_install_firmware_fail, True, 'cold', FW_AUTO_ERR_UKNOWN),
|
||
|
|
(mock_install_firmware_success, True, 'cold', FW_AUTO_INSTALLED)
|
||
|
|
]
|
||
|
|
|
||
|
|
test_data_ssd = [
|
||
|
|
(None, None, False, None, FW_AUTO_ERR_IMAGE),
|
||
|
|
(None, mock_update_notification_error, True, None, FW_AUTO_ERR_UKNOWN),
|
||
|
|
(mock_install_firmware_fail, mock_update_notification_cold_boot, True, 'cold', FW_AUTO_ERR_UKNOWN),
|
||
|
|
(mock_install_firmware_success, mock_update_notification_cold_boot, True, 'warm', FW_AUTO_ERR_BOOT_TYPE),
|
||
|
|
(mock_install_firmware_success, mock_update_notification_cold_boot, True, 'cold', FW_AUTO_INSTALLED),
|
||
|
|
(mock_install_firmware_success, mock_update_notification_warm_boot, True, 'warm', FW_AUTO_INSTALLED),
|
||
|
|
(mock_install_firmware_success, mock_update_notification_warm_boot, True, 'cold', FW_AUTO_INSTALLED)
|
||
|
|
]
|
||
|
|
|
||
|
|
@pytest.mark.parametrize('install_func, image_found, boot_type, expect', test_data_default)
|
||
|
|
def test_auto_update_firmware_default(monkeypatch, install_func, image_found, boot_type, expect):
|
||
|
|
|
||
|
|
def mock_path_exists(path):
|
||
|
|
return image_found
|
||
|
|
|
||
|
|
test_component = Component()
|
||
|
|
|
||
|
|
monkeypatch.setattr(test_component, 'install_firmware', install_func)
|
||
|
|
monkeypatch.setattr(os.path, 'exists', mock_path_exists)
|
||
|
|
|
||
|
|
result = test_component.auto_update_firmware(None, boot_type)
|
||
|
|
|
||
|
|
assert result == expect
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.parametrize('install_func, notify, image_found, boot_type, expect', test_data_ssd)
|
||
|
|
def test_auto_update_firmware_default(monkeypatch, install_func, notify, image_found, boot_type, expect):
|
||
|
|
|
||
|
|
def mock_path_exists(path):
|
||
|
|
return image_found
|
||
|
|
|
||
|
|
test_component_ssd = ComponentSSD()
|
||
|
|
|
||
|
|
monkeypatch.setattr(test_component_ssd, 'install_firmware', install_func)
|
||
|
|
monkeypatch.setattr(test_component_ssd, 'get_firmware_update_notification', notify)
|
||
|
|
monkeypatch.setattr(os.path, 'exists', mock_path_exists)
|
||
|
|
|
||
|
|
result = test_component_ssd.auto_update_firmware(None, boot_type)
|
||
|
|
|
||
|
|
assert result == expect
|
||
|
|
|