1
0
mirror of https://github.com/home-assistant/core.git synced 2025-12-24 12:59:34 +00:00

Test VIRTUAL_COUNT parameter (#100434)

This commit is contained in:
jan iversen
2023-09-15 15:02:24 +02:00
committed by GitHub
parent b4c095e944
commit 9eb0b844bc
3 changed files with 161 additions and 15 deletions

View File

@@ -21,6 +21,7 @@ from homeassistant.components.modbus.const import (
CONF_SWAP_NONE,
CONF_SWAP_WORD,
CONF_SWAP_WORD_BYTE,
CONF_VIRTUAL_COUNT,
CONF_ZERO_SUPPRESS,
MODBUS_DOMAIN,
DataType,
@@ -150,6 +151,16 @@ SLAVE_UNIQUE_ID = "ground_floor_sensor"
}
]
},
{
CONF_SENSORS: [
{
CONF_NAME: TEST_ENTITY_NAME,
CONF_ADDRESS: 51,
CONF_DATA_TYPE: DataType.INT32,
CONF_VIRTUAL_COUNT: 5,
}
]
},
],
)
async def test_config_sensor(hass: HomeAssistant, mock_modbus) -> None:
@@ -671,6 +682,21 @@ async def test_all_sensor(hass: HomeAssistant, mock_do_cycle, expected) -> None:
False,
["34899771392", "0"],
),
(
{
CONF_VIRTUAL_COUNT: 1,
CONF_UNIQUE_ID: SLAVE_UNIQUE_ID,
CONF_DATA_TYPE: DataType.FLOAT32,
},
[
0x5102,
0x0304,
int.from_bytes(struct.pack(">f", float("nan"))[0:2]),
int.from_bytes(struct.pack(">f", float("nan"))[2:4]),
],
False,
["34899771392", "0"],
),
(
{
CONF_SLAVE_COUNT: 0,
@@ -680,6 +706,15 @@ async def test_all_sensor(hass: HomeAssistant, mock_do_cycle, expected) -> None:
False,
["16909060"],
),
(
{
CONF_VIRTUAL_COUNT: 0,
CONF_UNIQUE_ID: SLAVE_UNIQUE_ID,
},
[0x0102, 0x0304],
False,
["16909060"],
),
(
{
CONF_SLAVE_COUNT: 1,
@@ -689,6 +724,15 @@ async def test_all_sensor(hass: HomeAssistant, mock_do_cycle, expected) -> None:
False,
["16909060", "67305985"],
),
(
{
CONF_VIRTUAL_COUNT: 1,
CONF_UNIQUE_ID: SLAVE_UNIQUE_ID,
},
[0x0102, 0x0304, 0x0403, 0x0201],
False,
["16909060", "67305985"],
),
(
{
CONF_SLAVE_COUNT: 3,
@@ -712,6 +756,29 @@ async def test_all_sensor(hass: HomeAssistant, mock_do_cycle, expected) -> None:
"219025152",
],
),
(
{
CONF_VIRTUAL_COUNT: 3,
CONF_UNIQUE_ID: SLAVE_UNIQUE_ID,
},
[
0x0102,
0x0304,
0x0506,
0x0708,
0x090A,
0x0B0C,
0x0D0E,
0x0F00,
],
False,
[
"16909060",
"84281096",
"151653132",
"219025152",
],
),
(
{
CONF_SLAVE_COUNT: 1,
@@ -721,6 +788,15 @@ async def test_all_sensor(hass: HomeAssistant, mock_do_cycle, expected) -> None:
True,
[STATE_UNAVAILABLE, STATE_UNKNOWN],
),
(
{
CONF_VIRTUAL_COUNT: 1,
CONF_UNIQUE_ID: SLAVE_UNIQUE_ID,
},
[0x0102, 0x0304, 0x0403, 0x0201],
True,
[STATE_UNAVAILABLE, STATE_UNKNOWN],
),
(
{
CONF_SLAVE_COUNT: 1,
@@ -730,9 +806,18 @@ async def test_all_sensor(hass: HomeAssistant, mock_do_cycle, expected) -> None:
False,
[STATE_UNAVAILABLE, STATE_UNKNOWN],
),
(
{
CONF_VIRTUAL_COUNT: 1,
CONF_UNIQUE_ID: SLAVE_UNIQUE_ID,
},
[],
False,
[STATE_UNAVAILABLE, STATE_UNKNOWN],
),
],
)
async def test_slave_sensor(hass: HomeAssistant, mock_do_cycle, expected) -> None:
async def test_virtual_sensor(hass: HomeAssistant, mock_do_cycle, expected) -> None:
"""Run test for sensor."""
entity_registry = er.async_get(hass)
for i in range(0, len(expected)):
@@ -766,7 +851,7 @@ async def test_slave_sensor(hass: HomeAssistant, mock_do_cycle, expected) -> Non
[
(
{
CONF_SLAVE_COUNT: 0,
CONF_VIRTUAL_COUNT: 0,
CONF_UNIQUE_ID: SLAVE_UNIQUE_ID,
CONF_SWAP: CONF_SWAP_BYTE,
CONF_DATA_TYPE: DataType.UINT16,
@@ -777,7 +862,7 @@ async def test_slave_sensor(hass: HomeAssistant, mock_do_cycle, expected) -> Non
),
(
{
CONF_SLAVE_COUNT: 0,
CONF_VIRTUAL_COUNT: 0,
CONF_UNIQUE_ID: SLAVE_UNIQUE_ID,
CONF_SWAP: CONF_SWAP_WORD,
CONF_DATA_TYPE: DataType.UINT32,
@@ -788,7 +873,7 @@ async def test_slave_sensor(hass: HomeAssistant, mock_do_cycle, expected) -> Non
),
(
{
CONF_SLAVE_COUNT: 0,
CONF_VIRTUAL_COUNT: 0,
CONF_UNIQUE_ID: SLAVE_UNIQUE_ID,
CONF_SWAP: CONF_SWAP_WORD,
CONF_DATA_TYPE: DataType.UINT64,
@@ -799,7 +884,7 @@ async def test_slave_sensor(hass: HomeAssistant, mock_do_cycle, expected) -> Non
),
(
{
CONF_SLAVE_COUNT: 1,
CONF_VIRTUAL_COUNT: 1,
CONF_UNIQUE_ID: SLAVE_UNIQUE_ID,
CONF_DATA_TYPE: DataType.UINT16,
CONF_SWAP: CONF_SWAP_BYTE,
@@ -810,7 +895,7 @@ async def test_slave_sensor(hass: HomeAssistant, mock_do_cycle, expected) -> Non
),
(
{
CONF_SLAVE_COUNT: 1,
CONF_VIRTUAL_COUNT: 1,
CONF_UNIQUE_ID: SLAVE_UNIQUE_ID,
CONF_DATA_TYPE: DataType.UINT32,
CONF_SWAP: CONF_SWAP_WORD,
@@ -821,7 +906,7 @@ async def test_slave_sensor(hass: HomeAssistant, mock_do_cycle, expected) -> Non
),
(
{
CONF_SLAVE_COUNT: 1,
CONF_VIRTUAL_COUNT: 1,
CONF_UNIQUE_ID: SLAVE_UNIQUE_ID,
CONF_DATA_TYPE: DataType.UINT64,
CONF_SWAP: CONF_SWAP_WORD,
@@ -832,7 +917,7 @@ async def test_slave_sensor(hass: HomeAssistant, mock_do_cycle, expected) -> Non
),
(
{
CONF_SLAVE_COUNT: 3,
CONF_VIRTUAL_COUNT: 3,
CONF_UNIQUE_ID: SLAVE_UNIQUE_ID,
CONF_DATA_TYPE: DataType.UINT16,
CONF_SWAP: CONF_SWAP_BYTE,
@@ -843,7 +928,7 @@ async def test_slave_sensor(hass: HomeAssistant, mock_do_cycle, expected) -> Non
),
(
{
CONF_SLAVE_COUNT: 3,
CONF_VIRTUAL_COUNT: 3,
CONF_UNIQUE_ID: SLAVE_UNIQUE_ID,
CONF_DATA_TYPE: DataType.UINT32,
CONF_SWAP: CONF_SWAP_WORD,
@@ -868,7 +953,7 @@ async def test_slave_sensor(hass: HomeAssistant, mock_do_cycle, expected) -> Non
),
(
{
CONF_SLAVE_COUNT: 3,
CONF_VIRTUAL_COUNT: 3,
CONF_UNIQUE_ID: SLAVE_UNIQUE_ID,
CONF_DATA_TYPE: DataType.UINT64,
CONF_SWAP: CONF_SWAP_WORD,
@@ -901,7 +986,9 @@ async def test_slave_sensor(hass: HomeAssistant, mock_do_cycle, expected) -> Non
),
],
)
async def test_slave_swap_sensor(hass: HomeAssistant, mock_do_cycle, expected) -> None:
async def test_virtual_swap_sensor(
hass: HomeAssistant, mock_do_cycle, expected
) -> None:
"""Run test for sensor."""
for i in range(0, len(expected)):
entity_id = f"{SENSOR_DOMAIN}.{TEST_ENTITY_NAME}".replace(" ", "_")
@@ -1230,7 +1317,7 @@ async def mock_restore(hass):
CONF_NAME: TEST_ENTITY_NAME,
CONF_ADDRESS: 51,
CONF_SCAN_INTERVAL: 0,
CONF_SLAVE_COUNT: 1,
CONF_VIRTUAL_COUNT: 1,
}
]
},