jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-13 08:17: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."""
# pylint: disable=protected-access
import asyncio
from typing import Optional
from unittest.mock import MagicMock, Mock, patch
from uuid import UUID
@ -124,23 +123,21 @@ async def async_setup_media_player_cast(hass: HomeAssistantType, info: Chromecas
return chromecast, entity
@asyncio.coroutine
def test_start_discovery_called_once(hass):
async def test_start_discovery_called_once(hass):
"""Test pychromecast.start_discovery called exactly once."""
with patch(
"homeassistant.components.cast.discovery.pychromecast.start_discovery",
return_value=(None, None),
) as start_discovery:
yield from async_setup_cast(hass)
await async_setup_cast(hass)
assert start_discovery.call_count == 1
yield from async_setup_cast(hass)
await async_setup_cast(hass)
assert start_discovery.call_count == 1
@asyncio.coroutine
def test_stop_discovery_called_on_stop(hass):
async def test_stop_discovery_called_on_stop(hass):
"""Test pychromecast.stop_discovery called on shutdown."""
browser = MagicMock(zc={})
@ -149,7 +146,7 @@ def test_stop_discovery_called_on_stop(hass):
return_value=(None, browser),
) as start_discovery:
# 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
@ -158,7 +155,7 @@ def test_stop_discovery_called_on_stop(hass):
) as stop_discovery:
# stop discovery should be called on shutdown
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)
@ -167,7 +164,7 @@ def test_stop_discovery_called_on_stop(hass):
return_value=(None, browser),
) as start_discovery:
# 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