github/esphome
2
0
Fork 0
mirror of https://github.com/esphome/esphome.git synced 2026-08-23 06:36:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

Make perf_counter mock inexhaustible for added timing calls

Browse Source
This commit is contained in:
J. Nick Koston
2026-08-21 00:42:02 -05:00
parent e097edc099
commit a15eec26f0
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/unit_tests/test_espota2.py
+4 -2
View File
@@ -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