From 9a1f7f020c2710ad2a16d222beab7ac93e7428f6 Mon Sep 17 00:00:00 2001
From: epenet <6771947+epenet@users.noreply.github.com>
Date: Thu, 4 Jul 2024 16:17:18 +0200
Subject: [PATCH] Add type hints to matrix events fixtures (#121213)

---
 tests/components/matrix/conftest.py          | 6 +++---
 tests/components/matrix/test_send_message.py | 6 +++---
 2 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/tests/components/matrix/conftest.py b/tests/components/matrix/conftest.py
index 11347302177..0b84aff5434 100644
--- a/tests/components/matrix/conftest.py
+++ b/tests/components/matrix/conftest.py
@@ -48,7 +48,7 @@ from homeassistant.const import (
     CONF_USERNAME,
     CONF_VERIFY_SSL,
 )
-from homeassistant.core import HomeAssistant
+from homeassistant.core import Event, HomeAssistant
 from homeassistant.setup import async_setup_component
 
 from tests.common import async_capture_events
@@ -294,13 +294,13 @@ async def matrix_bot(
 
 
 @pytest.fixture
-def matrix_events(hass: HomeAssistant):
+def matrix_events(hass: HomeAssistant) -> list[Event]:
     """Track event calls."""
     return async_capture_events(hass, MATRIX_DOMAIN)
 
 
 @pytest.fixture
-def command_events(hass: HomeAssistant):
+def command_events(hass: HomeAssistant) -> list[Event]:
     """Track event calls."""
     return async_capture_events(hass, EVENT_MATRIX_COMMAND)
 
diff --git a/tests/components/matrix/test_send_message.py b/tests/components/matrix/test_send_message.py
index cdea2270cf9..3db2877e789 100644
--- a/tests/components/matrix/test_send_message.py
+++ b/tests/components/matrix/test_send_message.py
@@ -10,7 +10,7 @@ from homeassistant.components.matrix import (
 )
 from homeassistant.components.matrix.const import FORMAT_HTML, SERVICE_SEND_MESSAGE
 from homeassistant.components.notify import ATTR_DATA, ATTR_MESSAGE, ATTR_TARGET
-from homeassistant.core import HomeAssistant
+from homeassistant.core import Event, HomeAssistant
 
 from .conftest import TEST_BAD_ROOM, TEST_JOINABLE_ROOMS
 
@@ -19,7 +19,7 @@ async def test_send_message(
     hass: HomeAssistant,
     matrix_bot: MatrixBot,
     image_path,
-    matrix_events,
+    matrix_events: list[Event],
     caplog: pytest.LogCaptureFixture,
 ) -> None:
     """Test the send_message service."""
@@ -63,7 +63,7 @@ async def test_send_message(
 async def test_unsendable_message(
     hass: HomeAssistant,
     matrix_bot: MatrixBot,
-    matrix_events,
+    matrix_events: list[Event],
     caplog: pytest.LogCaptureFixture,
 ) -> None:
     """Test the send_message service with an invalid room."""