Improve async_track_template_result callback typing (#97135)

homeassistant/components/bayesian/binary_sensor.py

@@ -27,7 +27,7 @@ from homeassistant.const import (
     STATE_UNAVAILABLE,
     STATE_UNKNOWN,
 )
-from homeassistant.core import Event, HomeAssistant, callback
+from homeassistant.core import HomeAssistant, callback
 from homeassistant.exceptions import ConditionError, TemplateError
 from homeassistant.helpers import condition
 import homeassistant.helpers.config_validation as cv
@@ -259,14 +259,13 @@ class BayesianBinarySensor(BinarySensorEntity):
 
         @callback
         def _async_template_result_changed(
-            event: Event | None, updates: list[TrackTemplateResult]
+            event: EventType[EventStateChangedData] | None,
+            updates: list[TrackTemplateResult],
         ) -> None:
             track_template_result = updates.pop()
             template = track_template_result.template
             result = track_template_result.result
-            entity: str | None = (
+            entity_id = None if event is None else event.data["entity_id"]
-                None if event is None else event.data.get(CONF_ENTITY_ID)
-            )
             if isinstance(result, TemplateError):
                 _LOGGER.error(
                     "TemplateError('%s') while processing template '%s' in entity '%s'",
@@ -283,8 +282,8 @@ class BayesianBinarySensor(BinarySensorEntity):
                 observation.observed = observed
 
                 # in some cases a template may update because of the absence of an entity
-                if entity is not None:
+                if entity_id is not None:
-                    observation.entity_id = entity
+                    observation.entity_id = entity_id
 
                 self.current_observations[observation.id] = observation
 

homeassistant/components/template/trigger.py

@@ -1,5 +1,7 @@
 """Offer template automation rules."""
+from datetime import timedelta
 import logging
+from typing import Any
 
 import voluptuous as vol
 
@@ -8,13 +10,15 @@ from homeassistant.const import CONF_FOR, CONF_PLATFORM, CONF_VALUE_TEMPLATE
 from homeassistant.core import CALLBACK_TYPE, HassJob, HomeAssistant, callback
 from homeassistant.helpers import config_validation as cv, template
 from homeassistant.helpers.event import (
+    EventStateChangedData,
     TrackTemplate,
+    TrackTemplateResult,
     async_call_later,
     async_track_template_result,
 )
 from homeassistant.helpers.template import Template, result_as_boolean
 from homeassistant.helpers.trigger import TriggerActionType, TriggerInfo
-from homeassistant.helpers.typing import ConfigType
+from homeassistant.helpers.typing import ConfigType, EventType
 
 _LOGGER = logging.getLogger(__name__)
 
@@ -59,7 +63,10 @@ async def async_attach_trigger(
         )
 
     @callback
-    def template_listener(event, updates):
+    def template_listener(
+        event: EventType[EventStateChangedData] | None,
+        updates: list[TrackTemplateResult],
+    ) -> None:
         """Listen for state changes and calls action."""
         nonlocal delay_cancel, armed
         result = updates.pop().result
@@ -88,9 +95,9 @@ async def async_attach_trigger(
         # Fire!
         armed = False
 
-        entity_id = event and event.data.get("entity_id")
+        entity_id = event and event.data["entity_id"]
-        from_s = event and event.data.get("old_state")
+        from_s = event and event.data["old_state"]
-        to_s = event and event.data.get("new_state")
+        to_s = event and event.data["new_state"]
 
         if entity_id is not None:
             description = f"{entity_id} via template"
@@ -110,7 +117,7 @@ async def async_attach_trigger(
         }
 
         @callback
-        def call_action(*_):
+        def call_action(*_: Any) -> None:
             """Call action with right context."""
             nonlocal trigger_variables
             hass.async_run_hass_job(
@@ -124,7 +131,7 @@ async def async_attach_trigger(
             return
 
         try:
-            period = cv.positive_time_period(
+            period: timedelta = cv.positive_time_period(
                 template.render_complex(time_delta, {"trigger": template_variables})
             )
         except (exceptions.TemplateError, vol.Invalid) as ex:

homeassistant/components/universal/media_player.py

@@ -87,6 +87,7 @@ from homeassistant.helpers.entity_platform import AddEntitiesCallback
 from homeassistant.helpers.event import (
     EventStateChangedData,
     TrackTemplate,
+    TrackTemplateResult,
     async_track_state_change_event,
     async_track_template_result,
 )
@@ -192,7 +193,10 @@ class UniversalMediaPlayer(MediaPlayerEntity):
             self.async_schedule_update_ha_state(True)
 
         @callback
-        def _async_on_template_update(event, updates):
+        def _async_on_template_update(
+            event: EventType[EventStateChangedData] | None,
+            updates: list[TrackTemplateResult],
+        ) -> None:
             """Update state when template state changes."""
             for data in updates:
                 template = data.template

homeassistant/components/websocket_api/commands.py

@@ -26,6 +26,7 @@ from homeassistant.exceptions import (
 from homeassistant.helpers import config_validation as cv, entity, template
 from homeassistant.helpers.dispatcher import async_dispatcher_connect
 from homeassistant.helpers.event import (
+    EventStateChangedData,
     TrackTemplate,
     TrackTemplateResult,
     async_track_template_result,
@@ -37,6 +38,7 @@ from homeassistant.helpers.json import (
     json_dumps,
 )
 from homeassistant.helpers.service import async_get_all_descriptions
+from homeassistant.helpers.typing import EventType
 from homeassistant.loader import (
     Integration,
     IntegrationNotFound,
@@ -535,7 +537,8 @@ async def handle_render_template(
 
     @callback
     def _template_listener(
-        event: Event | None, updates: list[TrackTemplateResult]
+        event: EventType[EventStateChangedData] | None,
+        updates: list[TrackTemplateResult],
     ) -> None:
         nonlocal info
         track_template_result = updates.pop()

homeassistant/helpers/event.py

@@ -888,9 +888,7 @@ class TrackTemplateResultInfo:
         self,
         hass: HomeAssistant,
         track_templates: Sequence[TrackTemplate],
-        action: Callable[
+        action: TrackTemplateResultListener,
-            [EventType[EventStateChangedData] | None, list[TrackTemplateResult]], None
-        ],
         has_super_template: bool = False,
     ) -> None:
         """Handle removal / refresh of tracker init."""
@@ -1209,7 +1207,7 @@ TrackTemplateResultListener = Callable[
         EventType[EventStateChangedData] | None,
         list[TrackTemplateResult],
     ],
-    None,
+    Coroutine[Any, Any, None] | None,
 ]
 """Type for the listener for template results.
 

homeassistant/helpers/template_entity.py

@@ -15,7 +15,6 @@ from homeassistant.components.sensor import (
     SensorEntity,
 )
 from homeassistant.const import (
-    ATTR_ENTITY_ID,
     ATTR_ENTITY_PICTURE,
     ATTR_FRIENDLY_NAME,
     ATTR_ICON,
@@ -33,7 +32,12 @@ from homeassistant.util.json import JSON_DECODE_EXCEPTIONS, json_loads
 
 from . import config_validation as cv
 from .entity import Entity
-from .event import TrackTemplate, TrackTemplateResult, async_track_template_result
+from .event import (
+    EventStateChangedData,
+    TrackTemplate,
+    TrackTemplateResult,
+    async_track_template_result,
+)
 from .script import Script, _VarsType
 from .template import (
     Template,
@@ -42,7 +46,7 @@ from .template import (
     render_complex,
     result_as_boolean,
 )
-from .typing import ConfigType
+from .typing import ConfigType, EventType
 
 _LOGGER = logging.getLogger(__name__)
 
@@ -327,14 +331,14 @@ class TemplateEntity(Entity):
     @callback
     def _handle_results(
         self,
-        event: Event | None,
+        event: EventType[EventStateChangedData] | None,
         updates: list[TrackTemplateResult],
     ) -> None:
         """Call back the results to the attributes."""
         if event:
             self.async_set_context(event.context)
 
-        entity_id = event and event.data.get(ATTR_ENTITY_ID)
+        entity_id = event and event.data["entity_id"]
 
         if entity_id and entity_id == self.entity_id:
             self._self_ref_update_count += 1

tests/helpers/test_event.py

@@ -965,7 +965,10 @@ async def test_track_template_result(hass: HomeAssistant) -> None:
         "{{(states.sensor.test.state|int) + test }}", hass
     )
 
-    def specific_run_callback(event, updates):
+    def specific_run_callback(
+        event: EventType[EventStateChangedData] | None,
+        updates: list[TrackTemplateResult],
+    ) -> None:
         track_result = updates.pop()
         specific_runs.append(int(track_result.result))
 
@@ -974,7 +977,10 @@ async def test_track_template_result(hass: HomeAssistant) -> None:
     )
 
     @ha.callback
-    def wildcard_run_callback(event, updates):
+    def wildcard_run_callback(
+        event: EventType[EventStateChangedData] | None,
+        updates: list[TrackTemplateResult],
+    ) -> None:
         track_result = updates.pop()
         wildcard_runs.append(
             (int(track_result.last_result or 0), int(track_result.result))
@@ -984,7 +990,10 @@ async def test_track_template_result(hass: HomeAssistant) -> None:
         hass, [TrackTemplate(template_condition, None)], wildcard_run_callback
     )
 
-    async def wildercard_run_callback(event, updates):
+    async def wildercard_run_callback(
+        event: EventType[EventStateChangedData] | None,
+        updates: list[TrackTemplateResult],
+    ) -> None:
         track_result = updates.pop()
         wildercard_runs.append(
             (int(track_result.last_result or 0), int(track_result.result))
@@ -1051,7 +1060,10 @@ async def test_track_template_result_none(hass: HomeAssistant) -> None:
         "{{(state_attr('sensor.test', 'battery')|int(default=0)) + test }}", hass
     )
 
-    def specific_run_callback(event, updates):
+    def specific_run_callback(
+        event: EventType[EventStateChangedData] | None,
+        updates: list[TrackTemplateResult],
+    ) -> None:
         track_result = updates.pop()
         result = int(track_result.result) if track_result.result is not None else None
         specific_runs.append(result)
@@ -1061,7 +1073,10 @@ async def test_track_template_result_none(hass: HomeAssistant) -> None:
     )
 
     @ha.callback
-    def wildcard_run_callback(event, updates):
+    def wildcard_run_callback(
+        event: EventType[EventStateChangedData] | None,
+        updates: list[TrackTemplateResult],
+    ) -> None:
         track_result = updates.pop()
         last_result = (
             int(track_result.last_result)
@@ -1075,7 +1090,10 @@ async def test_track_template_result_none(hass: HomeAssistant) -> None:
         hass, [TrackTemplate(template_condition, None)], wildcard_run_callback
     )
 
-    async def wildercard_run_callback(event, updates):
+    async def wildercard_run_callback(
+        event: EventType[EventStateChangedData] | None,
+        updates: list[TrackTemplateResult],
+    ) -> None:
         track_result = updates.pop()
         last_result = (
             int(track_result.last_result)
@@ -1122,7 +1140,10 @@ async def test_track_template_result_super_template(hass: HomeAssistant) -> None
         "{{(states.sensor.test.state|int) + test }}", hass
     )
 
-    def specific_run_callback(event, updates):
+    def specific_run_callback(
+        event: EventType[EventStateChangedData] | None,
+        updates: list[TrackTemplateResult],
+    ) -> None:
         for track_result in updates:
             if track_result.template is template_condition:
                 specific_runs.append(int(track_result.result))
@@ -1140,7 +1161,10 @@ async def test_track_template_result_super_template(hass: HomeAssistant) -> None
     )
 
     @ha.callback
-    def wildcard_run_callback(event, updates):
+    def wildcard_run_callback(
+        event: EventType[EventStateChangedData] | None,
+        updates: list[TrackTemplateResult],
+    ) -> None:
         for track_result in updates:
             if track_result.template is template_condition:
                 wildcard_runs.append(
@@ -1159,7 +1183,10 @@ async def test_track_template_result_super_template(hass: HomeAssistant) -> None
         has_super_template=True,
     )
 
-    async def wildercard_run_callback(event, updates):
+    async def wildercard_run_callback(
+        event: EventType[EventStateChangedData] | None,
+        updates: list[TrackTemplateResult],
+    ) -> None:
         for track_result in updates:
             if track_result.template is template_condition_var:
                 wildercard_runs.append(
@@ -1272,7 +1299,10 @@ async def test_track_template_result_super_template_initially_false(
     hass.states.async_set("sensor.test", "unavailable")
     await hass.async_block_till_done()
 
-    def specific_run_callback(event, updates):
+    def specific_run_callback(
+        event: EventType[EventStateChangedData] | None,
+        updates: list[TrackTemplateResult],
+    ) -> None:
         for track_result in updates:
             if track_result.template is template_condition:
                 specific_runs.append(int(track_result.result))
@@ -1290,7 +1320,10 @@ async def test_track_template_result_super_template_initially_false(
     )
 
     @ha.callback
-    def wildcard_run_callback(event, updates):
+    def wildcard_run_callback(
+        event: EventType[EventStateChangedData] | None,
+        updates: list[TrackTemplateResult],
+    ) -> None:
         for track_result in updates:
             if track_result.template is template_condition:
                 wildcard_runs.append(
@@ -1309,7 +1342,10 @@ async def test_track_template_result_super_template_initially_false(
         has_super_template=True,
     )
 
-    async def wildercard_run_callback(event, updates):
+    async def wildercard_run_callback(
+        event: EventType[EventStateChangedData] | None,
+        updates: list[TrackTemplateResult],
+    ) -> None:
         for track_result in updates:
             if track_result.template is template_condition_var:
                 wildercard_runs.append(
@@ -1434,7 +1470,10 @@ async def test_track_template_result_super_template_2(
 
         return result_as_boolean(result)
 
-    def specific_run_callback(event, updates):
+    def specific_run_callback(
+        event: EventType[EventStateChangedData] | None,
+        updates: list[TrackTemplateResult],
+    ) -> None:
         for track_result in updates:
             if track_result.template is template_condition:
                 specific_runs.append(int(track_result.result))
@@ -1454,7 +1493,10 @@ async def test_track_template_result_super_template_2(
     )
 
     @ha.callback
-    def wildcard_run_callback(event, updates):
+    def wildcard_run_callback(
+        event: EventType[EventStateChangedData] | None,
+        updates: list[TrackTemplateResult],
+    ) -> None:
         for track_result in updates:
             if track_result.template is template_condition:
                 wildcard_runs.append(
@@ -1475,7 +1517,10 @@ async def test_track_template_result_super_template_2(
         has_super_template=True,
     )
 
-    async def wildercard_run_callback(event, updates):
+    async def wildercard_run_callback(
+        event: EventType[EventStateChangedData] | None,
+        updates: list[TrackTemplateResult],
+    ) -> None:
         for track_result in updates:
             if track_result.template is template_condition_var:
                 wildercard_runs.append(
@@ -1580,7 +1625,10 @@ async def test_track_template_result_super_template_2_initially_false(
 
         return result_as_boolean(result)
 
-    def specific_run_callback(event, updates):
+    def specific_run_callback(
+        event: EventType[EventStateChangedData] | None,
+        updates: list[TrackTemplateResult],
+    ) -> None:
         for track_result in updates:
             if track_result.template is template_condition:
                 specific_runs.append(int(track_result.result))
@@ -1600,7 +1648,10 @@ async def test_track_template_result_super_template_2_initially_false(
     )
 
     @ha.callback
-    def wildcard_run_callback(event, updates):
+    def wildcard_run_callback(
+        event: EventType[EventStateChangedData] | None,
+        updates: list[TrackTemplateResult],
+    ) -> None:
         for track_result in updates:
             if track_result.template is template_condition:
                 wildcard_runs.append(
@@ -1621,7 +1672,10 @@ async def test_track_template_result_super_template_2_initially_false(
         has_super_template=True,
     )
 
-    async def wildercard_run_callback(event, updates):
+    async def wildercard_run_callback(
+        event: EventType[EventStateChangedData] | None,
+        updates: list[TrackTemplateResult],
+    ) -> None:
         for track_result in updates:
             if track_result.template is template_condition_var:
                 wildercard_runs.append(
@@ -1701,7 +1755,10 @@ async def test_track_template_result_complex(hass: HomeAssistant) -> None:
 """
     template_complex = Template(template_complex_str, hass)
 
-    def specific_run_callback(event, updates):
+    def specific_run_callback(
+        event: EventType[EventStateChangedData] | None,
+        updates: list[TrackTemplateResult],
+    ) -> None:
         specific_runs.append(updates.pop().result)
 
     hass.states.async_set("light.one", "on")
@@ -1854,7 +1911,10 @@ async def test_track_template_result_with_wildcard(hass: HomeAssistant) -> None:
 """
     template_complex = Template(template_complex_str, hass)
 
-    def specific_run_callback(event, updates):
+    def specific_run_callback(
+        event: EventType[EventStateChangedData] | None,
+        updates: list[TrackTemplateResult],
+    ) -> None:
         specific_runs.append(updates.pop().result)
 
     hass.states.async_set("cover.office_drapes", "closed")
@@ -1906,7 +1966,10 @@ async def test_track_template_result_with_group(hass: HomeAssistant) -> None:
 """
     template_complex = Template(template_complex_str, hass)
 
-    def specific_run_callback(event, updates):
+    def specific_run_callback(
+        event: EventType[EventStateChangedData] | None,
+        updates: list[TrackTemplateResult],
+    ) -> None:
         specific_runs.append(updates.pop().result)
 
     info = async_track_template_result(
@@ -1963,7 +2026,10 @@ async def test_track_template_result_and_conditional(hass: HomeAssistant) -> Non
 
     template = Template(template_str, hass)
 
-    def specific_run_callback(event, updates):
+    def specific_run_callback(
+        event: EventType[EventStateChangedData] | None,
+        updates: list[TrackTemplateResult],
+    ) -> None:
         specific_runs.append(updates.pop().result)
 
     info = async_track_template_result(
@@ -2028,7 +2094,10 @@ async def test_track_template_result_and_conditional_upper_case(
 
     template = Template(template_str, hass)
 
-    def specific_run_callback(event, updates):
+    def specific_run_callback(
+        event: EventType[EventStateChangedData] | None,
+        updates: list[TrackTemplateResult],
+    ) -> None:
         specific_runs.append(updates.pop().result)
 
     info = async_track_template_result(
@@ -2087,7 +2156,10 @@ async def test_track_template_result_iterator(hass: HomeAssistant) -> None:
     iterator_runs = []
 
     @ha.callback
-    def iterator_callback(event, updates):
+    def iterator_callback(
+        event: EventType[EventStateChangedData] | None,
+        updates: list[TrackTemplateResult],
+    ) -> None:
         iterator_runs.append(updates.pop().result)
 
     async_track_template_result(
@@ -2120,7 +2192,10 @@ async def test_track_template_result_iterator(hass: HomeAssistant) -> None:
     filter_runs = []
 
     @ha.callback
-    def filter_callback(event, updates):
+    def filter_callback(
+        event: EventType[EventStateChangedData] | None,
+        updates: list[TrackTemplateResult],
+    ) -> None:
         filter_runs.append(updates.pop().result)
 
     info = async_track_template_result(
@@ -2170,7 +2245,10 @@ async def test_track_template_result_errors(
     not_exist_runs = []
 
     @ha.callback
-    def syntax_error_listener(event, updates):
+    def syntax_error_listener(
+        event: EventType[EventStateChangedData] | None,
+        updates: list[TrackTemplateResult],
+    ) -> None:
         track_result = updates.pop()
         syntax_error_runs.append(
             (
@@ -2190,7 +2268,10 @@ async def test_track_template_result_errors(
     assert "TemplateSyntaxError" in caplog.text
 
     @ha.callback
-    def not_exist_runs_error_listener(event, updates):
+    def not_exist_runs_error_listener(
+        event: EventType[EventStateChangedData] | None,
+        updates: list[TrackTemplateResult],
+    ) -> None:
         template_track = updates.pop()
         not_exist_runs.append(
             (
@@ -2255,7 +2336,10 @@ async def test_track_template_result_transient_errors(
     sometimes_error_runs = []
 
     @ha.callback
-    def sometimes_error_listener(event, updates):
+    def sometimes_error_listener(
+        event: EventType[EventStateChangedData] | None,
+        updates: list[TrackTemplateResult],
+    ) -> None:
         track_result = updates.pop()
         sometimes_error_runs.append(
             (
@@ -2300,7 +2384,10 @@ async def test_static_string(hass: HomeAssistant) -> None:
     refresh_runs = []
 
     @ha.callback
-    def refresh_listener(event, updates):
+    def refresh_listener(
+        event: EventType[EventStateChangedData] | None,
+        updates: list[TrackTemplateResult],
+    ) -> None:
         refresh_runs.append(updates.pop().result)
 
     info = async_track_template_result(
@@ -2320,7 +2407,10 @@ async def test_track_template_rate_limit(hass: HomeAssistant) -> None:
     refresh_runs = []
 
     @ha.callback
-    def refresh_listener(event, updates):
+    def refresh_listener(
+        event: EventType[EventStateChangedData] | None,
+        updates: list[TrackTemplateResult],
+    ) -> None:
         refresh_runs.append(updates.pop().result)
 
     info = async_track_template_result(
@@ -2379,7 +2469,10 @@ async def test_track_template_rate_limit_super(hass: HomeAssistant) -> None:
     refresh_runs = []
 
     @ha.callback
-    def refresh_listener(event, updates):
+    def refresh_listener(
+        event: EventType[EventStateChangedData] | None,
+        updates: list[TrackTemplateResult],
+    ) -> None:
         for track_result in updates:
             if track_result.template is template_refresh:
                 refresh_runs.append(track_result.result)
@@ -2452,7 +2545,10 @@ async def test_track_template_rate_limit_super_2(hass: HomeAssistant) -> None:
     refresh_runs = []
 
     @ha.callback
-    def refresh_listener(event, updates):
+    def refresh_listener(
+        event: EventType[EventStateChangedData] | None,
+        updates: list[TrackTemplateResult],
+    ) -> None:
         for track_result in updates:
             if track_result.template is template_refresh:
                 refresh_runs.append(track_result.result)
@@ -2521,7 +2617,10 @@ async def test_track_template_rate_limit_super_3(hass: HomeAssistant) -> None:
     refresh_runs = []
 
     @ha.callback
-    def refresh_listener(event, updates):
+    def refresh_listener(
+        event: EventType[EventStateChangedData] | None,
+        updates: list[TrackTemplateResult],
+    ) -> None:
         for track_result in updates:
             if track_result.template is template_refresh:
                 refresh_runs.append(track_result.result)
@@ -2592,7 +2691,10 @@ async def test_track_template_rate_limit_suppress_listener(hass: HomeAssistant)
     refresh_runs = []
 
     @ha.callback
-    def refresh_listener(event, updates):
+    def refresh_listener(
+        event: EventType[EventStateChangedData] | None,
+        updates: list[TrackTemplateResult],
+    ) -> None:
         refresh_runs.append(updates.pop().result)
 
     info = async_track_template_result(
@@ -2689,7 +2791,10 @@ async def test_track_template_rate_limit_five(hass: HomeAssistant) -> None:
     refresh_runs = []
 
     @ha.callback
-    def refresh_listener(event, updates):
+    def refresh_listener(
+        event: EventType[EventStateChangedData] | None,
+        updates: list[TrackTemplateResult],
+    ) -> None:
         refresh_runs.append(updates.pop().result)
 
     info = async_track_template_result(
@@ -2725,7 +2830,10 @@ async def test_track_template_has_default_rate_limit(hass: HomeAssistant) -> Non
     refresh_runs = []
 
     @ha.callback
-    def refresh_listener(event, updates):
+    def refresh_listener(
+        event: EventType[EventStateChangedData] | None,
+        updates: list[TrackTemplateResult],
+    ) -> None:
         refresh_runs.append(updates.pop().result)
 
     info = async_track_template_result(
@@ -2766,7 +2874,10 @@ async def test_track_template_unavailable_states_has_default_rate_limit(
     refresh_runs = []
 
     @ha.callback
-    def refresh_listener(event, updates):
+    def refresh_listener(
+        event: EventType[EventStateChangedData] | None,
+        updates: list[TrackTemplateResult],
+    ) -> None:
         refresh_runs.append(updates.pop().result)
 
     info = async_track_template_result(
@@ -2807,7 +2918,10 @@ async def test_specifically_referenced_entity_is_not_rate_limited(
     refresh_runs = []
 
     @ha.callback
-    def refresh_listener(event, updates):
+    def refresh_listener(
+        event: EventType[EventStateChangedData] | None,
+        updates: list[TrackTemplateResult],
+    ) -> None:
         refresh_runs.append(updates.pop().result)
 
     info = async_track_template_result(
@@ -2850,7 +2964,10 @@ async def test_track_two_templates_with_different_rate_limits(
     }
 
     @ha.callback
-    def refresh_listener(event, updates):
+    def refresh_listener(
+        event: EventType[EventStateChangedData] | None,
+        updates: list[TrackTemplateResult],
+    ) -> None:
         for update in updates:
             refresh_runs[update.template].append(update.result)
 
@@ -2911,7 +3028,10 @@ async def test_string(hass: HomeAssistant) -> None:
     refresh_runs = []
 
     @ha.callback
-    def refresh_listener(event, updates):
+    def refresh_listener(
+        event: EventType[EventStateChangedData] | None,
+        updates: list[TrackTemplateResult],
+    ) -> None:
         refresh_runs.append(updates.pop().result)
 
     info = async_track_template_result(
@@ -2931,7 +3051,10 @@ async def test_track_template_result_refresh_cancel(hass: HomeAssistant) -> None
     refresh_runs = []
 
     @ha.callback
-    def refresh_listener(event, updates):
+    def refresh_listener(
+        event: EventType[EventStateChangedData] | None,
+        updates: list[TrackTemplateResult],
+    ) -> None:
         refresh_runs.append(updates.pop().result)
 
     info = async_track_template_result(
@@ -2993,7 +3116,10 @@ async def test_async_track_template_result_multiple_templates(
     refresh_runs = []
 
     @ha.callback
-    def refresh_listener(event, updates):
+    def refresh_listener(
+        event: EventType[EventStateChangedData] | None,
+        updates: list[TrackTemplateResult],
+    ) -> None:
         refresh_runs.append(updates)
 
     async_track_template_result(
@@ -3054,7 +3180,10 @@ async def test_async_track_template_result_multiple_templates_mixing_domain(
     refresh_runs = []
 
     @ha.callback
-    def refresh_listener(event, updates):
+    def refresh_listener(
+        event: EventType[EventStateChangedData] | None,
+        updates: list[TrackTemplateResult],
+    ) -> None:
         refresh_runs.append(updates)
 
     async_track_template_result(
@@ -3139,7 +3268,10 @@ async def test_track_template_with_time(hass: HomeAssistant) -> None:
     specific_runs = []
     template_complex = Template("{{ states.switch.test.state and now() }}", hass)
 
-    def specific_run_callback(event, updates):
+    def specific_run_callback(
+        event: EventType[EventStateChangedData] | None,
+        updates: list[TrackTemplateResult],
+    ) -> None:
         specific_runs.append(updates.pop().result)
 
     info = async_track_template_result(
@@ -3169,7 +3301,10 @@ async def test_track_template_with_time_default(hass: HomeAssistant) -> None:
     specific_runs = []
     template_complex = Template("{{ now() }}", hass)
 
-    def specific_run_callback(event, updates):
+    def specific_run_callback(
+        event: EventType[EventStateChangedData] | None,
+        updates: list[TrackTemplateResult],
+    ) -> None:
         specific_runs.append(updates.pop().result)
 
     info = async_track_template_result(
@@ -3218,7 +3353,10 @@ async def test_track_template_with_time_that_leaves_scope(hass: HomeAssistant) -
             hass,
         )
 
-        def specific_run_callback(event, updates):
+        def specific_run_callback(
+            event: EventType[EventStateChangedData] | None,
+            updates: list[TrackTemplateResult],
+        ) -> None:
             specific_runs.append(updates.pop().result)
 
         info = async_track_template_result(
@@ -3283,7 +3421,10 @@ async def test_async_track_template_result_multiple_templates_mixing_listeners(
     refresh_runs = []
 
     @ha.callback
-    def refresh_listener(event, updates):
+    def refresh_listener(
+        event: EventType[EventStateChangedData] | None,
+        updates: list[TrackTemplateResult],
+    ) -> None:
         refresh_runs.append(updates)
 
     now = dt_util.utcnow()