From 0d458e5d453b8cac82194f5a2841ca9172a7d1f7 Mon Sep 17 00:00:00 2001
From: marko1olo <96879250+marko1olo@users.noreply.github.com>
Date: Thu, 11 Jun 2026 06:15:16 +0400
Subject: [PATCH] test: add missing type hints to FakeProvider in test_utils
 (#631)

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

diff --git a/tests/test_utils.py b/tests/test_utils.py
index 7d78bb27d..ecf36f17f 100644
--- a/tests/test_utils.py
+++ b/tests/test_utils.py
@@ -6,13 +6,13 @@ from headroom import utils
 
 
 class FakeProvider:
-    def __init__(self, result):
+    def __init__(self, result: str | None) -> None:
         self.result = result
         self.calls: list[tuple[int, int, str, int]] = []
 
     def estimate_cost(
         self, input_tokens: int, output_tokens: int, model: str, cached_tokens: int = 0
-    ):
+    ) -> str | None:
         self.calls.append((input_tokens, output_tokens, model, cached_tokens))
         return self.result