60 lines
1.4 KiB
Python
60 lines
1.4 KiB
Python
|
"""The tests for the Modbus switch component."""
|
||
|
from datetime import timedelta
|
||
|
import logging
|
||
|
|
||
|
from homeassistant.components.modbus.const import CALL_TYPE_COIL, CONF_COILS
|
||
|
from homeassistant.components.switch import DOMAIN as SWITCH_DOMAIN
|
||
|
from homeassistant.const import CONF_NAME, CONF_SLAVE
|
||
|
|
||
|
from .conftest import run_base_read_test, setup_base_test
|
||
|
|
||
|
_LOGGER = logging.getLogger(__name__)
|
||
|
|
||
|
|
||
|
async def run_sensor_test(hass, use_mock_hub, value, expected):
|
||
|
"""Run test for given config."""
|
||
|
switch_name = "modbus_test_switch"
|
||
|
scan_interval = 5
|
||
|
entity_id, now, device = await setup_base_test(
|
||
|
switch_name,
|
||
|
hass,
|
||
|
use_mock_hub,
|
||
|
{
|
||
|
CONF_COILS: [
|
||
|
{CONF_NAME: switch_name, CALL_TYPE_COIL: 1234, CONF_SLAVE: 1},
|
||
|
]
|
||
|
},
|
||
|
SWITCH_DOMAIN,
|
||
|
scan_interval,
|
||
|
)
|
||
|
|
||
|
await run_base_read_test(
|
||
|
entity_id,
|
||
|
hass,
|
||
|
use_mock_hub,
|
||
|
CALL_TYPE_COIL,
|
||
|
value,
|
||
|
expected,
|
||
|
now + timedelta(seconds=scan_interval + 1),
|
||
|
)
|
||
|
|
||
|
|
||
|
async def test_read_coil_false(hass, mock_hub):
|
||
|
"""Test reading of switch coil."""
|
||
|
await run_sensor_test(
|
||
|
hass,
|
||
|
mock_hub,
|
||
|
[0x00],
|
||
|
expected="off",
|
||
|
)
|
||
|
|
||
|
|
||
|
async def test_read_coil_true(hass, mock_hub):
|
||
|
"""Test reading of switch coil."""
|
||
|
await run_sensor_test(
|
||
|
hass,
|
||
|
mock_hub,
|
||
|
[0xFF],
|
||
|
expected="on",
|
||
|
)
|