mirror of
https://github.com/home-assistant/supervisor.git
synced 2025-11-21 16:57:00 +00:00
cad14bf46e
Not all disks have all SMART attributes available, e.g. Sentry showed devices with missing "wctemp". In practice, any SMART attribute could be missing. Make sure we handle this gracefully.
103 lines
3.7 KiB
Python
103 lines
3.7 KiB
Python
"""Test UDisks2 NVMe Controller."""
|
|
|
|
from datetime import UTC, datetime
|
|
|
|
from dbus_fast.aio import MessageBus
|
|
import pytest
|
|
|
|
from supervisor.dbus.udisks2.nvme_controller import UDisks2NVMeController
|
|
|
|
from tests.common import mock_dbus_services
|
|
from tests.dbus_service_mocks.udisks2_nvme_controller import (
|
|
NVMeController as NVMeControllerService,
|
|
)
|
|
|
|
|
|
@pytest.fixture(name="nvme_controller_service")
|
|
async def fixture_nvme_controller_service(
|
|
dbus_session_bus: MessageBus,
|
|
) -> NVMeControllerService:
|
|
"""Mock NVMe Controller service."""
|
|
yield (
|
|
await mock_dbus_services(
|
|
{
|
|
"udisks2_nvme_controller": "/org/freedesktop/UDisks2/drives/Samsung_SSD_970_EVO_Plus_2TB_S40123456789ABC"
|
|
},
|
|
dbus_session_bus,
|
|
)
|
|
)["udisks2_nvme_controller"]
|
|
|
|
|
|
async def test_nvme_controller_info(
|
|
nvme_controller_service: NVMeControllerService, dbus_session_bus: MessageBus
|
|
):
|
|
"""Test NVMe Controller info."""
|
|
controller = UDisks2NVMeController(
|
|
"/org/freedesktop/UDisks2/drives/Samsung_SSD_970_EVO_Plus_2TB_S40123456789ABC"
|
|
)
|
|
|
|
assert controller.state is None
|
|
assert controller.unallocated_capacity is None
|
|
assert controller.smart_updated is None
|
|
assert controller.smart_temperature is None
|
|
|
|
await controller.connect(dbus_session_bus)
|
|
|
|
assert controller.state == "live"
|
|
assert controller.unallocated_capacity == 0
|
|
assert controller.smart_updated == datetime.fromtimestamp(1753906112, UTC)
|
|
assert controller.smart_temperature == 311
|
|
|
|
nvme_controller_service.emit_properties_changed({"SmartTemperature": 300})
|
|
await nvme_controller_service.ping()
|
|
await nvme_controller_service.ping()
|
|
|
|
assert controller.smart_temperature == 300
|
|
|
|
|
|
@pytest.mark.usefixtures("nvme_controller_service")
|
|
async def test_nvme_controller_smart_get_attributes(dbus_session_bus: MessageBus):
|
|
"""Test NVMe Controller smart get attributes."""
|
|
controller = UDisks2NVMeController(
|
|
"/org/freedesktop/UDisks2/drives/Samsung_SSD_970_EVO_Plus_2TB_S40123456789ABC"
|
|
)
|
|
await controller.connect(dbus_session_bus)
|
|
|
|
smart_log = await controller.smart_get_attributes()
|
|
assert smart_log.available_spare == 100
|
|
assert smart_log.percent_used == 1
|
|
assert smart_log.total_data_read == 22890461184000
|
|
assert smart_log.total_data_written == 27723431936000
|
|
assert smart_log.controller_busy_minutes == 2682
|
|
assert smart_log.temperature_sensors == [310, 305, 0, 0, 0, 0, 0, 0]
|
|
|
|
|
|
async def test_nvme_controller_smart_get_attributes_missing(
|
|
nvme_controller_service: NVMeControllerService, dbus_session_bus: MessageBus
|
|
):
|
|
"""Test NVMe Controller smart get attributes with missing vendor-specific attributes."""
|
|
# Simulate a drive that doesn't provide some optional attributes
|
|
nvme_controller_service.set_missing_attributes(
|
|
["wctemp", "cctemp", "warning_temp_time", "critical_temp_time", "temp_sensors"]
|
|
)
|
|
|
|
controller = UDisks2NVMeController(
|
|
"/org/freedesktop/UDisks2/drives/Samsung_SSD_970_EVO_Plus_2TB_S40123456789ABC"
|
|
)
|
|
await controller.connect(dbus_session_bus)
|
|
|
|
smart_log = await controller.smart_get_attributes()
|
|
# Core attributes should still be present
|
|
assert smart_log.available_spare == 100
|
|
assert smart_log.percent_used == 1
|
|
assert smart_log.total_data_read == 22890461184000
|
|
assert smart_log.total_data_written == 27723431936000
|
|
assert smart_log.controller_busy_minutes == 2682
|
|
|
|
# Optional attributes should be None
|
|
assert smart_log.warning_composite_temperature is None
|
|
assert smart_log.critical_composite_temperature is None
|
|
assert smart_log.warning_temperature_minutes is None
|
|
assert smart_log.critical_temperature_minutes is None
|
|
assert smart_log.temperature_sensors is None
|