mirror of
https://github.com/home-assistant/core.git
synced 2026-01-25 09:02:38 +01:00
117 lines
3.2 KiB
Python
117 lines
3.2 KiB
Python
"""Common fixtures for the Hikvision tests."""
|
|
|
|
from collections.abc import AsyncGenerator, Generator
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
import pytest
|
|
|
|
from homeassistant.components.hikvision import PLATFORMS
|
|
from homeassistant.components.hikvision.const import DOMAIN
|
|
from homeassistant.const import (
|
|
CONF_HOST,
|
|
CONF_PASSWORD,
|
|
CONF_PORT,
|
|
CONF_SSL,
|
|
CONF_USERNAME,
|
|
Platform,
|
|
)
|
|
|
|
from tests.common import MockConfigEntry
|
|
|
|
TEST_HOST = "192.168.1.100"
|
|
TEST_PORT = 80
|
|
TEST_USERNAME = "admin"
|
|
TEST_PASSWORD = "password123"
|
|
TEST_DEVICE_ID = "DS-2CD2142FWD-I20170101AAAA"
|
|
TEST_DEVICE_NAME = "Front Camera"
|
|
|
|
|
|
@pytest.fixture
|
|
def platforms() -> list[Platform]:
|
|
"""Platforms, which should be loaded during the test."""
|
|
return PLATFORMS
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
async def mock_patch_platforms(platforms: list[Platform]) -> AsyncGenerator[None]:
|
|
"""Fixture to set up platforms for tests."""
|
|
with patch(f"homeassistant.components.{DOMAIN}.PLATFORMS", platforms):
|
|
yield
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_setup_entry() -> Generator[MagicMock]:
|
|
"""Override async_setup_entry."""
|
|
with patch(
|
|
"homeassistant.components.hikvision.async_setup_entry", return_value=True
|
|
) as mock_setup_entry:
|
|
yield mock_setup_entry
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_config_entry() -> MockConfigEntry:
|
|
"""Return the default mocked config entry."""
|
|
return MockConfigEntry(
|
|
title=TEST_DEVICE_NAME,
|
|
domain=DOMAIN,
|
|
version=1,
|
|
minor_version=1,
|
|
data={
|
|
CONF_HOST: TEST_HOST,
|
|
CONF_PORT: TEST_PORT,
|
|
CONF_USERNAME: TEST_USERNAME,
|
|
CONF_PASSWORD: TEST_PASSWORD,
|
|
CONF_SSL: False,
|
|
},
|
|
unique_id=TEST_DEVICE_ID,
|
|
)
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_hikcamera() -> Generator[MagicMock]:
|
|
"""Return a mocked HikCamera."""
|
|
with (
|
|
patch(
|
|
"homeassistant.components.hikvision.HikCamera",
|
|
) as hikcamera_mock,
|
|
patch(
|
|
"homeassistant.components.hikvision.config_flow.HikCamera",
|
|
new=hikcamera_mock,
|
|
),
|
|
):
|
|
camera = hikcamera_mock.return_value
|
|
camera.get_id = TEST_DEVICE_ID
|
|
camera.get_name = TEST_DEVICE_NAME
|
|
camera.get_type = "Camera"
|
|
camera.current_event_states = {
|
|
"Motion": [(True, 1)],
|
|
"Line Crossing": [(False, 1)],
|
|
}
|
|
camera.fetch_attributes.return_value = (
|
|
False,
|
|
None,
|
|
None,
|
|
"2024-01-01T00:00:00Z",
|
|
)
|
|
camera.get_event_triggers.return_value = {}
|
|
|
|
# pyHik 0.4.0 methods
|
|
camera.get_channels.return_value = [1]
|
|
camera.get_snapshot.return_value = b"fake_image_data"
|
|
camera.get_stream_url.return_value = (
|
|
f"rtsp://{TEST_USERNAME}:{TEST_PASSWORD}"
|
|
f"@{TEST_HOST}:554/Streaming/Channels/1"
|
|
)
|
|
|
|
yield hikcamera_mock
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_hik_nvr(mock_hikcamera: MagicMock) -> MagicMock:
|
|
"""Return a mocked HikCamera configured as an NVR."""
|
|
camera = mock_hikcamera.return_value
|
|
camera.get_type = "NVR"
|
|
camera.current_event_states = {}
|
|
camera.get_event_triggers.return_value = {"Motion": [1, 2]}
|
|
return mock_hikcamera
|