82 lines
2.1 KiB
Python
82 lines
2.1 KiB
Python
"""Test Wallbox Lock component."""
|
|
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
|
|
from homeassistant.components.switch import SERVICE_TURN_OFF, SERVICE_TURN_ON
|
|
from homeassistant.const import ATTR_ENTITY_ID
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.exceptions import HomeAssistantError
|
|
|
|
from .conftest import http_404_error, http_429_error, setup_integration
|
|
from .const import MOCK_SWITCH_ENTITY_ID
|
|
|
|
from tests.common import MockConfigEntry
|
|
|
|
|
|
async def test_wallbox_switch_class(
|
|
hass: HomeAssistant, entry: MockConfigEntry, mock_wallbox
|
|
) -> None:
|
|
"""Test wallbox switch class."""
|
|
|
|
await setup_integration(hass, entry)
|
|
|
|
state = hass.states.get(MOCK_SWITCH_ENTITY_ID)
|
|
assert state
|
|
assert state.state == "on"
|
|
|
|
await hass.services.async_call(
|
|
"switch",
|
|
SERVICE_TURN_ON,
|
|
{
|
|
ATTR_ENTITY_ID: MOCK_SWITCH_ENTITY_ID,
|
|
},
|
|
blocking=True,
|
|
)
|
|
|
|
await hass.services.async_call(
|
|
"switch",
|
|
SERVICE_TURN_OFF,
|
|
{
|
|
ATTR_ENTITY_ID: MOCK_SWITCH_ENTITY_ID,
|
|
},
|
|
blocking=True,
|
|
)
|
|
|
|
|
|
async def test_wallbox_switch_class_error_handling(
|
|
hass: HomeAssistant, entry: MockConfigEntry, mock_wallbox
|
|
) -> None:
|
|
"""Test wallbox switch class connection error."""
|
|
|
|
await setup_integration(hass, entry)
|
|
|
|
with (
|
|
patch.object(mock_wallbox, "resumeChargingSession", side_effect=http_404_error),
|
|
pytest.raises(HomeAssistantError),
|
|
):
|
|
# Test behavior when a connection error occurs
|
|
await hass.services.async_call(
|
|
"switch",
|
|
SERVICE_TURN_ON,
|
|
{
|
|
ATTR_ENTITY_ID: MOCK_SWITCH_ENTITY_ID,
|
|
},
|
|
blocking=True,
|
|
)
|
|
|
|
with (
|
|
patch.object(mock_wallbox, "resumeChargingSession", side_effect=http_429_error),
|
|
pytest.raises(HomeAssistantError),
|
|
):
|
|
# Test behavior when a connection error occurs
|
|
await hass.services.async_call(
|
|
"switch",
|
|
SERVICE_TURN_ON,
|
|
{
|
|
ATTR_ENTITY_ID: MOCK_SWITCH_ENTITY_ID,
|
|
},
|
|
blocking=True,
|
|
)
|