mirror of
https://github.com/home-assistant/core.git
synced 2025-07-21 12:17:07 +00:00
Ensure entity platform in blackbird tests (#135715)
This commit is contained in:
parent
a39137c3fc
commit
f3683f0b5e
@ -10,7 +10,6 @@ from homeassistant.components.blackbird.const import DOMAIN, SERVICE_SETALLZONES
|
|||||||
from homeassistant.components.blackbird.media_player import (
|
from homeassistant.components.blackbird.media_player import (
|
||||||
DATA_BLACKBIRD,
|
DATA_BLACKBIRD,
|
||||||
PLATFORM_SCHEMA,
|
PLATFORM_SCHEMA,
|
||||||
setup_platform,
|
|
||||||
)
|
)
|
||||||
from homeassistant.components.media_player import (
|
from homeassistant.components.media_player import (
|
||||||
MediaPlayerEntity,
|
MediaPlayerEntity,
|
||||||
@ -18,6 +17,9 @@ from homeassistant.components.media_player import (
|
|||||||
)
|
)
|
||||||
from homeassistant.const import STATE_OFF, STATE_ON
|
from homeassistant.const import STATE_OFF, STATE_ON
|
||||||
from homeassistant.core import HomeAssistant
|
from homeassistant.core import HomeAssistant
|
||||||
|
from homeassistant.setup import async_setup_component
|
||||||
|
|
||||||
|
from tests.common import MockEntityPlatform
|
||||||
|
|
||||||
|
|
||||||
class AttrDict(dict):
|
class AttrDict(dict):
|
||||||
@ -181,21 +183,21 @@ async def setup_blackbird(hass: HomeAssistant, mock_blackbird: MockBlackbird) ->
|
|||||||
"homeassistant.components.blackbird.media_player.get_blackbird",
|
"homeassistant.components.blackbird.media_player.get_blackbird",
|
||||||
return_value=mock_blackbird,
|
return_value=mock_blackbird,
|
||||||
):
|
):
|
||||||
await hass.async_add_executor_job(
|
await async_setup_component(
|
||||||
setup_platform,
|
|
||||||
hass,
|
hass,
|
||||||
|
"media_player",
|
||||||
{
|
{
|
||||||
"platform": "blackbird",
|
"media_player": {
|
||||||
"port": "/dev/ttyUSB0",
|
"platform": "blackbird",
|
||||||
"zones": {3: {"name": "Zone name"}},
|
"port": "/dev/ttyUSB0",
|
||||||
"sources": {
|
"zones": {3: {"name": "Zone name"}},
|
||||||
1: {"name": "one"},
|
"sources": {
|
||||||
3: {"name": "three"},
|
1: {"name": "one"},
|
||||||
2: {"name": "two"},
|
3: {"name": "three"},
|
||||||
},
|
2: {"name": "two"},
|
||||||
|
},
|
||||||
|
}
|
||||||
},
|
},
|
||||||
lambda *args, **kwargs: None,
|
|
||||||
{},
|
|
||||||
)
|
)
|
||||||
await hass.async_block_till_done()
|
await hass.async_block_till_done()
|
||||||
|
|
||||||
@ -207,6 +209,7 @@ def media_player_entity(
|
|||||||
"""Return the media player entity."""
|
"""Return the media player entity."""
|
||||||
media_player = hass.data[DATA_BLACKBIRD]["/dev/ttyUSB0-3"]
|
media_player = hass.data[DATA_BLACKBIRD]["/dev/ttyUSB0-3"]
|
||||||
media_player.hass = hass
|
media_player.hass = hass
|
||||||
|
media_player.platform = MockEntityPlatform(hass)
|
||||||
media_player.entity_id = "media_player.zone_3"
|
media_player.entity_id = "media_player.zone_3"
|
||||||
return media_player
|
return media_player
|
||||||
|
|
||||||
@ -271,10 +274,6 @@ async def test_update(
|
|||||||
hass: HomeAssistant, media_player_entity: MediaPlayerEntity
|
hass: HomeAssistant, media_player_entity: MediaPlayerEntity
|
||||||
) -> None:
|
) -> None:
|
||||||
"""Test updating values from blackbird."""
|
"""Test updating values from blackbird."""
|
||||||
assert media_player_entity.state is None
|
|
||||||
assert media_player_entity.source is None
|
|
||||||
|
|
||||||
await hass.async_add_executor_job(media_player_entity.update)
|
|
||||||
|
|
||||||
assert media_player_entity.state == STATE_ON
|
assert media_player_entity.state == STATE_ON
|
||||||
assert media_player_entity.source == "one"
|
assert media_player_entity.source == "one"
|
||||||
@ -291,9 +290,6 @@ async def test_state(
|
|||||||
mock_blackbird: MockBlackbird,
|
mock_blackbird: MockBlackbird,
|
||||||
) -> None:
|
) -> None:
|
||||||
"""Test state property."""
|
"""Test state property."""
|
||||||
assert media_player_entity.state is None
|
|
||||||
|
|
||||||
await hass.async_add_executor_job(media_player_entity.update)
|
|
||||||
assert media_player_entity.state == STATE_ON
|
assert media_player_entity.state == STATE_ON
|
||||||
|
|
||||||
mock_blackbird.zones[3].power = False
|
mock_blackbird.zones[3].power = False
|
||||||
@ -315,8 +311,6 @@ async def test_source(
|
|||||||
hass: HomeAssistant, media_player_entity: MediaPlayerEntity
|
hass: HomeAssistant, media_player_entity: MediaPlayerEntity
|
||||||
) -> None:
|
) -> None:
|
||||||
"""Test source property."""
|
"""Test source property."""
|
||||||
assert media_player_entity.source is None
|
|
||||||
await hass.async_add_executor_job(media_player_entity.update)
|
|
||||||
assert media_player_entity.source == "one"
|
assert media_player_entity.source == "one"
|
||||||
|
|
||||||
|
|
||||||
@ -324,8 +318,6 @@ async def test_media_title(
|
|||||||
hass: HomeAssistant, media_player_entity: MediaPlayerEntity
|
hass: HomeAssistant, media_player_entity: MediaPlayerEntity
|
||||||
) -> None:
|
) -> None:
|
||||||
"""Test media title property."""
|
"""Test media title property."""
|
||||||
assert media_player_entity.media_title is None
|
|
||||||
await hass.async_add_executor_job(media_player_entity.update)
|
|
||||||
assert media_player_entity.media_title == "one"
|
assert media_player_entity.media_title == "one"
|
||||||
|
|
||||||
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user