mirror of
https://github.com/home-assistant/core.git
synced 2026-05-08 17:49:37 +01:00
Erik Montnemery
109 lines
3.5 KiB
Python
109 lines
3.5 KiB
Python
"""Tests for the legacy device tracker component."""
|
|
|
|
from unittest.mock import mock_open, patch
|
|
|
|
import pytest
|
|
|
|
from homeassistant.components import device_tracker
|
|
from homeassistant.components.device_tracker import legacy
|
|
from homeassistant.const import CONF_PLATFORM
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.setup import async_setup_component
|
|
from homeassistant.util.yaml import dump
|
|
|
|
from .common import MockScanner, mock_legacy_device_tracker_setup
|
|
|
|
from tests.common import assert_setup_component, patch_yaml_files
|
|
|
|
TEST_PLATFORM = {device_tracker.DOMAIN: {CONF_PLATFORM: "test"}}
|
|
|
|
|
|
def test_remove_device_from_config(hass: HomeAssistant) -> None:
|
|
"""Test the removal of a device from a config."""
|
|
yaml_devices = {
|
|
"test": {
|
|
"hide_if_away": True,
|
|
"mac": "00:11:22:33:44:55",
|
|
"name": "Test name",
|
|
"picture": "/local/test.png",
|
|
"track": True,
|
|
},
|
|
"test2": {
|
|
"hide_if_away": True,
|
|
"mac": "00:ab:cd:33:44:55",
|
|
"name": "Test2",
|
|
"picture": "/local/test2.png",
|
|
"track": True,
|
|
},
|
|
}
|
|
mopen = mock_open()
|
|
|
|
files = {legacy.YAML_DEVICES: dump(yaml_devices)}
|
|
with (
|
|
patch_yaml_files(files, True),
|
|
patch("homeassistant.components.device_tracker.legacy.open", mopen),
|
|
):
|
|
legacy.remove_device_from_config(hass, "test")
|
|
|
|
mopen().write.assert_called_once_with(
|
|
"test2:\n"
|
|
" hide_if_away: true\n"
|
|
" mac: 00:ab:cd:33:44:55\n"
|
|
" name: Test2\n"
|
|
" picture: /local/test2.png\n"
|
|
" track: true\n"
|
|
)
|
|
|
|
|
|
async def test_see_service_deprecation_warning(
|
|
hass: HomeAssistant,
|
|
caplog: pytest.LogCaptureFixture,
|
|
) -> None:
|
|
"""Test deprecation warning when calling device_tracker.see."""
|
|
mock_legacy_device_tracker_setup(hass, MockScanner())
|
|
with assert_setup_component(1, device_tracker.DOMAIN):
|
|
assert await async_setup_component(hass, device_tracker.DOMAIN, TEST_PLATFORM)
|
|
await hass.async_block_till_done()
|
|
|
|
await hass.services.async_call(
|
|
device_tracker.DOMAIN,
|
|
legacy.SERVICE_SEE,
|
|
{"dev_id": "test_device", "location_name": "Work"},
|
|
blocking=True,
|
|
)
|
|
|
|
assert (
|
|
"The device_tracker.see action is deprecated and will be removed in "
|
|
"Home Assistant Core 2027.5"
|
|
) in caplog.text
|
|
|
|
caplog.clear()
|
|
|
|
# Second call should not produce another warning
|
|
await hass.services.async_call(
|
|
device_tracker.DOMAIN,
|
|
legacy.SERVICE_SEE,
|
|
{"dev_id": "test_device", "location_name": "Work"},
|
|
blocking=True,
|
|
)
|
|
|
|
assert "deprecated" not in caplog.text
|
|
|
|
|
|
async def test_legacy_platform_setup_deprecation_warning(
|
|
hass: HomeAssistant,
|
|
caplog: pytest.LogCaptureFixture,
|
|
) -> None:
|
|
"""Test deprecation warning when setting up a legacy device tracker platform."""
|
|
mock_legacy_device_tracker_setup(hass, MockScanner())
|
|
with assert_setup_component(1, device_tracker.DOMAIN):
|
|
assert await async_setup_component(hass, device_tracker.DOMAIN, TEST_PLATFORM)
|
|
await hass.async_block_till_done()
|
|
|
|
assert (
|
|
"The legacy device tracker platform test.device_tracker is being set up; "
|
|
"legacy device trackers are deprecated and will be removed in Home "
|
|
"Assistant Core 2027.5, please migrate to an integration which "
|
|
"uses a modern config entry based device tracker"
|
|
) in caplog.text
|