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

Prevent race in pychromecast.start_discovery (#36350)

Browse Source 
* Workaround for race in pychromecast

* Fix tests
This commit is contained in:
Erik Montnemery 2020-06-01 23:25:06 +02:00 committed by GitHub
parent acbffb511d
commit 391983a0cf
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 42 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
homeassistant/components/cast/discovery.py
View File

@ -3,6 +3,7 @@ import logging
import threading import threading
import pychromecast import pychromecast
import zeroconf
from homeassistant.const import EVENT_HOMEASSISTANT_STOP from homeassistant.const import EVENT_HOMEASSISTANT_STOP
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
@ -84,10 +85,13 @@ def setup_internal_discovery(hass: HomeAssistant) -> None:
) )
_LOGGER.debug("Starting internal pychromecast discovery.") _LOGGER.debug("Starting internal pychromecast discovery.")
listener, browser = pychromecast.start_discovery( listener = pychromecast.discovery.CastListener(
internal_add_callback, internal_add_callback, internal_remove_callback
internal_remove_callback, )
ChromeCastZeroconf.get_zeroconf(), browser = zeroconf.ServiceBrowser(
ChromeCastZeroconf.get_zeroconf() or zeroconf.Zeroconf(),
"_googlecast._tcp.local.",
listener,
) )
def stop_discovery(event): def stop_discovery(event):

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

@ -80,16 +80,19 @@ async def async_setup_cast_internal_discovery(hass, config=None, discovery_info=
browser = MagicMock(zc={}) browser = MagicMock(zc={})
with patch( with patch(
"homeassistant.components.cast.discovery.pychromecast.start_discovery", "homeassistant.components.cast.discovery.pychromecast.discovery.CastListener",
return_value=(listener, browser), return_value=listener,
) as start_discovery: ) as cast_listener, patch(
"homeassistant.components.cast.discovery.zeroconf.ServiceBrowser",
return_value=browser,
):
add_entities = await async_setup_cast(hass, config, discovery_info) add_entities = await async_setup_cast(hass, config, discovery_info)
await hass.async_block_till_done() await hass.async_block_till_done()
await hass.async_block_till_done() await hass.async_block_till_done()
assert start_discovery.call_count == 1 assert cast_listener.call_count == 1
discovery_callback = start_discovery.call_args[0][0] discovery_callback = cast_listener.call_args[0][0]
def discover_chromecast(service_name: str, info: ChromecastInfo) -> None: def discover_chromecast(service_name: str, info: ChromecastInfo) -> None:
"""Discover a chromecast device.""" """Discover a chromecast device."""
@ -117,9 +120,12 @@ async def async_setup_media_player_cast(hass: HomeAssistantType, info: Chromecas
"homeassistant.components.cast.discovery.pychromecast.get_chromecast_from_service", "homeassistant.components.cast.discovery.pychromecast.get_chromecast_from_service",
return_value=chromecast, return_value=chromecast,
) as get_chromecast, patch( ) as get_chromecast, patch(
"homeassistant.components.cast.discovery.pychromecast.start_discovery", "homeassistant.components.cast.discovery.pychromecast.discovery.CastListener",
return_value=(listener, browser), return_value=listener,
) as start_discovery: ) as cast_listener, patch(
"homeassistant.components.cast.discovery.zeroconf.ServiceBrowser",
return_value=browser,
):
await async_setup_component( await async_setup_component(
hass, hass,
"media_player", "media_player",
@ -128,7 +134,7 @@ async def async_setup_media_player_cast(hass: HomeAssistantType, info: Chromecas
await hass.async_block_till_done() await hass.async_block_till_done()
discovery_callback = start_discovery.call_args[0][0] discovery_callback = cast_listener.call_args[0][0]
def discover_chromecast(service_name: str, info: ChromecastInfo) -> None: def discover_chromecast(service_name: str, info: ChromecastInfo) -> None:
"""Discover a chromecast device.""" """Discover a chromecast device."""
@ -153,15 +159,17 @@ async def async_setup_media_player_cast(hass: HomeAssistantType, info: Chromecas
async def test_start_discovery_called_once(hass): async 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.discovery.CastListener",
return_value=(None, Mock()), ) as cast_listener, patch(
) as start_discovery: "homeassistant.components.cast.discovery.zeroconf.ServiceBrowser",
return_value=Mock(),
):
await async_setup_cast(hass) await async_setup_cast(hass)
assert start_discovery.call_count == 1 assert cast_listener.call_count == 1
await async_setup_cast(hass) await async_setup_cast(hass)
assert start_discovery.call_count == 1 assert cast_listener.call_count == 1
async def test_stop_discovery_called_on_stop(hass): async def test_stop_discovery_called_on_stop(hass):
@ -169,13 +177,15 @@ async def test_stop_discovery_called_on_stop(hass):
browser = MagicMock(zc={}) browser = MagicMock(zc={})
with patch( with patch(
"homeassistant.components.cast.discovery.pychromecast.start_discovery", "homeassistant.components.cast.discovery.pychromecast.discovery.CastListener",
return_value=(None, browser), ) as cast_listener, patch(
) as start_discovery: "homeassistant.components.cast.discovery.zeroconf.ServiceBrowser",
return_value=browser,
):
# start_discovery should be called with empty config # start_discovery should be called with empty config
await async_setup_cast(hass, {}) await async_setup_cast(hass, {})
assert start_discovery.call_count == 1 assert cast_listener.call_count == 1
with patch( with patch(
"homeassistant.components.cast.discovery.pychromecast.stop_discovery" "homeassistant.components.cast.discovery.pychromecast.stop_discovery"
@ -187,13 +197,15 @@ async def test_stop_discovery_called_on_stop(hass):
stop_discovery.assert_called_once_with(browser) stop_discovery.assert_called_once_with(browser)
with patch( with patch(
"homeassistant.components.cast.discovery.pychromecast.start_discovery", "homeassistant.components.cast.discovery.pychromecast.discovery.CastListener",
return_value=(None, browser), ) as cast_listener, patch(
) as start_discovery: "homeassistant.components.cast.discovery.zeroconf.ServiceBrowser",
return_value=browser,
):
# start_discovery should be called again on re-startup # start_discovery should be called again on re-startup
await async_setup_cast(hass) await async_setup_cast(hass)
assert start_discovery.call_count == 1 assert cast_listener.call_count == 1
async def test_create_cast_device_without_uuid(hass): async def test_create_cast_device_without_uuid(hass):