evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Bump pylitterbot to 2022.9.1 (#78071)

Browse source
This commit is contained in:
Nathan Spencer 2022-09-08 11:13:20 -06:00 • committed by GitHub
parent 01189b023c
commit 03e6bd0811
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
4 changed files with 8 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

8
tests/components/litterrobot/conftest.py
View file

@ -17,13 +17,13 @@ from tests.common import MockConfigEntry
def create_mock_robot(
robot_data: dict | None = None, side_effect: Any | None = None
robot_data: dict | None, account: Account, side_effect: Any | None = None
) -> Robot:
"""Create a mock Litter-Robot device."""
if not robot_data:
robot_data = {}
robot = LitterRobot3(data={**ROBOT_DATA, **robot_data})
robot = LitterRobot3(data={**ROBOT_DATA, **robot_data}, account=account)
robot.start_cleaning = AsyncMock(side_effect=side_effect)
robot.set_power_status = AsyncMock(side_effect=side_effect)
robot.reset_waste_drawer = AsyncMock(side_effect=side_effect)
@ -44,7 +44,9 @@ def create_mock_account(
account = MagicMock(spec=Account)
account.connect = AsyncMock()
account.refresh_robots = AsyncMock()
account.robots = [] if skip_robots else [create_mock_robot(robot_data, side_effect)]
account.robots = (
[] if skip_robots else [create_mock_robot(robot_data, account, side_effect)]
)
return account