123 lines
3.4 KiB
Python
123 lines
3.4 KiB
Python
"""Test Wallbox Select component."""
|
|
|
|
from unittest.mock import Mock, patch
|
|
|
|
import pytest
|
|
|
|
from homeassistant.components.select import (
|
|
ATTR_OPTION,
|
|
DOMAIN as SELECT_DOMAIN,
|
|
SERVICE_SELECT_OPTION,
|
|
)
|
|
from homeassistant.components.wallbox.const import CHARGER_STATUS_ID_KEY, EcoSmartMode
|
|
from homeassistant.const import ATTR_ENTITY_ID
|
|
from homeassistant.core import HomeAssistant, HomeAssistantError
|
|
|
|
from . import (
|
|
authorisation_response,
|
|
http_404_error,
|
|
setup_integration_select,
|
|
test_response,
|
|
test_response_eco_mode,
|
|
test_response_full_solar,
|
|
test_response_no_power_boost,
|
|
)
|
|
from .const import MOCK_SELECT_ENTITY_ID
|
|
|
|
from tests.common import MockConfigEntry
|
|
|
|
TEST_OPTIONS = [
|
|
(EcoSmartMode.OFF, test_response),
|
|
(EcoSmartMode.ECO_MODE, test_response_eco_mode),
|
|
(EcoSmartMode.FULL_SOLAR, test_response_full_solar),
|
|
]
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_authenticate():
|
|
"""Fixture to patch Wallbox methods."""
|
|
with patch(
|
|
"homeassistant.components.wallbox.Wallbox.authenticate",
|
|
new=Mock(return_value=authorisation_response),
|
|
):
|
|
yield
|
|
|
|
|
|
@pytest.mark.parametrize(("mode", "response"), TEST_OPTIONS)
|
|
async def test_wallbox_select_solar_charging_class(
|
|
hass: HomeAssistant, entry: MockConfigEntry, mode, response, mock_authenticate
|
|
) -> None:
|
|
"""Test wallbox select class."""
|
|
|
|
with (
|
|
patch(
|
|
"homeassistant.components.wallbox.Wallbox.enableEcoSmart",
|
|
new=Mock(return_value={CHARGER_STATUS_ID_KEY: 193}),
|
|
),
|
|
patch(
|
|
"homeassistant.components.wallbox.Wallbox.disableEcoSmart",
|
|
new=Mock(return_value={CHARGER_STATUS_ID_KEY: 193}),
|
|
),
|
|
):
|
|
await setup_integration_select(hass, entry, response)
|
|
|
|
await hass.services.async_call(
|
|
SELECT_DOMAIN,
|
|
SERVICE_SELECT_OPTION,
|
|
{
|
|
ATTR_ENTITY_ID: MOCK_SELECT_ENTITY_ID,
|
|
ATTR_OPTION: mode,
|
|
},
|
|
blocking=True,
|
|
)
|
|
|
|
state = hass.states.get(MOCK_SELECT_ENTITY_ID)
|
|
assert state.state == mode
|
|
|
|
|
|
async def test_wallbox_select_no_power_boost_class(
|
|
hass: HomeAssistant, entry: MockConfigEntry
|
|
) -> None:
|
|
"""Test wallbox select class."""
|
|
|
|
await setup_integration_select(hass, entry, test_response_no_power_boost)
|
|
|
|
state = hass.states.get(MOCK_SELECT_ENTITY_ID)
|
|
assert state is None
|
|
|
|
|
|
@pytest.mark.parametrize(("mode", "response"), TEST_OPTIONS)
|
|
@pytest.mark.parametrize("error", [http_404_error, ConnectionError])
|
|
async def test_wallbox_select_class_error(
|
|
hass: HomeAssistant,
|
|
entry: MockConfigEntry,
|
|
mode,
|
|
response,
|
|
error,
|
|
mock_authenticate,
|
|
) -> None:
|
|
"""Test wallbox select class connection error."""
|
|
|
|
await setup_integration_select(hass, entry, response)
|
|
|
|
with (
|
|
patch(
|
|
"homeassistant.components.wallbox.Wallbox.disableEcoSmart",
|
|
new=Mock(side_effect=error),
|
|
),
|
|
patch(
|
|
"homeassistant.components.wallbox.Wallbox.enableEcoSmart",
|
|
new=Mock(side_effect=error),
|
|
),
|
|
pytest.raises(HomeAssistantError, match="Error communicating with Wallbox API"),
|
|
):
|
|
await hass.services.async_call(
|
|
SELECT_DOMAIN,
|
|
SERVICE_SELECT_OPTION,
|
|
{
|
|
ATTR_ENTITY_ID: MOCK_SELECT_ENTITY_ID,
|
|
ATTR_OPTION: mode,
|
|
},
|
|
blocking=True,
|
|
)
|