ha-core/tests/helpers/test_entity_values.py

"""Test the entity values helper."""
from collections import OrderedDict
from homeassistant.helpers.entity_values import EntityValues as EV

ent = 'test.test'


def test_override_single_value():
    """Test values with exact match."""
    store = EV({ent: {'key': 'value'}})
    assert store.get(ent) == {'key': 'value'}
    assert len(store._cache) == 1
    assert store.get(ent) == {'key': 'value'}
    assert len(store._cache) == 1


def test_override_by_domain():
    """Test values with domain match."""
    store = EV(domain={'test': {'key': 'value'}})
    assert store.get(ent) == {'key': 'value'}


def test_override_by_glob():
    """Test values with glob match."""
    store = EV(glob={'test.?e*': {'key': 'value'}})
    assert store.get(ent) == {'key': 'value'}


def test_glob_overrules_domain():
    """Test domain overrules glob match."""
    store = EV(
        domain={'test': {'key': 'domain'}},
        glob={'test.?e*': {'key': 'glob'}})
    assert store.get(ent) == {'key': 'glob'}


def test_exact_overrules_domain():
    """Test exact overrules domain match."""
    store = EV(
        exact={'test.test': {'key': 'exact'}},
        domain={'test': {'key': 'domain'}},
        glob={'test.?e*': {'key': 'glob'}})
    assert store.get(ent) == {'key': 'exact'}


def test_merging_values():
    """Test merging glob, domain and exact configs."""
    store = EV(
        exact={'test.test': {'exact_key': 'exact'}},
        domain={'test': {'domain_key': 'domain'}},
        glob={'test.?e*': {'glob_key': 'glob'}})
    assert store.get(ent) == {
        'exact_key': 'exact',
        'domain_key': 'domain',
        'glob_key': 'glob',
    }


def test_glob_order():
    """Test merging glob, domain and exact configs."""
    glob = OrderedDict()
    glob['test.*est'] = {"value": "first"}
    glob['test.*'] = {"value": "second"}

    store = EV(glob=glob)
    assert store.get(ent) == {
        'value': 'second'
    }