refactor(tests): Add strict type hints to second batch of test files

This commit is contained in:
2026-02-28 19:11:23 -05:00
parent f0415a40aa
commit 579ee8394f
10 changed files with 358 additions and 351 deletions

View File

@@ -1,17 +1,18 @@
import os
import pytest
from typing import Any
from datetime import datetime
from log_registry import LogRegistry
@pytest.fixture
def registry_setup(tmp_path):
def registry_setup(tmp_path: Any) -> Any:
registry_path = tmp_path / "log_registry.toml"
logs_dir = tmp_path / "logs"
logs_dir.mkdir()
registry = LogRegistry(str(registry_path))
return registry, logs_dir
def test_auto_whitelist_keywords(registry_setup):
def test_auto_whitelist_keywords(registry_setup: Any) -> None:
registry, logs_dir = registry_setup
session_id = "test_kw"
session_dir = logs_dir / session_id
@@ -24,7 +25,7 @@ def test_auto_whitelist_keywords(registry_setup):
assert registry.is_session_whitelisted(session_id)
assert "ERROR" in registry.data[session_id]["metadata"]["reason"]
def test_auto_whitelist_message_count(registry_setup):
def test_auto_whitelist_message_count(registry_setup: Any) -> None:
registry, logs_dir = registry_setup
session_id = "test_msg_count"
session_dir = logs_dir / session_id
@@ -37,7 +38,7 @@ def test_auto_whitelist_message_count(registry_setup):
assert registry.is_session_whitelisted(session_id)
assert registry.data[session_id]["metadata"]["message_count"] == 15
def test_auto_whitelist_large_size(registry_setup):
def test_auto_whitelist_large_size(registry_setup: Any) -> None:
registry, logs_dir = registry_setup
session_id = "test_large"
session_dir = logs_dir / session_id
@@ -50,7 +51,7 @@ def test_auto_whitelist_large_size(registry_setup):
assert registry.is_session_whitelisted(session_id)
assert "Large session size" in registry.data[session_id]["metadata"]["reason"]
def test_no_auto_whitelist_insignificant(registry_setup):
def test_no_auto_whitelist_insignificant(registry_setup: Any) -> None:
registry, logs_dir = registry_setup
session_id = "test_insignificant"
session_dir = logs_dir / session_id