Add tests for goalzero (#57008)

* Add tests for goalzero

* clean up

tests/components/goalzero/test_sensor.py

@@ -0,0 +1,112 @@
+"""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,
+    STATE_CLASS_MEASUREMENT,
+    STATE_CLASS_TOTAL_INCREASING,
+)
+from homeassistant.const import (
+    ATTR_DEVICE_CLASS,
+    ATTR_UNIT_OF_MEASUREMENT,
+    DEVICE_CLASS_BATTERY,
+    DEVICE_CLASS_CURRENT,
+    DEVICE_CLASS_ENERGY,
+    DEVICE_CLASS_POWER,
+    DEVICE_CLASS_SIGNAL_STRENGTH,
+    DEVICE_CLASS_TEMPERATURE,
+    DEVICE_CLASS_VOLTAGE,
+    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) == DEVICE_CLASS_POWER
+    assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == POWER_WATT
+    assert state.attributes.get(ATTR_STATE_CLASS) == STATE_CLASS_MEASUREMENT
+    state = hass.states.get(f"sensor.{DEFAULT_NAME}_amps_in")
+    assert state.state == "0.0"
+    assert state.attributes.get(ATTR_DEVICE_CLASS) == DEVICE_CLASS_CURRENT
+    assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == ELECTRIC_CURRENT_AMPERE
+    assert state.attributes.get(ATTR_STATE_CLASS) == STATE_CLASS_MEASUREMENT
+    state = hass.states.get(f"sensor.{DEFAULT_NAME}_watts_out")
+    assert state.state == "50.5"
+    assert state.attributes.get(ATTR_DEVICE_CLASS) == DEVICE_CLASS_POWER
+    assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == POWER_WATT
+    assert state.attributes.get(ATTR_STATE_CLASS) == STATE_CLASS_MEASUREMENT
+    state = hass.states.get(f"sensor.{DEFAULT_NAME}_amps_out")
+    assert state.state == "2.1"
+    assert state.attributes.get(ATTR_DEVICE_CLASS) == DEVICE_CLASS_CURRENT
+    assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == ELECTRIC_CURRENT_AMPERE
+    assert state.attributes.get(ATTR_STATE_CLASS) == STATE_CLASS_MEASUREMENT
+    state = hass.states.get(f"sensor.{DEFAULT_NAME}_wh_out")
+    assert state.state == "5.23"
+    assert state.attributes.get(ATTR_DEVICE_CLASS) == DEVICE_CLASS_ENERGY
+    assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == ENERGY_WATT_HOUR
+    assert state.attributes.get(ATTR_STATE_CLASS) == STATE_CLASS_TOTAL_INCREASING
+    state = hass.states.get(f"sensor.{DEFAULT_NAME}_wh_stored")
+    assert state.state == "1330"
+    assert state.attributes.get(ATTR_DEVICE_CLASS) == DEVICE_CLASS_ENERGY
+    assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == ENERGY_WATT_HOUR
+    assert state.attributes.get(ATTR_STATE_CLASS) == STATE_CLASS_MEASUREMENT
+    state = hass.states.get(f"sensor.{DEFAULT_NAME}_volts")
+    assert state.state == "12.0"
+    assert state.attributes.get(ATTR_DEVICE_CLASS) == DEVICE_CLASS_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) == DEVICE_CLASS_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) == DEVICE_CLASS_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) == DEVICE_CLASS_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