mirror of https://github.com/home-assistant/core
990 lines
34 KiB
Python
990 lines
34 KiB
Python
"""Test The generic (IP Camera) config flow."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import contextlib
|
|
import errno
|
|
from http import HTTPStatus
|
|
import os.path
|
|
from pathlib import Path
|
|
from unittest.mock import AsyncMock, MagicMock, PropertyMock, _patch, patch
|
|
|
|
import httpx
|
|
import pytest
|
|
import respx
|
|
|
|
from homeassistant import config_entries
|
|
from homeassistant.components.camera import async_get_image
|
|
from homeassistant.components.generic.config_flow import slug
|
|
from homeassistant.components.generic.const import (
|
|
CONF_CONFIRMED_OK,
|
|
CONF_CONTENT_TYPE,
|
|
CONF_FRAMERATE,
|
|
CONF_LIMIT_REFETCH_TO_URL_CHANGE,
|
|
CONF_STILL_IMAGE_URL,
|
|
CONF_STREAM_SOURCE,
|
|
DOMAIN,
|
|
)
|
|
from homeassistant.components.stream import (
|
|
CONF_RTSP_TRANSPORT,
|
|
CONF_USE_WALLCLOCK_AS_TIMESTAMPS,
|
|
)
|
|
from homeassistant.components.stream.worker import StreamWorkerError
|
|
from homeassistant.config_entries import ConfigEntryState, ConfigFlowResult
|
|
from homeassistant.const import (
|
|
CONF_AUTHENTICATION,
|
|
CONF_NAME,
|
|
CONF_PASSWORD,
|
|
CONF_USERNAME,
|
|
CONF_VERIFY_SSL,
|
|
HTTP_BASIC_AUTHENTICATION,
|
|
)
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.data_entry_flow import FlowResultType
|
|
from homeassistant.exceptions import HomeAssistantError
|
|
from homeassistant.helpers import entity_registry as er
|
|
|
|
from tests.common import MockConfigEntry
|
|
from tests.typing import ClientSessionGenerator
|
|
|
|
TESTDATA = {
|
|
CONF_STILL_IMAGE_URL: "http://127.0.0.1/testurl/1",
|
|
CONF_STREAM_SOURCE: "http://127.0.0.1/testurl/2",
|
|
CONF_AUTHENTICATION: HTTP_BASIC_AUTHENTICATION,
|
|
CONF_USERNAME: "fred_flintstone",
|
|
CONF_PASSWORD: "bambam",
|
|
CONF_FRAMERATE: 5,
|
|
CONF_VERIFY_SSL: False,
|
|
}
|
|
|
|
TESTDATA_OPTIONS = {
|
|
CONF_LIMIT_REFETCH_TO_URL_CHANGE: False,
|
|
**TESTDATA,
|
|
}
|
|
|
|
TESTDATA_YAML = {
|
|
CONF_NAME: "Yaml Defined Name",
|
|
**TESTDATA,
|
|
}
|
|
|
|
|
|
@respx.mock
|
|
async def test_form(
|
|
hass: HomeAssistant,
|
|
fakeimgbytes_png: bytes,
|
|
hass_client: ClientSessionGenerator,
|
|
user_flow: ConfigFlowResult,
|
|
mock_create_stream: _patch[MagicMock],
|
|
) -> None:
|
|
"""Test the form with a normal set of settings."""
|
|
|
|
respx.get("http://127.0.0.1/testurl/1").respond(stream=fakeimgbytes_png)
|
|
with (
|
|
mock_create_stream as mock_setup,
|
|
patch(
|
|
"homeassistant.components.generic.async_setup_entry", return_value=True
|
|
) as mock_setup_entry,
|
|
):
|
|
result1 = await hass.config_entries.flow.async_configure(
|
|
user_flow["flow_id"],
|
|
TESTDATA,
|
|
)
|
|
assert result1["type"] is FlowResultType.FORM
|
|
assert result1["step_id"] == "user_confirm_still"
|
|
client = await hass_client()
|
|
preview_url = result1["description_placeholders"]["preview_url"]
|
|
# Check the preview image works.
|
|
resp = await client.get(preview_url)
|
|
assert resp.status == HTTPStatus.OK
|
|
assert await resp.read() == fakeimgbytes_png
|
|
result2 = await hass.config_entries.flow.async_configure(
|
|
result1["flow_id"],
|
|
user_input={CONF_CONFIRMED_OK: True},
|
|
)
|
|
await hass.async_block_till_done()
|
|
assert result2["type"] is FlowResultType.CREATE_ENTRY
|
|
assert result2["title"] == "127_0_0_1"
|
|
assert result2["options"] == {
|
|
CONF_STILL_IMAGE_URL: "http://127.0.0.1/testurl/1",
|
|
CONF_STREAM_SOURCE: "http://127.0.0.1/testurl/2",
|
|
CONF_AUTHENTICATION: HTTP_BASIC_AUTHENTICATION,
|
|
CONF_USERNAME: "fred_flintstone",
|
|
CONF_PASSWORD: "bambam",
|
|
CONF_LIMIT_REFETCH_TO_URL_CHANGE: False,
|
|
CONF_CONTENT_TYPE: "image/png",
|
|
CONF_FRAMERATE: 5,
|
|
CONF_VERIFY_SSL: False,
|
|
}
|
|
|
|
await hass.async_block_till_done()
|
|
# Check that the preview image is disabled after.
|
|
resp = await client.get(preview_url)
|
|
assert resp.status == HTTPStatus.NOT_FOUND
|
|
assert len(mock_setup.mock_calls) == 1
|
|
assert len(mock_setup_entry.mock_calls) == 1
|
|
|
|
|
|
@respx.mock
|
|
@pytest.mark.usefixtures("fakeimg_png")
|
|
async def test_form_only_stillimage(
|
|
hass: HomeAssistant, user_flow: ConfigFlowResult
|
|
) -> None:
|
|
"""Test we complete ok if the user wants still images only."""
|
|
result = await hass.config_entries.flow.async_init(
|
|
DOMAIN, context={"source": config_entries.SOURCE_USER}
|
|
)
|
|
assert result["type"] is FlowResultType.FORM
|
|
assert result["errors"] == {}
|
|
|
|
data = TESTDATA.copy()
|
|
data.pop(CONF_STREAM_SOURCE)
|
|
with patch("homeassistant.components.generic.async_setup_entry", return_value=True):
|
|
result1 = await hass.config_entries.flow.async_configure(
|
|
user_flow["flow_id"],
|
|
data,
|
|
)
|
|
await hass.async_block_till_done()
|
|
assert result1["type"] is FlowResultType.FORM
|
|
assert result1["step_id"] == "user_confirm_still"
|
|
result2 = await hass.config_entries.flow.async_configure(
|
|
result1["flow_id"],
|
|
user_input={CONF_CONFIRMED_OK: True},
|
|
)
|
|
assert result2["type"] is FlowResultType.CREATE_ENTRY
|
|
assert result2["title"] == "127_0_0_1"
|
|
assert result2["options"] == {
|
|
CONF_STILL_IMAGE_URL: "http://127.0.0.1/testurl/1",
|
|
CONF_AUTHENTICATION: HTTP_BASIC_AUTHENTICATION,
|
|
CONF_USERNAME: "fred_flintstone",
|
|
CONF_PASSWORD: "bambam",
|
|
CONF_LIMIT_REFETCH_TO_URL_CHANGE: False,
|
|
CONF_CONTENT_TYPE: "image/png",
|
|
CONF_FRAMERATE: 5,
|
|
CONF_VERIFY_SSL: False,
|
|
}
|
|
|
|
assert respx.calls.call_count == 1
|
|
|
|
|
|
@respx.mock
|
|
async def test_form_reject_still_preview(
|
|
hass: HomeAssistant,
|
|
fakeimgbytes_png: bytes,
|
|
mock_create_stream: _patch[MagicMock],
|
|
user_flow: ConfigFlowResult,
|
|
) -> None:
|
|
"""Test we go back to the config screen if the user rejects the still preview."""
|
|
respx.get("http://127.0.0.1/testurl/1").respond(stream=fakeimgbytes_png)
|
|
with mock_create_stream:
|
|
result1 = await hass.config_entries.flow.async_configure(
|
|
user_flow["flow_id"],
|
|
TESTDATA,
|
|
)
|
|
assert result1["type"] is FlowResultType.FORM
|
|
assert result1["step_id"] == "user_confirm_still"
|
|
result2 = await hass.config_entries.flow.async_configure(
|
|
result1["flow_id"],
|
|
user_input={CONF_CONFIRMED_OK: False},
|
|
)
|
|
assert result2["type"] is FlowResultType.FORM
|
|
assert result2["step_id"] == "user"
|
|
|
|
|
|
@respx.mock
|
|
@pytest.mark.usefixtures("fakeimg_png")
|
|
async def test_form_still_preview_cam_off(
|
|
hass: HomeAssistant,
|
|
mock_create_stream: _patch[MagicMock],
|
|
user_flow: ConfigFlowResult,
|
|
hass_client: ClientSessionGenerator,
|
|
) -> None:
|
|
"""Test camera errors are triggered during preview."""
|
|
with (
|
|
patch(
|
|
"homeassistant.components.generic.camera.GenericCamera.is_on",
|
|
new_callable=PropertyMock(return_value=False),
|
|
),
|
|
mock_create_stream,
|
|
):
|
|
result1 = await hass.config_entries.flow.async_configure(
|
|
user_flow["flow_id"],
|
|
TESTDATA,
|
|
)
|
|
assert result1["type"] is FlowResultType.FORM
|
|
assert result1["step_id"] == "user_confirm_still"
|
|
preview_url = result1["description_placeholders"]["preview_url"]
|
|
# Try to view the image, should be unavailable.
|
|
client = await hass_client()
|
|
resp = await client.get(preview_url)
|
|
assert resp.status == HTTPStatus.SERVICE_UNAVAILABLE
|
|
|
|
|
|
@respx.mock
|
|
@pytest.mark.usefixtures("fakeimg_gif")
|
|
async def test_form_only_stillimage_gif(
|
|
hass: HomeAssistant, user_flow: ConfigFlowResult
|
|
) -> None:
|
|
"""Test we complete ok if the user wants a gif."""
|
|
data = TESTDATA.copy()
|
|
data.pop(CONF_STREAM_SOURCE)
|
|
with patch("homeassistant.components.generic.async_setup_entry", return_value=True):
|
|
result1 = await hass.config_entries.flow.async_configure(
|
|
user_flow["flow_id"],
|
|
data,
|
|
)
|
|
assert result1["type"] is FlowResultType.FORM
|
|
assert result1["step_id"] == "user_confirm_still"
|
|
result2 = await hass.config_entries.flow.async_configure(
|
|
result1["flow_id"],
|
|
user_input={CONF_CONFIRMED_OK: True},
|
|
)
|
|
await hass.async_block_till_done()
|
|
assert result2["type"] is FlowResultType.CREATE_ENTRY
|
|
assert result2["options"][CONF_CONTENT_TYPE] == "image/gif"
|
|
|
|
|
|
@respx.mock
|
|
async def test_form_only_svg_whitespace(
|
|
hass: HomeAssistant, fakeimgbytes_svg: bytes, user_flow: ConfigFlowResult
|
|
) -> None:
|
|
"""Test we complete ok if svg starts with whitespace, issue #68889."""
|
|
fakeimgbytes_wspace_svg = bytes(" \n ", encoding="utf-8") + fakeimgbytes_svg
|
|
respx.get("http://127.0.0.1/testurl/1").respond(stream=fakeimgbytes_wspace_svg)
|
|
data = TESTDATA.copy()
|
|
data.pop(CONF_STREAM_SOURCE)
|
|
with patch("homeassistant.components.generic.async_setup_entry", return_value=True):
|
|
result1 = await hass.config_entries.flow.async_configure(
|
|
user_flow["flow_id"],
|
|
data,
|
|
)
|
|
assert result1["type"] is FlowResultType.FORM
|
|
assert result1["step_id"] == "user_confirm_still"
|
|
result2 = await hass.config_entries.flow.async_configure(
|
|
result1["flow_id"],
|
|
user_input={CONF_CONFIRMED_OK: True},
|
|
)
|
|
await hass.async_block_till_done()
|
|
assert result2["type"] is FlowResultType.CREATE_ENTRY
|
|
|
|
|
|
@respx.mock
|
|
@pytest.mark.parametrize(
|
|
"image_file",
|
|
[
|
|
("sample1_animate.png"),
|
|
("sample2_jpeg_odd_header.jpg"),
|
|
("sample3_jpeg_odd_header.jpg"),
|
|
("sample4_K5-60mileAnim-320x240.gif"),
|
|
("sample5_webp.webp"),
|
|
],
|
|
)
|
|
async def test_form_only_still_sample(
|
|
hass: HomeAssistant, user_flow: ConfigFlowResult, image_file
|
|
) -> None:
|
|
"""Test various sample images #69037."""
|
|
image_path = os.path.join(os.path.dirname(__file__), image_file)
|
|
image_bytes = await hass.async_add_executor_job(Path(image_path).read_bytes)
|
|
respx.get("http://127.0.0.1/testurl/1").respond(stream=image_bytes)
|
|
data = TESTDATA.copy()
|
|
data.pop(CONF_STREAM_SOURCE)
|
|
with patch("homeassistant.components.generic.async_setup_entry", return_value=True):
|
|
result1 = await hass.config_entries.flow.async_configure(
|
|
user_flow["flow_id"],
|
|
data,
|
|
)
|
|
assert result1["type"] is FlowResultType.FORM
|
|
assert result1["step_id"] == "user_confirm_still"
|
|
result2 = await hass.config_entries.flow.async_configure(
|
|
result1["flow_id"],
|
|
user_input={CONF_CONFIRMED_OK: True},
|
|
)
|
|
await hass.async_block_till_done()
|
|
assert result2["type"] is FlowResultType.CREATE_ENTRY
|
|
|
|
|
|
@respx.mock
|
|
@pytest.mark.parametrize(
|
|
("template", "url", "expected_result", "expected_errors"),
|
|
[
|
|
# Test we can handle templates in strange parts of the url, #70961.
|
|
(
|
|
"http://localhost:812{{3}}/static/icons/favicon-apple-180x180.png",
|
|
"http://localhost:8123/static/icons/favicon-apple-180x180.png",
|
|
"user_confirm_still",
|
|
None,
|
|
),
|
|
(
|
|
"{% if 1 %}https://bla{% else %}https://yo{% endif %}",
|
|
"https://bla/",
|
|
"user_confirm_still",
|
|
None,
|
|
),
|
|
(
|
|
"http://{{example.org",
|
|
"http://example.org",
|
|
"user",
|
|
{"still_image_url": "template_error"},
|
|
),
|
|
(
|
|
"invalid1://invalid:4\\1",
|
|
"invalid1://invalid:4%5c1",
|
|
"user",
|
|
{"still_image_url": "malformed_url"},
|
|
),
|
|
(
|
|
"relative/urls/are/not/allowed.jpg",
|
|
"relative/urls/are/not/allowed.jpg",
|
|
"user",
|
|
{"still_image_url": "relative_url"},
|
|
),
|
|
],
|
|
)
|
|
async def test_still_template(
|
|
hass: HomeAssistant,
|
|
user_flow: ConfigFlowResult,
|
|
fakeimgbytes_png: bytes,
|
|
template,
|
|
url,
|
|
expected_result,
|
|
expected_errors,
|
|
) -> None:
|
|
"""Test we can handle various templates."""
|
|
with contextlib.suppress(httpx.InvalidURL):
|
|
# There is no need to mock the request if its an
|
|
# invalid url because we will never make the request
|
|
respx.get(url).respond(stream=fakeimgbytes_png)
|
|
data = TESTDATA.copy()
|
|
data.pop(CONF_STREAM_SOURCE)
|
|
data[CONF_STILL_IMAGE_URL] = template
|
|
with patch("homeassistant.components.generic.async_setup_entry", return_value=True):
|
|
result2 = await hass.config_entries.flow.async_configure(
|
|
user_flow["flow_id"],
|
|
data,
|
|
)
|
|
await hass.async_block_till_done()
|
|
assert result2["step_id"] == expected_result
|
|
assert result2.get("errors") == expected_errors
|
|
|
|
|
|
@respx.mock
|
|
@pytest.mark.usefixtures("fakeimg_png")
|
|
async def test_form_rtsp_mode(
|
|
hass: HomeAssistant,
|
|
user_flow: ConfigFlowResult,
|
|
mock_create_stream: _patch[MagicMock],
|
|
) -> None:
|
|
"""Test we complete ok if the user enters a stream url."""
|
|
data = TESTDATA.copy()
|
|
data[CONF_RTSP_TRANSPORT] = "tcp"
|
|
data[CONF_STREAM_SOURCE] = "rtsp://127.0.0.1/testurl/2"
|
|
with (
|
|
mock_create_stream as mock_setup,
|
|
patch("homeassistant.components.generic.async_setup_entry", return_value=True),
|
|
):
|
|
result1 = await hass.config_entries.flow.async_configure(
|
|
user_flow["flow_id"], data
|
|
)
|
|
assert result1["type"] is FlowResultType.FORM
|
|
assert result1["step_id"] == "user_confirm_still"
|
|
result2 = await hass.config_entries.flow.async_configure(
|
|
result1["flow_id"],
|
|
user_input={CONF_CONFIRMED_OK: True},
|
|
)
|
|
assert result2["type"] is FlowResultType.CREATE_ENTRY
|
|
assert result2["title"] == "127_0_0_1"
|
|
assert result2["options"] == {
|
|
CONF_STILL_IMAGE_URL: "http://127.0.0.1/testurl/1",
|
|
CONF_AUTHENTICATION: HTTP_BASIC_AUTHENTICATION,
|
|
CONF_STREAM_SOURCE: "rtsp://127.0.0.1/testurl/2",
|
|
CONF_RTSP_TRANSPORT: "tcp",
|
|
CONF_USERNAME: "fred_flintstone",
|
|
CONF_PASSWORD: "bambam",
|
|
CONF_LIMIT_REFETCH_TO_URL_CHANGE: False,
|
|
CONF_CONTENT_TYPE: "image/png",
|
|
CONF_FRAMERATE: 5,
|
|
CONF_VERIFY_SSL: False,
|
|
}
|
|
|
|
await hass.async_block_till_done()
|
|
assert len(mock_setup.mock_calls) == 1
|
|
|
|
|
|
async def test_form_only_stream(
|
|
hass: HomeAssistant,
|
|
fakeimgbytes_jpg: bytes,
|
|
user_flow: ConfigFlowResult,
|
|
mock_create_stream: _patch[MagicMock],
|
|
) -> None:
|
|
"""Test we complete ok if the user wants stream only."""
|
|
data = TESTDATA.copy()
|
|
data.pop(CONF_STILL_IMAGE_URL)
|
|
data[CONF_STREAM_SOURCE] = "rtsp://user:pass@127.0.0.1/testurl/2"
|
|
with mock_create_stream as mock_setup:
|
|
result1 = await hass.config_entries.flow.async_configure(
|
|
user_flow["flow_id"],
|
|
data,
|
|
)
|
|
assert result1["type"] is FlowResultType.CREATE_ENTRY
|
|
assert result1["title"] == "127_0_0_1"
|
|
assert result1["options"] == {
|
|
CONF_AUTHENTICATION: HTTP_BASIC_AUTHENTICATION,
|
|
CONF_STREAM_SOURCE: "rtsp://user:pass@127.0.0.1/testurl/2",
|
|
CONF_USERNAME: "fred_flintstone",
|
|
CONF_PASSWORD: "bambam",
|
|
CONF_LIMIT_REFETCH_TO_URL_CHANGE: False,
|
|
CONF_CONTENT_TYPE: "image/jpeg",
|
|
CONF_FRAMERATE: 5,
|
|
CONF_VERIFY_SSL: False,
|
|
}
|
|
|
|
await hass.async_block_till_done()
|
|
|
|
with patch(
|
|
"homeassistant.components.camera._async_get_stream_image",
|
|
return_value=fakeimgbytes_jpg,
|
|
):
|
|
image_obj = await async_get_image(hass, "camera.127_0_0_1")
|
|
assert image_obj.content == fakeimgbytes_jpg
|
|
assert len(mock_setup.mock_calls) == 1
|
|
|
|
|
|
async def test_form_still_and_stream_not_provided(
|
|
hass: HomeAssistant, user_flow: ConfigFlowResult
|
|
) -> None:
|
|
"""Test we show a suitable error if neither still or stream URL are provided."""
|
|
result2 = await hass.config_entries.flow.async_configure(
|
|
user_flow["flow_id"],
|
|
{
|
|
CONF_AUTHENTICATION: HTTP_BASIC_AUTHENTICATION,
|
|
CONF_FRAMERATE: 5,
|
|
CONF_VERIFY_SSL: False,
|
|
},
|
|
)
|
|
assert result2["type"] is FlowResultType.FORM
|
|
assert result2["errors"] == {"base": "no_still_image_or_stream_url"}
|
|
|
|
|
|
@respx.mock
|
|
@pytest.mark.parametrize(
|
|
("side_effect", "expected_message"),
|
|
[
|
|
(httpx.TimeoutException, {"still_image_url": "unable_still_load"}),
|
|
(
|
|
httpx.HTTPStatusError("", request=None, response=httpx.Response(401)),
|
|
{"still_image_url": "unable_still_load_auth"},
|
|
),
|
|
(
|
|
httpx.HTTPStatusError("", request=None, response=httpx.Response(403)),
|
|
{"still_image_url": "unable_still_load_auth"},
|
|
),
|
|
(
|
|
httpx.HTTPStatusError("", request=None, response=httpx.Response(404)),
|
|
{"still_image_url": "unable_still_load_not_found"},
|
|
),
|
|
(
|
|
httpx.HTTPStatusError("", request=None, response=httpx.Response(500)),
|
|
{"still_image_url": "unable_still_load_server_error"},
|
|
),
|
|
(
|
|
httpx.HTTPStatusError("", request=None, response=httpx.Response(503)),
|
|
{"still_image_url": "unable_still_load_server_error"},
|
|
),
|
|
( # Errors without specific handler should show the general message.
|
|
httpx.HTTPStatusError("", request=None, response=httpx.Response(507)),
|
|
{"still_image_url": "unable_still_load"},
|
|
),
|
|
],
|
|
)
|
|
async def test_form_image_http_exceptions(
|
|
side_effect,
|
|
expected_message,
|
|
hass: HomeAssistant,
|
|
user_flow: ConfigFlowResult,
|
|
mock_create_stream: _patch[MagicMock],
|
|
) -> None:
|
|
"""Test we handle image http exceptions."""
|
|
respx.get("http://127.0.0.1/testurl/1").side_effect = [
|
|
side_effect,
|
|
]
|
|
|
|
with mock_create_stream:
|
|
result2 = await hass.config_entries.flow.async_configure(
|
|
user_flow["flow_id"],
|
|
TESTDATA,
|
|
)
|
|
await hass.async_block_till_done()
|
|
|
|
assert result2["type"] is FlowResultType.FORM
|
|
assert result2["errors"] == expected_message
|
|
|
|
|
|
@respx.mock
|
|
async def test_form_stream_invalidimage(
|
|
hass: HomeAssistant,
|
|
user_flow: ConfigFlowResult,
|
|
mock_create_stream: _patch[MagicMock],
|
|
) -> None:
|
|
"""Test we handle invalid image when a stream is specified."""
|
|
respx.get("http://127.0.0.1/testurl/1").respond(stream=b"invalid")
|
|
with mock_create_stream:
|
|
result2 = await hass.config_entries.flow.async_configure(
|
|
user_flow["flow_id"],
|
|
TESTDATA,
|
|
)
|
|
await hass.async_block_till_done()
|
|
|
|
assert result2["type"] is FlowResultType.FORM
|
|
assert result2["errors"] == {"still_image_url": "invalid_still_image"}
|
|
|
|
|
|
@respx.mock
|
|
async def test_form_stream_invalidimage2(
|
|
hass: HomeAssistant,
|
|
user_flow: ConfigFlowResult,
|
|
mock_create_stream: _patch[MagicMock],
|
|
) -> None:
|
|
"""Test we handle invalid image when a stream is specified."""
|
|
respx.get("http://127.0.0.1/testurl/1").respond(content=None)
|
|
with mock_create_stream:
|
|
result2 = await hass.config_entries.flow.async_configure(
|
|
user_flow["flow_id"],
|
|
TESTDATA,
|
|
)
|
|
await hass.async_block_till_done()
|
|
|
|
assert result2["type"] is FlowResultType.FORM
|
|
assert result2["errors"] == {"still_image_url": "unable_still_load_no_image"}
|
|
|
|
|
|
@respx.mock
|
|
async def test_form_stream_invalidimage3(
|
|
hass: HomeAssistant,
|
|
user_flow: ConfigFlowResult,
|
|
mock_create_stream: _patch[MagicMock],
|
|
) -> None:
|
|
"""Test we handle invalid image when a stream is specified."""
|
|
respx.get("http://127.0.0.1/testurl/1").respond(content=bytes([0xFF]))
|
|
with mock_create_stream:
|
|
result2 = await hass.config_entries.flow.async_configure(
|
|
user_flow["flow_id"],
|
|
TESTDATA,
|
|
)
|
|
await hass.async_block_till_done()
|
|
|
|
assert result2["type"] is FlowResultType.FORM
|
|
assert result2["errors"] == {"still_image_url": "invalid_still_image"}
|
|
|
|
|
|
@respx.mock
|
|
@pytest.mark.usefixtures("fakeimg_png")
|
|
async def test_form_stream_timeout(
|
|
hass: HomeAssistant, user_flow: ConfigFlowResult
|
|
) -> None:
|
|
"""Test we handle invalid auth."""
|
|
with patch(
|
|
"homeassistant.components.generic.config_flow.create_stream"
|
|
) as create_stream:
|
|
create_stream.return_value.start = AsyncMock()
|
|
create_stream.return_value.add_provider.return_value.part_recv = AsyncMock()
|
|
create_stream.return_value.add_provider.return_value.part_recv.return_value = (
|
|
False
|
|
)
|
|
result2 = await hass.config_entries.flow.async_configure(
|
|
user_flow["flow_id"],
|
|
TESTDATA,
|
|
)
|
|
assert result2["type"] is FlowResultType.FORM
|
|
assert result2["errors"] == {"stream_source": "timeout"}
|
|
|
|
|
|
@respx.mock
|
|
async def test_form_stream_not_set_up(hass: HomeAssistant, user_flow) -> None:
|
|
"""Test we handle if stream has not been set up."""
|
|
TESTDATA_ONLY_STREAM = TESTDATA.copy()
|
|
TESTDATA_ONLY_STREAM.pop(CONF_STILL_IMAGE_URL)
|
|
|
|
with patch(
|
|
"homeassistant.components.generic.config_flow.create_stream",
|
|
side_effect=HomeAssistantError("Stream integration is not set up."),
|
|
):
|
|
result1 = await hass.config_entries.flow.async_configure(
|
|
user_flow["flow_id"],
|
|
TESTDATA_ONLY_STREAM,
|
|
)
|
|
await hass.async_block_till_done()
|
|
|
|
assert result1["type"] is FlowResultType.FORM
|
|
assert result1["errors"] == {"stream_source": "stream_not_set_up"}
|
|
|
|
|
|
@respx.mock
|
|
async def test_form_stream_other_error(hass: HomeAssistant, user_flow) -> None:
|
|
"""Test the unknown error for streams."""
|
|
TESTDATA_ONLY_STREAM = TESTDATA.copy()
|
|
TESTDATA_ONLY_STREAM.pop(CONF_STILL_IMAGE_URL)
|
|
|
|
with (
|
|
patch(
|
|
"homeassistant.components.generic.config_flow.create_stream",
|
|
side_effect=HomeAssistantError("Some other error."),
|
|
),
|
|
pytest.raises(HomeAssistantError),
|
|
):
|
|
await hass.config_entries.flow.async_configure(
|
|
user_flow["flow_id"],
|
|
TESTDATA_ONLY_STREAM,
|
|
)
|
|
await hass.async_block_till_done()
|
|
|
|
|
|
@respx.mock
|
|
@pytest.mark.usefixtures("fakeimg_png")
|
|
async def test_form_stream_worker_error(
|
|
hass: HomeAssistant, user_flow: ConfigFlowResult
|
|
) -> None:
|
|
"""Test we handle a StreamWorkerError and pass the message through."""
|
|
with patch(
|
|
"homeassistant.components.generic.config_flow.create_stream",
|
|
side_effect=StreamWorkerError("Some message"),
|
|
):
|
|
result2 = await hass.config_entries.flow.async_configure(
|
|
user_flow["flow_id"],
|
|
TESTDATA,
|
|
)
|
|
assert result2["type"] is FlowResultType.FORM
|
|
assert result2["errors"] == {"stream_source": "unknown_with_details"}
|
|
assert result2["description_placeholders"] == {"error": "Some message"}
|
|
|
|
|
|
@respx.mock
|
|
async def test_form_stream_permission_error(
|
|
hass: HomeAssistant, fakeimgbytes_png: bytes, user_flow: ConfigFlowResult
|
|
) -> None:
|
|
"""Test we handle permission error."""
|
|
respx.get("http://127.0.0.1/testurl/1").respond(stream=fakeimgbytes_png)
|
|
with patch(
|
|
"homeassistant.components.generic.config_flow.create_stream",
|
|
side_effect=PermissionError(),
|
|
):
|
|
result2 = await hass.config_entries.flow.async_configure(
|
|
user_flow["flow_id"],
|
|
TESTDATA,
|
|
)
|
|
assert result2["type"] is FlowResultType.FORM
|
|
assert result2["errors"] == {"stream_source": "stream_not_permitted"}
|
|
|
|
|
|
@respx.mock
|
|
@pytest.mark.usefixtures("fakeimg_png")
|
|
async def test_form_no_route_to_host(
|
|
hass: HomeAssistant, user_flow: ConfigFlowResult
|
|
) -> None:
|
|
"""Test we handle no route to host."""
|
|
with patch(
|
|
"homeassistant.components.generic.config_flow.create_stream",
|
|
side_effect=OSError(errno.EHOSTUNREACH, "No route to host"),
|
|
):
|
|
result2 = await hass.config_entries.flow.async_configure(
|
|
user_flow["flow_id"],
|
|
TESTDATA,
|
|
)
|
|
assert result2["type"] is FlowResultType.FORM
|
|
assert result2["errors"] == {"stream_source": "stream_no_route_to_host"}
|
|
|
|
|
|
@respx.mock
|
|
@pytest.mark.usefixtures("fakeimg_png")
|
|
async def test_form_stream_io_error(
|
|
hass: HomeAssistant, user_flow: ConfigFlowResult
|
|
) -> None:
|
|
"""Test we handle no io error when setting up stream."""
|
|
with patch(
|
|
"homeassistant.components.generic.config_flow.create_stream",
|
|
side_effect=OSError(errno.EIO, "Input/output error"),
|
|
):
|
|
result2 = await hass.config_entries.flow.async_configure(
|
|
user_flow["flow_id"],
|
|
TESTDATA,
|
|
)
|
|
assert result2["type"] is FlowResultType.FORM
|
|
assert result2["errors"] == {"stream_source": "stream_io_error"}
|
|
|
|
|
|
@respx.mock
|
|
@pytest.mark.usefixtures("fakeimg_png")
|
|
async def test_form_oserror(hass: HomeAssistant, user_flow: ConfigFlowResult) -> None:
|
|
"""Test we handle OS error when setting up stream."""
|
|
with (
|
|
patch(
|
|
"homeassistant.components.generic.config_flow.create_stream",
|
|
side_effect=OSError("Some other OSError"),
|
|
),
|
|
pytest.raises(OSError),
|
|
):
|
|
await hass.config_entries.flow.async_configure(
|
|
user_flow["flow_id"],
|
|
TESTDATA,
|
|
)
|
|
|
|
|
|
@respx.mock
|
|
async def test_options_template_error(
|
|
hass: HomeAssistant, fakeimgbytes_png: bytes, mock_create_stream: _patch[MagicMock]
|
|
) -> None:
|
|
"""Test the options flow with a template error."""
|
|
respx.get("http://127.0.0.1/testurl/1").respond(stream=fakeimgbytes_png)
|
|
respx.get("http://127.0.0.1/testurl/2").respond(stream=fakeimgbytes_png)
|
|
|
|
mock_entry = MockConfigEntry(
|
|
title="Test Camera",
|
|
domain=DOMAIN,
|
|
data={},
|
|
options=TESTDATA,
|
|
)
|
|
|
|
mock_entry.add_to_hass(hass)
|
|
await hass.config_entries.async_setup(mock_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
result = await hass.config_entries.options.async_init(mock_entry.entry_id)
|
|
assert result["type"] is FlowResultType.FORM
|
|
assert result["step_id"] == "init"
|
|
|
|
# try updating the still image url
|
|
data = TESTDATA.copy()
|
|
data[CONF_STILL_IMAGE_URL] = "http://127.0.0.1/testurl/2"
|
|
with mock_create_stream:
|
|
result2 = await hass.config_entries.options.async_configure(
|
|
result["flow_id"],
|
|
user_input=data,
|
|
)
|
|
assert result2["type"] is FlowResultType.FORM
|
|
assert result2["step_id"] == "confirm_still"
|
|
|
|
result2a = await hass.config_entries.options.async_configure(
|
|
result2["flow_id"], user_input={CONF_CONFIRMED_OK: True}
|
|
)
|
|
assert result2a["type"] is FlowResultType.CREATE_ENTRY
|
|
|
|
result3 = await hass.config_entries.options.async_init(mock_entry.entry_id)
|
|
assert result3["type"] is FlowResultType.FORM
|
|
assert result3["step_id"] == "init"
|
|
|
|
# verify that an invalid template reports the correct UI error.
|
|
data[CONF_STILL_IMAGE_URL] = "http://127.0.0.1/testurl/{{1/0}}"
|
|
result4 = await hass.config_entries.options.async_configure(
|
|
result3["flow_id"],
|
|
user_input=data,
|
|
)
|
|
assert result4.get("type") is FlowResultType.FORM
|
|
assert result4["errors"] == {"still_image_url": "template_error"}
|
|
|
|
# verify that an invalid template reports the correct UI error.
|
|
data[CONF_STILL_IMAGE_URL] = "http://127.0.0.1/testurl/1"
|
|
data[CONF_STREAM_SOURCE] = "http://127.0.0.2/testurl/{{1/0}}"
|
|
result5 = await hass.config_entries.options.async_configure(
|
|
result4["flow_id"],
|
|
user_input=data,
|
|
)
|
|
|
|
assert result5.get("type") is FlowResultType.FORM
|
|
assert result5["errors"] == {"stream_source": "template_error"}
|
|
|
|
# verify that an relative stream url is rejected.
|
|
data[CONF_STILL_IMAGE_URL] = "http://127.0.0.1/testurl/1"
|
|
data[CONF_STREAM_SOURCE] = "relative/stream.mjpeg"
|
|
result6 = await hass.config_entries.options.async_configure(
|
|
result5["flow_id"],
|
|
user_input=data,
|
|
)
|
|
assert result6.get("type") is FlowResultType.FORM
|
|
assert result6["errors"] == {"stream_source": "relative_url"}
|
|
|
|
# verify that an malformed stream url is rejected.
|
|
data[CONF_STILL_IMAGE_URL] = "http://127.0.0.1/testurl/1"
|
|
data[CONF_STREAM_SOURCE] = "http://example.com:45:56"
|
|
result7 = await hass.config_entries.options.async_configure(
|
|
result6["flow_id"],
|
|
user_input=data,
|
|
)
|
|
assert result7.get("type") is FlowResultType.FORM
|
|
assert result7["errors"] == {"stream_source": "malformed_url"}
|
|
|
|
|
|
async def test_slug(hass: HomeAssistant, caplog: pytest.LogCaptureFixture) -> None:
|
|
"""Test that the slug function generates an error in case of invalid template.
|
|
|
|
Other paths in the slug function are already tested by other tests.
|
|
"""
|
|
result = slug(hass, "http://127.0.0.2/testurl/{{1/0}}")
|
|
assert result is None
|
|
assert "Syntax error in" in caplog.text
|
|
|
|
result = slug(hass, "http://example.com:999999999999/stream")
|
|
assert result is None
|
|
assert "Syntax error in" in caplog.text
|
|
|
|
|
|
@respx.mock
|
|
async def test_options_only_stream(
|
|
hass: HomeAssistant, fakeimgbytes_png: bytes, mock_create_stream: _patch[MagicMock]
|
|
) -> None:
|
|
"""Test the options flow without a still_image_url."""
|
|
respx.get("http://127.0.0.1/testurl/2").respond(stream=fakeimgbytes_png)
|
|
data = TESTDATA.copy()
|
|
data.pop(CONF_STILL_IMAGE_URL)
|
|
|
|
mock_entry = MockConfigEntry(
|
|
title="Test Camera",
|
|
domain=DOMAIN,
|
|
data={},
|
|
options=data,
|
|
)
|
|
mock_entry.add_to_hass(hass)
|
|
await hass.config_entries.async_setup(mock_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
result = await hass.config_entries.options.async_init(mock_entry.entry_id)
|
|
assert result["type"] is FlowResultType.FORM
|
|
assert result["step_id"] == "init"
|
|
|
|
# try updating the config options
|
|
with mock_create_stream:
|
|
result2 = await hass.config_entries.options.async_configure(
|
|
result["flow_id"],
|
|
user_input=data,
|
|
)
|
|
assert result2["type"] is FlowResultType.FORM
|
|
assert result2["step_id"] == "confirm_still"
|
|
|
|
result3 = await hass.config_entries.options.async_configure(
|
|
result2["flow_id"], user_input={CONF_CONFIRMED_OK: True}
|
|
)
|
|
assert result3["type"] is FlowResultType.CREATE_ENTRY
|
|
assert result3["data"][CONF_CONTENT_TYPE] == "image/jpeg"
|
|
|
|
|
|
@pytest.mark.usefixtures("fakeimg_png")
|
|
async def test_unload_entry(hass: HomeAssistant) -> None:
|
|
"""Test unloading the generic IP Camera entry."""
|
|
mock_entry = MockConfigEntry(domain=DOMAIN, options=TESTDATA)
|
|
mock_entry.add_to_hass(hass)
|
|
|
|
await hass.config_entries.async_setup(mock_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
assert mock_entry.state is ConfigEntryState.LOADED
|
|
|
|
await hass.config_entries.async_unload(mock_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
assert mock_entry.state is ConfigEntryState.NOT_LOADED
|
|
|
|
|
|
async def test_reload_on_title_change(hass: HomeAssistant) -> None:
|
|
"""Test the integration gets reloaded when the title is updated."""
|
|
|
|
test_data = TESTDATA_OPTIONS
|
|
test_data[CONF_CONTENT_TYPE] = "image/png"
|
|
mock_entry = MockConfigEntry(
|
|
domain=DOMAIN, unique_id="54321", options=test_data, title="My Title"
|
|
)
|
|
mock_entry.add_to_hass(hass)
|
|
|
|
await hass.config_entries.async_setup(mock_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
assert mock_entry.state is ConfigEntryState.LOADED
|
|
assert hass.states.get("camera.my_title").attributes["friendly_name"] == "My Title"
|
|
|
|
hass.config_entries.async_update_entry(mock_entry, title="New Title")
|
|
assert mock_entry.title == "New Title"
|
|
await hass.async_block_till_done()
|
|
|
|
assert hass.states.get("camera.my_title").attributes["friendly_name"] == "New Title"
|
|
|
|
|
|
async def test_migrate_existing_ids(
|
|
hass: HomeAssistant, entity_registry: er.EntityRegistry
|
|
) -> None:
|
|
"""Test that existing ids are migrated for issue #70568."""
|
|
|
|
test_data = TESTDATA_OPTIONS.copy()
|
|
test_data[CONF_CONTENT_TYPE] = "image/png"
|
|
old_unique_id = "54321"
|
|
entity_id = "camera.sample_camera"
|
|
|
|
mock_entry = MockConfigEntry(
|
|
domain=DOMAIN, unique_id=old_unique_id, options=test_data, title="My Title"
|
|
)
|
|
new_unique_id = mock_entry.entry_id
|
|
mock_entry.add_to_hass(hass)
|
|
|
|
entity_entry = entity_registry.async_get_or_create(
|
|
"camera",
|
|
DOMAIN,
|
|
old_unique_id,
|
|
suggested_object_id="sample camera",
|
|
config_entry=mock_entry,
|
|
)
|
|
assert entity_entry.entity_id == entity_id
|
|
assert entity_entry.unique_id == old_unique_id
|
|
|
|
await hass.config_entries.async_setup(mock_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
entity_entry = entity_registry.async_get(entity_id)
|
|
assert entity_entry.unique_id == new_unique_id
|
|
|
|
|
|
@respx.mock
|
|
@pytest.mark.usefixtures("fakeimg_png")
|
|
async def test_use_wallclock_as_timestamps_option(
|
|
hass: HomeAssistant, mock_create_stream: _patch[MagicMock]
|
|
) -> None:
|
|
"""Test the use_wallclock_as_timestamps option flow."""
|
|
|
|
mock_entry = MockConfigEntry(
|
|
title="Test Camera",
|
|
domain=DOMAIN,
|
|
data={},
|
|
options=TESTDATA,
|
|
)
|
|
|
|
mock_entry.add_to_hass(hass)
|
|
await hass.config_entries.async_setup(mock_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
result = await hass.config_entries.options.async_init(
|
|
mock_entry.entry_id, context={"show_advanced_options": True}
|
|
)
|
|
assert result["type"] is FlowResultType.FORM
|
|
assert result["step_id"] == "init"
|
|
with (
|
|
patch("homeassistant.components.generic.async_setup_entry", return_value=True),
|
|
mock_create_stream,
|
|
):
|
|
result2 = await hass.config_entries.options.async_configure(
|
|
result["flow_id"],
|
|
user_input={CONF_USE_WALLCLOCK_AS_TIMESTAMPS: True, **TESTDATA},
|
|
)
|
|
assert result2["type"] is FlowResultType.FORM
|
|
# Test what happens if user rejects the preview
|
|
result3 = await hass.config_entries.options.async_configure(
|
|
result2["flow_id"], user_input={CONF_CONFIRMED_OK: False}
|
|
)
|
|
assert result3["type"] is FlowResultType.FORM
|
|
assert result3["step_id"] == "init"
|
|
with (
|
|
patch("homeassistant.components.generic.async_setup_entry", return_value=True),
|
|
mock_create_stream,
|
|
):
|
|
result4 = await hass.config_entries.options.async_configure(
|
|
result3["flow_id"],
|
|
user_input={CONF_USE_WALLCLOCK_AS_TIMESTAMPS: True, **TESTDATA},
|
|
)
|
|
assert result4["type"] is FlowResultType.FORM
|
|
assert result4["step_id"] == "confirm_still"
|
|
result5 = await hass.config_entries.options.async_configure(
|
|
result4["flow_id"],
|
|
user_input={CONF_CONFIRMED_OK: True},
|
|
)
|
|
assert result5["type"] is FlowResultType.CREATE_ENTRY
|