"""Test the Fibaro config flow.""" from unittest.mock import Mock import pytest from requests.exceptions import HTTPError from homeassistant import config_entries from homeassistant.components.fibaro import DOMAIN from homeassistant.components.fibaro.config_flow import _normalize_url from homeassistant.components.fibaro.const import CONF_IMPORT_PLUGINS from homeassistant.const import CONF_PASSWORD, CONF_URL, CONF_USERNAME from homeassistant.core import HomeAssistant from homeassistant.data_entry_flow import FlowResult, FlowResultType from .conftest import TEST_NAME, TEST_PASSWORD, TEST_URL, TEST_USERNAME from tests.common import MockConfigEntry pytestmark = pytest.mark.usefixtures("mock_setup_entry", "mock_fibaro_client") async def _recovery_after_failure_works( hass: HomeAssistant, mock_fibaro_client: Mock, result: FlowResult ) -> None: mock_fibaro_client.connect.side_effect = None mock_fibaro_client.connect.return_value = True result = await hass.config_entries.flow.async_configure( result["flow_id"], { CONF_URL: TEST_URL, CONF_USERNAME: TEST_USERNAME, CONF_PASSWORD: TEST_PASSWORD, }, ) assert result["type"] is FlowResultType.CREATE_ENTRY assert result["title"] == TEST_NAME assert result["data"] == { CONF_URL: TEST_URL, CONF_USERNAME: TEST_USERNAME, CONF_PASSWORD: TEST_PASSWORD, CONF_IMPORT_PLUGINS: False, } async def _recovery_after_reauth_failure_works( hass: HomeAssistant, mock_fibaro_client: Mock, result: FlowResult ) -> None: mock_fibaro_client.connect.side_effect = None mock_fibaro_client.connect.return_value = True result = await hass.config_entries.flow.async_configure( result["flow_id"], user_input={CONF_PASSWORD: "other_fake_password"}, ) assert result["type"] is FlowResultType.ABORT assert result["reason"] == "reauth_successful" async def test_config_flow_user_initiated_success(hass: HomeAssistant) -> None: """Successful flow manually initialized by the user.""" result = await hass.config_entries.flow.async_init( DOMAIN, context={"source": config_entries.SOURCE_USER} ) assert result["type"] is FlowResultType.FORM assert result["step_id"] == "user" assert result["errors"] == {} result = await hass.config_entries.flow.async_configure( result["flow_id"], { CONF_URL: TEST_URL, CONF_USERNAME: TEST_USERNAME, CONF_PASSWORD: TEST_PASSWORD, }, ) assert result["type"] is FlowResultType.CREATE_ENTRY assert result["title"] == TEST_NAME assert result["data"] == { CONF_URL: TEST_URL, CONF_USERNAME: TEST_USERNAME, CONF_PASSWORD: TEST_PASSWORD, CONF_IMPORT_PLUGINS: False, } async def test_config_flow_user_initiated_auth_failure( hass: HomeAssistant, mock_fibaro_client: Mock ) -> None: """Authentication failure in flow manually initialized by the user.""" result = await hass.config_entries.flow.async_init( DOMAIN, context={"source": config_entries.SOURCE_USER} ) assert result["type"] is FlowResultType.FORM assert result["step_id"] == "user" assert result["errors"] == {} mock_fibaro_client.connect.side_effect = HTTPError(response=Mock(status_code=403)) result = await hass.config_entries.flow.async_configure( result["flow_id"], { CONF_URL: TEST_URL, CONF_USERNAME: TEST_USERNAME, CONF_PASSWORD: TEST_PASSWORD, }, ) assert result["type"] is FlowResultType.FORM assert result["step_id"] == "user" assert result["errors"] == {"base": "invalid_auth"} await _recovery_after_failure_works(hass, mock_fibaro_client, result) async def test_config_flow_user_initiated_unknown_failure_1( hass: HomeAssistant, mock_fibaro_client: Mock ) -> None: """Unknown failure in flow manually initialized by the user.""" result = await hass.config_entries.flow.async_init( DOMAIN, context={"source": config_entries.SOURCE_USER} ) assert result["type"] is FlowResultType.FORM assert result["step_id"] == "user" assert result["errors"] == {} mock_fibaro_client.connect.side_effect = HTTPError(response=Mock(status_code=500)) result = await hass.config_entries.flow.async_configure( result["flow_id"], { CONF_URL: TEST_URL, CONF_USERNAME: TEST_USERNAME, CONF_PASSWORD: TEST_PASSWORD, }, ) assert result["type"] is FlowResultType.FORM assert result["step_id"] == "user" assert result["errors"] == {"base": "cannot_connect"} await _recovery_after_failure_works(hass, mock_fibaro_client, result) async def test_config_flow_user_initiated_unknown_failure_2( hass: HomeAssistant, mock_fibaro_client: Mock ) -> None: """Unknown failure in flow manually initialized by the user.""" result = await hass.config_entries.flow.async_init( DOMAIN, context={"source": config_entries.SOURCE_USER} ) assert result["type"] is FlowResultType.FORM assert result["step_id"] == "user" assert result["errors"] == {} mock_fibaro_client.connect.side_effect = Exception() result = await hass.config_entries.flow.async_configure( result["flow_id"], { CONF_URL: TEST_URL, CONF_USERNAME: TEST_USERNAME, CONF_PASSWORD: TEST_PASSWORD, }, ) assert result["type"] is FlowResultType.FORM assert result["step_id"] == "user" assert result["errors"] == {"base": "cannot_connect"} await _recovery_after_failure_works(hass, mock_fibaro_client, result) async def test_reauth_success( hass: HomeAssistant, mock_config_entry: MockConfigEntry ) -> None: """Successful reauth flow initialized by the user.""" result = await mock_config_entry.start_reauth_flow(hass) assert result["type"] is FlowResultType.FORM assert result["step_id"] == "reauth_confirm" assert result["errors"] == {} result = await hass.config_entries.flow.async_configure( result["flow_id"], user_input={CONF_PASSWORD: "other_fake_password"}, ) assert result["type"] is FlowResultType.ABORT assert result["reason"] == "reauth_successful" async def test_reauth_connect_failure( hass: HomeAssistant, mock_config_entry: MockConfigEntry, mock_fibaro_client: Mock, ) -> None: """Successful reauth flow initialized by the user.""" result = await mock_config_entry.start_reauth_flow(hass) assert result["type"] is FlowResultType.FORM assert result["step_id"] == "reauth_confirm" assert result["errors"] == {} mock_fibaro_client.connect.side_effect = Exception() result = await hass.config_entries.flow.async_configure( result["flow_id"], user_input={CONF_PASSWORD: "other_fake_password"}, ) assert result["type"] is FlowResultType.FORM assert result["step_id"] == "reauth_confirm" assert result["errors"] == {"base": "cannot_connect"} await _recovery_after_reauth_failure_works(hass, mock_fibaro_client, result) async def test_reauth_auth_failure( hass: HomeAssistant, mock_config_entry: MockConfigEntry, mock_fibaro_client: Mock, ) -> None: """Successful reauth flow initialized by the user.""" result = await mock_config_entry.start_reauth_flow(hass) assert result["type"] is FlowResultType.FORM assert result["step_id"] == "reauth_confirm" assert result["errors"] == {} mock_fibaro_client.connect.side_effect = HTTPError(response=Mock(status_code=403)) result = await hass.config_entries.flow.async_configure( result["flow_id"], user_input={CONF_PASSWORD: "other_fake_password"}, ) assert result["type"] is FlowResultType.FORM assert result["step_id"] == "reauth_confirm" assert result["errors"] == {"base": "invalid_auth"} await _recovery_after_reauth_failure_works(hass, mock_fibaro_client, result) @pytest.mark.parametrize("url_path", ["/api/", "/api", "/", ""]) async def test_normalize_url(url_path: str) -> None: """Test that the url is normalized for different entered values.""" assert _normalize_url(f"http://192.168.1.1{url_path}") == "http://192.168.1.1/api/"