mirror of https://github.com/home-assistant/core
101 lines
3.0 KiB
Python
101 lines
3.0 KiB
Python
"""Common helpers for gree test cases."""
|
|
import asyncio
|
|
import logging
|
|
from unittest.mock import AsyncMock, Mock
|
|
|
|
from greeclimate.discovery import Listener
|
|
|
|
from homeassistant.components.gree.const import DISCOVERY_TIMEOUT, DOMAIN as GREE_DOMAIN
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.setup import async_setup_component
|
|
|
|
from tests.common import MockConfigEntry
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
class FakeDiscovery:
|
|
"""Mock class replacing Gree device discovery."""
|
|
|
|
def __init__(self, timeout: int = DISCOVERY_TIMEOUT) -> None:
|
|
"""Initialize the class."""
|
|
self.mock_devices = [build_device_mock()]
|
|
self.last_mock_infos = []
|
|
self.timeout = timeout
|
|
self._listeners = []
|
|
self.scan_count = 0
|
|
|
|
def add_listener(self, listener: Listener) -> None:
|
|
"""Add an event listener."""
|
|
self._listeners.append(listener)
|
|
|
|
async def scan(self, wait_for: int = 0, bcast_ifaces=None):
|
|
"""Search for devices, return mocked data."""
|
|
self.scan_count += 1
|
|
_LOGGER.info("CALLED SCAN %d TIMES", self.scan_count)
|
|
|
|
mock_infos = [x.device_info for x in self.mock_devices]
|
|
|
|
new_infos = []
|
|
updated_infos = []
|
|
for info in mock_infos:
|
|
if not [i for i in self.last_mock_infos if info.mac == i.mac]:
|
|
new_infos.append(info)
|
|
else:
|
|
last_info = next(i for i in self.last_mock_infos if info.mac == i.mac)
|
|
if info.ip != last_info.ip:
|
|
updated_infos.append(info)
|
|
|
|
self.last_mock_infos = mock_infos
|
|
for listener in self._listeners:
|
|
[await listener.device_found(x) for x in new_infos]
|
|
[await listener.device_update(x) for x in updated_infos]
|
|
|
|
if wait_for:
|
|
await asyncio.sleep(wait_for)
|
|
|
|
return new_infos
|
|
|
|
|
|
def build_device_info_mock(
|
|
name="fake-device-1", ipAddress="1.1.1.1", mac="aabbcc112233"
|
|
):
|
|
"""Build mock device info structure."""
|
|
mock = Mock(ip=ipAddress, port=7000, mac=mac)
|
|
mock.name = name
|
|
return mock
|
|
|
|
|
|
def build_device_mock(name="fake-device-1", ipAddress="1.1.1.1", mac="aabbcc112233"):
|
|
"""Build mock device object."""
|
|
mock = Mock(
|
|
device_info=build_device_info_mock(name, ipAddress, mac),
|
|
name=name,
|
|
bind=AsyncMock(),
|
|
update_state=AsyncMock(),
|
|
push_state_update=AsyncMock(),
|
|
temperature_units=0,
|
|
mode=0,
|
|
fan_speed=0,
|
|
horizontal_swing=0,
|
|
vertical_swing=0,
|
|
target_temperature=25,
|
|
current_temperature=25,
|
|
power=False,
|
|
sleep=False,
|
|
quiet=False,
|
|
turbo=False,
|
|
power_save=False,
|
|
steady_heat=False,
|
|
)
|
|
return mock
|
|
|
|
|
|
async def async_setup_gree(hass: HomeAssistant) -> MockConfigEntry:
|
|
"""Set up the gree platform."""
|
|
entry = MockConfigEntry(domain=GREE_DOMAIN)
|
|
entry.add_to_hass(hass)
|
|
await async_setup_component(hass, GREE_DOMAIN, {GREE_DOMAIN: {"climate": {}}})
|
|
await hass.async_block_till_done()
|
|
return entry
|