core/tests/components/zone/test_config_flow.py

"""Tests for zone config flow."""

from homeassistant.components.zone import config_flow
from homeassistant.components.zone.const import CONF_PASSIVE, DOMAIN, HOME_ZONE
from homeassistant.const import (
    CONF_NAME, CONF_LATITUDE, CONF_LONGITUDE, CONF_ICON, CONF_RADIUS)

from tests.common import MockConfigEntry


async def test_flow_works(hass):
    """Test that config flow works."""
    flow = config_flow.ZoneFlowHandler()
    flow.hass = hass

    result = await flow.async_step_init(user_input={
        CONF_NAME: 'Name',
        CONF_LATITUDE: '1.1',
        CONF_LONGITUDE: '2.2',
        CONF_RADIUS: '100',
        CONF_ICON: 'mdi:home',
        CONF_PASSIVE: True
        })

    assert result['type'] == 'create_entry'
    assert result['title'] == 'Name'
    assert result['data'] == {
        CONF_NAME: 'Name',
        CONF_LATITUDE: '1.1',
        CONF_LONGITUDE: '2.2',
        CONF_RADIUS: '100',
        CONF_ICON: 'mdi:home',
        CONF_PASSIVE: True
    }


async def test_flow_requires_unique_name(hass):
    """Test that config flow verifies that each zones name is unique."""
    MockConfigEntry(domain=DOMAIN, data={
        CONF_NAME: 'Name'
    }).add_to_hass(hass)
    flow = config_flow.ZoneFlowHandler()
    flow.hass = hass

    result = await flow.async_step_init(user_input={CONF_NAME: 'Name'})
    assert result['errors'] == {'base': 'name_exists'}


async def test_flow_requires_name_different_from_home(hass):
    """Test that config flow verifies that each zones name is unique."""
    flow = config_flow.ZoneFlowHandler()
    flow.hass = hass

    result = await flow.async_step_init(user_input={CONF_NAME: HOME_ZONE})
    assert result['errors'] == {'base': 'name_exists'}