evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Test speedtest.net config entry lifecycle (#100280)

Browse source
This commit is contained in:
Joost Lekkerkerker 2023-09-13 12:40:26 +02:00 • committed by GitHub
parent aedd06b9a9
commit 29d8be510e
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 15 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

18
tests/components/speedtestdotnet/test_init.py
View file

@ -5,7 +5,11 @@ from unittest.mock import MagicMock
import speedtest
from homeassistant.components.speedtestdotnet.const import DOMAIN
from homeassistant.components.speedtestdotnet.const import (
CONF_SERVER_ID,
CONF_SERVER_NAME,
DOMAIN,
)
from homeassistant.config_entries import ConfigEntryState
from homeassistant.const import STATE_UNAVAILABLE
from homeassistant.core import HomeAssistant
@ -27,16 +31,24 @@ async def test_setup_failed(hass: HomeAssistant, mock_api: MagicMock) -> None:
assert entry.state is ConfigEntryState.SETUP_RETRY
async def test_unload_entry(hass: HomeAssistant) -> None:
"""Test removing SpeedTestDotNet."""
async def test_entry_lifecycle(hass: HomeAssistant, mock_api: MagicMock) -> None:
"""Test the SpeedTestDotNet entry lifecycle."""
entry = MockConfigEntry(
domain=DOMAIN,
data={},
options={
CONF_SERVER_NAME: "Country1 - Sponsor1 - Server1",
CONF_SERVER_ID: "1",
},
)
entry.add_to_hass(hass)
await hass.config_entries.async_setup(entry.entry_id)
await hass.async_block_till_done()
assert entry.state == ConfigEntryState.LOADED
assert hass.data[DOMAIN]
assert await hass.config_entries.async_unload(entry.entry_id)
await hass.async_block_till_done()