diff --git a/tests/components/airnow/conftest.py b/tests/components/airnow/conftest.py index 4e9d1698e8c..0356c6f3395 100644 --- a/tests/components/airnow/conftest.py +++ b/tests/components/airnow/conftest.py @@ -6,7 +6,6 @@ import pytest from homeassistant.components.airnow import DOMAIN from homeassistant.const import CONF_API_KEY, CONF_LATITUDE, CONF_LONGITUDE, CONF_RADIUS -from homeassistant.setup import async_setup_component from tests.common import MockConfigEntry, load_fixture @@ -60,8 +59,6 @@ def mock_api_get_fixture(data): async def setup_airnow_fixture(hass, config, mock_api_get): """Define a fixture to set up AirNow.""" with patch("pyairnow.WebServiceAPI._get", mock_api_get), patch( - "homeassistant.components.airnow.config_flow.WebServiceAPI._get", mock_api_get - ), patch("homeassistant.components.airnow.PLATFORMS", []): - assert await async_setup_component(hass, DOMAIN, config) - await hass.async_block_till_done() + "homeassistant.components.airnow.PLATFORMS", [] + ): yield diff --git a/tests/components/airnow/test_diagnostics.py b/tests/components/airnow/test_diagnostics.py index ecf6acc1c80..50ff3ed2b32 100644 --- a/tests/components/airnow/test_diagnostics.py +++ b/tests/components/airnow/test_diagnostics.py @@ -15,6 +15,7 @@ async def test_entry_diagnostics( snapshot: SnapshotAssertion, ) -> None: """Test config entry diagnostics.""" + assert await hass.config_entries.async_setup(config_entry.entry_id) assert ( await get_diagnostics_for_config_entry(hass, hass_client, config_entry) == snapshot