"""Test the Ubiquiti airOS config flow.""" from typing import Any from unittest.mock import AsyncMock from airos.exceptions import ( ConnectionAuthenticationError, DeviceConnectionError, KeyDataMissingError, ) import pytest from homeassistant.components.airos.const import DOMAIN from homeassistant.config_entries import SOURCE_USER from homeassistant.const import CONF_HOST, CONF_PASSWORD, CONF_USERNAME from homeassistant.core import HomeAssistant from homeassistant.data_entry_flow import FlowResultType from tests.common import MockConfigEntry MOCK_CONFIG = { CONF_HOST: "1.1.1.1", CONF_USERNAME: "ubnt", CONF_PASSWORD: "test-password", } async def test_form_creates_entry( hass: HomeAssistant, mock_setup_entry: AsyncMock, mock_airos_client: AsyncMock, ap_fixture: dict[str, Any], ) -> None: """Test we get the form and create the appropriate entry.""" result = await hass.config_entries.flow.async_init( DOMAIN, context={"source": SOURCE_USER} ) assert result["type"] is FlowResultType.FORM assert result["errors"] == {} result = await hass.config_entries.flow.async_configure( result["flow_id"], MOCK_CONFIG, ) assert result["type"] is FlowResultType.CREATE_ENTRY assert result["title"] == "NanoStation 5AC ap name" assert result["result"].unique_id == "01:23:45:67:89:AB" assert result["data"] == MOCK_CONFIG assert len(mock_setup_entry.mock_calls) == 1 async def test_form_duplicate_entry( hass: HomeAssistant, mock_airos_client: AsyncMock, mock_config_entry: MockConfigEntry, mock_setup_entry: AsyncMock, ) -> None: """Test the form does not allow duplicate entries.""" mock_config_entry.add_to_hass(hass) result = await hass.config_entries.flow.async_init( DOMAIN, context={"source": SOURCE_USER} ) assert result["type"] is FlowResultType.FORM assert not result["errors"] assert result["step_id"] == "user" result = await hass.config_entries.flow.async_configure( result["flow_id"], MOCK_CONFIG, ) assert result["type"] is FlowResultType.ABORT assert result["reason"] == "already_configured" @pytest.mark.parametrize( ("exception", "error"), [ (ConnectionAuthenticationError, "invalid_auth"), (DeviceConnectionError, "cannot_connect"), (KeyDataMissingError, "key_data_missing"), (Exception, "unknown"), ], ) async def test_form_exception_handling( hass: HomeAssistant, mock_setup_entry: AsyncMock, mock_airos_client: AsyncMock, exception: Exception, error: str, ) -> None: """Test we handle exceptions.""" mock_airos_client.login.side_effect = exception result = await hass.config_entries.flow.async_init( DOMAIN, context={"source": SOURCE_USER} ) result = await hass.config_entries.flow.async_configure( result["flow_id"], MOCK_CONFIG, ) assert result["type"] is FlowResultType.FORM assert result["errors"] == {"base": error} mock_airos_client.login.side_effect = None result = await hass.config_entries.flow.async_configure( result["flow_id"], MOCK_CONFIG, ) assert result["type"] is FlowResultType.CREATE_ENTRY assert result["title"] == "NanoStation 5AC ap name" assert result["data"] == MOCK_CONFIG assert len(mock_setup_entry.mock_calls) == 1