GeoNet NZ Quakes code improvements (#32338)
* code quality improvements * code quality improvements and fixed tests * explicitly set unique ids * improve unique id creation * remove entities from entity registry * added test for removing entities from entity registry * revert entity registry handling from sensor and test code * check for entity registry removal in geolocation test case * make import absolute; isort * change quality scale
This commit is contained in:
parent
dd7d8d4792
commit
2abdfc9da6
11 changed files with 152 additions and 147 deletions
21
tests/components/geonetnz_quakes/test_init.py
Normal file
21
tests/components/geonetnz_quakes/test_init.py
Normal file
|
@ -0,0 +1,21 @@
|
|||
"""Define tests for the GeoNet NZ Quakes general setup."""
|
||||
from asynctest import patch
|
||||
|
||||
from homeassistant.components.geonetnz_quakes import DOMAIN, FEED
|
||||
|
||||
|
||||
async def test_component_unload_config_entry(hass, config_entry):
|
||||
"""Test that loading and unloading of a config entry works."""
|
||||
config_entry.add_to_hass(hass)
|
||||
with patch(
|
||||
"aio_geojson_geonetnz_quakes.GeonetnzQuakesFeedManager.update"
|
||||
) as mock_feed_manager_update:
|
||||
# Load config entry.
|
||||
assert await hass.config_entries.async_setup(config_entry.entry_id)
|
||||
await hass.async_block_till_done()
|
||||
assert mock_feed_manager_update.call_count == 1
|
||||
assert hass.data[DOMAIN][FEED][config_entry.entry_id] is not None
|
||||
# Unload config entry.
|
||||
assert await hass.config_entries.async_unload(config_entry.entry_id)
|
||||
await hass.async_block_till_done()
|
||||
assert hass.data[DOMAIN][FEED].get(config_entry.entry_id) is None
|
Loading…
Add table
Add a link
Reference in a new issue