Update nextbus stop tag to accept strings (#27765)

This commit is contained in:
Grodesh 2019-11-06 01:24:11 -05:00 committed by Martin Hjelmare
parent a63e976496
commit f8712b4d7f
2 changed files with 2 additions and 2 deletions

View file

@ -173,7 +173,7 @@ class NextBusDepartureSensor(Entity):
"""Update sensor with new departures times."""
# Note: using Multi because there is a bug with the single stop impl
results = self._client.get_predictions_for_multi_stops(
[{"stop_tag": int(self.stop), "route_tag": self.route}], self.agency
[{"stop_tag": self.stop, "route_tag": self.route}], self.agency
)
self._log_debug("Predictions results: %s", results)

View file

@ -113,7 +113,7 @@ async def test_verify_valid_state(
"""Verify all attributes are set from a valid response."""
await assert_setup_sensor(hass, CONFIG_BASIC)
mock_nextbus_predictions.assert_called_once_with(
[{"stop_tag": int(VALID_STOP), "route_tag": VALID_ROUTE}], VALID_AGENCY
[{"stop_tag": VALID_STOP, "route_tag": VALID_ROUTE}], VALID_AGENCY
)
state = hass.states.get(SENSOR_ID_SHORT)