d4ed65e0f5
* Add power binary_sensor support to Sonos * Prepare for future unloading of config entries * Remove unnecessary calls to super() inits * Add binary_sensor to tests, remove invalid test for empty battery payload * Move sensor added_to_hass to common sensor class * Avoid dispatching sensors if no battery * Use proper attributes property * Remove power source fallback * Update homeassistant/components/sonos/speaker.py Co-authored-by: Martin Hjelmare <marhje52@gmail.com> Co-authored-by: Martin Hjelmare <marhje52@gmail.com>
57 lines
2.3 KiB
Python
57 lines
2.3 KiB
Python
"""Tests for the Sonos battery sensor platform."""
|
|
from pysonos.exceptions import NotSupportedException
|
|
|
|
from homeassistant.components.sonos import DOMAIN
|
|
from homeassistant.components.sonos.binary_sensor import ATTR_BATTERY_POWER_SOURCE
|
|
from homeassistant.const import STATE_ON
|
|
from homeassistant.setup import async_setup_component
|
|
|
|
|
|
async def setup_platform(hass, config_entry, config):
|
|
"""Set up the media player platform for testing."""
|
|
config_entry.add_to_hass(hass)
|
|
assert await async_setup_component(hass, DOMAIN, config)
|
|
await hass.async_block_till_done()
|
|
|
|
|
|
async def test_entity_registry_unsupported(hass, config_entry, config, soco):
|
|
"""Test sonos device without battery registered in the device registry."""
|
|
soco.get_battery_info.side_effect = NotSupportedException
|
|
|
|
await setup_platform(hass, config_entry, config)
|
|
|
|
entity_registry = await hass.helpers.entity_registry.async_get_registry()
|
|
|
|
assert "media_player.zone_a" in entity_registry.entities
|
|
assert "sensor.zone_a_battery" not in entity_registry.entities
|
|
assert "binary_sensor.zone_a_power" not in entity_registry.entities
|
|
|
|
|
|
async def test_entity_registry_supported(hass, config_entry, config, soco):
|
|
"""Test sonos device with battery registered in the device registry."""
|
|
await setup_platform(hass, config_entry, config)
|
|
|
|
entity_registry = await hass.helpers.entity_registry.async_get_registry()
|
|
|
|
assert "media_player.zone_a" in entity_registry.entities
|
|
assert "sensor.zone_a_battery" in entity_registry.entities
|
|
assert "binary_sensor.zone_a_power" in entity_registry.entities
|
|
|
|
|
|
async def test_battery_attributes(hass, config_entry, config, soco):
|
|
"""Test sonos device with battery state."""
|
|
await setup_platform(hass, config_entry, config)
|
|
|
|
entity_registry = await hass.helpers.entity_registry.async_get_registry()
|
|
|
|
battery = entity_registry.entities["sensor.zone_a_battery"]
|
|
battery_state = hass.states.get(battery.entity_id)
|
|
assert battery_state.state == "100"
|
|
assert battery_state.attributes.get("unit_of_measurement") == "%"
|
|
|
|
power = entity_registry.entities["binary_sensor.zone_a_power"]
|
|
power_state = hass.states.get(power.entity_id)
|
|
assert power_state.state == STATE_ON
|
|
assert (
|
|
power_state.attributes.get(ATTR_BATTERY_POWER_SOURCE) == "SONOS_CHARGING_RING"
|
|
)
|