hass-core/tests/components/fully_kiosk/test_switch.py

"""Test the Fully Kiosk Browser switches."""

from unittest.mock import MagicMock

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
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"


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


def call_service(hass, service, entity_id):
    """Call any service on entity."""
    return hass.services.async_call(
        switch.DOMAIN, service, {ATTR_ENTITY_ID: entity_id}, blocking=True
    )