c063f14c24
* Add configuration flow for Buienradar integration * Update buienradar camera tests to work with config flow * Update buienradar weather tests to work with config flow * Update buienradar sensor tests to work with config flow * Remove buienradar config_flow tests to pass tests * Add config flow tests for buienradar integration * Increase test coverage for buienradar config_flow tests * Move data into domain * Remove forecast option * Move data to options * Remove options from config flow * Adjust tests * Adjust string * Fix pylint issues * Rework review comments * Handle import * Change config flow to setup camera or weather * Fix tests * Remove translated file * Fix pylint * Fix flake8 * Fix unload * Minor name changes * Update homeassistant/components/buienradar/config_flow.py Co-authored-by: Ties de Kock <ties@tiesdekock.nl> * Remove asynctest * Add translation * Disable sensors by default * Remove integration name from translations * Remove import method * Drop selection between platforms, disable camera by default * Minor fix in configured_instances * Bugfix in weather * Rework import * Change unique ids of camera * Fix in import * Fix camera tests * Fix sensor test * Fix sensor test 2 * Fix config flow tests * Add option flow * Add tests for option flow * Add import tests * Some cleanups * Apply suggestions from code review Apply code suggestions Co-authored-by: Franck Nijhof <git@frenck.dev> * Fix isort,black,mypy * Small tweaks and added typing to new parts * Fix review comments (1) * Apply suggestions from code review Co-authored-by: Martin Hjelmare <marhje52@gmail.com> * Fix review comments (2) * Fix issues * Fix unique id * Improve tests * Extend tests * Fix issue with unload * Address review comments * Add warning when loading platform * Add load/unload test Co-authored-by: Ties de Kock <ties@tiesdekock.nl> Co-authored-by: Franck Nijhof <git@frenck.dev> Co-authored-by: Martin Hjelmare <marhje52@gmail.com>
120 lines
3.2 KiB
Python
120 lines
3.2 KiB
Python
"""Tests for the buienradar component."""
|
|
from unittest.mock import patch
|
|
|
|
from homeassistant.components.buienradar import async_setup
|
|
from homeassistant.components.buienradar.const import DOMAIN
|
|
from homeassistant.const import CONF_LATITUDE, CONF_LONGITUDE
|
|
from homeassistant.helpers.entity_registry import async_get_registry
|
|
|
|
from tests.common import MockConfigEntry
|
|
|
|
TEST_LATITUDE = 51.5288504
|
|
TEST_LONGITUDE = 5.4002156
|
|
|
|
|
|
async def test_import_all(hass):
|
|
"""Test import of all platforms."""
|
|
config = {
|
|
"weather 1": [{"platform": "buienradar", "name": "test1"}],
|
|
"sensor 1": [{"platform": "buienradar", "timeframe": 30, "name": "test2"}],
|
|
"camera 1": [
|
|
{
|
|
"platform": "buienradar",
|
|
"country_code": "BE",
|
|
"delta": 300,
|
|
"name": "test3",
|
|
}
|
|
],
|
|
}
|
|
|
|
with patch(
|
|
"homeassistant.components.buienradar.async_setup_entry", return_value=True
|
|
):
|
|
await async_setup(hass, config)
|
|
await hass.async_block_till_done()
|
|
|
|
conf_entries = hass.config_entries.async_entries(DOMAIN)
|
|
|
|
assert len(conf_entries) == 1
|
|
|
|
entry = conf_entries[0]
|
|
|
|
assert entry.state == "loaded"
|
|
assert entry.data == {
|
|
"latitude": hass.config.latitude,
|
|
"longitude": hass.config.longitude,
|
|
"timeframe": 30,
|
|
"country_code": "BE",
|
|
"delta": 300,
|
|
"name": "test2",
|
|
}
|
|
|
|
|
|
async def test_import_camera(hass):
|
|
"""Test import of camera platform."""
|
|
entity_registry = await async_get_registry(hass)
|
|
entity_registry.async_get_or_create(
|
|
domain="camera",
|
|
platform="buienradar",
|
|
unique_id="512_NL",
|
|
original_name="test_name",
|
|
)
|
|
await hass.async_block_till_done()
|
|
|
|
config = {
|
|
"camera 1": [{"platform": "buienradar", "country_code": "NL", "dimension": 512}]
|
|
}
|
|
|
|
with patch(
|
|
"homeassistant.components.buienradar.async_setup_entry", return_value=True
|
|
):
|
|
await async_setup(hass, config)
|
|
await hass.async_block_till_done()
|
|
|
|
conf_entries = hass.config_entries.async_entries(DOMAIN)
|
|
|
|
assert len(conf_entries) == 1
|
|
|
|
entry = conf_entries[0]
|
|
|
|
assert entry.state == "loaded"
|
|
assert entry.data == {
|
|
"latitude": hass.config.latitude,
|
|
"longitude": hass.config.longitude,
|
|
"timeframe": 60,
|
|
"country_code": "NL",
|
|
"delta": 600,
|
|
"name": "Buienradar",
|
|
}
|
|
|
|
entity_id = entity_registry.async_get_entity_id(
|
|
"camera",
|
|
"buienradar",
|
|
f"{hass.config.latitude:2.6f}{hass.config.longitude:2.6f}",
|
|
)
|
|
assert entity_id
|
|
entity = entity_registry.async_get(entity_id)
|
|
assert entity.original_name == "test_name"
|
|
|
|
|
|
async def test_load_unload(hass):
|
|
"""Test options flow."""
|
|
entry = MockConfigEntry(
|
|
domain=DOMAIN,
|
|
data={
|
|
CONF_LATITUDE: TEST_LATITUDE,
|
|
CONF_LONGITUDE: TEST_LONGITUDE,
|
|
},
|
|
unique_id=DOMAIN,
|
|
)
|
|
entry.add_to_hass(hass)
|
|
|
|
await hass.config_entries.async_setup(entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
assert entry.state == "loaded"
|
|
|
|
await hass.config_entries.async_unload(entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
assert entry.state == "not_loaded"
|