Sensor test

tests/components/wolflink/test_sensor.py

@@ -0,0 +1,153 @@
+"""Test the Wolf SmartSet Service Sensor platform."""
+
+from unittest.mock import MagicMock
+
+import pytest
+
+from homeassistant.components.wolflink.const import COORDINATOR, DEVICE_ID, DOMAIN
+from homeassistant.components.wolflink.sensor import (
+    WolfLinkEnergy,
+    WolfLinkHours,
+    WolfLinkPercentage,
+    WolfLinkPower,
+    WolfLinkPressure,
+    WolfLinkSensor,
+    WolfLinkState,
+    WolfLinkTemperature,
+)
+from homeassistant.const import (
+    PERCENTAGE,
+    UnitOfEnergy,
+    UnitOfPower,
+    UnitOfPressure,
+    UnitOfTemperature,
+    UnitOfTime,
+)
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers import device_registry as dr
+
+from .const import CONFIG
+
+from tests.common import MockConfigEntry
+
+
+@pytest.fixture
+def mock_coordinator(hass: HomeAssistant) -> MagicMock:
+    """Mock the Wolf SmartSet Service Coordinator."""
+    coordinator = MagicMock()
+    coordinator.data = {}
+    hass.data[DOMAIN] = {CONFIG[DEVICE_ID]: {COORDINATOR: coordinator}}
+    return coordinator
+
+
+async def mock_config_entry(
+    hass: HomeAssistant, device_registry: dr.DeviceRegistry
+) -> None:
+    """Test already configured while creating entry."""
+    config_entry = MockConfigEntry(
+        domain=DOMAIN, unique_id=CONFIG[DEVICE_ID], data=CONFIG
+    )
+    config_entry.add_to_hass(hass)
+
+
+def test_wolflink_sensor_initialization(mock_coordinator) -> None:
+    """Test WolflinkSensor initialization."""
+    parameter = MagicMock()
+    parameter.name = "Outside Temperature"
+    parameter.parameter_id = "outside_temp"
+    sensor = WolfLinkSensor(parameter, "mock_device_id")
+    assert sensor._attr_name == "Outside Temperature"
+    assert sensor._attr_unique_id == "mock_device_id:outside_temp"
+
+
+def test_wolflink_sensor_native_value(mock_coordinator) -> None:
+    """Test WolflinkSensor native value."""
+    parameter = MagicMock()
+    parameter.parameter_id = "outside_temp"
+    sensor = WolfLinkSensor(mock_coordinator, parameter, "mock_device_id")
+    mock_coordinator.data = {"outside_temp": [None, 20]}
+    assert sensor.native_value == 20
+
+
+def test_wolflink_sensor_extra_state_attributes(mock_coordinator) -> None:
+    """Test WolflinkSensor extra state attributes."""
+    parameter = MagicMock()
+    parameter.parameter_id = "outside_temp"
+    parameter.value_id = "value_id"
+    parameter.parent = "parent"
+    sensor = WolfLinkSensor(mock_coordinator, parameter, "mock_device_id")
+    attributes = sensor.extra_state_attributes
+    assert attributes["parameter_id"] == "outside_temp"
+    assert attributes["value_id"] == "value_id"
+    assert attributes["parent"] == "parent"
+
+
+def test_wolflink_temperature_initialization(mock_coordinator) -> None:
+    """Test WolflinkTemperature initialization."""
+    parameter = MagicMock()
+    parameter.name = "Outside Temperature"
+    parameter.parameter_id = "outside_temp"
+    sensor = WolfLinkTemperature(mock_coordinator, parameter, "mock_device_id")
+    assert sensor._attr_device_class == "temperature"
+    assert sensor._attr_native_unit_of_measurement == UnitOfTemperature.CELSIUS
+
+
+def test_wolflink_pressure_initialization(mock_coordinator) -> None:
+    """Test WolflinkPressure initialization."""
+    parameter = MagicMock()
+    parameter.name = "Pressure"
+    parameter.parameter_id = "pressure"
+    sensor = WolfLinkPressure(mock_coordinator, parameter, "mock_device_id")
+    assert sensor._attr_device_class == "pressure"
+    assert sensor._attr_native_unit_of_measurement == UnitOfPressure.BAR
+
+
+def test_wolflink_power_initialization(mock_coordinator) -> None:
+    """Test WolflinkPower initialization."""
+    parameter = MagicMock()
+    parameter.name = "Power"
+    parameter.parameter_id = "power"
+    sensor = WolfLinkPower(mock_coordinator, parameter, "mock_device_id")
+    assert sensor._attr_device_class == "power"
+    assert sensor._attr_native_unit_of_measurement == UnitOfPower.KILO_WATT
+
+
+def test_wolflink_energy_initialization(mock_coordinator) -> None:
+    """Test WolflinkEnergy initialization."""
+    parameter = MagicMock()
+    parameter.name = "Energy"
+    parameter.parameter_id = "energy"
+    sensor = WolfLinkEnergy(mock_coordinator, parameter, "mock_device_id")
+    assert sensor._attr_device_class == "energy"
+    assert sensor._attr_native_unit_of_measurement == UnitOfEnergy.KILO_WATT_HOUR
+
+
+def test_wolflink_percentage_initialization(mock_coordinator) -> None:
+    """Test WolflinkPercentage initialization."""
+    parameter = MagicMock()
+    parameter.name = "Percentage"
+    parameter.parameter_id = "percentage"
+    parameter.unit = PERCENTAGE
+    sensor = WolfLinkPercentage(mock_coordinator, parameter, "mock_device_id")
+    assert sensor.native_unit_of_measurement == PERCENTAGE
+
+
+def test_wolflink_state_initialization(mock_coordinator) -> None:
+    """Test WolflinkState initialization."""
+    parameter = MagicMock()
+    parameter.name = "State"
+    parameter.parameter_id = "state"
+    parameter.items = [MagicMock(value=1, name="On"), MagicMock(value=0, name="Off")]
+    sensor = WolfLinkState(mock_coordinator, parameter, "mock_device_id")
+    mock_coordinator.data = {"state": [None, 1]}
+    assert sensor.native_value == "On"
+
+
+def test_wolflink_hours_initialization(mock_coordinator) -> None:
+    """Test WolflinkHours initialization."""
+    parameter = MagicMock()
+    parameter.name = "Hours"
+    parameter.parameter_id = "hours"
+    sensor = WolfLinkHours(mock_coordinator, parameter, "mock_device_id")
+    assert sensor._attr_icon == "mdi:clock"
+    assert sensor._attr_native_unit_of_measurement == UnitOfTime.HOURS