From 02046fcdb4612c9a9a563bb4a391e523e379d6cd Mon Sep 17 00:00:00 2001
From: epenet <6771947+epenet@users.noreply.github.com>
Date: Sun, 3 Nov 2024 17:29:33 +0100
Subject: [PATCH] Fix advantage_air CI failure (#129735)

---
 tests/components/advantage_air/test_binary_sensor.py | 4 ++--
 tests/components/advantage_air/test_sensor.py        | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/tests/components/advantage_air/test_binary_sensor.py b/tests/components/advantage_air/test_binary_sensor.py
index 13bbadb38f9..7a7b2f8df5b 100644
--- a/tests/components/advantage_air/test_binary_sensor.py
+++ b/tests/components/advantage_air/test_binary_sensor.py
@@ -85,7 +85,7 @@ async def test_binary_sensor_async_setup_entry(
         dt_util.utcnow() + timedelta(seconds=RELOAD_AFTER_UPDATE_DELAY + 1),
     )
     await hass.async_block_till_done(wait_background_tasks=True)
-    assert len(mock_get.mock_calls) == 2
+    assert len(mock_get.mock_calls) == 3
 
     state = hass.states.get(entity_id)
     assert state
@@ -116,7 +116,7 @@ async def test_binary_sensor_async_setup_entry(
         dt_util.utcnow() + timedelta(seconds=RELOAD_AFTER_UPDATE_DELAY + 1),
     )
     await hass.async_block_till_done(wait_background_tasks=True)
-    assert len(mock_get.mock_calls) == 2
+    assert len(mock_get.mock_calls) == 3
 
     state = hass.states.get(entity_id)
     assert state
diff --git a/tests/components/advantage_air/test_sensor.py b/tests/components/advantage_air/test_sensor.py
index 06243921a64..4389e67228a 100644
--- a/tests/components/advantage_air/test_sensor.py
+++ b/tests/components/advantage_air/test_sensor.py
@@ -140,7 +140,7 @@ async def test_sensor_platform_disabled_entity(
         dt_util.utcnow() + timedelta(seconds=RELOAD_AFTER_UPDATE_DELAY + 1),
     )
     await hass.async_block_till_done(wait_background_tasks=True)
-    assert len(mock_get.mock_calls) == 2
+    assert len(mock_get.mock_calls) == 3
 
     state = hass.states.get(entity_id)
     assert state