* Mark entities as unavailable when they are removed but are still registered * Add sync_entity_lifecycle to collection helper * Remove debug print * Lint * Fix tests * Fix tests * Update zha * Update zone * Fix tests * Update hyperion * Update rfxtrx * Fix tests * Pass force_remove=True from integrations Co-authored-by: Erik <erik@montnemery.com>
37 lines
1.3 KiB
Python
37 lines
1.3 KiB
Python
"""Tests for init module."""
|
|
from homeassistant.components.nws.const import DOMAIN
|
|
from homeassistant.components.weather import DOMAIN as WEATHER_DOMAIN
|
|
from homeassistant.const import STATE_UNAVAILABLE
|
|
|
|
from tests.common import MockConfigEntry
|
|
from tests.components.nws.const import NWS_CONFIG
|
|
|
|
|
|
async def test_unload_entry(hass, mock_simple_nws):
|
|
"""Test that nws setup with config yaml."""
|
|
entry = MockConfigEntry(
|
|
domain=DOMAIN,
|
|
data=NWS_CONFIG,
|
|
)
|
|
entry.add_to_hass(hass)
|
|
|
|
await hass.config_entries.async_setup(entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
assert len(hass.states.async_entity_ids(WEATHER_DOMAIN)) == 1
|
|
assert DOMAIN in hass.data
|
|
|
|
assert len(hass.data[DOMAIN]) == 1
|
|
entries = hass.config_entries.async_entries(DOMAIN)
|
|
assert len(entries) == 1
|
|
|
|
assert await hass.config_entries.async_unload(entries[0].entry_id)
|
|
entities = hass.states.async_entity_ids(WEATHER_DOMAIN)
|
|
assert len(entities) == 1
|
|
for entity in entities:
|
|
assert hass.states.get(entity).state == STATE_UNAVAILABLE
|
|
assert DOMAIN not in hass.data
|
|
|
|
assert await hass.config_entries.async_remove(entries[0].entry_id)
|
|
await hass.async_block_till_done()
|
|
assert len(hass.states.async_entity_ids(WEATHER_DOMAIN)) == 0
|