From 5892a1dbe2b346ac22487045e0c573163cf613cf Mon Sep 17 00:00:00 2001
From: "J. Nick Koston" <nick@koston.org>
Date: Sun, 29 Jun 2025 08:40:26 -0500
Subject: [PATCH] tests

---
 .../test_api_conditional_memory.py            | 19 ++++++++++---------
 1 file changed, 10 insertions(+), 9 deletions(-)

diff --git a/tests/integration/test_api_conditional_memory.py b/tests/integration/test_api_conditional_memory.py
index b2b235b400..b85e8d91af 100644
--- a/tests/integration/test_api_conditional_memory.py
+++ b/tests/integration/test_api_conditional_memory.py
@@ -9,6 +9,7 @@ from aioesphomeapi import (
     EntityState,
     SensorInfo,
     TextSensorInfo,
+    UserService,
     UserServiceArgType,
 )
 import pytest
@@ -39,11 +40,11 @@ async def test_api_conditional_memory(
             )
 
             # Find our entities
-            client_connected = None
-            client_disconnected_event = None
-            service_called_sensor = None
-            service_arg_sensor = None
-            last_client_info = None
+            client_connected: BinarySensorInfo | None = None
+            client_disconnected_event: BinarySensorInfo | None = None
+            service_called_sensor: BinarySensorInfo | None = None
+            service_arg_sensor: SensorInfo | None = None
+            last_client_info: TextSensorInfo | None = None
 
             for entity in entity_info:
                 if isinstance(entity, BinarySensorInfo):
@@ -73,8 +74,8 @@ async def test_api_conditional_memory(
             assert len(services) == 2, f"Expected 2 services, found {len(services)}"
 
             # Find our services
-            simple_service = None
-            service_with_args = None
+            simple_service: UserService | None = None
+            service_with_args: UserService | None = None
 
             for service in services:
                 if service.name == "test_simple_service":
@@ -98,7 +99,7 @@ async def test_api_conditional_memory(
             assert arg_types["arg_float"] == UserServiceArgType.FLOAT
 
             # Track state changes
-            states = {}
+            states: dict[int, EntityState] = {}
             states_future: asyncio.Future[None] = loop.create_future()
 
             def on_state(state: EntityState) -> None:
@@ -175,7 +176,7 @@ async def test_api_conditional_memory(
         # After disconnecting first client, reconnect and verify triggers work
         async with api_client_connected() as client2:
             # Subscribe to states with new client
-            states2 = {}
+            states2: dict[int, EntityState] = {}
             connected_future: asyncio.Future[None] = loop.create_future()
 
             def on_state2(state: EntityState) -> None: