"""Tests for Ghost config flow.""" from unittest.mock import AsyncMock from aioghost.exceptions import GhostAuthError, GhostConnectionError import pytest from homeassistant.components.ghost.const import ( CONF_ADMIN_API_KEY, CONF_API_URL, DOMAIN, ) from homeassistant.config_entries import SOURCE_USER from homeassistant.core import HomeAssistant from homeassistant.data_entry_flow import FlowResultType from .conftest import API_KEY, API_URL, SITE_UUID from tests.common import MockConfigEntry NEW_API_KEY = "new_key_id:new_key_secret" NEW_API_URL = "https://new.ghost.io" @pytest.mark.usefixtures("mock_setup_entry") async def test_form_user(hass: HomeAssistant, mock_ghost_api: AsyncMock) -> None: """Test the user config flow.""" result = await hass.config_entries.flow.async_init( DOMAIN, context={"source": 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_API_URL: API_URL, CONF_ADMIN_API_KEY: API_KEY, }, ) assert result["type"] is FlowResultType.CREATE_ENTRY assert result["title"] == "Test Ghost" assert result["result"].unique_id == SITE_UUID assert result["data"] == { CONF_API_URL: API_URL, CONF_ADMIN_API_KEY: API_KEY, } async def test_form_invalid_api_key_format(hass: HomeAssistant) -> None: """Test error on invalid API key format.""" result = await hass.config_entries.flow.async_init( DOMAIN, context={"source": SOURCE_USER} ) result = await hass.config_entries.flow.async_configure( result["flow_id"], { CONF_API_URL: API_URL, CONF_ADMIN_API_KEY: "invalid-no-colon", }, ) assert result["type"] is FlowResultType.FORM assert result["errors"] == {"base": "invalid_api_key"} @pytest.mark.usefixtures("mock_setup_entry") async def test_form_already_configured( hass: HomeAssistant, mock_config_entry, mock_ghost_api: AsyncMock ) -> None: """Test error when already configured.""" # Add existing entry to hass mock_config_entry.add_to_hass(hass) # Try to configure a second entry with same URL result = await hass.config_entries.flow.async_init( DOMAIN, context={"source": SOURCE_USER} ) assert result["type"] is FlowResultType.FORM result = await hass.config_entries.flow.async_configure( result["flow_id"], { CONF_API_URL: API_URL, CONF_ADMIN_API_KEY: API_KEY, }, ) assert result["type"] is FlowResultType.ABORT assert result["reason"] == "already_configured" @pytest.mark.parametrize( ("side_effect", "error_key"), [ (GhostAuthError("Invalid API key"), "invalid_auth"), (GhostConnectionError("Connection failed"), "cannot_connect"), (RuntimeError("Unexpected"), "unknown"), ], ) @pytest.mark.usefixtures("mock_setup_entry") async def test_form_errors_can_recover( hass: HomeAssistant, mock_ghost_api: AsyncMock, side_effect: Exception, error_key: str, ) -> None: """Test errors and recovery during setup.""" result = await hass.config_entries.flow.async_init( DOMAIN, context={"source": SOURCE_USER} ) mock_ghost_api.get_site.side_effect = side_effect result = await hass.config_entries.flow.async_configure( result["flow_id"], { CONF_API_URL: API_URL, CONF_ADMIN_API_KEY: API_KEY, }, ) assert result["type"] is FlowResultType.FORM assert result["errors"] == {"base": error_key} mock_ghost_api.get_site.side_effect = None result = await hass.config_entries.flow.async_configure( result["flow_id"], { CONF_API_URL: API_URL, CONF_ADMIN_API_KEY: API_KEY, }, ) assert result["type"] is FlowResultType.CREATE_ENTRY assert result["title"] == "Test Ghost" assert result["result"].unique_id == SITE_UUID assert result["data"] == { CONF_API_URL: API_URL, CONF_ADMIN_API_KEY: API_KEY, } @pytest.mark.usefixtures("mock_ghost_api", "mock_setup_entry") async def test_reauth_flow( hass: HomeAssistant, mock_config_entry: MockConfigEntry, ) -> None: """Test reauth flow.""" mock_config_entry.add_to_hass(hass) result = await mock_config_entry.start_reauth_flow(hass) assert result["type"] is FlowResultType.FORM assert result["step_id"] == "reauth_confirm" result = await hass.config_entries.flow.async_configure( result["flow_id"], {CONF_ADMIN_API_KEY: NEW_API_KEY}, ) assert result["type"] is FlowResultType.ABORT assert result["reason"] == "reauth_successful" assert mock_config_entry.data[CONF_ADMIN_API_KEY] == NEW_API_KEY assert len(hass.config_entries.async_entries()) == 1 @pytest.mark.parametrize( ("side_effect", "error_key"), [ (GhostAuthError("Invalid API key"), "invalid_auth"), (GhostConnectionError("Connection failed"), "cannot_connect"), (RuntimeError("Unexpected"), "unknown"), ], ) @pytest.mark.usefixtures("mock_setup_entry") async def test_reauth_flow_errors_can_recover( hass: HomeAssistant, mock_config_entry: MockConfigEntry, mock_ghost_api: AsyncMock, side_effect: Exception, error_key: str, ) -> None: """Test reauth flow errors and recovery.""" mock_config_entry.add_to_hass(hass) result = await mock_config_entry.start_reauth_flow(hass) mock_ghost_api.get_site.side_effect = side_effect result = await hass.config_entries.flow.async_configure( result["flow_id"], {CONF_ADMIN_API_KEY: NEW_API_KEY}, ) assert result["type"] is FlowResultType.FORM assert result["errors"] == {"base": error_key} mock_ghost_api.get_site.side_effect = None result = await hass.config_entries.flow.async_configure( result["flow_id"], {CONF_ADMIN_API_KEY: NEW_API_KEY}, ) assert result["type"] is FlowResultType.ABORT assert result["reason"] == "reauth_successful" assert mock_config_entry.data[CONF_ADMIN_API_KEY] == NEW_API_KEY assert len(hass.config_entries.async_entries()) == 1 async def test_reauth_flow_invalid_api_key_format( hass: HomeAssistant, mock_config_entry: MockConfigEntry, ) -> None: """Test reauth flow with invalid API key format.""" mock_config_entry.add_to_hass(hass) result = await mock_config_entry.start_reauth_flow(hass) result = await hass.config_entries.flow.async_configure( result["flow_id"], {CONF_ADMIN_API_KEY: "invalid-no-colon"}, ) assert result["type"] is FlowResultType.FORM assert result["errors"] == {"base": "invalid_api_key"} @pytest.mark.usefixtures("mock_ghost_api", "mock_setup_entry") async def test_reconfigure_flow( hass: HomeAssistant, mock_config_entry: MockConfigEntry, ) -> None: """Test reconfigure flow.""" mock_config_entry.add_to_hass(hass) result = await mock_config_entry.start_reconfigure_flow(hass) assert result["type"] is FlowResultType.FORM assert result["step_id"] == "reconfigure" result = await hass.config_entries.flow.async_configure( result["flow_id"], { CONF_API_URL: NEW_API_URL, CONF_ADMIN_API_KEY: NEW_API_KEY, }, ) assert result["type"] is FlowResultType.ABORT assert result["reason"] == "reconfigure_successful" assert mock_config_entry.data[CONF_API_URL] == NEW_API_URL assert mock_config_entry.data[CONF_ADMIN_API_KEY] == NEW_API_KEY @pytest.mark.parametrize( ("side_effect", "error_key"), [ (GhostAuthError("Invalid API key"), "invalid_auth"), (GhostConnectionError("Connection failed"), "cannot_connect"), (RuntimeError("Unexpected"), "unknown"), ], ) @pytest.mark.usefixtures("mock_setup_entry") async def test_reconfigure_flow_errors_can_recover( hass: HomeAssistant, mock_config_entry: MockConfigEntry, mock_ghost_api: AsyncMock, side_effect: Exception, error_key: str, ) -> None: """Test reconfigure flow errors and recovery.""" mock_config_entry.add_to_hass(hass) result = await mock_config_entry.start_reconfigure_flow(hass) mock_ghost_api.get_site.side_effect = side_effect result = await hass.config_entries.flow.async_configure( result["flow_id"], { CONF_API_URL: NEW_API_URL, CONF_ADMIN_API_KEY: NEW_API_KEY, }, ) assert result["type"] is FlowResultType.FORM assert result["errors"] == {"base": error_key} mock_ghost_api.get_site.side_effect = None result = await hass.config_entries.flow.async_configure( result["flow_id"], { CONF_API_URL: NEW_API_URL, CONF_ADMIN_API_KEY: NEW_API_KEY, }, ) assert result["type"] is FlowResultType.ABORT assert result["reason"] == "reconfigure_successful" assert mock_config_entry.data[CONF_API_URL] == NEW_API_URL assert mock_config_entry.data[CONF_ADMIN_API_KEY] == NEW_API_KEY @pytest.mark.usefixtures("mock_ghost_api", "mock_setup_entry") async def test_reconfigure_flow_invalid_api_key_format( hass: HomeAssistant, mock_config_entry: MockConfigEntry, ) -> None: """Test reconfigure flow with invalid API key format.""" mock_config_entry.add_to_hass(hass) result = await mock_config_entry.start_reconfigure_flow(hass) result = await hass.config_entries.flow.async_configure( result["flow_id"], { CONF_API_URL: NEW_API_URL, CONF_ADMIN_API_KEY: "invalid-no-colon", }, ) assert result["type"] is FlowResultType.FORM assert result["errors"] == {"base": "invalid_api_key"} result = await hass.config_entries.flow.async_configure( result["flow_id"], { CONF_API_URL: NEW_API_URL, CONF_ADMIN_API_KEY: NEW_API_KEY, }, ) assert result["type"] is FlowResultType.ABORT assert result["reason"] == "reconfigure_successful" assert mock_config_entry.data[CONF_API_URL] == NEW_API_URL assert mock_config_entry.data[CONF_ADMIN_API_KEY] == NEW_API_KEY @pytest.mark.usefixtures("mock_setup_entry") async def test_reconfigure_flow_unique_id_mismatch( hass: HomeAssistant, mock_config_entry: MockConfigEntry, mock_ghost_api: AsyncMock, ) -> None: """Test reconfigure flow aborts on unique ID mismatch.""" mock_config_entry.add_to_hass(hass) mock_ghost_api.get_site.return_value = { "title": "Different Ghost", "url": NEW_API_URL, "site_uuid": "different-uuid", } result = await mock_config_entry.start_reconfigure_flow(hass) result = await hass.config_entries.flow.async_configure( result["flow_id"], { CONF_API_URL: NEW_API_URL, CONF_ADMIN_API_KEY: NEW_API_KEY, }, ) assert result["type"] is FlowResultType.ABORT assert result["reason"] == "unique_id_mismatch"