Report modbus buffer too small or too big to unpack (#57838)

2021-10-21 00:22:01 +02:00 · 2021-10-21 00:22:01 +02:00 · cca7da77ad
commit cca7da77ad
parent 38586d2cf1
5 changed files with 54 additions and 4 deletions
--- a/tests/components/modbus/test_sensor.py
+++ b/tests/components/modbus/test_sensor.py
@ -559,6 +559,44 @@ async def test_all_sensor(hass, mock_do_cycle, expected):
    assert hass.states.get(ENTITY_ID).state == expected


+@pytest.mark.parametrize(
+    "do_config",
+    [
+        {
+            CONF_SENSORS: [
+                {
+                    CONF_NAME: TEST_ENTITY_NAME,
+                    CONF_ADDRESS: 51,
+                    CONF_SCAN_INTERVAL: 1,
+                },
+            ],
+        },
+    ],
+)
+@pytest.mark.parametrize(
+    "config_addon,register_words",
+    [
+        (
+            {
+                CONF_COUNT: 1,
+                CONF_DATA_TYPE: DataType.INT16,
+            },
+            [7, 9],
+        ),
+        (
+            {
+                CONF_COUNT: 2,
+                CONF_DATA_TYPE: DataType.INT32,
+            },
+            [7],
+        ),
+    ],
+)
+async def test_wrong_unpack(hass, mock_do_cycle):
+    """Run test for sensor."""
+    assert hass.states.get(ENTITY_ID).state == STATE_UNAVAILABLE
+
+
@pytest.mark.parametrize(
    "do_config",
    [