From 7d7f5272fec56ebf95b8124349f3e8950cb48741 Mon Sep 17 00:00:00 2001
From: Franck Nijhof <git@frenck.dev>
Date: Sat, 25 Dec 2021 10:52:31 +0100
Subject: [PATCH] Add device info to Luftdaten (#62692)

---
 homeassistant/components/luftdaten/sensor.py |  8 ++++++
 tests/components/luftdaten/test_sensor.py    | 27 +++++++++++++++-----
 2 files changed, 28 insertions(+), 7 deletions(-)

diff --git a/homeassistant/components/luftdaten/sensor.py b/homeassistant/components/luftdaten/sensor.py
index 422e5ed7117..69ad30eb926 100644
--- a/homeassistant/components/luftdaten/sensor.py
+++ b/homeassistant/components/luftdaten/sensor.py
@@ -20,6 +20,7 @@ from homeassistant.const import (
     TEMP_CELSIUS,
 )
 from homeassistant.core import HomeAssistant
+from homeassistant.helpers.entity import DeviceInfo
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 from homeassistant.helpers.update_coordinator import (
     CoordinatorEntity,
@@ -114,6 +115,13 @@ class LuftdatenSensor(CoordinatorEntity, SensorEntity):
         self._attr_extra_state_attributes = {
             ATTR_SENSOR_ID: sensor_id,
         }
+        self._attr_device_info = DeviceInfo(
+            configuration_url=f"https://devices.sensor.community/sensors/{sensor_id}/settings",
+            identifiers={(DOMAIN, str(sensor_id))},
+            name=f"Sensor {sensor_id}",
+            manufacturer="Luftdaten.info",
+        )
+
         if show_on_map:
             self._attr_extra_state_attributes[ATTR_LONGITUDE] = coordinator.data[
                 "longitude"
diff --git a/tests/components/luftdaten/test_sensor.py b/tests/components/luftdaten/test_sensor.py
index ae016615047..46570d6cb69 100644
--- a/tests/components/luftdaten/test_sensor.py
+++ b/tests/components/luftdaten/test_sensor.py
@@ -1,4 +1,5 @@
 """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,
@@ -15,7 +16,7 @@ from homeassistant.const import (
     TEMP_CELSIUS,
 )
 from homeassistant.core import HomeAssistant
-from homeassistant.helpers import entity_registry as er
+from homeassistant.helpers import device_registry as dr, entity_registry as er
 
 from tests.common import MockConfigEntry
 
@@ -26,10 +27,11 @@ async def test_luftdaten_sensors(
 ) -> 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 not entry.device_id
+    assert entry.device_id
     assert entry.unique_id == "12345_temperature"
 
     state = hass.states.get("sensor.temperature")
@@ -43,7 +45,7 @@ async def test_luftdaten_sensors(
 
     entry = entity_registry.async_get("sensor.humidity")
     assert entry
-    assert not entry.device_id
+    assert entry.device_id
     assert entry.unique_id == "12345_humidity"
 
     state = hass.states.get("sensor.humidity")
@@ -57,7 +59,7 @@ async def test_luftdaten_sensors(
 
     entry = entity_registry.async_get("sensor.pressure")
     assert entry
-    assert not entry.device_id
+    assert entry.device_id
     assert entry.unique_id == "12345_pressure"
 
     state = hass.states.get("sensor.pressure")
@@ -71,7 +73,7 @@ async def test_luftdaten_sensors(
 
     entry = entity_registry.async_get("sensor.pressure_at_sealevel")
     assert entry
-    assert not entry.device_id
+    assert entry.device_id
     assert entry.unique_id == "12345_pressure_at_sealevel"
 
     state = hass.states.get("sensor.pressure_at_sealevel")
@@ -85,7 +87,7 @@ async def test_luftdaten_sensors(
 
     entry = entity_registry.async_get("sensor.pm10")
     assert entry
-    assert not entry.device_id
+    assert entry.device_id
     assert entry.unique_id == "12345_P1"
 
     state = hass.states.get("sensor.pm10")
@@ -102,7 +104,7 @@ async def test_luftdaten_sensors(
 
     entry = entity_registry.async_get("sensor.pm2_5")
     assert entry
-    assert not entry.device_id
+    assert entry.device_id
     assert entry.unique_id == "12345_P2"
 
     state = hass.states.get("sensor.pm2_5")
@@ -116,3 +118,14 @@ async def test_luftdaten_sensors(
         == 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 == "Luftdaten.info"
+    assert device_entry.name == "Sensor 12345"
+    assert (
+        device_entry.configuration_url
+        == "https://devices.sensor.community/sensors/12345/settings"
+    )