77 lines
2.9 KiB
Python
77 lines
2.9 KiB
Python
"""Tests for the Somfy config flow."""
|
|
import asyncio
|
|
from unittest.mock import Mock, patch
|
|
|
|
from pymfy.api.somfy_api import SomfyApi
|
|
|
|
from homeassistant import data_entry_flow
|
|
from homeassistant.components.somfy import config_flow, DOMAIN
|
|
from homeassistant.components.somfy.config_flow import register_flow_implementation
|
|
from tests.common import MockConfigEntry, mock_coro
|
|
|
|
CLIENT_SECRET_VALUE = "5678"
|
|
|
|
CLIENT_ID_VALUE = "1234"
|
|
|
|
AUTH_URL = "http://somfy.com"
|
|
|
|
|
|
async def test_abort_if_no_configuration(hass):
|
|
"""Check flow abort when no configuration."""
|
|
flow = config_flow.SomfyFlowHandler()
|
|
flow.hass = hass
|
|
result = await flow.async_step_user()
|
|
assert result["type"] == data_entry_flow.RESULT_TYPE_ABORT
|
|
assert result["reason"] == "missing_configuration"
|
|
|
|
|
|
async def test_abort_if_existing_entry(hass):
|
|
"""Check flow abort when an entry already exist."""
|
|
flow = config_flow.SomfyFlowHandler()
|
|
flow.hass = hass
|
|
MockConfigEntry(domain=DOMAIN).add_to_hass(hass)
|
|
result = await flow.async_step_import()
|
|
assert result["type"] == data_entry_flow.RESULT_TYPE_ABORT
|
|
assert result["reason"] == "already_setup"
|
|
result = await flow.async_step_user()
|
|
assert result["type"] == data_entry_flow.RESULT_TYPE_ABORT
|
|
assert result["reason"] == "already_setup"
|
|
|
|
|
|
async def test_full_flow(hass):
|
|
"""Check classic use case."""
|
|
hass.data[DOMAIN] = {}
|
|
register_flow_implementation(hass, CLIENT_ID_VALUE, CLIENT_SECRET_VALUE)
|
|
flow = config_flow.SomfyFlowHandler()
|
|
flow.hass = hass
|
|
hass.config.api = Mock(base_url="https://example.com")
|
|
flow._get_authorization_url = Mock(return_value=mock_coro((AUTH_URL, "state")))
|
|
result = await flow.async_step_import()
|
|
assert result["type"] == data_entry_flow.RESULT_TYPE_EXTERNAL_STEP
|
|
assert result["url"] == AUTH_URL
|
|
result = await flow.async_step_auth("my_super_code")
|
|
assert result["type"] == data_entry_flow.RESULT_TYPE_EXTERNAL_STEP_DONE
|
|
assert result["step_id"] == "creation"
|
|
assert flow.code == "my_super_code"
|
|
with patch.object(
|
|
SomfyApi, "request_token", return_value={"access_token": "super_token"}
|
|
):
|
|
result = await flow.async_step_creation()
|
|
assert result["type"] == data_entry_flow.RESULT_TYPE_CREATE_ENTRY
|
|
assert result["data"]["refresh_args"] == {
|
|
"client_id": CLIENT_ID_VALUE,
|
|
"client_secret": CLIENT_SECRET_VALUE,
|
|
}
|
|
assert result["title"] == "Somfy"
|
|
assert result["data"]["token"] == {"access_token": "super_token"}
|
|
|
|
|
|
async def test_abort_if_authorization_timeout(hass):
|
|
"""Check Somfy authorization timeout."""
|
|
flow = config_flow.SomfyFlowHandler()
|
|
flow.hass = hass
|
|
flow._get_authorization_url = Mock(side_effect=asyncio.TimeoutError)
|
|
result = await flow.async_step_auth()
|
|
assert result["type"] == data_entry_flow.RESULT_TYPE_ABORT
|
|
assert result["reason"] == "authorize_url_timeout"
|