70 lines
2.1 KiB
Python
70 lines
2.1 KiB
Python
"""Test the UniFi Protect button platform."""
|
|
# pylint: disable=protected-access
|
|
from __future__ import annotations
|
|
|
|
from unittest.mock import AsyncMock
|
|
|
|
import pytest
|
|
from pyunifiprotect.data import Camera
|
|
|
|
from homeassistant.components.unifiprotect.const import DEFAULT_ATTRIBUTION
|
|
from homeassistant.const import ATTR_ATTRIBUTION, ATTR_ENTITY_ID, Platform
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers import entity_registry as er
|
|
|
|
from .conftest import MockEntityFixture, assert_entity_counts, enable_entity
|
|
|
|
|
|
@pytest.fixture(name="camera")
|
|
async def camera_fixture(
|
|
hass: HomeAssistant, mock_entry: MockEntityFixture, mock_camera: Camera
|
|
):
|
|
"""Fixture for a single camera for testing the button platform."""
|
|
|
|
camera_obj = mock_camera.copy(deep=True)
|
|
camera_obj._api = mock_entry.api
|
|
camera_obj.channels[0]._api = mock_entry.api
|
|
camera_obj.channels[1]._api = mock_entry.api
|
|
camera_obj.channels[2]._api = mock_entry.api
|
|
camera_obj.name = "Test Camera"
|
|
|
|
mock_entry.api.bootstrap.cameras = {
|
|
camera_obj.id: camera_obj,
|
|
}
|
|
|
|
await hass.config_entries.async_setup(mock_entry.entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
assert_entity_counts(hass, Platform.BUTTON, 1, 0)
|
|
|
|
return (camera_obj, "button.test_camera_reboot_device")
|
|
|
|
|
|
async def test_button(
|
|
hass: HomeAssistant,
|
|
mock_entry: MockEntityFixture,
|
|
camera: tuple[Camera, str],
|
|
):
|
|
"""Test button entity."""
|
|
|
|
mock_entry.api.reboot_device = AsyncMock()
|
|
|
|
unique_id = f"{camera[0].id}"
|
|
entity_id = camera[1]
|
|
|
|
entity_registry = er.async_get(hass)
|
|
entity = entity_registry.async_get(entity_id)
|
|
assert entity
|
|
assert entity.disabled
|
|
assert entity.unique_id == unique_id
|
|
|
|
await enable_entity(hass, mock_entry.entry.entry_id, entity_id)
|
|
state = hass.states.get(entity_id)
|
|
assert state
|
|
assert state.attributes[ATTR_ATTRIBUTION] == DEFAULT_ATTRIBUTION
|
|
|
|
await hass.services.async_call(
|
|
"button", "press", {ATTR_ENTITY_ID: entity_id}, blocking=True
|
|
)
|
|
mock_entry.api.reboot_device.assert_called_once()
|