043f3e640c
* use lamdba to return the library function * Rename mocks * lambda number * Lambda button * Add missing * Remove context manager
71 lines
2.5 KiB
Python
71 lines
2.5 KiB
Python
"""Test the Tessie number platform."""
|
|
|
|
from unittest.mock import patch
|
|
|
|
from homeassistant.components.number import DOMAIN as NUMBER_DOMAIN, SERVICE_SET_VALUE
|
|
from homeassistant.components.tessie.number import DESCRIPTIONS
|
|
from homeassistant.const import ATTR_ENTITY_ID
|
|
from homeassistant.core import HomeAssistant
|
|
|
|
from .common import TEST_VEHICLE_STATE_ONLINE, setup_platform
|
|
|
|
|
|
async def test_numbers(hass: HomeAssistant) -> None:
|
|
"""Tests that the number entities are correct."""
|
|
|
|
assert len(hass.states.async_all("number")) == 0
|
|
|
|
await setup_platform(hass)
|
|
|
|
assert len(hass.states.async_all("number")) == len(DESCRIPTIONS)
|
|
|
|
assert hass.states.get("number.test_charge_current").state == str(
|
|
TEST_VEHICLE_STATE_ONLINE["charge_state"]["charge_current_request"]
|
|
)
|
|
|
|
assert hass.states.get("number.test_charge_limit").state == str(
|
|
TEST_VEHICLE_STATE_ONLINE["charge_state"]["charge_limit_soc"]
|
|
)
|
|
|
|
assert hass.states.get("number.test_speed_limit").state == str(
|
|
TEST_VEHICLE_STATE_ONLINE["vehicle_state"]["speed_limit_mode"][
|
|
"current_limit_mph"
|
|
]
|
|
)
|
|
|
|
# Test number set value functions
|
|
with patch(
|
|
"homeassistant.components.tessie.number.set_charging_amps",
|
|
) as mock_set_charging_amps:
|
|
await hass.services.async_call(
|
|
NUMBER_DOMAIN,
|
|
SERVICE_SET_VALUE,
|
|
{ATTR_ENTITY_ID: ["number.test_charge_current"], "value": 16},
|
|
blocking=True,
|
|
)
|
|
assert hass.states.get("number.test_charge_current").state == "16.0"
|
|
mock_set_charging_amps.assert_called_once()
|
|
|
|
with patch(
|
|
"homeassistant.components.tessie.number.set_charge_limit",
|
|
) as mock_set_charge_limit:
|
|
await hass.services.async_call(
|
|
NUMBER_DOMAIN,
|
|
SERVICE_SET_VALUE,
|
|
{ATTR_ENTITY_ID: ["number.test_charge_limit"], "value": 80},
|
|
blocking=True,
|
|
)
|
|
assert hass.states.get("number.test_charge_limit").state == "80.0"
|
|
mock_set_charge_limit.assert_called_once()
|
|
|
|
with patch(
|
|
"homeassistant.components.tessie.number.set_speed_limit",
|
|
) as mock_set_speed_limit:
|
|
await hass.services.async_call(
|
|
NUMBER_DOMAIN,
|
|
SERVICE_SET_VALUE,
|
|
{ATTR_ENTITY_ID: ["number.test_speed_limit"], "value": 60},
|
|
blocking=True,
|
|
)
|
|
assert hass.states.get("number.test_speed_limit").state == "60.0"
|
|
mock_set_speed_limit.assert_called_once()
|