mirror of
https://github.com/home-assistant/core.git
synced 2026-10-01 10:03:26 +01:00
176 lines
6.8 KiB
Python
176 lines
6.8 KiB
Python
"""Test the Fully Kiosk Browser switches."""
|
|
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
import pytest
|
|
|
|
from homeassistant.components import switch
|
|
from homeassistant.components.fully_kiosk.const import DOMAIN
|
|
from homeassistant.const import ATTR_ENTITY_ID
|
|
from homeassistant.core import HomeAssistant, ServiceResponse
|
|
from homeassistant.helpers import device_registry as dr, entity_registry as er
|
|
|
|
from tests.common import MockConfigEntry, async_fire_mqtt_message
|
|
from tests.typing import MqttMockHAClient
|
|
|
|
|
|
async def test_switches(
|
|
hass: HomeAssistant,
|
|
entity_registry: er.EntityRegistry,
|
|
device_registry: dr.DeviceRegistry,
|
|
mock_fully_kiosk: MagicMock,
|
|
init_integration: MockConfigEntry,
|
|
) -> None:
|
|
"""Test Fully Kiosk switches."""
|
|
entity = hass.states.get("switch.amazon_fire_screensaver")
|
|
assert entity
|
|
assert entity.state == "off"
|
|
entry = entity_registry.async_get("switch.amazon_fire_screensaver")
|
|
assert entry
|
|
assert entry.unique_id == "abcdef-123456-screensaver"
|
|
await call_service(hass, "turn_on", "switch.amazon_fire_screensaver")
|
|
assert len(mock_fully_kiosk.startScreensaver.mock_calls) == 1
|
|
await call_service(hass, "turn_off", "switch.amazon_fire_screensaver")
|
|
assert len(mock_fully_kiosk.stopScreensaver.mock_calls) == 1
|
|
|
|
entity = hass.states.get("switch.amazon_fire_maintenance_mode")
|
|
assert entity
|
|
assert entity.state == "off"
|
|
entry = entity_registry.async_get("switch.amazon_fire_maintenance_mode")
|
|
assert entry
|
|
assert entry.unique_id == "abcdef-123456-maintenance"
|
|
await call_service(hass, "turn_on", "switch.amazon_fire_maintenance_mode")
|
|
assert len(mock_fully_kiosk.enableLockedMode.mock_calls) == 1
|
|
await call_service(hass, "turn_off", "switch.amazon_fire_maintenance_mode")
|
|
assert len(mock_fully_kiosk.disableLockedMode.mock_calls) == 1
|
|
|
|
entity = hass.states.get("switch.amazon_fire_kiosk_lock")
|
|
assert entity
|
|
assert entity.state == "on"
|
|
entry = entity_registry.async_get("switch.amazon_fire_kiosk_lock")
|
|
assert entry
|
|
assert entry.unique_id == "abcdef-123456-kiosk"
|
|
await call_service(hass, "turn_off", "switch.amazon_fire_kiosk_lock")
|
|
assert len(mock_fully_kiosk.unlockKiosk.mock_calls) == 1
|
|
await call_service(hass, "turn_on", "switch.amazon_fire_kiosk_lock")
|
|
assert len(mock_fully_kiosk.lockKiosk.mock_calls) == 1
|
|
|
|
entity = hass.states.get("switch.amazon_fire_motion_detection")
|
|
assert entity
|
|
assert entity.state == "off"
|
|
entry = entity_registry.async_get("switch.amazon_fire_motion_detection")
|
|
assert entry
|
|
assert entry.unique_id == "abcdef-123456-motion-detection"
|
|
await call_service(hass, "turn_on", "switch.amazon_fire_motion_detection")
|
|
assert len(mock_fully_kiosk.enableMotionDetection.mock_calls) == 1
|
|
await call_service(hass, "turn_off", "switch.amazon_fire_motion_detection")
|
|
assert len(mock_fully_kiosk.disableMotionDetection.mock_calls) == 1
|
|
|
|
entity = hass.states.get("switch.amazon_fire_screen")
|
|
assert entity
|
|
assert entity.state == "on"
|
|
entry = entity_registry.async_get("switch.amazon_fire_screen")
|
|
assert entry
|
|
assert entry.unique_id == "abcdef-123456-screenOn"
|
|
await call_service(hass, "turn_off", "switch.amazon_fire_screen")
|
|
assert len(mock_fully_kiosk.screenOff.mock_calls) == 1
|
|
await call_service(hass, "turn_on", "switch.amazon_fire_screen")
|
|
assert len(mock_fully_kiosk.screenOn.mock_calls) == 1
|
|
|
|
assert entry.device_id
|
|
device_entry = device_registry.async_get(entry.device_id)
|
|
assert device_entry
|
|
assert device_entry.configuration_url == "http://192.168.1.234:2323"
|
|
assert device_entry.entry_type is None
|
|
assert device_entry.hw_version is None
|
|
assert device_entry.identifiers == {(DOMAIN, "abcdef-123456")}
|
|
assert device_entry.manufacturer == "amzn"
|
|
assert device_entry.model == "KFDOWI"
|
|
assert device_entry.name == "Amazon Fire"
|
|
assert device_entry.sw_version == "1.42.5"
|
|
|
|
|
|
async def test_switches_mqtt_update(
|
|
hass: HomeAssistant,
|
|
mock_fully_kiosk: MagicMock,
|
|
mqtt_mock: MqttMockHAClient,
|
|
init_integration: MockConfigEntry,
|
|
) -> None:
|
|
"""Test push updates over MQTT."""
|
|
assert has_subscribed(mqtt_mock, "fully/event/onScreensaverStart/abcdef-123456")
|
|
assert has_subscribed(mqtt_mock, "fully/event/onScreensaverStop/abcdef-123456")
|
|
assert has_subscribed(mqtt_mock, "fully/event/screenOff/abcdef-123456")
|
|
assert has_subscribed(mqtt_mock, "fully/event/screenOn/abcdef-123456")
|
|
|
|
entity = hass.states.get("switch.amazon_fire_screensaver")
|
|
assert entity
|
|
assert entity.state == "off"
|
|
|
|
entity = hass.states.get("switch.amazon_fire_screen")
|
|
assert entity
|
|
assert entity.state == "on"
|
|
|
|
async_fire_mqtt_message(
|
|
hass,
|
|
"fully/event/onScreensaverStart/abcdef-123456",
|
|
'{"deviceId": "abcdef-123456","event": "onScreensaverStart"}',
|
|
)
|
|
entity = hass.states.get("switch.amazon_fire_screensaver")
|
|
assert entity.state == "on"
|
|
|
|
async_fire_mqtt_message(
|
|
hass,
|
|
"fully/event/onScreensaverStop/abcdef-123456",
|
|
'{"deviceId": "abcdef-123456","event": "onScreensaverStop"}',
|
|
)
|
|
entity = hass.states.get("switch.amazon_fire_screensaver")
|
|
assert entity.state == "off"
|
|
|
|
async_fire_mqtt_message(
|
|
hass,
|
|
"fully/event/screenOff/abcdef-123456",
|
|
'{"deviceId": "abcdef-123456","event": "screenOff"}',
|
|
)
|
|
entity = hass.states.get("switch.amazon_fire_screen")
|
|
assert entity.state == "off"
|
|
|
|
async_fire_mqtt_message(
|
|
hass,
|
|
"fully/event/screenOn/abcdef-123456",
|
|
'{"deviceId": "abcdef-123456","event": "screenOn"}',
|
|
)
|
|
entity = hass.states.get("switch.amazon_fire_screen")
|
|
assert entity.state == "on"
|
|
|
|
|
|
@pytest.mark.usefixtures("mock_fully_kiosk")
|
|
async def test_switches_mqtt_config_entry_not_enabled(
|
|
hass: HomeAssistant, mock_config_entry: MockConfigEntry
|
|
) -> None:
|
|
"""Test no MQTT subscriptions are made without an enabled MQTT config entry."""
|
|
hass.config.components.add("mqtt")
|
|
mock_config_entry.add_to_hass(hass)
|
|
with patch("homeassistant.components.mqtt.async_subscribe") as mock_subscribe:
|
|
await hass.config_entries.async_setup(mock_config_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
mock_subscribe.assert_not_called()
|
|
assert hass.states.get("switch.amazon_fire_screensaver")
|
|
|
|
|
|
def has_subscribed(mqtt_mock: MqttMockHAClient, topic: str) -> bool:
|
|
"""Check if MQTT topic has subscription."""
|
|
for call in mqtt_mock.async_subscribe.call_args_list:
|
|
if call.args[0] == topic:
|
|
return True
|
|
return False
|
|
|
|
|
|
async def call_service(
|
|
hass: HomeAssistant, service: str, entity_id: str
|
|
) -> ServiceResponse:
|
|
"""Call any service on entity."""
|
|
return await hass.services.async_call(
|
|
switch.DOMAIN, service, {ATTR_ENTITY_ID: entity_id}, blocking=True
|
|
)
|