"""Test Simplepush config flow.""" from unittest.mock import patch import pytest from simplepush import UnknownError from homeassistant import config_entries, data_entry_flow from homeassistant.components.simplepush.const import CONF_DEVICE_KEY, CONF_SALT, DOMAIN from homeassistant.const import CONF_NAME, CONF_PASSWORD from homeassistant.core import HomeAssistant from tests.common import MockConfigEntry MOCK_CONFIG = { CONF_DEVICE_KEY: "abc", CONF_NAME: "simplepush", } @pytest.fixture(autouse=True) def simplepush_setup_fixture(): """Patch simplepush setup entry.""" with patch( "homeassistant.components.simplepush.async_setup_entry", return_value=True ): yield @pytest.fixture(autouse=True) def mock_api_request(): """Patch simplepush api request.""" with patch("homeassistant.components.simplepush.config_flow.send"), patch( "homeassistant.components.simplepush.config_flow.send_encrypted" ): yield async def test_flow_successful(hass: HomeAssistant) -> None: """Test user initialized flow with minimum config.""" result = await hass.config_entries.flow.async_init( DOMAIN, context={"source": config_entries.SOURCE_USER}, ) result = await hass.config_entries.flow.async_configure( result["flow_id"], user_input=MOCK_CONFIG, ) assert result["type"] == data_entry_flow.FlowResultType.CREATE_ENTRY assert result["title"] == "simplepush" assert result["data"] == MOCK_CONFIG async def test_flow_with_password(hass: HomeAssistant) -> None: """Test user initialized flow with password and salt.""" mock_config_pass = {**MOCK_CONFIG, CONF_PASSWORD: "password", CONF_SALT: "salt"} result = await hass.config_entries.flow.async_init( DOMAIN, context={"source": config_entries.SOURCE_USER}, ) result = await hass.config_entries.flow.async_configure( result["flow_id"], user_input=mock_config_pass, ) assert result["type"] == data_entry_flow.FlowResultType.CREATE_ENTRY assert result["title"] == "simplepush" assert result["data"] == mock_config_pass async def test_flow_user_device_key_already_configured(hass: HomeAssistant) -> None: """Test user initialized flow with duplicate device key.""" entry = MockConfigEntry( domain=DOMAIN, data=MOCK_CONFIG, unique_id="abc", ) entry.add_to_hass(hass) result = await hass.config_entries.flow.async_init( DOMAIN, context={"source": config_entries.SOURCE_USER}, ) result = await hass.config_entries.flow.async_configure( result["flow_id"], user_input=MOCK_CONFIG, ) assert result["type"] == data_entry_flow.FlowResultType.ABORT assert result["reason"] == "already_configured" async def test_flow_user_name_already_configured(hass: HomeAssistant) -> None: """Test user initialized flow with duplicate name.""" entry = MockConfigEntry( domain=DOMAIN, data=MOCK_CONFIG, unique_id="abc", ) entry.add_to_hass(hass) new_entry = MOCK_CONFIG.copy() new_entry[CONF_DEVICE_KEY] = "abc1" result = await hass.config_entries.flow.async_init( DOMAIN, context={"source": config_entries.SOURCE_USER}, ) result = await hass.config_entries.flow.async_configure( result["flow_id"], user_input=MOCK_CONFIG, ) assert result["type"] == data_entry_flow.FlowResultType.ABORT assert result["reason"] == "already_configured" async def test_error_on_connection_failure(hass: HomeAssistant) -> None: """Test when connection to api fails.""" with patch( "homeassistant.components.simplepush.config_flow.send", side_effect=UnknownError, ): result = await hass.config_entries.flow.async_init( DOMAIN, context={"source": config_entries.SOURCE_USER}, ) result = await hass.config_entries.flow.async_configure( result["flow_id"], user_input=MOCK_CONFIG, ) assert result["type"] == data_entry_flow.FlowResultType.FORM assert result["errors"] == {"base": "cannot_connect"}