core/tests/components/somfy/test_config_flow.py

"""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"