tests: add tests for filter_dict_keys

inveniosoftware · Feb 19, 2024 · f7e18cf · f7e18cf
1 parent a316741
commit f7e18cf
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 4 deletions.
diff --git a/tests/test_dictutils.py b/tests/test_dictutils.py
@@ -12,7 +12,12 @@
 
 import pytest
 
-from invenio_records.dictutils import clear_none, dict_lookup, dict_merge
+from invenio_records.dictutils import (
+    clear_none,
+    dict_lookup,
+    dict_merge,
+    filter_dict_keys,
+)
 
 
 def test_clear_none():
@@ -78,3 +83,17 @@ def test_dict_merge():
         "foo2": "bar2",
         "metadata": {"field1": 3, "field2": "test"},
     }
+
+
+def test_filter_dict_keys():
+    """Test filter dict keys."""
+    source = {
+        "foo1": {"bar1": 1, "bar2": 2},
+        "foo2": 1,
+        "foo3": {"bar1": {"foo4": 0}, "bar2": 1, "bar3": 2},
+        "foo4": {},
+    }
+
+    assert filter_dict_keys(
+        source, ["foo1.bar1", "foo2", "foo3.bar1.foo4", "foo3.bar3"]
+    ) == {"foo1": {"bar1": 1}, "foo2": 1, "foo3": {"bar1": {"foo4": 0}, "bar3": 2}}
diff --git a/tests/test_resolver.py b/tests/test_resolver.py
@@ -41,9 +41,9 @@ def local_ref_resolver_store_factory():
 
 @pytest.fixture(scope="module")
 def app_config(app_config):
-    app_config[
-        "RECORDS_REFRESOLVER_CLS"
-    ] = "invenio_records.resolver.InvenioRefResolver"
+    app_config["RECORDS_REFRESOLVER_CLS"] = (
+        "invenio_records.resolver.InvenioRefResolver"
+    )
     app_config["RECORDS_REFRESOLVER_STORE"] = local_ref_resolver_store_factory()
     return app_config