From a6cfbd4072e3bfb8004a686cfdd545dcdef007fb Mon Sep 17 00:00:00 2001 From: Ian Date: Fri, 17 Dec 2021 00:39:55 -0800 Subject: [PATCH] Nextbus upcoming sort as integer (#61416) --- homeassistant/components/nextbus/sensor.py | 2 +- tests/components/nextbus/test_sensor.py | 3 ++- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/homeassistant/components/nextbus/sensor.py b/homeassistant/components/nextbus/sensor.py index 604bd0c6602..28fbaeb75d3 100644 --- a/homeassistant/components/nextbus/sensor.py +++ b/homeassistant/components/nextbus/sensor.py @@ -218,7 +218,7 @@ class NextBusDepartureSensor(SensorEntity): # Generate list of upcoming times self._attributes["upcoming"] = ", ".join( - sorted(p["minutes"] for p in predictions) + sorted((p["minutes"] for p in predictions), key=int) ) latest_prediction = maybe_first(predictions) diff --git a/tests/components/nextbus/test_sensor.py b/tests/components/nextbus/test_sensor.py index 016afed2b0f..f113d5c83da 100644 --- a/tests/components/nextbus/test_sensor.py +++ b/tests/components/nextbus/test_sensor.py @@ -40,6 +40,7 @@ BASIC_RESULTS = { {"minutes": "1", "epochTime": "1553807371000"}, {"minutes": "2", "epochTime": "1553807372000"}, {"minutes": "3", "epochTime": "1553807373000"}, + {"minutes": "10", "epochTime": "1553807380000"}, ], }, } @@ -128,7 +129,7 @@ async def test_verify_valid_state( assert state.attributes["route"] == VALID_ROUTE_TITLE assert state.attributes["stop"] == VALID_STOP_TITLE assert state.attributes["direction"] == "Outbound" - assert state.attributes["upcoming"] == "1, 2, 3" + assert state.attributes["upcoming"] == "1, 2, 3, 10" async def test_message_dict(