mirror of https://github.com/home-assistant/core
79 lines
2.5 KiB
Python
79 lines
2.5 KiB
Python
"""Test VoIP devices."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from voip_utils import CallInfo
|
|
|
|
from homeassistant.components.voip import DOMAIN
|
|
from homeassistant.components.voip.devices import VoIPDevice, VoIPDevices
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers import device_registry as dr
|
|
|
|
|
|
async def test_device_registry_info(
|
|
hass: HomeAssistant,
|
|
voip_devices: VoIPDevices,
|
|
call_info: CallInfo,
|
|
device_registry: dr.DeviceRegistry,
|
|
) -> None:
|
|
"""Test info in device registry."""
|
|
voip_device = voip_devices.async_get_or_create(call_info)
|
|
assert not voip_device.async_allow_call(hass)
|
|
|
|
device = device_registry.async_get_device(
|
|
identifiers={(DOMAIN, call_info.caller_ip)}
|
|
)
|
|
assert device is not None
|
|
assert device.name == call_info.caller_ip
|
|
assert device.manufacturer == "Grandstream"
|
|
assert device.model == "HT801"
|
|
assert device.sw_version == "1.0.17.5"
|
|
|
|
# Test we update the device if the fw updates
|
|
call_info.headers["user-agent"] = "Grandstream HT801 2.0.0.0"
|
|
voip_device = voip_devices.async_get_or_create(call_info)
|
|
|
|
assert not voip_device.async_allow_call(hass)
|
|
|
|
device = device_registry.async_get_device(
|
|
identifiers={(DOMAIN, call_info.caller_ip)}
|
|
)
|
|
assert device.sw_version == "2.0.0.0"
|
|
|
|
|
|
async def test_device_registry_info_from_unknown_phone(
|
|
hass: HomeAssistant,
|
|
voip_devices: VoIPDevices,
|
|
call_info: CallInfo,
|
|
device_registry: dr.DeviceRegistry,
|
|
) -> None:
|
|
"""Test info in device registry from unknown phone."""
|
|
call_info.headers["user-agent"] = "Unknown"
|
|
voip_device = voip_devices.async_get_or_create(call_info)
|
|
assert not voip_device.async_allow_call(hass)
|
|
|
|
device = device_registry.async_get_device(
|
|
identifiers={(DOMAIN, call_info.caller_ip)}
|
|
)
|
|
assert device.manufacturer is None
|
|
assert device.model == "Unknown"
|
|
assert device.sw_version is None
|
|
|
|
|
|
async def test_remove_device_registry_entry(
|
|
hass: HomeAssistant,
|
|
voip_device: VoIPDevice,
|
|
voip_devices: VoIPDevices,
|
|
device_registry: dr.DeviceRegistry,
|
|
) -> None:
|
|
"""Test removing a device registry entry."""
|
|
assert voip_device.voip_id in voip_devices.devices
|
|
assert hass.states.get("switch.192_168_1_210_allow_calls") is not None
|
|
|
|
device_registry.async_remove_device(voip_device.device_id)
|
|
await hass.async_block_till_done()
|
|
await hass.async_block_till_done()
|
|
|
|
assert hass.states.get("switch.192_168_1_210_allow_calls") is None
|
|
assert voip_device.voip_id not in voip_devices.devices
|