From c6210b68bd2661f0e5e0b3ddcf9a50dafdb0e5d7 Mon Sep 17 00:00:00 2001
From: dougiteixeira <31328123+dougiteixeira@users.noreply.github.com>
Date: Fri, 30 Jun 2023 15:49:00 -0300
Subject: [PATCH] Fix device source for Riemann sum integral (#95622)

Fix Device Source
---
 homeassistant/components/integration/sensor.py | 1 +
 tests/components/integration/test_sensor.py    | 1 +
 2 files changed, 2 insertions(+)

diff --git a/homeassistant/components/integration/sensor.py b/homeassistant/components/integration/sensor.py
index ad0f96dd540..af4248e5e3b 100644
--- a/homeassistant/components/integration/sensor.py
+++ b/homeassistant/components/integration/sensor.py
@@ -162,6 +162,7 @@ async def async_setup_entry(
     ):
         device_info = DeviceInfo(
             identifiers=device.identifiers,
+            connections=device.connections,
         )
     else:
         device_info = None
diff --git a/tests/components/integration/test_sensor.py b/tests/components/integration/test_sensor.py
index 5b3734bd1be..a552d401681 100644
--- a/tests/components/integration/test_sensor.py
+++ b/tests/components/integration/test_sensor.py
@@ -688,6 +688,7 @@ async def test_device_id(hass: HomeAssistant) -> None:
     source_device_entry = device_registry.async_get_or_create(
         config_entry_id=source_config_entry.entry_id,
         identifiers={("sensor", "identifier_test")},
+        connections={("mac", "30:31:32:33:34:35")},
     )
     source_entity = entity_registry.async_get_or_create(
         "sensor",