Use new enums for prometheus tests (#62550)

This commit is contained in:
Dave T 2021-12-22 13:31:20 +00:00 committed by GitHub
parent d7de3fbfce
commit 38e95ca663
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -13,12 +13,11 @@ from homeassistant.components.demo.light import DemoLight
from homeassistant.components.demo.number import DemoNumber from homeassistant.components.demo.number import DemoNumber
from homeassistant.components.demo.sensor import DemoSensor from homeassistant.components.demo.sensor import DemoSensor
import homeassistant.components.prometheus as prometheus import homeassistant.components.prometheus as prometheus
from homeassistant.components.sensor import SensorDeviceClass
from homeassistant.const import ( from homeassistant.const import (
CONCENTRATION_MICROGRAMS_PER_CUBIC_METER, CONCENTRATION_MICROGRAMS_PER_CUBIC_METER,
CONTENT_TYPE_TEXT_PLAIN, CONTENT_TYPE_TEXT_PLAIN,
DEGREE, DEGREE,
DEVICE_CLASS_POWER,
DEVICE_CLASS_TEMPERATURE,
ENERGY_KILO_WATT_HOUR, ENERGY_KILO_WATT_HOUR,
EVENT_STATE_CHANGED, EVENT_STATE_CHANGED,
TEMP_FAHRENHEIT, TEMP_FAHRENHEIT,
@ -75,7 +74,13 @@ async def test_view_empty_namespace(hass, hass_client):
client = await setup_prometheus_client(hass, hass_client, "") client = await setup_prometheus_client(hass, hass_client, "")
sensor2 = DemoSensor( sensor2 = DemoSensor(
None, "Radio Energy", 14, DEVICE_CLASS_POWER, None, ENERGY_KILO_WATT_HOUR, None None,
"Radio Energy",
14,
SensorDeviceClass.POWER,
None,
ENERGY_KILO_WATT_HOUR,
None,
) )
sensor2.hass = hass sensor2.hass = hass
sensor2.entity_id = "sensor.radio_energy" sensor2.entity_id = "sensor.radio_energy"
@ -149,7 +154,13 @@ async def test_sensor_unit(hass, hass_client):
await sensor1.async_update_ha_state() await sensor1.async_update_ha_state()
sensor2 = DemoSensor( sensor2 = DemoSensor(
None, "Radio Energy", 14, DEVICE_CLASS_POWER, None, ENERGY_KILO_WATT_HOUR, None None,
"Radio Energy",
14,
SensorDeviceClass.POWER,
None,
ENERGY_KILO_WATT_HOUR,
None,
) )
sensor2.hass = hass sensor2.hass = hass
sensor2.entity_id = "sensor.radio_energy" sensor2.entity_id = "sensor.radio_energy"
@ -272,14 +283,26 @@ async def test_sensor_device_class(hass, hass_client):
await hass.async_block_till_done() await hass.async_block_till_done()
sensor1 = DemoSensor( sensor1 = DemoSensor(
None, "Fahrenheit", 50, DEVICE_CLASS_TEMPERATURE, None, TEMP_FAHRENHEIT, None None,
"Fahrenheit",
50,
SensorDeviceClass.TEMPERATURE,
None,
TEMP_FAHRENHEIT,
None,
) )
sensor1.hass = hass sensor1.hass = hass
sensor1.entity_id = "sensor.fahrenheit" sensor1.entity_id = "sensor.fahrenheit"
await sensor1.async_update_ha_state() await sensor1.async_update_ha_state()
sensor2 = DemoSensor( sensor2 = DemoSensor(
None, "Radio Energy", 14, DEVICE_CLASS_POWER, None, ENERGY_KILO_WATT_HOUR, None None,
"Radio Energy",
14,
SensorDeviceClass.POWER,
None,
ENERGY_KILO_WATT_HOUR,
None,
) )
sensor2.hass = hass sensor2.hass = hass
sensor2.entity_id = "sensor.radio_energy" sensor2.entity_id = "sensor.radio_energy"