mirror of
https://github.com/home-assistant/core.git
synced 2026-03-01 14:25:31 +00:00
410 lines
13 KiB
Python
410 lines
13 KiB
Python
"""Tests for the Mastodon config flow."""
|
|
|
|
from unittest.mock import AsyncMock, patch
|
|
|
|
from mastodon.Mastodon import (
|
|
MastodonNetworkError,
|
|
MastodonNotFoundError,
|
|
MastodonUnauthorizedError,
|
|
)
|
|
import pytest
|
|
|
|
from homeassistant.components.mastodon.const import CONF_BASE_URL, DOMAIN
|
|
from homeassistant.config_entries import SOURCE_USER
|
|
from homeassistant.const import CONF_ACCESS_TOKEN, CONF_CLIENT_ID, CONF_CLIENT_SECRET
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.data_entry_flow import FlowResultType
|
|
|
|
from tests.common import MockConfigEntry
|
|
|
|
|
|
async def test_full_flow(
|
|
hass: HomeAssistant,
|
|
mock_mastodon_client: AsyncMock,
|
|
mock_setup_entry: AsyncMock,
|
|
) -> None:
|
|
"""Test full 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"
|
|
|
|
result = await hass.config_entries.flow.async_configure(
|
|
result["flow_id"],
|
|
{
|
|
CONF_BASE_URL: "https://mastodon.social",
|
|
CONF_CLIENT_ID: "client_id",
|
|
CONF_CLIENT_SECRET: "client_secret",
|
|
CONF_ACCESS_TOKEN: "access_token",
|
|
},
|
|
)
|
|
assert result["type"] is FlowResultType.CREATE_ENTRY
|
|
assert result["title"] == "@trwnh@mastodon.social"
|
|
assert result["data"] == {
|
|
CONF_BASE_URL: "https://mastodon.social",
|
|
CONF_CLIENT_ID: "client_id",
|
|
CONF_CLIENT_SECRET: "client_secret",
|
|
CONF_ACCESS_TOKEN: "access_token",
|
|
}
|
|
assert result["result"].unique_id == "trwnh_mastodon_social"
|
|
|
|
|
|
async def test_full_flow_with_path(
|
|
hass: HomeAssistant,
|
|
mock_mastodon_client: AsyncMock,
|
|
mock_setup_entry: AsyncMock,
|
|
) -> None:
|
|
"""Test full flow, where a path is accidentally specified."""
|
|
result = await hass.config_entries.flow.async_init(
|
|
DOMAIN,
|
|
context={"source": SOURCE_USER},
|
|
)
|
|
assert result["type"] is FlowResultType.FORM
|
|
assert result["step_id"] == "user"
|
|
|
|
result = await hass.config_entries.flow.async_configure(
|
|
result["flow_id"],
|
|
{
|
|
CONF_BASE_URL: "https://mastodon.social/home",
|
|
CONF_CLIENT_ID: "client_id",
|
|
CONF_CLIENT_SECRET: "client_secret",
|
|
CONF_ACCESS_TOKEN: "access_token",
|
|
},
|
|
)
|
|
assert result["type"] is FlowResultType.CREATE_ENTRY
|
|
assert result["title"] == "@trwnh@mastodon.social"
|
|
assert result["data"] == {
|
|
CONF_BASE_URL: "https://mastodon.social",
|
|
CONF_CLIENT_ID: "client_id",
|
|
CONF_CLIENT_SECRET: "client_secret",
|
|
CONF_ACCESS_TOKEN: "access_token",
|
|
}
|
|
assert result["result"].unique_id == "trwnh_mastodon_social"
|
|
|
|
|
|
async def test_full_flow_fallback_to_instance_v1(
|
|
hass: HomeAssistant,
|
|
mock_mastodon_client: AsyncMock,
|
|
mock_setup_entry: AsyncMock,
|
|
) -> None:
|
|
"""Test full flow where instance_v2 fails and falls back to instance_v1."""
|
|
mock_mastodon_client.instance_v2.side_effect = MastodonNotFoundError(
|
|
"Instance API v2 not found"
|
|
)
|
|
|
|
result = await hass.config_entries.flow.async_init(
|
|
DOMAIN,
|
|
context={"source": SOURCE_USER},
|
|
)
|
|
assert result["type"] is FlowResultType.FORM
|
|
assert result["step_id"] == "user"
|
|
|
|
result = await hass.config_entries.flow.async_configure(
|
|
result["flow_id"],
|
|
{
|
|
CONF_BASE_URL: "https://mastodon.social",
|
|
CONF_CLIENT_ID: "client_id",
|
|
CONF_CLIENT_SECRET: "client_secret",
|
|
CONF_ACCESS_TOKEN: "access_token",
|
|
},
|
|
)
|
|
assert result["type"] is FlowResultType.CREATE_ENTRY
|
|
assert result["title"] == "@trwnh@mastodon.social"
|
|
assert result["data"] == {
|
|
CONF_BASE_URL: "https://mastodon.social",
|
|
CONF_CLIENT_ID: "client_id",
|
|
CONF_CLIENT_SECRET: "client_secret",
|
|
CONF_ACCESS_TOKEN: "access_token",
|
|
}
|
|
assert result["result"].unique_id == "trwnh_mastodon_social"
|
|
|
|
mock_mastodon_client.instance_v2.assert_called_once()
|
|
mock_mastodon_client.instance_v1.assert_called_once()
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("exception", "error"),
|
|
[
|
|
(MastodonNetworkError, "network_error"),
|
|
(MastodonUnauthorizedError, "unauthorized_error"),
|
|
(Exception, "unknown"),
|
|
],
|
|
)
|
|
async def test_flow_errors(
|
|
hass: HomeAssistant,
|
|
mock_mastodon_client: AsyncMock,
|
|
mock_setup_entry: AsyncMock,
|
|
exception: Exception,
|
|
error: str,
|
|
) -> None:
|
|
"""Test flow errors."""
|
|
mock_mastodon_client.account_verify_credentials.side_effect = exception
|
|
|
|
result = await hass.config_entries.flow.async_init(
|
|
DOMAIN,
|
|
context={"source": SOURCE_USER},
|
|
)
|
|
assert result["type"] is FlowResultType.FORM
|
|
assert result["step_id"] == "user"
|
|
|
|
result = await hass.config_entries.flow.async_configure(
|
|
result["flow_id"],
|
|
{
|
|
CONF_BASE_URL: "https://mastodon.social",
|
|
CONF_CLIENT_ID: "client_id",
|
|
CONF_CLIENT_SECRET: "client_secret",
|
|
CONF_ACCESS_TOKEN: "access_token",
|
|
},
|
|
)
|
|
|
|
assert result["type"] is FlowResultType.FORM
|
|
assert result["errors"] == {"base": error}
|
|
|
|
mock_mastodon_client.account_verify_credentials.side_effect = None
|
|
|
|
result = await hass.config_entries.flow.async_configure(
|
|
result["flow_id"],
|
|
{
|
|
CONF_BASE_URL: "https://mastodon.social",
|
|
CONF_CLIENT_ID: "client_id",
|
|
CONF_CLIENT_SECRET: "client_secret",
|
|
CONF_ACCESS_TOKEN: "access_token",
|
|
},
|
|
)
|
|
assert result["type"] is FlowResultType.CREATE_ENTRY
|
|
|
|
|
|
async def test_duplicate(
|
|
hass: HomeAssistant,
|
|
mock_mastodon_client: AsyncMock,
|
|
mock_setup_entry: AsyncMock,
|
|
mock_config_entry: MockConfigEntry,
|
|
) -> None:
|
|
"""Test duplicate flow."""
|
|
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 result["step_id"] == "user"
|
|
|
|
result = await hass.config_entries.flow.async_configure(
|
|
result["flow_id"],
|
|
{
|
|
CONF_BASE_URL: "https://mastodon.social",
|
|
CONF_CLIENT_ID: "client_id",
|
|
CONF_CLIENT_SECRET: "client_secret",
|
|
CONF_ACCESS_TOKEN: "access_token",
|
|
},
|
|
)
|
|
|
|
assert result["type"] is FlowResultType.ABORT
|
|
assert result["reason"] == "already_configured"
|
|
|
|
|
|
async def test_reauth_flow(
|
|
hass: HomeAssistant,
|
|
mock_mastodon_client: AsyncMock,
|
|
mock_setup_entry: AsyncMock,
|
|
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_ACCESS_TOKEN: "token2"},
|
|
)
|
|
|
|
assert result["type"] is FlowResultType.ABORT
|
|
assert result["reason"] == "reauth_successful"
|
|
assert mock_config_entry.data[CONF_ACCESS_TOKEN] == "token2"
|
|
|
|
|
|
async def test_reauth_flow_wrong_account(
|
|
hass: HomeAssistant,
|
|
mock_mastodon_client: AsyncMock,
|
|
mock_setup_entry: AsyncMock,
|
|
mock_config_entry: MockConfigEntry,
|
|
) -> None:
|
|
"""Test reauth flow with wrong account."""
|
|
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"
|
|
|
|
with patch(
|
|
"homeassistant.components.mastodon.config_flow.construct_mastodon_username",
|
|
return_value="BAD_USERNAME",
|
|
):
|
|
result = await hass.config_entries.flow.async_configure(
|
|
result["flow_id"],
|
|
{CONF_ACCESS_TOKEN: "token2"},
|
|
)
|
|
|
|
assert result["type"] is FlowResultType.ABORT
|
|
assert result["reason"] == "wrong_account"
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("exception", "error"),
|
|
[
|
|
(MastodonNetworkError, "network_error"),
|
|
(MastodonUnauthorizedError, "unauthorized_error"),
|
|
(Exception, "unknown"),
|
|
],
|
|
)
|
|
async def test_reauth_flow_exceptions(
|
|
hass: HomeAssistant,
|
|
mock_mastodon_client: AsyncMock,
|
|
mock_setup_entry: AsyncMock,
|
|
mock_config_entry: MockConfigEntry,
|
|
exception: Exception,
|
|
error: str,
|
|
) -> None:
|
|
"""Test reauth flow errors."""
|
|
mock_config_entry.add_to_hass(hass)
|
|
mock_mastodon_client.account_verify_credentials.side_effect = exception
|
|
|
|
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_ACCESS_TOKEN: "token"},
|
|
)
|
|
|
|
assert result["type"] is FlowResultType.FORM
|
|
assert result["step_id"] == "reauth_confirm"
|
|
assert result["errors"] == {"base": error}
|
|
|
|
mock_mastodon_client.account_verify_credentials.side_effect = None
|
|
|
|
result = await hass.config_entries.flow.async_configure(
|
|
result["flow_id"],
|
|
{CONF_ACCESS_TOKEN: "token"},
|
|
)
|
|
assert result["type"] is FlowResultType.ABORT
|
|
assert result["reason"] == "reauth_successful"
|
|
|
|
|
|
async def test_reconfigure_flow(
|
|
hass: HomeAssistant,
|
|
mock_mastodon_client: AsyncMock,
|
|
mock_setup_entry: AsyncMock,
|
|
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_CLIENT_ID: "client_id2",
|
|
CONF_CLIENT_SECRET: "client_secret2",
|
|
CONF_ACCESS_TOKEN: "access_token2",
|
|
},
|
|
)
|
|
|
|
assert result["type"] is FlowResultType.ABORT
|
|
assert result["reason"] == "reconfigure_successful"
|
|
assert mock_config_entry.data[CONF_BASE_URL] == "https://mastodon.social"
|
|
assert mock_config_entry.data[CONF_CLIENT_ID] == "client_id2"
|
|
assert mock_config_entry.data[CONF_CLIENT_SECRET] == "client_secret2"
|
|
assert mock_config_entry.data[CONF_ACCESS_TOKEN] == "access_token2"
|
|
|
|
|
|
async def test_reconfigure_flow_wrong_account(
|
|
hass: HomeAssistant,
|
|
mock_mastodon_client: AsyncMock,
|
|
mock_setup_entry: AsyncMock,
|
|
mock_config_entry: MockConfigEntry,
|
|
) -> None:
|
|
"""Test reconfigure flow with wrong account."""
|
|
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"
|
|
|
|
with patch(
|
|
"homeassistant.components.mastodon.config_flow.construct_mastodon_username",
|
|
return_value="WRONG_USERNAME",
|
|
):
|
|
result = await hass.config_entries.flow.async_configure(
|
|
result["flow_id"],
|
|
{
|
|
CONF_CLIENT_ID: "client_id",
|
|
CONF_CLIENT_SECRET: "client_secret",
|
|
CONF_ACCESS_TOKEN: "access_token2",
|
|
},
|
|
)
|
|
|
|
assert result["type"] is FlowResultType.ABORT
|
|
assert result["reason"] == "wrong_account"
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("exception", "error"),
|
|
[
|
|
(MastodonNetworkError, "network_error"),
|
|
(MastodonUnauthorizedError, "unauthorized_error"),
|
|
(Exception, "unknown"),
|
|
],
|
|
)
|
|
async def test_reconfigure_flow_exceptions(
|
|
hass: HomeAssistant,
|
|
mock_mastodon_client: AsyncMock,
|
|
mock_setup_entry: AsyncMock,
|
|
mock_config_entry: MockConfigEntry,
|
|
exception: Exception,
|
|
error: str,
|
|
) -> None:
|
|
"""Test reconfigure flow errors."""
|
|
mock_config_entry.add_to_hass(hass)
|
|
mock_mastodon_client.account_verify_credentials.side_effect = exception
|
|
|
|
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_CLIENT_ID: "client_id",
|
|
CONF_CLIENT_SECRET: "client_secret",
|
|
CONF_ACCESS_TOKEN: "access_token",
|
|
},
|
|
)
|
|
|
|
assert result["type"] is FlowResultType.FORM
|
|
assert result["step_id"] == "reconfigure"
|
|
assert result["errors"] == {"base": error}
|
|
|
|
mock_mastodon_client.account_verify_credentials.side_effect = None
|
|
|
|
result = await hass.config_entries.flow.async_configure(
|
|
result["flow_id"],
|
|
{
|
|
CONF_CLIENT_ID: "client_id",
|
|
CONF_CLIENT_SECRET: "client_secret",
|
|
CONF_ACCESS_TOKEN: "access_token",
|
|
},
|
|
)
|
|
assert result["type"] is FlowResultType.ABORT
|
|
assert result["reason"] == "reconfigure_successful"
|