jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-13 16:27:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

Migrate cast tests from coroutine to async/await (#30374)

Browse Source
This commit is contained in:
Franck Nijhof 2020-01-02 00:21:42 +01:00 committed by Andrew Sayre
parent c837f185f7
commit c5298dc4dc
1 changed files with 7 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
tests/components/cast/test_media_player.py
View File

@ -1,6 +1,5 @@
"""The tests for the Cast Media player platform.""" """The tests for the Cast Media player platform."""
# pylint: disable=protected-access # pylint: disable=protected-access
import asyncio
from typing import Optional from typing import Optional
from unittest.mock import MagicMock, Mock, patch from unittest.mock import MagicMock, Mock, patch
from uuid import UUID from uuid import UUID
@ -124,23 +123,21 @@ async def async_setup_media_player_cast(hass: HomeAssistantType, info: Chromecas
return chromecast, entity return chromecast, entity
@asyncio.coroutine async def test_start_discovery_called_once(hass):
def test_start_discovery_called_once(hass):
"""Test pychromecast.start_discovery called exactly once.""" """Test pychromecast.start_discovery called exactly once."""
with patch( with patch(
"homeassistant.components.cast.discovery.pychromecast.start_discovery", "homeassistant.components.cast.discovery.pychromecast.start_discovery",
return_value=(None, None), return_value=(None, None),
) as start_discovery: ) as start_discovery:
yield from async_setup_cast(hass) await async_setup_cast(hass)
assert start_discovery.call_count == 1 assert start_discovery.call_count == 1
yield from async_setup_cast(hass) await async_setup_cast(hass)
assert start_discovery.call_count == 1 assert start_discovery.call_count == 1
@asyncio.coroutine async def test_stop_discovery_called_on_stop(hass):
def test_stop_discovery_called_on_stop(hass):
"""Test pychromecast.stop_discovery called on shutdown.""" """Test pychromecast.stop_discovery called on shutdown."""
browser = MagicMock(zc={}) browser = MagicMock(zc={})
@ -149,7 +146,7 @@ def test_stop_discovery_called_on_stop(hass):
return_value=(None, browser), return_value=(None, browser),
) as start_discovery: ) as start_discovery:
# start_discovery should be called with empty config # start_discovery should be called with empty config
yield from async_setup_cast(hass, {}) await async_setup_cast(hass, {})
assert start_discovery.call_count == 1 assert start_discovery.call_count == 1
@ -158,7 +155,7 @@ def test_stop_discovery_called_on_stop(hass):
) as stop_discovery: ) as stop_discovery:
# stop discovery should be called on shutdown # stop discovery should be called on shutdown
hass.bus.async_fire(EVENT_HOMEASSISTANT_STOP) hass.bus.async_fire(EVENT_HOMEASSISTANT_STOP)
yield from hass.async_block_till_done() await hass.async_block_till_done()
stop_discovery.assert_called_once_with(browser) stop_discovery.assert_called_once_with(browser)
@ -167,7 +164,7 @@ def test_stop_discovery_called_on_stop(hass):
return_value=(None, browser), return_value=(None, browser),
) as start_discovery: ) as start_discovery:
# start_discovery should be called again on re-startup # start_discovery should be called again on re-startup
yield from async_setup_cast(hass) await async_setup_cast(hass)
assert start_discovery.call_count == 1 assert start_discovery.call_count == 1