jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-13 08:17:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add device tracker test for Vodafone Station (#134334)

Browse Source
This commit is contained in:
Simone Chemelli 2025-01-06 17:17:50 -05:00 committed by GitHub
parent 89c73f56b1
commit 111ef13a3f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 53 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tests/components/vodafone_station/const.py
View File

@ -19,18 +19,19 @@ MOCK_CONFIG = {
MOCK_USER_DATA = MOCK_CONFIG[DOMAIN][CONF_DEVICES][0] MOCK_USER_DATA = MOCK_CONFIG[DOMAIN][CONF_DEVICES][0]
DEVICE_1_MAC = "xx:xx:xx:xx:xx:xx"
DEVICE_DATA_QUERY = { DEVICE_1 = {
"xx:xx:xx:xx:xx:xx": VodafoneStationDevice( DEVICE_1_MAC: VodafoneStationDevice(
connected=True, connected=True,
connection_type="wifi", connection_type="wifi",
ip_address="192.168.1.10", ip_address="192.168.1.10",
name="WifiDevice0", name="WifiDevice0",
mac="xx:xx:xx:xx:xx:xx", mac=DEVICE_1_MAC,
type="laptop", type="laptop",
wifi="2.4G", wifi="2.4G",
), ),
} }
DEVICE_DATA_QUERY = DEVICE_1
SERIAL = "m123456789" SERIAL = "m123456789"

48
tests/components/vodafone_station/test_device_tracker.py Normal file
View File

@ -0,0 +1,48 @@
"""Define tests for the Vodafone Station device tracker."""
from unittest.mock import AsyncMock
from freezegun.api import FrozenDateTimeFactory
import pytest
from homeassistant.components.vodafone_station.const import DOMAIN, SCAN_INTERVAL
from homeassistant.components.vodafone_station.coordinator import CONSIDER_HOME_SECONDS
from homeassistant.const import STATE_HOME, STATE_NOT_HOME
from homeassistant.core import HomeAssistant
from .const import DEVICE_1, DEVICE_1_MAC, DEVICE_DATA_QUERY, MOCK_USER_DATA
from tests.common import MockConfigEntry, async_fire_time_changed
@pytest.mark.usefixtures("entity_registry_enabled_by_default")
async def test_coordinator_consider_home(
hass: HomeAssistant,
freezer: FrozenDateTimeFactory,
mock_vodafone_station_router: AsyncMock,
) -> None:
"""Test if device is considered not_home when disconnected."""
entry = MockConfigEntry(domain=DOMAIN, data=MOCK_USER_DATA)
entry.add_to_hass(hass)
device_tracker = f"device_tracker.vodafone_station_{DEVICE_1_MAC.replace(":", "_")}"
await hass.config_entries.async_setup(entry.entry_id)
await hass.async_block_till_done()
state = hass.states.get(device_tracker)
assert state
assert state.state == STATE_HOME
DEVICE_1[DEVICE_1_MAC].connected = False
DEVICE_DATA_QUERY.update(DEVICE_1)
mock_vodafone_station_router.get_devices_data.return_value = DEVICE_DATA_QUERY
freezer.tick(SCAN_INTERVAL + CONSIDER_HOME_SECONDS)
async_fire_time_changed(hass)
await hass.async_block_till_done(wait_background_tasks=True)
state = hass.states.get(device_tracker)
assert state
assert state.state == STATE_NOT_HOME