Add Sense Devices for entities

homeassistant/components/sense/__init__.py

@@ -113,7 +113,6 @@ async def async_setup_entry(hass: HomeAssistant, entry: SenseConfigEntry) -> boo
     )
 
     await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
-
     return True
 
 

homeassistant/components/sense/binary_sensor.py

@@ -10,6 +10,7 @@
 )
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers import entity_registry as er
+from homeassistant.helpers.device_registry import DeviceInfo
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 from homeassistant.helpers.update_coordinator import CoordinatorEntity
 
@@ -66,6 +67,14 @@ def __init__(
         self._attr_unique_id = f"{sense_monitor_id}-{self._id}"
         self._attr_icon = sense_to_mdi(device.icon)
         self._device = device
+        self._attr_device_info = DeviceInfo(
+            name=f"Sense {sense_monitor_id} - {device.name}",
+            identifiers={(DOMAIN, f"{sense_monitor_id}:{device.id}")},
+            model="Sense",
+            manufacturer="Sense Labs, Inc.",
+            configuration_url="https://home.sense.com",
+            via_device=(DOMAIN, f"{sense_monitor_id}"),
+        )
 
     @property
     def old_unique_id(self) -> str:

homeassistant/components/sense/sensor.py

@@ -195,6 +195,13 @@ def __init__(
         self._attr_name = f"L{index + 1} Voltage"
         self._data = data
         self._voltage_index = index
+        self._attr_device_info = DeviceInfo(
+            name=f"Sense {sense_monitor_id}",
+            identifiers={(DOMAIN, sense_monitor_id)},
+            model="Sense",
+            manufacturer="Sense Labs, Inc.",
+            configuration_url="https://home.sense.com",
+        )
 
     @property
     def native_value(self) -> float:
@@ -276,6 +283,14 @@ def __init__(
         self._id = device.id
         self._attr_icon = sense_to_mdi(device.icon)
         self._device = device
+        self._attr_device_info = DeviceInfo(
+            name=f"Sense {sense_monitor_id} - {device.name}",
+            identifiers={(DOMAIN, f"{sense_monitor_id}:{device.id}")},
+            model="Sense",
+            manufacturer="Sense Labs, Inc.",
+            configuration_url="https://home.sense.com",
+            via_device=(DOMAIN, f"{sense_monitor_id}"),
+        )
 
     @property
     def native_value(self) -> float:

tests/components/sense/conftest.py

@@ -7,14 +7,17 @@
 from unittest.mock import AsyncMock, MagicMock, PropertyMock, patch
 
 import pytest
+from sense_energy import Scale
 
 from homeassistant.components.sense.binary_sensor import SenseDevice
 from homeassistant.components.sense.const import DOMAIN
 
 from .const import (
+    DEVICE_1_DAY_ENERGY,
     DEVICE_1_ID,
     DEVICE_1_NAME,
     DEVICE_1_POWER,
+    DEVICE_2_DAY_ENERGY,
     DEVICE_2_ID,
     DEVICE_2_NAME,
     DEVICE_2_POWER,
@@ -68,12 +71,14 @@ def mock_sense() -> Generator[MagicMock]:
         device_1.icon = "car"
         device_1.is_on = False
         device_1.power_w = DEVICE_1_POWER
+        device_1.energy_kwh[Scale.DAY] = DEVICE_1_DAY_ENERGY
 
         device_2 = SenseDevice(DEVICE_2_ID)
         device_2.name = DEVICE_2_NAME
         device_2.icon = "stove"
         device_2.is_on = False
         device_2.power_w = DEVICE_2_POWER
+        device_2.energy_kwh[Scale.DAY] = DEVICE_2_DAY_ENERGY
         type(gateway).devices = PropertyMock(return_value=[device_1, device_2])
 
         yield gateway

tests/components/sense/const.py

@@ -15,10 +15,12 @@
 DEVICE_1_ID = "abc123"
 DEVICE_1_ICON = "car-electric"
 DEVICE_1_POWER = 100.0
+DEVICE_1_DAY_ENERGY = 500
 
 DEVICE_2_NAME = "Oven"
 DEVICE_2_ID = "def456"
 DEVICE_2_ICON = "stove"
 DEVICE_2_POWER = 50.0
+DEVICE_2_DAY_ENERGY = 42
 
 MONITOR_ID = "12345"

tests/components/sense/test_sensor.py

@@ -15,7 +15,7 @@
 from homeassistant.util.dt import utcnow
 
 from . import setup_platform
-from .const import DEVICE_1_NAME, DEVICE_1_POWER, DEVICE_2_NAME, DEVICE_2_POWER
+from .const import DEVICE_1_NAME, DEVICE_2_NAME, DEVICE_2_POWER
 
 from tests.common import MockConfigEntry, async_fire_time_changed, snapshot_platform
 
@@ -52,17 +52,6 @@ async def test_device_power_sensors(
     state = hass.states.get(f"sensor.{DEVICE_2_NAME.lower()}_{CONSUMPTION_ID}")
     assert state.state == "0"
 
-    device_1.power_w = DEVICE_1_POWER
-    async_fire_time_changed(hass, utcnow() + timedelta(seconds=ACTIVE_UPDATE_RATE))
-    await hass.async_block_till_done()
-
-    state = hass.states.get(f"sensor.{DEVICE_1_NAME.lower()}_{CONSUMPTION_ID}")
-    assert state.state == f"{DEVICE_1_POWER:.1f}"
-
-    state = hass.states.get(f"sensor.{DEVICE_2_NAME.lower()}_{CONSUMPTION_ID}")
-    assert state.state == "0"
-
-    device_1.power_w = 0
     device_2.power_w = DEVICE_2_POWER
     async_fire_time_changed(hass, utcnow() + timedelta(seconds=ACTIVE_UPDATE_RATE))
     await hass.async_block_till_done()