From 3ba90776c0d72f7c2d3646f8d9979c85a5ed3337 Mon Sep 17 00:00:00 2001 From: jan iversen Date: Thu, 17 Jun 2021 10:57:20 +0200 Subject: [PATCH] Add autospec to modbus mock, in order to use getattr (#51813) --- tests/components/modbus/conftest.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/tests/components/modbus/conftest.py b/tests/components/modbus/conftest.py index 7f67a0653fb..43221b219e5 100644 --- a/tests/components/modbus/conftest.py +++ b/tests/components/modbus/conftest.py @@ -97,12 +97,9 @@ async def base_test( mock_sync = mock.MagicMock() with mock.patch( - "homeassistant.components.modbus.modbus.ModbusTcpClient", return_value=mock_sync - ), mock.patch( - "homeassistant.components.modbus.modbus.ModbusSerialClient", + "homeassistant.components.modbus.modbus.ModbusTcpClient", + autospec=True, return_value=mock_sync, - ), mock.patch( - "homeassistant.components.modbus.modbus.ModbusUdpClient", return_value=mock_sync ): # Setup inputs for the sensor