Small improvements to the AdGuard tests (#133073)

This commit is contained in:
Franck Nijhof 2024-12-12 21:12:11 +01:00 committed by GitHub
parent 3baa432bae
commit 839f06b2dc
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 46 additions and 43 deletions

View File

@ -1 +1 @@
"""Tests for the AdGuard Home component.""" """Tests for the AdGuard Home integration."""

View File

@ -59,9 +59,9 @@ async def test_connection_error(
) )
assert result assert result
assert result.get("type") is FlowResultType.FORM assert result["type"] is FlowResultType.FORM
assert result.get("step_id") == "user" assert result["step_id"] == "user"
assert result.get("errors") == {"base": "cannot_connect"} assert result["errors"] == {"base": "cannot_connect"}
async def test_full_flow_implementation( async def test_full_flow_implementation(
@ -83,25 +83,27 @@ async def test_full_flow_implementation(
) )
assert result assert result
assert result.get("flow_id") assert result["flow_id"]
assert result.get("type") is FlowResultType.FORM assert result["type"] is FlowResultType.FORM
assert result.get("step_id") == "user" assert result["step_id"] == "user"
result2 = await hass.config_entries.flow.async_configure( result = await hass.config_entries.flow.async_configure(
result["flow_id"], user_input=FIXTURE_USER_INPUT result["flow_id"], user_input=FIXTURE_USER_INPUT
) )
assert result2 assert result
assert result2.get("type") is FlowResultType.CREATE_ENTRY assert result["type"] is FlowResultType.CREATE_ENTRY
assert result2.get("title") == FIXTURE_USER_INPUT[CONF_HOST]
data = result2.get("data") config_entry = result["result"]
assert data assert config_entry.title == FIXTURE_USER_INPUT[CONF_HOST]
assert data[CONF_HOST] == FIXTURE_USER_INPUT[CONF_HOST] assert config_entry.data == {
assert data[CONF_PASSWORD] == FIXTURE_USER_INPUT[CONF_PASSWORD] CONF_HOST: FIXTURE_USER_INPUT[CONF_HOST],
assert data[CONF_PORT] == FIXTURE_USER_INPUT[CONF_PORT] CONF_PASSWORD: FIXTURE_USER_INPUT[CONF_PASSWORD],
assert data[CONF_SSL] == FIXTURE_USER_INPUT[CONF_SSL] CONF_PORT: FIXTURE_USER_INPUT[CONF_PORT],
assert data[CONF_USERNAME] == FIXTURE_USER_INPUT[CONF_USERNAME] CONF_SSL: FIXTURE_USER_INPUT[CONF_SSL],
assert data[CONF_VERIFY_SSL] == FIXTURE_USER_INPUT[CONF_VERIFY_SSL] CONF_USERNAME: FIXTURE_USER_INPUT[CONF_USERNAME],
CONF_VERIFY_SSL: FIXTURE_USER_INPUT[CONF_VERIFY_SSL],
}
assert not config_entry.options
async def test_integration_already_exists(hass: HomeAssistant) -> None: async def test_integration_already_exists(hass: HomeAssistant) -> None:
@ -116,8 +118,8 @@ async def test_integration_already_exists(hass: HomeAssistant) -> None:
context={"source": config_entries.SOURCE_USER}, context={"source": config_entries.SOURCE_USER},
) )
assert result assert result
assert result.get("type") is FlowResultType.ABORT assert result["type"] is FlowResultType.ABORT
assert result.get("reason") == "already_configured" assert result["reason"] == "already_configured"
async def test_hassio_already_configured(hass: HomeAssistant) -> None: async def test_hassio_already_configured(hass: HomeAssistant) -> None:
@ -141,8 +143,8 @@ async def test_hassio_already_configured(hass: HomeAssistant) -> None:
context={"source": config_entries.SOURCE_HASSIO}, context={"source": config_entries.SOURCE_HASSIO},
) )
assert result assert result
assert result.get("type") is FlowResultType.ABORT assert result["type"] is FlowResultType.ABORT
assert result.get("reason") == "already_configured" assert result["reason"] == "already_configured"
async def test_hassio_ignored(hass: HomeAssistant) -> None: async def test_hassio_ignored(hass: HomeAssistant) -> None:
@ -166,8 +168,8 @@ async def test_hassio_ignored(hass: HomeAssistant) -> None:
context={"source": config_entries.SOURCE_HASSIO}, context={"source": config_entries.SOURCE_HASSIO},
) )
assert result assert result
assert result.get("type") is FlowResultType.ABORT assert result["type"] is FlowResultType.ABORT
assert result.get("reason") == "already_configured" assert result["reason"] == "already_configured"
async def test_hassio_confirm( async def test_hassio_confirm(
@ -195,24 +197,25 @@ async def test_hassio_confirm(
context={"source": config_entries.SOURCE_HASSIO}, context={"source": config_entries.SOURCE_HASSIO},
) )
assert result assert result
assert result.get("type") is FlowResultType.FORM assert result["type"] is FlowResultType.FORM
assert result.get("step_id") == "hassio_confirm" assert result["step_id"] == "hassio_confirm"
assert result.get("description_placeholders") == {"addon": "AdGuard Home Addon"} assert result["description_placeholders"] == {"addon": "AdGuard Home Addon"}
result2 = await hass.config_entries.flow.async_configure(result["flow_id"], {}) result = await hass.config_entries.flow.async_configure(result["flow_id"], {})
assert result2 assert result
assert result2.get("type") is FlowResultType.CREATE_ENTRY assert result["type"] is FlowResultType.CREATE_ENTRY
assert result2.get("title") == "AdGuard Home Addon"
data = result2.get("data") config_entry = result["result"]
assert data assert config_entry.title == "AdGuard Home Addon"
assert data[CONF_HOST] == "mock-adguard" assert config_entry.data == {
assert data[CONF_PASSWORD] is None CONF_HOST: "mock-adguard",
assert data[CONF_PORT] == 3000 CONF_PASSWORD: None,
assert data[CONF_SSL] is False CONF_PORT: 3000,
assert data[CONF_USERNAME] is None CONF_SSL: False,
assert data[CONF_VERIFY_SSL] CONF_USERNAME: None,
CONF_VERIFY_SSL: True,
}
async def test_hassio_connection_error( async def test_hassio_connection_error(
@ -241,6 +244,6 @@ async def test_hassio_connection_error(
result = await hass.config_entries.flow.async_configure(result["flow_id"], {}) result = await hass.config_entries.flow.async_configure(result["flow_id"], {})
assert result assert result
assert result.get("type") is FlowResultType.FORM assert result["type"] is FlowResultType.FORM
assert result.get("step_id") == "hassio_confirm" assert result["step_id"] == "hassio_confirm"
assert result.get("errors") == {"base": "cannot_connect"} assert result["errors"] == {"base": "cannot_connect"}