core/tests/components/luftdaten/test_sensor.py

"""Tests for the sensors provided by the Luftdaten integration."""
from homeassistant.components.luftdaten.const import DOMAIN
from homeassistant.components.sensor import (
    ATTR_STATE_CLASS,
    SensorDeviceClass,
    SensorStateClass,
)
from homeassistant.const import (
    ATTR_DEVICE_CLASS,
    ATTR_FRIENDLY_NAME,
    ATTR_ICON,
    ATTR_UNIT_OF_MEASUREMENT,
    CONCENTRATION_MICROGRAMS_PER_CUBIC_METER,
    PERCENTAGE,
    PRESSURE_PA,
    TEMP_CELSIUS,
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers import device_registry as dr, entity_registry as er

from tests.common import MockConfigEntry


async def test_luftdaten_sensors(
    hass: HomeAssistant,
    init_integration: MockConfigEntry,
) -> None:
    """Test the Luftdaten sensors."""
    entity_registry = er.async_get(hass)
    device_registry = dr.async_get(hass)

    entry = entity_registry.async_get("sensor.temperature")
    assert entry
    assert entry.device_id
    assert entry.unique_id == "12345_temperature"

    state = hass.states.get("sensor.temperature")
    assert state
    assert state.state == "22.3"
    assert state.attributes.get(ATTR_FRIENDLY_NAME) == "Temperature"
    assert state.attributes.get(ATTR_DEVICE_CLASS) == SensorDeviceClass.TEMPERATURE
    assert state.attributes.get(ATTR_STATE_CLASS) is SensorStateClass.MEASUREMENT
    assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == TEMP_CELSIUS
    assert ATTR_ICON not in state.attributes

    entry = entity_registry.async_get("sensor.humidity")
    assert entry
    assert entry.device_id
    assert entry.unique_id == "12345_humidity"

    state = hass.states.get("sensor.humidity")
    assert state
    assert state.state == "34.7"
    assert state.attributes.get(ATTR_FRIENDLY_NAME) == "Humidity"
    assert state.attributes.get(ATTR_DEVICE_CLASS) == SensorDeviceClass.HUMIDITY
    assert state.attributes.get(ATTR_STATE_CLASS) is SensorStateClass.MEASUREMENT
    assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == PERCENTAGE
    assert ATTR_ICON not in state.attributes

    entry = entity_registry.async_get("sensor.pressure")
    assert entry
    assert entry.device_id
    assert entry.unique_id == "12345_pressure"

    state = hass.states.get("sensor.pressure")
    assert state
    assert state.state == "98545.0"
    assert state.attributes.get(ATTR_FRIENDLY_NAME) == "Pressure"
    assert state.attributes.get(ATTR_DEVICE_CLASS) == SensorDeviceClass.PRESSURE
    assert state.attributes.get(ATTR_STATE_CLASS) is SensorStateClass.MEASUREMENT
    assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == PRESSURE_PA
    assert ATTR_ICON not in state.attributes

    entry = entity_registry.async_get("sensor.pressure_at_sealevel")
    assert entry
    assert entry.device_id
    assert entry.unique_id == "12345_pressure_at_sealevel"

    state = hass.states.get("sensor.pressure_at_sealevel")
    assert state
    assert state.state == "103102.13"
    assert state.attributes.get(ATTR_FRIENDLY_NAME) == "Pressure at sealevel"
    assert state.attributes.get(ATTR_DEVICE_CLASS) == SensorDeviceClass.PRESSURE
    assert state.attributes.get(ATTR_STATE_CLASS) is SensorStateClass.MEASUREMENT
    assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == PRESSURE_PA
    assert ATTR_ICON not in state.attributes

    entry = entity_registry.async_get("sensor.pm10")
    assert entry
    assert entry.device_id
    assert entry.unique_id == "12345_P1"

    state = hass.states.get("sensor.pm10")
    assert state
    assert state.state == "8.5"
    assert state.attributes.get(ATTR_FRIENDLY_NAME) == "PM10"
    assert state.attributes.get(ATTR_DEVICE_CLASS) == SensorDeviceClass.PM10
    assert state.attributes.get(ATTR_STATE_CLASS) is SensorStateClass.MEASUREMENT
    assert (
        state.attributes.get(ATTR_UNIT_OF_MEASUREMENT)
        == CONCENTRATION_MICROGRAMS_PER_CUBIC_METER
    )
    assert ATTR_ICON not in state.attributes

    entry = entity_registry.async_get("sensor.pm2_5")
    assert entry
    assert entry.device_id
    assert entry.unique_id == "12345_P2"

    state = hass.states.get("sensor.pm2_5")
    assert state
    assert state.state == "4.07"
    assert state.attributes.get(ATTR_FRIENDLY_NAME) == "PM2.5"
    assert state.attributes.get(ATTR_DEVICE_CLASS) == SensorDeviceClass.PM25
    assert state.attributes.get(ATTR_STATE_CLASS) is SensorStateClass.MEASUREMENT
    assert (
        state.attributes.get(ATTR_UNIT_OF_MEASUREMENT)
        == CONCENTRATION_MICROGRAMS_PER_CUBIC_METER
    )
    assert ATTR_ICON not in state.attributes

    assert entry.device_id
    device_entry = device_registry.async_get(entry.device_id)
    assert device_entry
    assert device_entry.identifiers == {(DOMAIN, "12345")}
    assert device_entry.manufacturer == "Sensor.Community"
    assert device_entry.name == "Sensor 12345"
    assert (
        device_entry.configuration_url
        == "https://devices.sensor.community/sensors/12345/settings"
    )