Home-Assistant
/
core
mirror of https://github.com/home-assistant/core.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Migrate xiaomi_miio tests from coroutine to async/await (#30329)

pull/30341/head
Franck Nijhof 2019-12-31 23:49:00 +01:00 committed by Andrew Sayre
parent 2ac5537495
commit 194cb8dbf5
1 changed files with 24 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

52
tests/components/xiaomi_miio/test_vacuum.py
View File

@ -1,5 +1,4 @@
"""The tests for the Xiaomi vacuum platform.""" """The tests for the Xiaomi vacuum platform."""
import asyncio
from datetime import time, timedelta from datetime import time, timedelta
from unittest import mock from unittest import mock
@ -149,11 +148,10 @@ def mirobo_errors_fixture():
yield mock_vacuum yield mock_vacuum
@asyncio.coroutine async def test_xiaomi_exceptions(hass, caplog, mock_mirobo_errors):
def test_xiaomi_exceptions(hass, caplog, mock_mirobo_errors):
"""Test vacuum supported features.""" """Test vacuum supported features."""
entity_name = "test_vacuum_cleaner_error" entity_name = "test_vacuum_cleaner_error"
yield from async_setup_component( await async_setup_component(
hass, hass,
DOMAIN, DOMAIN,
{ {
@ -165,7 +163,7 @@ def test_xiaomi_exceptions(hass, caplog, mock_mirobo_errors):
} }
}, },
) )
yield from hass.async_block_till_done() await hass.async_block_till_done()
assert "Initializing with host 127.0.0.1 (token 12345...)" in caplog.text assert "Initializing with host 127.0.0.1 (token 12345...)" in caplog.text
assert mock_mirobo_errors.status.call_count == 1 assert mock_mirobo_errors.status.call_count == 1
@ -173,13 +171,12 @@ def test_xiaomi_exceptions(hass, caplog, mock_mirobo_errors):
assert "Got OSError while fetching the state" in caplog.text assert "Got OSError while fetching the state" in caplog.text
@asyncio.coroutine async def test_xiaomi_vacuum_services(hass, caplog, mock_mirobo_is_got_error):
def test_xiaomi_vacuum_services(hass, caplog, mock_mirobo_is_got_error):
"""Test vacuum supported features.""" """Test vacuum supported features."""
entity_name = "test_vacuum_cleaner_1" entity_name = "test_vacuum_cleaner_1"
entity_id = "{}.{}".format(DOMAIN, entity_name) entity_id = "{}.{}".format(DOMAIN, entity_name)
yield from async_setup_component( await async_setup_component(
hass, hass,
DOMAIN, DOMAIN,
{ {
@ -191,7 +188,7 @@ def test_xiaomi_vacuum_services(hass, caplog, mock_mirobo_is_got_error):
} }
}, },
) )
yield from hass.async_block_till_done() await hass.async_block_till_done()
assert "Initializing with host 127.0.0.1 (token 12345...)" in caplog.text assert "Initializing with host 127.0.0.1 (token 12345...)" in caplog.text
@ -223,7 +220,7 @@ def test_xiaomi_vacuum_services(hass, caplog, mock_mirobo_is_got_error):
assert state.attributes.get(ATTR_CLEANING_TOTAL_TIME) == 695 assert state.attributes.get(ATTR_CLEANING_TOTAL_TIME) == 695
# Call services # Call services
yield from hass.services.async_call( await hass.services.async_call(
DOMAIN, SERVICE_START, {"entity_id": entity_id}, blocking=True DOMAIN, SERVICE_START, {"entity_id": entity_id}, blocking=True
) )
mock_mirobo_is_got_error.assert_has_calls( mock_mirobo_is_got_error.assert_has_calls(
@ -232,28 +229,28 @@ def test_xiaomi_vacuum_services(hass, caplog, mock_mirobo_is_got_error):
mock_mirobo_is_got_error.assert_has_calls(STATUS_CALLS, any_order=True) mock_mirobo_is_got_error.assert_has_calls(STATUS_CALLS, any_order=True)
mock_mirobo_is_got_error.reset_mock() mock_mirobo_is_got_error.reset_mock()
yield from hass.services.async_call( await hass.services.async_call(
DOMAIN, SERVICE_STOP, {"entity_id": entity_id}, blocking=True DOMAIN, SERVICE_STOP, {"entity_id": entity_id}, blocking=True
) )
mock_mirobo_is_got_error.assert_has_calls([mock.call.stop()], any_order=True) mock_mirobo_is_got_error.assert_has_calls([mock.call.stop()], any_order=True)
mock_mirobo_is_got_error.assert_has_calls(STATUS_CALLS, any_order=True) mock_mirobo_is_got_error.assert_has_calls(STATUS_CALLS, any_order=True)
mock_mirobo_is_got_error.reset_mock() mock_mirobo_is_got_error.reset_mock()
yield from hass.services.async_call( await hass.services.async_call(
DOMAIN, SERVICE_RETURN_TO_BASE, {"entity_id": entity_id}, blocking=True DOMAIN, SERVICE_RETURN_TO_BASE, {"entity_id": entity_id}, blocking=True
) )
mock_mirobo_is_got_error.assert_has_calls([mock.call.home()], any_order=True) mock_mirobo_is_got_error.assert_has_calls([mock.call.home()], any_order=True)
mock_mirobo_is_got_error.assert_has_calls(STATUS_CALLS, any_order=True) mock_mirobo_is_got_error.assert_has_calls(STATUS_CALLS, any_order=True)
mock_mirobo_is_got_error.reset_mock() mock_mirobo_is_got_error.reset_mock()
yield from hass.services.async_call( await hass.services.async_call(
DOMAIN, SERVICE_LOCATE, {"entity_id": entity_id}, blocking=True DOMAIN, SERVICE_LOCATE, {"entity_id": entity_id}, blocking=True
) )
mock_mirobo_is_got_error.assert_has_calls([mock.call.find()], any_order=True) mock_mirobo_is_got_error.assert_has_calls([mock.call.find()], any_order=True)
mock_mirobo_is_got_error.assert_has_calls(STATUS_CALLS, any_order=True) mock_mirobo_is_got_error.assert_has_calls(STATUS_CALLS, any_order=True)
mock_mirobo_is_got_error.reset_mock() mock_mirobo_is_got_error.reset_mock()
yield from hass.services.async_call( await hass.services.async_call(
DOMAIN, SERVICE_CLEAN_SPOT, {"entity_id": entity_id}, blocking=True DOMAIN, SERVICE_CLEAN_SPOT, {"entity_id": entity_id}, blocking=True
) )
mock_mirobo_is_got_error.assert_has_calls([mock.call.spot()], any_order=True) mock_mirobo_is_got_error.assert_has_calls([mock.call.spot()], any_order=True)
@ -261,7 +258,7 @@ def test_xiaomi_vacuum_services(hass, caplog, mock_mirobo_is_got_error):
mock_mirobo_is_got_error.reset_mock() mock_mirobo_is_got_error.reset_mock()
# Set speed service: # Set speed service:
yield from hass.services.async_call( await hass.services.async_call(
DOMAIN, DOMAIN,
SERVICE_SET_FAN_SPEED, SERVICE_SET_FAN_SPEED,
{"entity_id": entity_id, "fan_speed": 60}, {"entity_id": entity_id, "fan_speed": 60},
@ -273,7 +270,7 @@ def test_xiaomi_vacuum_services(hass, caplog, mock_mirobo_is_got_error):
mock_mirobo_is_got_error.assert_has_calls(STATUS_CALLS, any_order=True) mock_mirobo_is_got_error.assert_has_calls(STATUS_CALLS, any_order=True)
mock_mirobo_is_got_error.reset_mock() mock_mirobo_is_got_error.reset_mock()
yield from hass.services.async_call( await hass.services.async_call(
DOMAIN, DOMAIN,
SERVICE_SET_FAN_SPEED, SERVICE_SET_FAN_SPEED,
{"entity_id": entity_id, "fan_speed": "turbo"}, {"entity_id": entity_id, "fan_speed": "turbo"},
@ -286,7 +283,7 @@ def test_xiaomi_vacuum_services(hass, caplog, mock_mirobo_is_got_error):
mock_mirobo_is_got_error.reset_mock() mock_mirobo_is_got_error.reset_mock()
assert "ERROR" not in caplog.text assert "ERROR" not in caplog.text
yield from hass.services.async_call( await hass.services.async_call(
DOMAIN, DOMAIN,
SERVICE_SET_FAN_SPEED, SERVICE_SET_FAN_SPEED,
{"entity_id": entity_id, "fan_speed": "invent"}, {"entity_id": entity_id, "fan_speed": "invent"},
@ -294,7 +291,7 @@ def test_xiaomi_vacuum_services(hass, caplog, mock_mirobo_is_got_error):
) )
assert "ERROR" in caplog.text assert "ERROR" in caplog.text
yield from hass.services.async_call( await hass.services.async_call(
DOMAIN, DOMAIN,
SERVICE_SEND_COMMAND, SERVICE_SEND_COMMAND,
{"entity_id": entity_id, "command": "raw"}, {"entity_id": entity_id, "command": "raw"},
@ -306,7 +303,7 @@ def test_xiaomi_vacuum_services(hass, caplog, mock_mirobo_is_got_error):
mock_mirobo_is_got_error.assert_has_calls(STATUS_CALLS, any_order=True) mock_mirobo_is_got_error.assert_has_calls(STATUS_CALLS, any_order=True)
mock_mirobo_is_got_error.reset_mock() mock_mirobo_is_got_error.reset_mock()
yield from hass.services.async_call( await hass.services.async_call(
DOMAIN, DOMAIN,
SERVICE_SEND_COMMAND, SERVICE_SEND_COMMAND,
{"entity_id": entity_id, "command": "raw", "params": {"k1": 2}}, {"entity_id": entity_id, "command": "raw", "params": {"k1": 2}},
@ -319,13 +316,12 @@ def test_xiaomi_vacuum_services(hass, caplog, mock_mirobo_is_got_error):
mock_mirobo_is_got_error.reset_mock() mock_mirobo_is_got_error.reset_mock()
@asyncio.coroutine async def test_xiaomi_specific_services(hass, caplog, mock_mirobo_is_on):
def test_xiaomi_specific_services(hass, caplog, mock_mirobo_is_on):
"""Test vacuum supported features.""" """Test vacuum supported features."""
entity_name = "test_vacuum_cleaner_2" entity_name = "test_vacuum_cleaner_2"
entity_id = "{}.{}".format(DOMAIN, entity_name) entity_id = "{}.{}".format(DOMAIN, entity_name)
yield from async_setup_component( await async_setup_component(
hass, hass,
DOMAIN, DOMAIN,
{ {
@ -337,7 +333,7 @@ def test_xiaomi_specific_services(hass, caplog, mock_mirobo_is_on):
} }
}, },
) )
yield from hass.async_block_till_done() await hass.async_block_till_done()
assert "Initializing with host 192.168.1.100 (token 12345" in caplog.text assert "Initializing with host 192.168.1.100 (token 12345" in caplog.text
@ -366,7 +362,7 @@ def test_xiaomi_specific_services(hass, caplog, mock_mirobo_is_on):
assert state.attributes.get(ATTR_CLEANING_TOTAL_TIME) == 675 assert state.attributes.get(ATTR_CLEANING_TOTAL_TIME) == 675
# Xiaomi vacuum specific services: # Xiaomi vacuum specific services:
yield from hass.services.async_call( await hass.services.async_call(
XIAOMI_DOMAIN, XIAOMI_DOMAIN,
SERVICE_START_REMOTE_CONTROL, SERVICE_START_REMOTE_CONTROL,
{ATTR_ENTITY_ID: entity_id}, {ATTR_ENTITY_ID: entity_id},
@ -378,7 +374,7 @@ def test_xiaomi_specific_services(hass, caplog, mock_mirobo_is_on):
mock_mirobo_is_on.reset_mock() mock_mirobo_is_on.reset_mock()
control = {"duration": 1000, "rotation": -40, "velocity": -0.1} control = {"duration": 1000, "rotation": -40, "velocity": -0.1}
yield from hass.services.async_call( await hass.services.async_call(
XIAOMI_DOMAIN, SERVICE_MOVE_REMOTE_CONTROL, control, blocking=True XIAOMI_DOMAIN, SERVICE_MOVE_REMOTE_CONTROL, control, blocking=True
) )
mock_mirobo_is_on.manual_control.assert_has_calls( mock_mirobo_is_on.manual_control.assert_has_calls(
@ -387,7 +383,7 @@ def test_xiaomi_specific_services(hass, caplog, mock_mirobo_is_on):
mock_mirobo_is_on.assert_has_calls(STATUS_CALLS, any_order=True) mock_mirobo_is_on.assert_has_calls(STATUS_CALLS, any_order=True)
mock_mirobo_is_on.reset_mock() mock_mirobo_is_on.reset_mock()
yield from hass.services.async_call( await hass.services.async_call(
XIAOMI_DOMAIN, SERVICE_STOP_REMOTE_CONTROL, {}, blocking=True XIAOMI_DOMAIN, SERVICE_STOP_REMOTE_CONTROL, {}, blocking=True
) )
mock_mirobo_is_on.assert_has_calls([mock.call.manual_stop()], any_order=True) mock_mirobo_is_on.assert_has_calls([mock.call.manual_stop()], any_order=True)
@ -395,7 +391,7 @@ def test_xiaomi_specific_services(hass, caplog, mock_mirobo_is_on):
mock_mirobo_is_on.reset_mock() mock_mirobo_is_on.reset_mock()
control_once = {"duration": 2000, "rotation": 120, "velocity": 0.1} control_once = {"duration": 2000, "rotation": 120, "velocity": 0.1}
yield from hass.services.async_call( await hass.services.async_call(
XIAOMI_DOMAIN, SERVICE_MOVE_REMOTE_CONTROL_STEP, control_once, blocking=True XIAOMI_DOMAIN, SERVICE_MOVE_REMOTE_CONTROL_STEP, control_once, blocking=True
) )
mock_mirobo_is_on.manual_control_once.assert_has_calls( mock_mirobo_is_on.manual_control_once.assert_has_calls(
@ -405,7 +401,7 @@ def test_xiaomi_specific_services(hass, caplog, mock_mirobo_is_on):
mock_mirobo_is_on.reset_mock() mock_mirobo_is_on.reset_mock()
control = {"zone": [[123, 123, 123, 123]], "repeats": 2} control = {"zone": [[123, 123, 123, 123]], "repeats": 2}
yield from hass.services.async_call( await hass.services.async_call(
XIAOMI_DOMAIN, SERVICE_CLEAN_ZONE, control, blocking=True XIAOMI_DOMAIN, SERVICE_CLEAN_ZONE, control, blocking=True
) )
mock_mirobo_is_on.zoned_clean.assert_has_calls( mock_mirobo_is_on.zoned_clean.assert_has_calls(