Ensure gdacs tests add config entry before updating it (#110431)

This commit is contained in:
J. Nick Koston 2024-02-12 15:56:48 -06:00 committed by GitHub
parent b3d9e15fb4
commit 4e76e6b2b0
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -97,10 +97,10 @@ async def test_setup(
) as mock_feed_update:
mock_feed_update.return_value = "OK", [mock_entry_1, mock_entry_2, mock_entry_3]
config_entry.add_to_hass(hass)
hass.config_entries.async_update_entry(
config_entry, data=config_entry.data | CONFIG
)
config_entry.add_to_hass(hass)
assert await hass.config_entries.async_setup(config_entry.entry_id)
# Artificially trigger update and collect events.
hass.bus.async_fire(EVENT_HOMEASSISTANT_START)
@ -232,10 +232,10 @@ async def test_setup_imperial(
"aio_georss_client.feed.GeoRssFeed.last_timestamp", create=True
):
mock_feed_update.return_value = "OK", [mock_entry_1]
config_entry.add_to_hass(hass)
hass.config_entries.async_update_entry(
config_entry, data=config_entry.data | CONFIG
)
config_entry.add_to_hass(hass)
assert await hass.config_entries.async_setup(
config_entry.entry_id
) # Artificially trigger update and collect events.