9a79320861
* Mark executor jobs as background unless created from a tracked task If the current task is not tracked the executor job should not be a background task to avoid delaying startup and shutdown. Currently any executor job created in a untracked task or background task would end up being tracked and delaying startup/shutdown * import exec has the same issue * Avoid tracking import executor jobs There is no reason to track these jobs as they are always awaited and we do not want to support fire and forget import executor jobs * fix xiaomi_miio * lots of fire time changed without background await * revert changes moved to other PR * more * more * more * m * m * p * fix fire and forget tests * scrape * sonos * system * more * capture callback before block * coverage * more * more races * more races * more * missed some * more fixes * missed some more * fix * remove unneeded * one more race * two
104 lines
3 KiB
Python
104 lines
3 KiB
Python
"""Tests for SpeedTest integration."""
|
|
|
|
from datetime import timedelta
|
|
from unittest.mock import MagicMock
|
|
|
|
import speedtest
|
|
|
|
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
|
|
import homeassistant.util.dt as dt_util
|
|
|
|
from tests.common import MockConfigEntry, async_fire_time_changed
|
|
|
|
|
|
async def test_setup_failed(hass: HomeAssistant, mock_api: MagicMock) -> None:
|
|
"""Test SpeedTestDotNet failed due to an error."""
|
|
|
|
entry = MockConfigEntry(
|
|
domain=DOMAIN,
|
|
)
|
|
entry.add_to_hass(hass)
|
|
|
|
mock_api.side_effect = speedtest.ConfigRetrievalError
|
|
await hass.config_entries.async_setup(entry.entry_id)
|
|
assert entry.state is ConfigEntryState.SETUP_RETRY
|
|
|
|
|
|
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()
|
|
|
|
assert entry.state is ConfigEntryState.NOT_LOADED
|
|
assert DOMAIN not in hass.data
|
|
|
|
|
|
async def test_server_not_found(hass: HomeAssistant, mock_api: MagicMock) -> None:
|
|
"""Test configured server id is not found."""
|
|
|
|
entry = MockConfigEntry(
|
|
domain=DOMAIN,
|
|
options={},
|
|
)
|
|
entry.add_to_hass(hass)
|
|
|
|
await hass.config_entries.async_setup(entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
assert hass.data[DOMAIN]
|
|
|
|
mock_api.return_value.get_servers.side_effect = speedtest.NoMatchedServers
|
|
async_fire_time_changed(
|
|
hass,
|
|
dt_util.utcnow() + timedelta(minutes=61),
|
|
)
|
|
await hass.async_block_till_done(wait_background_tasks=True)
|
|
state = hass.states.get("sensor.speedtest_ping")
|
|
assert state is not None
|
|
assert state.state == STATE_UNAVAILABLE
|
|
|
|
|
|
async def test_get_best_server_error(hass: HomeAssistant, mock_api: MagicMock) -> None:
|
|
"""Test configured server id is not found."""
|
|
|
|
entry = MockConfigEntry(
|
|
domain=DOMAIN,
|
|
)
|
|
entry.add_to_hass(hass)
|
|
|
|
await hass.config_entries.async_setup(entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
assert hass.data[DOMAIN]
|
|
|
|
mock_api.return_value.get_best_server.side_effect = (
|
|
speedtest.SpeedtestBestServerFailure(
|
|
"Unable to connect to servers to test latency."
|
|
)
|
|
)
|
|
await hass.data[DOMAIN].async_refresh()
|
|
await hass.async_block_till_done()
|
|
state = hass.states.get("sensor.speedtest_ping")
|
|
assert state is not None
|
|
assert state.state == STATE_UNAVAILABLE
|