a1bc929421
* Migrate tibber notify service * Tests and repair flow * Use notify repair flow helper * Cleanup strings after using helper, use HomeAssistantError * Add entry state assertions to unload test * Update comment * Update comment
21 lines
820 B
Python
21 lines
820 B
Python
"""Test loading of the Tibber config entry."""
|
|
|
|
from unittest.mock import MagicMock
|
|
|
|
from homeassistant.components.recorder import Recorder
|
|
from homeassistant.components.tibber import DOMAIN
|
|
from homeassistant.config_entries import ConfigEntryState
|
|
from homeassistant.core import HomeAssistant
|
|
|
|
|
|
async def test_entry_unload(
|
|
recorder_mock: Recorder, hass: HomeAssistant, mock_tibber_setup: MagicMock
|
|
) -> None:
|
|
"""Test unloading the entry."""
|
|
entry = hass.config_entries.async_entry_for_domain_unique_id(DOMAIN, "tibber")
|
|
assert entry.state == ConfigEntryState.LOADED
|
|
|
|
await hass.config_entries.async_unload(entry.entry_id)
|
|
mock_tibber_setup.rt_disconnect.assert_called_once()
|
|
await hass.async_block_till_done(wait_background_tasks=True)
|
|
assert entry.state == ConfigEntryState.NOT_LOADED
|