152 lines
6.0 KiB
Python
152 lines
6.0 KiB
Python
"""The button tests for the Mazda Connected Services integration."""
|
|
|
|
from pymazda import MazdaException
|
|
import pytest
|
|
|
|
from homeassistant.components.button import DOMAIN as BUTTON_DOMAIN, SERVICE_PRESS
|
|
from homeassistant.const import ATTR_ENTITY_ID, ATTR_FRIENDLY_NAME, ATTR_ICON
|
|
from homeassistant.exceptions import HomeAssistantError
|
|
from homeassistant.helpers import entity_registry as er
|
|
|
|
from . import init_integration
|
|
|
|
|
|
async def test_button_setup_non_electric_vehicle(hass) -> None:
|
|
"""Test creation of button entities."""
|
|
await init_integration(hass)
|
|
|
|
entity_registry = er.async_get(hass)
|
|
|
|
entry = entity_registry.async_get("button.my_mazda3_start_engine")
|
|
assert entry
|
|
assert entry.unique_id == "JM000000000000000_start_engine"
|
|
state = hass.states.get("button.my_mazda3_start_engine")
|
|
assert state
|
|
assert state.attributes.get(ATTR_FRIENDLY_NAME) == "My Mazda3 Start Engine"
|
|
assert state.attributes.get(ATTR_ICON) == "mdi:engine"
|
|
|
|
entry = entity_registry.async_get("button.my_mazda3_stop_engine")
|
|
assert entry
|
|
assert entry.unique_id == "JM000000000000000_stop_engine"
|
|
state = hass.states.get("button.my_mazda3_stop_engine")
|
|
assert state
|
|
assert state.attributes.get(ATTR_FRIENDLY_NAME) == "My Mazda3 Stop Engine"
|
|
assert state.attributes.get(ATTR_ICON) == "mdi:engine-off"
|
|
|
|
entry = entity_registry.async_get("button.my_mazda3_turn_on_hazard_lights")
|
|
assert entry
|
|
assert entry.unique_id == "JM000000000000000_turn_on_hazard_lights"
|
|
state = hass.states.get("button.my_mazda3_turn_on_hazard_lights")
|
|
assert state
|
|
assert state.attributes.get(ATTR_FRIENDLY_NAME) == "My Mazda3 Turn On Hazard Lights"
|
|
assert state.attributes.get(ATTR_ICON) == "mdi:hazard-lights"
|
|
|
|
entry = entity_registry.async_get("button.my_mazda3_turn_off_hazard_lights")
|
|
assert entry
|
|
assert entry.unique_id == "JM000000000000000_turn_off_hazard_lights"
|
|
state = hass.states.get("button.my_mazda3_turn_off_hazard_lights")
|
|
assert state
|
|
assert (
|
|
state.attributes.get(ATTR_FRIENDLY_NAME) == "My Mazda3 Turn Off Hazard Lights"
|
|
)
|
|
assert state.attributes.get(ATTR_ICON) == "mdi:hazard-lights"
|
|
|
|
# Since this is a non-electric vehicle, electric vehicle buttons should not be created
|
|
entry = entity_registry.async_get("button.my_mazda3_refresh_vehicle_status")
|
|
assert entry is None
|
|
state = hass.states.get("button.my_mazda3_refresh_vehicle_status")
|
|
assert state is None
|
|
|
|
|
|
async def test_button_setup_electric_vehicle(hass) -> None:
|
|
"""Test creation of button entities for an electric vehicle."""
|
|
await init_integration(hass, electric_vehicle=True)
|
|
|
|
entity_registry = er.async_get(hass)
|
|
|
|
entry = entity_registry.async_get("button.my_mazda3_start_engine")
|
|
assert entry
|
|
assert entry.unique_id == "JM000000000000000_start_engine"
|
|
state = hass.states.get("button.my_mazda3_start_engine")
|
|
assert state
|
|
assert state.attributes.get(ATTR_FRIENDLY_NAME) == "My Mazda3 Start Engine"
|
|
assert state.attributes.get(ATTR_ICON) == "mdi:engine"
|
|
|
|
entry = entity_registry.async_get("button.my_mazda3_stop_engine")
|
|
assert entry
|
|
assert entry.unique_id == "JM000000000000000_stop_engine"
|
|
state = hass.states.get("button.my_mazda3_stop_engine")
|
|
assert state
|
|
assert state.attributes.get(ATTR_FRIENDLY_NAME) == "My Mazda3 Stop Engine"
|
|
assert state.attributes.get(ATTR_ICON) == "mdi:engine-off"
|
|
|
|
entry = entity_registry.async_get("button.my_mazda3_turn_on_hazard_lights")
|
|
assert entry
|
|
assert entry.unique_id == "JM000000000000000_turn_on_hazard_lights"
|
|
state = hass.states.get("button.my_mazda3_turn_on_hazard_lights")
|
|
assert state
|
|
assert state.attributes.get(ATTR_FRIENDLY_NAME) == "My Mazda3 Turn On Hazard Lights"
|
|
assert state.attributes.get(ATTR_ICON) == "mdi:hazard-lights"
|
|
|
|
entry = entity_registry.async_get("button.my_mazda3_turn_off_hazard_lights")
|
|
assert entry
|
|
assert entry.unique_id == "JM000000000000000_turn_off_hazard_lights"
|
|
state = hass.states.get("button.my_mazda3_turn_off_hazard_lights")
|
|
assert state
|
|
assert (
|
|
state.attributes.get(ATTR_FRIENDLY_NAME) == "My Mazda3 Turn Off Hazard Lights"
|
|
)
|
|
assert state.attributes.get(ATTR_ICON) == "mdi:hazard-lights"
|
|
|
|
entry = entity_registry.async_get("button.my_mazda3_refresh_status")
|
|
assert entry
|
|
assert entry.unique_id == "JM000000000000000_refresh_vehicle_status"
|
|
state = hass.states.get("button.my_mazda3_refresh_status")
|
|
assert state
|
|
assert state.attributes.get(ATTR_FRIENDLY_NAME) == "My Mazda3 Refresh Status"
|
|
assert state.attributes.get(ATTR_ICON) == "mdi:refresh"
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"entity_id_suffix, api_method_name",
|
|
[
|
|
("start_engine", "start_engine"),
|
|
("stop_engine", "stop_engine"),
|
|
("turn_on_hazard_lights", "turn_on_hazard_lights"),
|
|
("turn_off_hazard_lights", "turn_off_hazard_lights"),
|
|
("refresh_status", "refresh_vehicle_status"),
|
|
],
|
|
)
|
|
async def test_button_press(hass, entity_id_suffix, api_method_name) -> None:
|
|
"""Test pressing the button entities."""
|
|
client_mock = await init_integration(hass, electric_vehicle=True)
|
|
|
|
await hass.services.async_call(
|
|
BUTTON_DOMAIN,
|
|
SERVICE_PRESS,
|
|
{ATTR_ENTITY_ID: f"button.my_mazda3_{entity_id_suffix}"},
|
|
blocking=True,
|
|
)
|
|
await hass.async_block_till_done()
|
|
|
|
api_method = getattr(client_mock, api_method_name)
|
|
api_method.assert_called_once_with(12345)
|
|
|
|
|
|
async def test_button_press_error(hass) -> None:
|
|
"""Test the Mazda API raising an error when a button entity is pressed."""
|
|
client_mock = await init_integration(hass)
|
|
|
|
client_mock.start_engine.side_effect = MazdaException("Test error")
|
|
|
|
with pytest.raises(HomeAssistantError) as err:
|
|
await hass.services.async_call(
|
|
BUTTON_DOMAIN,
|
|
SERVICE_PRESS,
|
|
{ATTR_ENTITY_ID: "button.my_mazda3_start_engine"},
|
|
blocking=True,
|
|
)
|
|
await hass.async_block_till_done()
|
|
|
|
assert str(err.value) == "Test error"
|