mirror of
https://github.com/home-assistant/core.git
synced 2025-04-23 16:57:53 +00:00
Local ip tests (#82051)
This commit is contained in:
parent
3d29638804
commit
13577981f9
@ -1,12 +1,15 @@
|
||||
"""Tests for the local_ip config_flow."""
|
||||
from __future__ import annotations
|
||||
|
||||
from homeassistant import data_entry_flow
|
||||
from homeassistant.components.local_ip.const import DOMAIN
|
||||
from homeassistant.config_entries import SOURCE_USER
|
||||
from homeassistant.core import HomeAssistant
|
||||
|
||||
from tests.common import MockConfigEntry
|
||||
|
||||
|
||||
async def test_config_flow(hass, mock_get_source_ip):
|
||||
async def test_config_flow(hass: HomeAssistant, mock_get_source_ip) -> None:
|
||||
"""Test we can finish a config flow."""
|
||||
result = await hass.config_entries.flow.async_init(
|
||||
DOMAIN, context={"source": SOURCE_USER}
|
||||
@ -21,7 +24,7 @@ async def test_config_flow(hass, mock_get_source_ip):
|
||||
assert state
|
||||
|
||||
|
||||
async def test_already_setup(hass, mock_get_source_ip):
|
||||
async def test_already_setup(hass: HomeAssistant, mock_get_source_ip) -> None:
|
||||
"""Test we abort if already setup."""
|
||||
MockConfigEntry(
|
||||
domain=DOMAIN,
|
||||
|
@ -1,20 +1,29 @@
|
||||
"""Tests for the local_ip component."""
|
||||
from __future__ import annotations
|
||||
|
||||
from homeassistant import config_entries
|
||||
from homeassistant.components.local_ip import DOMAIN
|
||||
from homeassistant.components.network import async_get_source_ip
|
||||
from homeassistant.components.zeroconf import MDNS_TARGET_IP
|
||||
from homeassistant.core import HomeAssistant
|
||||
|
||||
from tests.common import MockConfigEntry
|
||||
|
||||
|
||||
async def test_basic_setup(hass, mock_get_source_ip):
|
||||
async def test_basic_setup(hass: HomeAssistant, mock_get_source_ip) -> None:
|
||||
"""Test component setup creates entry from config."""
|
||||
entry = MockConfigEntry(domain=DOMAIN, data={})
|
||||
entry.add_to_hass(hass)
|
||||
|
||||
await hass.config_entries.async_setup(entry.entry_id)
|
||||
await hass.async_block_till_done()
|
||||
assert entry.state == config_entries.ConfigEntryState.LOADED
|
||||
|
||||
local_ip = await async_get_source_ip(hass, target_ip=MDNS_TARGET_IP)
|
||||
state = hass.states.get(f"sensor.{DOMAIN}")
|
||||
assert state
|
||||
assert state.state == local_ip
|
||||
|
||||
assert await hass.config_entries.async_unload(entry.entry_id)
|
||||
await hass.async_block_till_done()
|
||||
assert entry.state is config_entries.ConfigEntryState.NOT_LOADED
|
||||
|
Loading…
x
Reference in New Issue
Block a user