Media source to verify domain to avoid KeyError (#67137)

This commit is contained in:
Paulus Schoutsen
2022-02-23 22:16:11 -08:00
parent b21d954e50
commit b0d043c55b
3 changed files with 16 additions and 7 deletions
@@ -98,6 +98,10 @@ async def test_async_unresolve_media(hass):
with pytest.raises(media_source.Unresolvable):
await media_source.async_resolve_media(hass, "invalid")
# Test invalid media source
with pytest.raises(media_source.Unresolvable):
await media_source.async_resolve_media(hass, "media-source://media_source2")
async def test_websocket_browse_media(hass, hass_ws_client):
"""Test browse media websocket."""
@@ -54,7 +54,7 @@ async def test_async_browse_media(hass):
# Test invalid base
with pytest.raises(media_source.BrowseError) as excinfo:
await media_source.async_browse_media(hass, f"{const.URI_SCHEME}{DOMAIN}/")
assert str(excinfo.value) == "Not a media source item"
assert str(excinfo.value) == "Invalid media source URI"
# Test successful listing
media = await media_source.async_browse_media(