From dec1b0fc4c1de284d104a44c526d66348b10d823 Mon Sep 17 00:00:00 2001 From: Thomas Protzner Date: Mon, 12 Aug 2024 17:26:26 +0000 Subject: [PATCH] reset mock --- tests/components/husqvarna_automower/test_init.py | 15 ++++----------- 1 file changed, 4 insertions(+), 11 deletions(-) diff --git a/tests/components/husqvarna_automower/test_init.py b/tests/components/husqvarna_automower/test_init.py index c92d4953e37..3585eb7872e 100644 --- a/tests/components/husqvarna_automower/test_init.py +++ b/tests/components/husqvarna_automower/test_init.py @@ -148,16 +148,12 @@ async def test_websocket_not_available( ) # Initial call count and range for reconnection attempts - start_call_count = mock_automower_client.auth.websocket_connect.call_count - test_range = 945 - start_call_count + mock_automower_client.reset_mock() # Perform reconnection attempts - for count in range(1, test_range): + for count in range(1, 945): await hass.async_block_till_done() - assert ( - mock_automower_client.auth.websocket_connect.call_count - == start_call_count + count - ) + assert mock_automower_client.auth.websocket_connect.call_count == count assert mock_config_entry.state is ConfigEntryState.LOADED # Simulate a successful connection @@ -174,10 +170,7 @@ async def test_websocket_not_available( freezer.tick(timedelta(seconds=0)) async_fire_time_changed(hass) await hass.async_block_till_done() - assert ( - mock_automower_client.auth.websocket_connect.call_count - == start_call_count + test_range + 1 - ) + assert mock_automower_client.auth.websocket_connect.call_count == 946 async def test_device_info(