mirror of
https://github.com/home-assistant/core.git
synced 2025-04-29 03:37:51 +00:00
8aa25af014
* Create tests for sense integration * Rearrange files * Update to use snapshots * Update tests/components/sense/__init__.py Co-authored-by: epenet <6771947+epenet@users.noreply.github.com> * Update tests/components/sense/__init__.py Co-authored-by: epenet <6771947+epenet@users.noreply.github.com> * Update tests/components/sense/test_binary_sensor.py Co-authored-by: epenet <6771947+epenet@users.noreply.github.com> * Update tests/components/sense/test_sensor.py Co-authored-by: epenet <6771947+epenet@users.noreply.github.com> * Add missing imports --------- Co-authored-by: epenet <6771947+epenet@users.noreply.github.com>
71 lines
2.1 KiB
Python
71 lines
2.1 KiB
Python
"""Common methods for Sense."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from collections.abc import Generator
|
|
import datetime
|
|
from unittest.mock import AsyncMock, MagicMock, PropertyMock, patch
|
|
|
|
import pytest
|
|
|
|
from homeassistant.components.sense.const import DOMAIN
|
|
|
|
from .const import (
|
|
DEVICE_1_DATA,
|
|
DEVICE_1_NAME,
|
|
DEVICE_2_DATA,
|
|
DEVICE_2_NAME,
|
|
MOCK_CONFIG,
|
|
MONITOR_ID,
|
|
)
|
|
|
|
from tests.common import MockConfigEntry
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_setup_entry() -> Generator[AsyncMock]:
|
|
"""Override async_setup_entry."""
|
|
with patch(
|
|
"homeassistant.components.sense.async_setup_entry", return_value=True
|
|
) as mock_setup_entry:
|
|
yield mock_setup_entry
|
|
|
|
|
|
@pytest.fixture
|
|
def config_entry() -> MockConfigEntry:
|
|
"""Mock sense config entry."""
|
|
return MockConfigEntry(
|
|
domain=DOMAIN,
|
|
data=MOCK_CONFIG,
|
|
unique_id="test-email",
|
|
)
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_sense() -> Generator[MagicMock]:
|
|
"""Mock an ASyncSenseable object with a split foundation."""
|
|
with patch("homeassistant.components.sense.ASyncSenseable", autospec=True) as mock:
|
|
gateway = mock.return_value
|
|
gateway._devices = [DEVICE_1_NAME, DEVICE_2_NAME]
|
|
gateway.sense_monitor_id = MONITOR_ID
|
|
gateway.get_monitor_data.return_value = None
|
|
gateway.get_discovered_device_data.return_value = [DEVICE_1_DATA, DEVICE_2_DATA]
|
|
gateway.update_realtime.return_value = None
|
|
type(gateway).active_power = PropertyMock(return_value=100)
|
|
type(gateway).active_solar_power = PropertyMock(return_value=500)
|
|
type(gateway).active_voltage = PropertyMock(return_value=[120, 240])
|
|
gateway.get_trend.return_value = 15
|
|
gateway.trend_start.return_value = datetime.datetime.fromisoformat(
|
|
"2024-01-01 01:01:00+00:00"
|
|
)
|
|
|
|
def get_realtime():
|
|
yield {"devices": []}
|
|
yield {"devices": [DEVICE_1_DATA]}
|
|
while True:
|
|
yield {"devices": [DEVICE_1_DATA, DEVICE_2_DATA]}
|
|
|
|
gateway.get_realtime.side_effect = get_realtime()
|
|
|
|
yield gateway
|