fix mypy checks

Author: lilydu

packages/common/src/common/logging/test_filter.py

@@ -9,23 +9,23 @@
 from .filter import ConsoleFilter
 
 
-def test_default_pattern():
+def test_default_pattern() -> None:
     filter = ConsoleFilter()
     record = MagicMock(spec=logging.LogRecord)
     record.name = "test"
 
     assert filter.filter(record) is True
 
 
-def test_exact_match():
+def test_exact_match() -> None:
     filter = ConsoleFilter("test")
     record = MagicMock(spec=logging.LogRecord)
     record.name = "test"
 
     assert filter.filter(record) is True
 
 
-def test_wildcard_prefix():
+def test_wildcard_prefix() -> None:
     filter = ConsoleFilter("test*")
 
     matching_record = MagicMock(spec=logging.LogRecord)
@@ -37,7 +37,7 @@ def test_wildcard_prefix():
     assert filter.filter(non_matching_record) is False
 
 
-def test_wildcard_suffix():
+def test_wildcard_suffix() -> None:
     filter = ConsoleFilter("*test")
 
     matching_record = MagicMock(spec=logging.LogRecord)
@@ -49,7 +49,7 @@ def test_wildcard_suffix():
     assert filter.filter(non_matching_record) is False
 
 
-def test_wildcard_middle():
+def test_wildcard_middle() -> None:
     filter = ConsoleFilter("my*test")
 
     matching_record = MagicMock(spec=logging.LogRecord)
@@ -61,7 +61,7 @@ def test_wildcard_middle():
     assert filter.filter(non_matching_record) is False
 
 
-def test_multiple_wildcards():
+def test_multiple_wildcards() -> None:
     filter = ConsoleFilter("my*log*test")
 
     matching_record = MagicMock(spec=logging.LogRecord)
@@ -73,7 +73,7 @@ def test_multiple_wildcards():
     assert filter.filter(partial_match_record) is False
 
 
-def test_case_sensitivity():
+def test_case_sensitivity() -> None:
     filter = ConsoleFilter("Test")
 
     upper_record = MagicMock(spec=logging.LogRecord)

packages/common/src/common/logging/test_formatter.py

@@ -4,19 +4,19 @@
 """
 
 import logging
-from typing import Union
+from typing import Collection, Union
 
 from .ansi import ANSI
 from .formatter import ConsoleFormatter
 
 
-def create_record(name: str, level: int, msg: Union[str, dict, list]) -> logging.LogRecord:
+def create_record(name: str, level: int, msg: Union[str, dict[str, Collection[str]], list[Collection[str]]]) -> logging.LogRecord:
     record = logging.LogRecord(name=name, level=level, pathname="test.py", lineno=1, msg=msg, args=(), exc_info=None)
     record.levelname = logging.getLevelName(level)
     return record
 
 
-def test_error_formatting():
+def test_error_formatting() -> None:
     formatter = ConsoleFormatter()
     record = create_record("test", logging.ERROR, "Error message")
 
@@ -25,7 +25,7 @@ def test_error_formatting():
     assert result == expected
 
 
-def test_warning_formatting():
+def test_warning_formatting() -> None:
     formatter = ConsoleFormatter()
     record = create_record("test", logging.WARNING, "Warning message")
 
@@ -36,7 +36,7 @@ def test_warning_formatting():
     assert result == expected
 
 
-def test_info_formatting():
+def test_info_formatting() -> None:
     formatter = ConsoleFormatter()
     record = create_record("test", logging.INFO, "Info message")
 
@@ -45,7 +45,7 @@ def test_info_formatting():
     assert result == expected
 
 
-def test_debug_formatting():
+def test_debug_formatting() -> None:
     formatter = ConsoleFormatter()
     record = create_record("test", logging.DEBUG, "Debug message")
 
@@ -54,7 +54,7 @@ def test_debug_formatting():
     assert result == expected
 
 
-def test_dict_message_formatting():
+def test_dict_message_formatting() -> None:
     formatter = ConsoleFormatter()
     dict_msg = {"key": "value", "nested": {"inner": "data"}}
     record = create_record("test", logging.INFO, dict_msg)
@@ -69,7 +69,7 @@ def test_dict_message_formatting():
     assert '"inner": "data"' in result
 
 
-def test_list_message_formatting():
+def test_list_message_formatting() -> None:
     formatter = ConsoleFormatter()
     list_msg = ["item1", "item2", {"key": "value"}]
     record = create_record("test", logging.INFO, list_msg)
@@ -84,7 +84,7 @@ def test_list_message_formatting():
     assert '"key": "value"' in result
 
 
-def test_multiline_message_formatting():
+def test_multiline_message_formatting() -> None:
     formatter = ConsoleFormatter()
     multiline_msg = "Line 1\nLine 2\nLine 3"
     record = create_record("test", logging.INFO, multiline_msg)
@@ -99,7 +99,7 @@ def test_multiline_message_formatting():
     assert result == expected
 
 
-def test_unknown_level_formatting():
+def test_unknown_level_formatting() -> None:
     formatter = ConsoleFormatter()
     record = create_record("test", 123, "Custom level message")
 

packages/common/src/common/storage/test_list_local_storage.py

@@ -6,12 +6,12 @@
 from .list_local_storage import ListLocalStorage
 
 
-def test_get_undefined_empty_index():
-    storage = ListLocalStorage()
+def test_get_undefined_empty_index() -> None:
+    storage: ListLocalStorage[int] = ListLocalStorage()
     assert storage.get(0) is None
 
 
-def test_push_and_get():
+def test_push_and_get() -> None:
     storage = ListLocalStorage[int]()
     storage.append(1)
     storage.append(2)
@@ -20,39 +20,39 @@ def test_push_and_get():
     assert storage.length() == 2
 
 
-def test_set_and_overwrite():
+def test_set_and_overwrite() -> None:
     storage = ListLocalStorage[int]([1, 2, 3])
     storage.set(1, 42)
     assert storage.get(1) == 42
     assert storage.items() == [1, 42, 3]
 
 
-def test_delete_by_index():
+def test_delete_by_index() -> None:
     storage = ListLocalStorage[int]([1, 2, 3])
     storage.delete(1)
     assert storage.items() == [1, 3]
     assert storage.length() == 2
 
 
-def test_pop():
+def test_pop() -> None:
     storage = ListLocalStorage[int]([1, 2, 3])
     assert storage.pop() == 3
     assert storage.items() == [1, 2]
     assert storage.length() == 2
 
 
-def test_get_all_values():
+def test_get_all_values() -> None:
     storage = ListLocalStorage[int]([1, 2, 3])
     assert storage.items() == [1, 2, 3]
 
 
-def test_filter_with_where():
+def test_filter_with_where() -> None:
     storage = ListLocalStorage[int]([1, 2, 3, 4])
     even = storage.filter(lambda v, i: v % 2 == 0)
     assert even == [2, 4]
 
 
-def test_mixed_operations():
+def test_mixed_operations() -> None:
     storage = ListLocalStorage[str]()
     storage.append("a")
     storage.append("b")

packages/common/src/common/storage/test_local_storage.py

@@ -6,21 +6,21 @@
 from .local_storage import LocalStorage, LocalStorageOptions
 
 
-def test_get_undefined():
-    storage = LocalStorage()
+def test_get_undefined() -> None:
+    storage: LocalStorage[int] = LocalStorage()
     assert storage.get("test") is None
 
 
-def test_set_get_delete():
-    storage = LocalStorage()
+def test_set_get_delete() -> None:
+    storage: LocalStorage[str] = LocalStorage()
     storage.set("testing", "123")
     assert storage.get("testing") == "123"
     storage.delete("testing")
     assert storage.get("testing") is None
 
 
-def test_max_size():
-    storage = LocalStorage(options=LocalStorageOptions(max=3))
+def test_max_size() -> None:
+    storage: LocalStorage[int] = LocalStorage(options=LocalStorageOptions(max=3))
 
     storage.set("a", 1)
     storage.set("b", 2)