chore: add benchmarks for adding and expiring records (#1489)

bdraco · web-flow · commit d6f1fda45ec1 · 2025-01-22T11:29:15.000-10:00
diff --git a/tests/benchmarks/test_cache.py b/tests/benchmarks/test_cache.py
@@ -0,0 +1,48 @@
+from pytest_codspeed import BenchmarkFixture
+
+from zeroconf import DNSCache, DNSPointer, current_time_millis
+from zeroconf.const import _CLASS_IN, _TYPE_PTR
+
+
+def test_add_expire_1000_records(benchmark: BenchmarkFixture) -> None:
+    """Benchmark for DNSCache to expire 10000 records."""
+    cache = DNSCache()
+    now = current_time_millis()
+    records = [
+        DNSPointer(
+            name=f"test{id}.local.",
+            type_=_TYPE_PTR,
+            class_=_CLASS_IN,
+            ttl=60,
+            alias=f"test{id}.local.",
+            created=now,
+        )
+        for id in range(1000)
+    ]
+
+    @benchmark
+    def _expire_records() -> None:
+        cache.async_add_records(records)
+        cache.async_expire(now + 61_000)
+
+
+def test_expire_no_records_to_expire(benchmark: BenchmarkFixture) -> None:
+    """Benchmark for DNSCache with 1000 records none to expire."""
+    cache = DNSCache()
+    now = current_time_millis()
+    cache.async_add_records(
+        DNSPointer(
+            name=f"test{id}.local.",
+            type_=_TYPE_PTR,
+            class_=_CLASS_IN,
+            ttl=60,
+            alias=f"test{id}.local.",
+            created=now,
+        )
+        for id in range(1000)
+    )
+    cache.async_expire(now)
+
+    @benchmark
+    def _expire_records() -> None:
+        cache.async_expire(now)
