Add config flow and device registry to fritzbox integration (#31240)

* add config flow * fix pylint * update lib * Update config_flow.py * remote devices layer in config * add default host * avoid double setups of entities * remove async_setup_platform * store entities in hass.data * pass fritz connection together with config_entry * fritz connections try no4 (or is it even more) * fix comments * add unloading * fixed comments * Update config_flow.py * Update const.py * Update config_flow.py * Update __init__.py * Update config_flow.py * Update __init__.py * Update __init__.py * Update config_flow.py * Update __init__.py * Update __init__.py * Update __init__.py * Update config_flow.py * add init tests * test unloading * add switch tests * add sensor tests * add climate tests * test target temperature * mock config to package * comments * test binary sensor state * add config flow tests * comments * add missing tests * minor * remove string title * deprecate yaml * don't change yaml * get devices async * minor * add devices again * comments fixed * unique_id fixes * fix patches * Fix schema Co-authored-by: Martin Hjelmare <marhje52@gmail.com>
2020-04-20 15:00:07 +02:00 · 2020-04-20 15:00:07 +02:00 · c87ecf0ff6
commit c87ecf0ff6
parent 2123f6f133
22 changed files with 1403 additions and 271 deletions
--- a/tests/components/fritzbox/test_init.py
+++ b/tests/components/fritzbox/test_init.py
@ -0,0 +1,76 @@
+"""Tests for the AVM Fritz!Box integration."""
+from unittest.mock import Mock, call
+
+from homeassistant.components.fritzbox.const import DOMAIN as FB_DOMAIN
+from homeassistant.components.switch import DOMAIN as SWITCH_DOMAIN
+from homeassistant.config_entries import ENTRY_STATE_LOADED, ENTRY_STATE_NOT_LOADED
+from homeassistant.const import CONF_DEVICES, CONF_HOST, CONF_PASSWORD, CONF_USERNAME
+from homeassistant.helpers.typing import HomeAssistantType
+from homeassistant.setup import async_setup_component
+
+from . import MOCK_CONFIG, FritzDeviceSwitchMock
+
+from tests.common import MockConfigEntry
+
+
+async def test_setup(hass: HomeAssistantType, fritz: Mock):
+    """Test setup of integration."""
+    assert await async_setup_component(hass, FB_DOMAIN, MOCK_CONFIG)
+    await hass.async_block_till_done()
+    entries = hass.config_entries.async_entries()
+    assert entries
+    assert entries[0].data[CONF_HOST] == "fake_host"
+    assert entries[0].data[CONF_PASSWORD] == "fake_pass"
+    assert entries[0].data[CONF_USERNAME] == "fake_user"
+    assert fritz.call_count == 1
+    assert fritz.call_args_list == [
+        call(host="fake_host", password="fake_pass", user="fake_user")
+    ]
+
+
+async def test_setup_duplicate_config(hass: HomeAssistantType, fritz: Mock, caplog):
+    """Test duplicate config of integration."""
+    DUPLICATE = {
+        FB_DOMAIN: {
+            CONF_DEVICES: [
+                MOCK_CONFIG[FB_DOMAIN][CONF_DEVICES][0],
+                MOCK_CONFIG[FB_DOMAIN][CONF_DEVICES][0],
+            ]
+        }
+    }
+    assert not await async_setup_component(hass, FB_DOMAIN, DUPLICATE)
+    await hass.async_block_till_done()
+    assert not hass.states.async_entity_ids()
+    assert not hass.states.async_all()
+    assert "duplicate host entries found" in caplog.text
+
+
+async def test_unload(hass: HomeAssistantType, fritz: Mock):
+    """Test unload of integration."""
+    fritz().get_devices.return_value = [FritzDeviceSwitchMock()]
+    entity_id = f"{SWITCH_DOMAIN}.fake_name"
+
+    entry = MockConfigEntry(
+        domain=FB_DOMAIN,
+        data=MOCK_CONFIG[FB_DOMAIN][CONF_DEVICES][0],
+        unique_id=entity_id,
+    )
+    entry.add_to_hass(hass)
+
+    config_entries = hass.config_entries.async_entries(FB_DOMAIN)
+    assert len(config_entries) == 1
+    assert entry is config_entries[0]
+
+    assert await async_setup_component(hass, FB_DOMAIN, {}) is True
+    await hass.async_block_till_done()
+
+    assert entry.state == ENTRY_STATE_LOADED
+    state = hass.states.get(entity_id)
+    assert state
+
+    await hass.config_entries.async_unload(entry.entry_id)
+
+    assert fritz().logout.call_count == 1
+    assert entry.state == ENTRY_STATE_NOT_LOADED
+    state = hass.states.get(entity_id)
+    assert state is None