core/tests/helpers/test_intent.py

"""Tests for the intent helpers."""

import pytest
import voluptuous as vol

from homeassistant.core import State
from homeassistant.helpers import config_validation as cv, intent


class MockIntentHandler(intent.IntentHandler):
    """Provide a mock intent handler."""

    def __init__(self, slot_schema):
        """Initialize the mock handler."""
        self.slot_schema = slot_schema


def test_async_match_state():
    """Test async_match_state helper."""
    state1 = State("light.kitchen", "on")
    state2 = State("switch.kitchen", "on")

    state = intent.async_match_state(None, "kitch", [state1, state2])
    assert state is state1


def test_async_validate_slots():
    """Test async_validate_slots of IntentHandler."""
    handler1 = MockIntentHandler({vol.Required("name"): cv.string})

    with pytest.raises(vol.error.MultipleInvalid):
        handler1.async_validate_slots({})
    with pytest.raises(vol.error.MultipleInvalid):
        handler1.async_validate_slots({"name": 1})
    with pytest.raises(vol.error.MultipleInvalid):
        handler1.async_validate_slots({"name": "kitchen"})
    handler1.async_validate_slots({"name": {"value": "kitchen"}})
    handler1.async_validate_slots(
        {"name": {"value": "kitchen"}, "probability": {"value": "0.5"}}
    )


def test_fuzzy_match():
    """Test _fuzzymatch."""
    state1 = State("light.living_room_northwest", "off")
    state2 = State("light.living_room_north", "off")
    state3 = State("light.living_room_northeast", "off")
    state4 = State("light.living_room_west", "off")
    state5 = State("light.living_room", "off")
    states = [state1, state2, state3, state4, state5]

    state = intent._fuzzymatch("Living Room", states, lambda state: state.name)
    assert state == state5

    state = intent._fuzzymatch(
        "Living Room Northwest", states, lambda state: state.name
    )
    assert state == state1