From 9a1bb1df15edf0ca9b8a067517effbd36fd7b843 Mon Sep 17 00:00:00 2001
From: "J. Nick Koston" <nick@home-assistant.io>
Date: Fri, 21 Aug 2026 00:41:53 -0500
Subject: [PATCH] Make perf_counter mock inexhaustible for added timing calls

---
 tests/unit_tests/test_espota2.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/tests/unit_tests/test_espota2.py b/tests/unit_tests/test_espota2.py
index db4a4b1117..6db1e70593 100644
--- a/tests/unit_tests/test_espota2.py
+++ b/tests/unit_tests/test_espota2.py
@@ -6,6 +6,7 @@ from collections.abc import Generator
 import gzip
 import hashlib
 import io
+import itertools
 from pathlib import Path
 import socket
 import struct
@@ -53,8 +54,9 @@ def mock_sleep() -> Generator[Mock]:
 @pytest.fixture
 def mock_time(mock_sleep: Mock) -> Generator[None]:
     """Mock time-related functions for consistent testing."""
-    # Provide enough values for multiple calls (tests may call perform_ota multiple times)
-    with patch("time.perf_counter", side_effect=[0, 1, 0, 1, 0, 1]):
+    # Monotonically increasing, never exhausted regardless of how many timing
+    # windows perform_ota measures or how many times a test calls it
+    with patch("time.perf_counter", side_effect=itertools.count()):
         yield