Add some tests

Author: sergey-misuk-personal

src/country_workspace/utils/fields.py

@@ -14,6 +14,9 @@
 RecordPreprocessor = Callable[[Record], Record]
 
 
+TO_REMOVE = "_h_c", "_h_f", "_i_c", "_i_f"
+
+
 def clean_field_name(v: str) -> str:
     """Normalize a field name by removing specific substrings (case-insensitive) and converting it to lowercase.
 
@@ -24,8 +27,7 @@ def clean_field_name(v: str) -> str:
         str: The cleaned field name.
 
     """
-    to_remove = ("_h_c", "_h_f", "_i_c", "_i_f")
-    return reduce(lambda name, substr: name.replace(substr, ""), to_remove, v.lower())
+    return reduce(lambda name, substr: name.replace(substr, ""), TO_REMOVE, v.lower())
 
 
 class ExtraFieldInRecordError(Exception):

tests/utils/test_utils_fields.py

@@ -1,8 +1,13 @@
-import pytest
+from unittest.mock import MagicMock
 
-from country_workspace.utils.fields import clean_field_name
+import pytest
 
-TO_REMOVE = ("_h_c", "_h_f", "_i_c", "_i_f")
+from country_workspace.utils.fields import (
+    clean_field_name,
+    TO_REMOVE,
+    ExtraFieldInRecordError,
+    create_json_record_preprocessor,
+)
 
 
 @pytest.mark.parametrize(
@@ -15,3 +20,18 @@
 )
 def test_clean_field_name(input_value, expected_output):
     assert clean_field_name(input_value) == expected_output
+
+
+def test_extra_field_in_record_error_format() -> None:
+    assert ", ".join(extra_fields := ("a", "b", "c")) in str(ExtraFieldInRecordError(*extra_fields))
+
+
+def test_create_json_record_preprocessor_raises_on_extra_fields() -> None:
+    config = {
+        "fail_if_alien": True,
+    }
+    checker = MagicMock()
+    preprocessor = create_json_record_preprocessor(config, checker)
+
+    with pytest.raises(ExtraFieldInRecordError):
+        preprocessor({"foo": "bar"})