From 8e9b5eb4e19810e76b9050500c94fed7539e54fb Mon Sep 17 00:00:00 2001 From: Franck Nijhof Date: Tue, 26 Nov 2024 00:45:47 +0100 Subject: [PATCH] Extend tests for Stookwijzer init (#131589) --- tests/components/stookwijzer/test_init.py | 35 +++++++++++++++++++++++ 1 file changed, 35 insertions(+) diff --git a/tests/components/stookwijzer/test_init.py b/tests/components/stookwijzer/test_init.py index 0774def7813..0df9b55d1a9 100644 --- a/tests/components/stookwijzer/test_init.py +++ b/tests/components/stookwijzer/test_init.py @@ -14,6 +14,41 @@ from homeassistant.helpers import entity_registry as er, issue_registry as ir from tests.common import MockConfigEntry +async def test_load_unload_config_entry( + hass: HomeAssistant, + mock_config_entry: MockConfigEntry, + mock_stookwijzer: MagicMock, +) -> None: + """Test the Stookwijzer configuration entry loading and unloading.""" + mock_config_entry.add_to_hass(hass) + await hass.config_entries.async_setup(mock_config_entry.entry_id) + await hass.async_block_till_done() + + assert mock_config_entry.state is ConfigEntryState.LOADED + assert len(mock_stookwijzer.return_value.async_update.mock_calls) == 1 + + await hass.config_entries.async_unload(mock_config_entry.entry_id) + await hass.async_block_till_done() + + assert mock_config_entry.state is ConfigEntryState.NOT_LOADED + + +async def test_config_entry_not_ready( + hass: HomeAssistant, + mock_config_entry: MockConfigEntry, + mock_stookwijzer: MagicMock, +) -> None: + """Test the Stookwijzer configuration entry loading and unloading.""" + mock_stookwijzer.return_value.advice = None + + mock_config_entry.add_to_hass(hass) + await hass.config_entries.async_setup(mock_config_entry.entry_id) + await hass.async_block_till_done() + + assert mock_config_entry.state is ConfigEntryState.SETUP_RETRY + assert len(mock_stookwijzer.return_value.async_update.mock_calls) == 1 + + async def test_migrate_entry( hass: HomeAssistant, mock_v1_config_entry: MockConfigEntry,