mirror of
https://github.com/home-assistant/core.git
synced 2025-11-04 00:19:31 +00:00
111 lines
3.5 KiB
Python
111 lines
3.5 KiB
Python
"""Test Prowl config flow."""
|
|
|
|
from unittest.mock import AsyncMock
|
|
|
|
import prowlpy
|
|
|
|
from homeassistant import config_entries
|
|
from homeassistant.components.prowl.const import DOMAIN
|
|
from homeassistant.const import CONF_API_KEY, CONF_NAME
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.data_entry_flow import FlowResultType
|
|
|
|
from .conftest import BAD_API_RESPONSE, CONF_INPUT, INVALID_API_KEY_ERROR, TIMEOUT_ERROR
|
|
|
|
|
|
async def test_flow_user(hass: HomeAssistant, mock_prowlpy: AsyncMock) -> None:
|
|
"""Test user initialized flow."""
|
|
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=CONF_INPUT,
|
|
)
|
|
|
|
assert mock_prowlpy.verify_key.call_count > 0
|
|
assert result["type"] is FlowResultType.CREATE_ENTRY
|
|
assert result["title"] == CONF_INPUT[CONF_NAME]
|
|
assert result["data"] == {CONF_API_KEY: CONF_INPUT[CONF_API_KEY]}
|
|
|
|
|
|
async def test_flow_duplicate_api_key(
|
|
hass: HomeAssistant, mock_prowlpy: AsyncMock
|
|
) -> None:
|
|
"""Test user initialized flow."""
|
|
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=CONF_INPUT,
|
|
)
|
|
|
|
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=CONF_INPUT,
|
|
)
|
|
assert result["type"] is FlowResultType.ABORT
|
|
|
|
|
|
async def test_flow_user_bad_key(hass: HomeAssistant, mock_prowlpy: AsyncMock) -> None:
|
|
"""Test user submitting a bad API key."""
|
|
mock_prowlpy.verify_key.side_effect = prowlpy.APIError("Invalid API key")
|
|
|
|
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=CONF_INPUT,
|
|
)
|
|
|
|
assert mock_prowlpy.verify_key.call_count > 0
|
|
assert result["type"] is FlowResultType.FORM
|
|
assert result["errors"] == INVALID_API_KEY_ERROR
|
|
|
|
|
|
async def test_flow_user_prowl_timeout(
|
|
hass: HomeAssistant, mock_prowlpy: AsyncMock
|
|
) -> None:
|
|
"""Test Prowl API timeout."""
|
|
mock_prowlpy.verify_key.side_effect = TimeoutError
|
|
|
|
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=CONF_INPUT,
|
|
)
|
|
|
|
assert mock_prowlpy.verify_key.call_count > 0
|
|
assert result["type"] is FlowResultType.FORM
|
|
assert result["errors"] == TIMEOUT_ERROR
|
|
|
|
|
|
async def test_flow_api_failure(hass: HomeAssistant, mock_prowlpy: AsyncMock) -> None:
|
|
"""Test Prowl API failure."""
|
|
mock_prowlpy.verify_key.side_effect = prowlpy.APIError(BAD_API_RESPONSE)
|
|
|
|
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=CONF_INPUT,
|
|
)
|
|
|
|
assert mock_prowlpy.verify_key.call_count > 0
|
|
assert result["type"] is FlowResultType.FORM
|
|
assert result["errors"] == BAD_API_RESPONSE
|