add apply_key_filter tests

getsentry · May 10, 2024 · f1cf889 · f1cf889
1 parent 369644a
commit f1cf889
Showing 1 changed file with 39 additions and 0 deletions.
diff --git a/tests/sentry/utils/test_json.py b/tests/sentry/utils/test_json.py
@@ -100,3 +100,42 @@ def test_prune_empty_keys_keeps_falsy_values(self):
 
     def test_prune_empty_keys_none_input(self):
         assert json.prune_empty_keys(None) is None
+
+    def test_apply_key_filter_with_key_list(self):
+        dog_data = {
+            "dogs_are_great": True,
+            "good_dogs": "all",
+            "bad_dogs": None,
+        }
+        keep_keys = ["dogs_are_great", "good_dogs"]
+
+        assert json.apply_key_filter(dog_data, keep_keys=keep_keys,) == {
+            "dogs_are_great": True,
+            "good_dogs": "all",
+        }
+
+    def test_apply_key_filter_with_callback(self):
+        dog_data = {
+            "dogs_are_great": True,
+            "good_dogs": "all",
+            "bad_dogs": None,
+        }
+        keep_keys = ["dogs_are_great", "good_dogs"]
+
+        assert json.apply_key_filter(dog_data, key_filter=lambda key: key in keep_keys,) == {
+            "dogs_are_great": True,
+            "good_dogs": "all",
+        }
+
+    def test_apply_key_filter_no_filter(self):
+        dog_data = {
+            "dogs_are_great": True,
+            "good_dogs": "all",
+            "bad_dogs": None,
+        }
+
+        assert json.apply_key_filter(dog_data,) == {
+            "dogs_are_great": True,
+            "good_dogs": "all",
+            "bad_dogs": None,
+        }