106 lines
5.3 KiB
Python
106 lines
5.3 KiB
Python
"""Sensor tests for the Goalzero integration."""
|
|
from homeassistant.components.goalzero.const import DEFAULT_NAME
|
|
from homeassistant.components.goalzero.sensor import SENSOR_TYPES
|
|
from homeassistant.components.sensor import (
|
|
ATTR_STATE_CLASS,
|
|
DOMAIN,
|
|
SensorDeviceClass,
|
|
SensorStateClass,
|
|
)
|
|
from homeassistant.const import (
|
|
ATTR_DEVICE_CLASS,
|
|
ATTR_UNIT_OF_MEASUREMENT,
|
|
ELECTRIC_CURRENT_AMPERE,
|
|
ELECTRIC_POTENTIAL_VOLT,
|
|
ENERGY_WATT_HOUR,
|
|
PERCENTAGE,
|
|
POWER_WATT,
|
|
SIGNAL_STRENGTH_DECIBELS,
|
|
TEMP_CELSIUS,
|
|
TIME_MINUTES,
|
|
TIME_SECONDS,
|
|
)
|
|
from homeassistant.core import HomeAssistant
|
|
|
|
from . import async_setup_platform
|
|
|
|
from tests.test_util.aiohttp import AiohttpClientMocker
|
|
|
|
|
|
async def test_sensors(hass: HomeAssistant, aioclient_mock: AiohttpClientMocker):
|
|
"""Test we get sensor data."""
|
|
for description in SENSOR_TYPES:
|
|
description.entity_registry_enabled_default = True
|
|
await async_setup_platform(hass, aioclient_mock, DOMAIN)
|
|
|
|
state = hass.states.get(f"sensor.{DEFAULT_NAME}_watts_in")
|
|
assert state.state == "0.0"
|
|
assert state.attributes.get(ATTR_DEVICE_CLASS) == SensorDeviceClass.POWER
|
|
assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == POWER_WATT
|
|
assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT
|
|
state = hass.states.get(f"sensor.{DEFAULT_NAME}_amps_in")
|
|
assert state.state == "0.0"
|
|
assert state.attributes.get(ATTR_DEVICE_CLASS) == SensorDeviceClass.CURRENT
|
|
assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == ELECTRIC_CURRENT_AMPERE
|
|
assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT
|
|
state = hass.states.get(f"sensor.{DEFAULT_NAME}_watts_out")
|
|
assert state.state == "50.5"
|
|
assert state.attributes.get(ATTR_DEVICE_CLASS) == SensorDeviceClass.POWER
|
|
assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == POWER_WATT
|
|
assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT
|
|
state = hass.states.get(f"sensor.{DEFAULT_NAME}_amps_out")
|
|
assert state.state == "2.1"
|
|
assert state.attributes.get(ATTR_DEVICE_CLASS) == SensorDeviceClass.CURRENT
|
|
assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == ELECTRIC_CURRENT_AMPERE
|
|
assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT
|
|
state = hass.states.get(f"sensor.{DEFAULT_NAME}_wh_out")
|
|
assert state.state == "5.23"
|
|
assert state.attributes.get(ATTR_DEVICE_CLASS) == SensorDeviceClass.ENERGY
|
|
assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == ENERGY_WATT_HOUR
|
|
assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.TOTAL_INCREASING
|
|
state = hass.states.get(f"sensor.{DEFAULT_NAME}_wh_stored")
|
|
assert state.state == "1330"
|
|
assert state.attributes.get(ATTR_DEVICE_CLASS) == SensorDeviceClass.ENERGY
|
|
assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == ENERGY_WATT_HOUR
|
|
assert state.attributes.get(ATTR_STATE_CLASS) == SensorStateClass.MEASUREMENT
|
|
state = hass.states.get(f"sensor.{DEFAULT_NAME}_volts")
|
|
assert state.state == "12.0"
|
|
assert state.attributes.get(ATTR_DEVICE_CLASS) == SensorDeviceClass.VOLTAGE
|
|
assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == ELECTRIC_POTENTIAL_VOLT
|
|
assert state.attributes.get(ATTR_STATE_CLASS) is None
|
|
state = hass.states.get(f"sensor.{DEFAULT_NAME}_state_of_charge_percent")
|
|
assert state.state == "95"
|
|
assert state.attributes.get(ATTR_DEVICE_CLASS) == SensorDeviceClass.BATTERY
|
|
assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == PERCENTAGE
|
|
assert state.attributes.get(ATTR_STATE_CLASS) is None
|
|
state = hass.states.get(f"sensor.{DEFAULT_NAME}_time_to_empty_full")
|
|
assert state.state == "-1"
|
|
assert state.attributes.get(ATTR_DEVICE_CLASS) == TIME_MINUTES
|
|
assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == TIME_MINUTES
|
|
assert state.attributes.get(ATTR_STATE_CLASS) is None
|
|
state = hass.states.get(f"sensor.{DEFAULT_NAME}_temperature")
|
|
assert state.state == "25"
|
|
assert state.attributes.get(ATTR_DEVICE_CLASS) == SensorDeviceClass.TEMPERATURE
|
|
assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == TEMP_CELSIUS
|
|
assert state.attributes.get(ATTR_STATE_CLASS) is None
|
|
state = hass.states.get(f"sensor.{DEFAULT_NAME}_wifi_strength")
|
|
assert state.state == "-62"
|
|
assert state.attributes.get(ATTR_DEVICE_CLASS) == SensorDeviceClass.SIGNAL_STRENGTH
|
|
assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == SIGNAL_STRENGTH_DECIBELS
|
|
assert state.attributes.get(ATTR_STATE_CLASS) is None
|
|
state = hass.states.get(f"sensor.{DEFAULT_NAME}_total_run_time")
|
|
assert state.state == "1720984"
|
|
assert state.attributes.get(ATTR_DEVICE_CLASS) is None
|
|
assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == TIME_SECONDS
|
|
assert state.attributes.get(ATTR_STATE_CLASS) is None
|
|
state = hass.states.get(f"sensor.{DEFAULT_NAME}_wi_fi_ssid")
|
|
assert state.state == "wifi"
|
|
assert state.attributes.get(ATTR_DEVICE_CLASS) is None
|
|
assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) is None
|
|
assert state.attributes.get(ATTR_STATE_CLASS) is None
|
|
state = hass.states.get(f"sensor.{DEFAULT_NAME}_ip_address")
|
|
assert state.state == "1.2.3.4"
|
|
assert state.attributes.get(ATTR_DEVICE_CLASS) is None
|
|
assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) is None
|
|
assert state.attributes.get(ATTR_STATE_CLASS) is None
|