test: add secret filtering

tests/__init__.py

@@ -1,3 +1,3 @@
-from . import test_format_duration
+from . import test_filter_secrets, test_format_duration
 
-__all__ = ["test_format_duration"]
+__all__ = ["test_format_duration", "test_filter_secrets"]

tests/test_filter_secrets.py

@@ -0,0 +1,21 @@
+import unittest
+
+import utils
+
+
+class TestFilterSecrets(unittest.TestCase):
+    def test_filter_secrets(self):
+        secret = "PLACEHOLDER_TOKEN"
+        self.assertFalse(
+            secret in utils.filter_secrets(f"HELLO{secret}WORLD", {"TOKEN": secret})
+        )
+        self.assertFalse(secret in utils.filter_secrets(secret, {"TOKEN": secret}))
+        self.assertFalse(
+            secret in utils.filter_secrets(f"123{secret}", {"TOKEN": secret})
+        )
+        self.assertFalse(
+            secret in utils.filter_secrets(f"{secret}{secret}", {"TOKEN": secret})
+        )
+        self.assertFalse(
+            secret in utils.filter_secrets(f"{secret}@#(*&*$)", {"TOKEN": secret})
+        )

utils.py

@@ -112,8 +112,8 @@ async def invalid_user_handler(interaction):
     )
 
 
-def filter_secrets(text: str) -> str:
-    for secret_name, secret in constants.SECRETS.items():
+def filter_secrets(text: str, secrets=constants.SECRETS) -> str:
+    for secret_name, secret in secrets.items():
         if not secret:
             continue
         text = text.replace(secret, f"<{secret_name}>")