"""Test Wyoming switch devices."""
from homeassistant.components.wyoming.devices import SatelliteDevice
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import STATE_OFF, STATE_ON
from homeassistant.core import HomeAssistant

from . import reload_satellite


async def test_muted(
    hass: HomeAssistant,
    satellite_config_entry: ConfigEntry,
    satellite_device: SatelliteDevice,
) -> None:
    """Test satellite muted."""
    muted_id = satellite_device.get_muted_entity_id(hass)
    assert muted_id

    state = hass.states.get(muted_id)
    assert state is not None
    assert state.state == STATE_OFF
    assert not satellite_device.is_muted

    await hass.services.async_call(
        "switch",
        "turn_on",
        {"entity_id": muted_id},
        blocking=True,
    )

    state = hass.states.get(muted_id)
    assert state is not None
    assert state.state == STATE_ON
    assert satellite_device.is_muted

    # test restore
    satellite_device = await reload_satellite(hass, satellite_config_entry.entry_id)

    state = hass.states.get(muted_id)
    assert state is not None
    assert state.state == STATE_ON