1
0
mirror of https://github.com/home-assistant/core.git synced 2025-12-24 12:59:34 +00:00

Add litterrobot integration (#45886)

This commit is contained in:
Nathan Spencer
2021-02-22 11:53:57 -07:00
committed by GitHub
parent 668574c48f
commit e70d896e1b
18 changed files with 676 additions and 0 deletions

View File

@@ -0,0 +1,35 @@
"""Configure pytest for Litter-Robot tests."""
from unittest.mock import AsyncMock, MagicMock
from pylitterbot import Robot
import pytest
from homeassistant.components import litterrobot
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
from .common import ROBOT_DATA
def create_mock_robot(hass):
"""Create a mock Litter-Robot device."""
robot = Robot(data=ROBOT_DATA)
robot.start_cleaning = AsyncMock()
robot.set_power_status = AsyncMock()
robot.reset_waste_drawer = AsyncMock()
robot.set_sleep_mode = AsyncMock()
return robot
@pytest.fixture()
def mock_hub(hass):
"""Mock a Litter-Robot hub."""
hub = MagicMock(
hass=hass,
account=MagicMock(),
logged_in=True,
coordinator=MagicMock(spec=DataUpdateCoordinator),
spec=litterrobot.LitterRobotHub,
)
hub.coordinator.last_update_success = True
hub.account.robots = [create_mock_robot(hass)]
return hub