mirror of https://github.com/home-assistant/core
60 lines
1.6 KiB
Python
60 lines
1.6 KiB
Python
"""Test reproduce state for Number entities."""
|
|
|
|
import pytest
|
|
|
|
from homeassistant.components.number.const import (
|
|
ATTR_MAX,
|
|
ATTR_MIN,
|
|
DOMAIN,
|
|
SERVICE_SET_VALUE,
|
|
)
|
|
from homeassistant.core import HomeAssistant, State
|
|
from homeassistant.helpers.state import async_reproduce_state
|
|
|
|
from tests.common import async_mock_service
|
|
|
|
VALID_NUMBER1 = "19.0"
|
|
VALID_NUMBER2 = "99.9"
|
|
|
|
|
|
async def test_reproducing_states(
|
|
hass: HomeAssistant, caplog: pytest.LogCaptureFixture
|
|
) -> None:
|
|
"""Test reproducing Number states."""
|
|
|
|
hass.states.async_set(
|
|
"number.test_number", VALID_NUMBER1, {ATTR_MIN: 5, ATTR_MAX: 100}
|
|
)
|
|
|
|
# These calls should do nothing as entities already in desired state
|
|
await async_reproduce_state(
|
|
hass,
|
|
[
|
|
State("number.test_number", VALID_NUMBER1),
|
|
# Should not raise
|
|
State("number.non_existing", "234"),
|
|
],
|
|
)
|
|
|
|
assert hass.states.get("number.test_number").state == VALID_NUMBER1
|
|
|
|
# Test reproducing with different state
|
|
calls = async_mock_service(hass, DOMAIN, SERVICE_SET_VALUE)
|
|
await async_reproduce_state(
|
|
hass,
|
|
[
|
|
State("number.test_number", VALID_NUMBER2),
|
|
# Should not raise
|
|
State("number.non_existing", "234"),
|
|
],
|
|
)
|
|
|
|
assert len(calls) == 1
|
|
assert calls[0].domain == DOMAIN
|
|
assert calls[0].data == {"entity_id": "number.test_number", "value": VALID_NUMBER2}
|
|
|
|
# Test invalid state
|
|
await async_reproduce_state(hass, [State("number.test_number", "invalid_state")])
|
|
|
|
assert len(calls) == 1
|