From e158546425b51378a9319058d2eebbdfc0e697f4 Mon Sep 17 00:00:00 2001
From: Kevin Stillhammer <kevin.stillhammer@gmail.com>
Date: Wed, 30 Nov 2022 22:40:17 +0100
Subject: [PATCH] Add missing @callback decorator in here_travel_time sensor
(#83009)
---
homeassistant/components/here_travel_time/sensor.py | 5 +++--
tests/components/here_travel_time/test_sensor.py | 2 --
2 files changed, 3 insertions(+), 4 deletions(-)
diff --git a/homeassistant/components/here_travel_time/sensor.py b/homeassistant/components/here_travel_time/sensor.py
index 18433a86560..e432511c486 100644
--- a/homeassistant/components/here_travel_time/sensor.py
+++ b/homeassistant/components/here_travel_time/sensor.py
@@ -21,7 +21,7 @@ from homeassistant.const import (
TIME_MINUTES,
UnitOfLength,
)
-from homeassistant.core import HomeAssistant
+from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import DeviceEntryType
from homeassistant.helpers.entity import DeviceInfo
from homeassistant.helpers.entity_platform import AddEntitiesCallback
@@ -99,7 +99,7 @@ async def async_setup_entry(
async_add_entities(sensors)
-class HERETravelTimeSensor(RestoreSensor, CoordinatorEntity):
+class HERETravelTimeSensor(CoordinatorEntity, RestoreSensor):
"""Representation of a HERE travel time sensor."""
def __init__(
@@ -136,6 +136,7 @@ class HERETravelTimeSensor(RestoreSensor, CoordinatorEntity):
self.async_on_remove(async_at_start(self.hass, _update_at_start))
+ @callback
def _handle_coordinator_update(self) -> None:
"""Handle updated data from the coordinator."""
if self.coordinator.data is not None:
diff --git a/tests/components/here_travel_time/test_sensor.py b/tests/components/here_travel_time/test_sensor.py
index 0ad86a04992..34859956032 100644
--- a/tests/components/here_travel_time/test_sensor.py
+++ b/tests/components/here_travel_time/test_sensor.py
@@ -558,8 +558,6 @@ async def test_restore_state(hass):
await hass.config_entries.async_setup(mock_entry.entry_id)
await hass.async_block_till_done()
- print(hass.states.async_all())
-
# restore from cache
state = hass.states.get("sensor.test_duration")
assert state.state == "1234"