1f9f201571
* Initial work on config_flow * Finish config flow * Pylint checks, make sure the import only happens once * Added support for unloading, small fixes * Check in the hassfest output files * Flake8 fixes * pylint mistake after flake8 fixes * Work on comments * Abort the import if it is already imported * More comments resolved * Added testcases for velbus config flow * Fix pylint and flake8 * Added connection test to the config flow * More sugestions * renamed the abort reason * excluded all but the config_flow.py from the velbus component in coveragerc * Rewrote testcases with a patched version of _test_connection * Docstyle fixes * Updated the velbus testcases * just yield * flake8 fixes
98 lines
3.2 KiB
Python
98 lines
3.2 KiB
Python
"""Tests for the Velbus config flow."""
|
|
from unittest.mock import patch, Mock
|
|
|
|
import pytest
|
|
|
|
from homeassistant import data_entry_flow
|
|
from homeassistant.components.velbus import config_flow
|
|
from homeassistant.const import CONF_PORT, CONF_NAME
|
|
|
|
from tests.common import MockConfigEntry
|
|
|
|
PORT_SERIAL = '/dev/ttyACME100'
|
|
PORT_TCP = '127.0.1.0.1:3788'
|
|
|
|
|
|
@pytest.fixture(name='controller_assert')
|
|
def mock_controller_assert():
|
|
"""Mock the velbus controller with an assert."""
|
|
with patch('velbus.Controller', side_effect=Exception()):
|
|
yield
|
|
|
|
|
|
@pytest.fixture(name='controller')
|
|
def mock_controller():
|
|
"""Mock a successfull velbus controller."""
|
|
controller = Mock()
|
|
with patch('velbus.Controller', return_value=controller):
|
|
yield controller
|
|
|
|
|
|
def init_config_flow(hass):
|
|
"""Init a configuration flow."""
|
|
flow = config_flow.VelbusConfigFlow()
|
|
flow.hass = hass
|
|
return flow
|
|
|
|
|
|
async def test_user(hass, controller):
|
|
"""Test user config."""
|
|
flow = init_config_flow(hass)
|
|
|
|
result = await flow.async_step_user()
|
|
assert result['type'] == data_entry_flow.RESULT_TYPE_FORM
|
|
assert result['step_id'] == 'user'
|
|
|
|
result = await flow.async_step_user({
|
|
CONF_NAME: 'Velbus Test Serial', CONF_PORT: PORT_SERIAL})
|
|
assert result['type'] == data_entry_flow.RESULT_TYPE_CREATE_ENTRY
|
|
assert result['title'] == 'velbus_test_serial'
|
|
assert result['data'][CONF_PORT] == PORT_SERIAL
|
|
|
|
result = await flow.async_step_user({
|
|
CONF_NAME: 'Velbus Test TCP', CONF_PORT: PORT_TCP})
|
|
assert result['type'] == data_entry_flow.RESULT_TYPE_CREATE_ENTRY
|
|
assert result['title'] == 'velbus_test_tcp'
|
|
assert result['data'][CONF_PORT] == PORT_TCP
|
|
|
|
|
|
async def test_user_fail(hass, controller_assert):
|
|
"""Test user config."""
|
|
flow = init_config_flow(hass)
|
|
|
|
result = await flow.async_step_user({
|
|
CONF_NAME: 'Velbus Test Serial', CONF_PORT: PORT_SERIAL})
|
|
assert result['type'] == data_entry_flow.RESULT_TYPE_FORM
|
|
assert result['errors'] == {CONF_PORT: 'connection_failed'}
|
|
|
|
result = await flow.async_step_user({
|
|
CONF_NAME: 'Velbus Test TCP', CONF_PORT: PORT_TCP})
|
|
assert result['type'] == data_entry_flow.RESULT_TYPE_FORM
|
|
assert result['errors'] == {CONF_PORT: 'connection_failed'}
|
|
|
|
|
|
async def test_import(hass, controller):
|
|
"""Test import step."""
|
|
flow = init_config_flow(hass)
|
|
|
|
result = await flow.async_step_import({CONF_PORT: PORT_TCP})
|
|
assert result['type'] == data_entry_flow.RESULT_TYPE_CREATE_ENTRY
|
|
assert result['title'] == 'velbus_import'
|
|
|
|
|
|
async def test_abort_if_already_setup(hass):
|
|
"""Test we abort if Daikin is already setup."""
|
|
flow = init_config_flow(hass)
|
|
MockConfigEntry(domain='velbus',
|
|
data={CONF_PORT: PORT_TCP,
|
|
CONF_NAME: 'velbus home'}).add_to_hass(hass)
|
|
|
|
result = await flow.async_step_import(
|
|
{CONF_PORT: PORT_TCP, CONF_NAME: 'velbus import test'})
|
|
assert result['type'] == data_entry_flow.RESULT_TYPE_ABORT
|
|
assert result['reason'] == 'port_exists'
|
|
|
|
result = await flow.async_step_user(
|
|
{CONF_PORT: PORT_TCP, CONF_NAME: 'velbus import test'})
|
|
assert result['type'] == data_entry_flow.RESULT_TYPE_FORM
|
|
assert result['errors'] == {'port': 'port_exists'}
|