mirror of
https://github.com/home-assistant/core.git
synced 2026-07-10 16:19:29 +01:00
Muhammad Hamza Khan
92 lines
3.0 KiB
Python
92 lines
3.0 KiB
Python
"""Tests for syncthing config flow."""
|
|
|
|
from unittest.mock import patch
|
|
|
|
from aiosyncthing.exceptions import UnauthorizedError
|
|
|
|
from homeassistant import config_entries
|
|
from homeassistant.components.syncthing.const import DOMAIN
|
|
from homeassistant.const import CONF_TOKEN, CONF_URL, CONF_VERIFY_SSL
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.data_entry_flow import FlowResultType
|
|
|
|
from . import MOCK_ENTRY, SERVER_ID, TOKEN, URL, VERIFY_SSL
|
|
|
|
from tests.common import MockConfigEntry
|
|
|
|
|
|
async def test_show_setup_form(hass: HomeAssistant) -> None:
|
|
"""Test that the setup form is served."""
|
|
|
|
result = await hass.config_entries.flow.async_init(
|
|
DOMAIN, context={"source": config_entries.SOURCE_USER}
|
|
)
|
|
assert result["type"] is FlowResultType.FORM
|
|
assert result["errors"] == {}
|
|
assert result["step_id"] == "user"
|
|
|
|
|
|
async def test_flow_successful(hass: HomeAssistant) -> None:
|
|
"""Test with required fields only."""
|
|
with (
|
|
patch("aiosyncthing.system.System.status", return_value={"myID": SERVER_ID}),
|
|
patch(
|
|
"homeassistant.components.syncthing.async_setup_entry",
|
|
return_value=True,
|
|
) as mock_setup_entry,
|
|
):
|
|
result = await hass.config_entries.flow.async_init(
|
|
DOMAIN,
|
|
context={"source": "user"},
|
|
data=MOCK_ENTRY,
|
|
)
|
|
assert result["type"] is FlowResultType.CREATE_ENTRY
|
|
assert result["title"] == URL
|
|
assert result["data"][CONF_URL] == URL
|
|
assert result["data"][CONF_TOKEN] == TOKEN
|
|
assert result["data"][CONF_VERIFY_SSL] == VERIFY_SSL
|
|
assert len(mock_setup_entry.mock_calls) == 1
|
|
|
|
|
|
async def test_flow_already_configured(
|
|
hass: HomeAssistant, entry: MockConfigEntry
|
|
) -> None:
|
|
"""Test the server ID is already configured."""
|
|
with patch("aiosyncthing.system.System.status", return_value={"myID": SERVER_ID}):
|
|
result = await hass.config_entries.flow.async_init(
|
|
DOMAIN,
|
|
context={"source": "user"},
|
|
data=MOCK_ENTRY,
|
|
)
|
|
|
|
assert result["type"] is FlowResultType.ABORT
|
|
assert result["reason"] == "already_configured"
|
|
|
|
|
|
async def test_flow_invalid_auth(hass: HomeAssistant) -> None:
|
|
"""Test invalid auth."""
|
|
|
|
with patch("aiosyncthing.system.System.status", side_effect=UnauthorizedError):
|
|
result = await hass.config_entries.flow.async_init(
|
|
DOMAIN,
|
|
context={"source": "user"},
|
|
data=MOCK_ENTRY,
|
|
)
|
|
|
|
assert result["type"] is FlowResultType.FORM
|
|
assert result["errors"] == {"token": "invalid_auth"}
|
|
|
|
|
|
async def test_flow_cannot_connect(hass: HomeAssistant) -> None:
|
|
"""Test cannot connect."""
|
|
|
|
with patch("aiosyncthing.system.System.status", side_effect=Exception):
|
|
result = await hass.config_entries.flow.async_init(
|
|
DOMAIN,
|
|
context={"source": "user"},
|
|
data=MOCK_ENTRY,
|
|
)
|
|
|
|
assert result["type"] is FlowResultType.FORM
|
|
assert result["errors"] == {"base": "cannot_connect"}
|