Store runtime data inside the config entry in IPP (#116765)

* store runtime data inside the config entry

* improve tests
This commit is contained in:
Michael
2024-05-08 08:16:06 +02:00
committed by GitHub
parent 3844e2d533
commit 2891a63281
4 changed files with 14 additions and 19 deletions

View File

@@ -4,7 +4,7 @@ from unittest.mock import AsyncMock, MagicMock, patch
from pyipp import IPPConnectionError
from homeassistant.components.ipp.const import DOMAIN
from homeassistant.components.ipp.coordinator import IPPDataUpdateCoordinator
from homeassistant.config_entries import ConfigEntryState
from homeassistant.core import HomeAssistant
@@ -37,10 +37,9 @@ async def test_load_unload_config_entry(
await hass.config_entries.async_setup(mock_config_entry.entry_id)
await hass.async_block_till_done()
assert mock_config_entry.entry_id in hass.data[DOMAIN]
assert mock_config_entry.state is ConfigEntryState.LOADED
assert isinstance(mock_config_entry.runtime_data, IPPDataUpdateCoordinator)
await hass.config_entries.async_unload(mock_config_entry.entry_id)
await hass.async_block_till_done()
assert mock_config_entry.entry_id not in hass.data[DOMAIN]
assert mock_config_entry.state is ConfigEntryState.NOT_LOADED