jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-23 13:17:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

Improve MQTT number test coverage (#44870)

Browse Source
This commit is contained in:
Erik Montnemery 2021-01-06 13:27:05 +01:00 committed by GitHub
parent 1a65ab0b80
commit 93ae65d704
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 29 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
tests/components/mqtt/test_number.py
View File

@ -69,8 +69,8 @@ async def test_run_number_setup(hass, mqtt_mock):
assert state.state == "20.5" assert state.state == "20.5"
async def test_run_number_service(hass, mqtt_mock): async def test_run_number_service_optimistic(hass, mqtt_mock):
"""Test that set_value service works.""" """Test that set_value service works in optimistic mode."""
topic = "test/number" topic = "test/number"
await async_setup_component( await async_setup_component(
hass, hass,
@ -79,6 +79,7 @@ async def test_run_number_service(hass, mqtt_mock):
) )
await hass.async_block_till_done() await hass.async_block_till_done()
# Integer
await hass.services.async_call( await hass.services.async_call(
NUMBER_DOMAIN, NUMBER_DOMAIN,
SERVICE_SET_VALUE, SERVICE_SET_VALUE,
@ -91,6 +92,32 @@ async def test_run_number_service(hass, mqtt_mock):
state = hass.states.get("number.test_number") state = hass.states.get("number.test_number")
assert state.state == "30" assert state.state == "30"
# Float with no decimal -> integer
await hass.services.async_call(
NUMBER_DOMAIN,
SERVICE_SET_VALUE,
{ATTR_ENTITY_ID: "number.test_number", ATTR_VALUE: 42.0},
blocking=True,
)
mqtt_mock.async_publish.assert_called_once_with(topic, "42", 0, False)
mqtt_mock.async_publish.reset_mock()
state = hass.states.get("number.test_number")
assert state.state == "42"
# Float with decimal -> float
await hass.services.async_call(
NUMBER_DOMAIN,
SERVICE_SET_VALUE,
{ATTR_ENTITY_ID: "number.test_number", ATTR_VALUE: 42.1},
blocking=True,
)
mqtt_mock.async_publish.assert_called_once_with(topic, "42.1", 0, False)
mqtt_mock.async_publish.reset_mock()
state = hass.states.get("number.test_number")
assert state.state == "42.1"
async def test_availability_when_connection_lost(hass, mqtt_mock): async def test_availability_when_connection_lost(hass, mqtt_mock):
"""Test availability after MQTT disconnection.""" """Test availability after MQTT disconnection."""