mirror of
https://github.com/home-assistant/core.git
synced 2026-02-23 03:17:06 +00:00
c8fc2dc440
* add camera to prosegur * add tests * address review * better tests * clean * clean * fix tests * leftover from merge * sorting missing * Update homeassistant/components/prosegur/services.yaml Co-authored-by: Paulus Schoutsen <paulus@home-assistant.io> --------- Co-authored-by: Paulus Schoutsen <paulus@home-assistant.io>
70 lines
2.1 KiB
Python
70 lines
2.1 KiB
Python
"""The camera tests for the prosegur platform."""
|
|
import logging
|
|
from unittest.mock import AsyncMock
|
|
|
|
from pyprosegur.exceptions import ProsegurException
|
|
import pytest
|
|
|
|
from homeassistant.components import camera
|
|
from homeassistant.components.camera import Image
|
|
from homeassistant.components.prosegur.const import DOMAIN
|
|
from homeassistant.const import ATTR_ENTITY_ID
|
|
from homeassistant.exceptions import HomeAssistantError
|
|
|
|
|
|
async def test_camera(hass, init_integration):
|
|
"""Test prosegur get_image."""
|
|
|
|
image = await camera.async_get_image(hass, "camera.test_cam")
|
|
|
|
assert image == Image(content_type="image/jpeg", content=b"ABC")
|
|
|
|
|
|
async def test_camera_fail(hass, init_integration, mock_install, caplog):
|
|
"""Test prosegur get_image fails."""
|
|
|
|
mock_install.get_image = AsyncMock(
|
|
return_value=b"ABC", side_effect=ProsegurException()
|
|
)
|
|
|
|
with caplog.at_level(logging.ERROR, logger="homeassistant.components.prosegur"):
|
|
try:
|
|
await camera.async_get_image(hass, "camera.test_cam")
|
|
except HomeAssistantError as exc:
|
|
assert str(exc) == "Unable to get image"
|
|
else:
|
|
assert pytest.fail()
|
|
|
|
assert "Image test_cam doesn't exist" in caplog.text
|
|
|
|
|
|
async def test_request_image(hass, init_integration, mock_install):
|
|
"""Test the camera request image service."""
|
|
|
|
await hass.services.async_call(
|
|
DOMAIN,
|
|
"request_image",
|
|
{ATTR_ENTITY_ID: "camera.test_cam"},
|
|
)
|
|
await hass.async_block_till_done()
|
|
|
|
assert mock_install.request_image.called
|
|
|
|
|
|
async def test_request_image_fail(hass, init_integration, mock_install, caplog):
|
|
"""Test the camera request image service fails."""
|
|
|
|
mock_install.request_image = AsyncMock(side_effect=ProsegurException())
|
|
|
|
with caplog.at_level(logging.ERROR, logger="homeassistant.components.prosegur"):
|
|
await hass.services.async_call(
|
|
DOMAIN,
|
|
"request_image",
|
|
{ATTR_ENTITY_ID: "camera.test_cam"},
|
|
)
|
|
await hass.async_block_till_done()
|
|
|
|
assert mock_install.request_image.called
|
|
|
|
assert "Could not request image from camera test_cam" in caplog.text
|