96 lines
2.7 KiB
Python
96 lines
2.7 KiB
Python
"""Provide tests for mysensors notify platform."""
|
|
from __future__ import annotations
|
|
|
|
from collections.abc import Callable
|
|
from unittest.mock import MagicMock, call
|
|
|
|
from mysensors.sensor import Sensor
|
|
|
|
from homeassistant.components.notify import DOMAIN as NOTIFY_DOMAIN
|
|
from homeassistant.core import HomeAssistant
|
|
|
|
from tests.common import MockConfigEntry
|
|
|
|
|
|
async def test_text_type(
|
|
hass: HomeAssistant,
|
|
text_node: Sensor,
|
|
transport_write: MagicMock,
|
|
integration: MockConfigEntry,
|
|
) -> None:
|
|
"""Test a text type child."""
|
|
# Test without target.
|
|
await hass.services.async_call(
|
|
NOTIFY_DOMAIN, "mysensors", {"message": "Hello World"}, blocking=True
|
|
)
|
|
|
|
assert transport_write.call_count == 1
|
|
assert transport_write.call_args == call("1;1;1;0;47;Hello World\n")
|
|
|
|
# Test with target.
|
|
await hass.services.async_call(
|
|
NOTIFY_DOMAIN,
|
|
"mysensors",
|
|
{"message": "Hello", "target": "Text Node 1 1"},
|
|
blocking=True,
|
|
)
|
|
|
|
assert transport_write.call_count == 2
|
|
assert transport_write.call_args == call("1;1;1;0;47;Hello\n")
|
|
|
|
transport_write.reset_mock()
|
|
|
|
# Test a message longer than 25 characters.
|
|
await hass.services.async_call(
|
|
NOTIFY_DOMAIN,
|
|
"mysensors",
|
|
{
|
|
"message": "This is a long message that will be split",
|
|
"target": "Text Node 1 1",
|
|
},
|
|
blocking=True,
|
|
)
|
|
|
|
assert transport_write.call_count == 2
|
|
assert transport_write.call_args_list == [
|
|
call("1;1;1;0;47;This is a long message th\n"),
|
|
call("1;1;1;0;47;at will be split\n"),
|
|
]
|
|
|
|
|
|
async def test_text_type_discovery(
|
|
hass: HomeAssistant,
|
|
text_node: Sensor,
|
|
transport_write: MagicMock,
|
|
receive_message: Callable[[str], None],
|
|
) -> None:
|
|
"""Test text type discovery."""
|
|
receive_message("1;2;0;0;36;\n")
|
|
receive_message("1;2;1;0;47;test\n")
|
|
receive_message("1;2;1;0;47;test2\n") # Test that more than one set message works.
|
|
await hass.async_block_till_done()
|
|
|
|
# Test targeting the discovered child.
|
|
await hass.services.async_call(
|
|
NOTIFY_DOMAIN,
|
|
"mysensors",
|
|
{"message": "Hello", "target": "Text Node 1 2"},
|
|
blocking=True,
|
|
)
|
|
|
|
assert transport_write.call_count == 1
|
|
assert transport_write.call_args == call("1;2;1;0;47;Hello\n")
|
|
|
|
transport_write.reset_mock()
|
|
|
|
# Test targeting all notify children.
|
|
await hass.services.async_call(
|
|
NOTIFY_DOMAIN, "mysensors", {"message": "Hello World"}, blocking=True
|
|
)
|
|
|
|
assert transport_write.call_count == 2
|
|
assert transport_write.call_args_list == [
|
|
call("1;1;1;0;47;Hello World\n"),
|
|
call("1;2;1;0;47;Hello World\n"),
|
|
]
|