added tests

Author: James Gutu

debug_toolbar/panels/sql/tracking.py

@@ -128,10 +128,8 @@ def _decode(self, param):
 
         # make sure datetime, date and time are converted to string by force_str
         CONVERT_TYPES = (datetime.datetime, datetime.date, datetime.time)
-        try:
-            return force_str(param, strings_only=not isinstance(param, CONVERT_TYPES))
-        except UnicodeDecodeError:
-            return "(encoded string)"
+        return force_str(param, strings_only=not isinstance(param, CONVERT_TYPES))
+
 
     def _last_executed_query(self, sql, params):
         """Get the last executed query from the connection."""

debug_toolbar/sanitize.py

@@ -1,8 +1,14 @@
 from django.utils.encoding import DjangoUnicodeDecodeError, force_str as force_string
 
 
-def force_str(value):
+# def force_str(s, encoding="utf-8", strings_only=False, errors="strict"):
+#     try:
+#         return force_string(s, encoding, strings_only, errors)
+#     except DjangoUnicodeDecodeError:
+#         return "Django Debug Toolbar was unable to parse value."
+
+def force_str(s, *args, **kwargs):
     try:
-        return force_string(value)
+        return force_string(s, *args, **kwargs)
     except DjangoUnicodeDecodeError:
         return "Django Debug Toolbar was unable to parse value."

tests/test_sanitize.py

@@ -0,0 +1,20 @@
+import unittest
+
+from debug_toolbar.sanitize import force_str
+
+
+class ForceStrTestCase(unittest.TestCase):
+    def test_success_convert(self):
+        input = 0
+
+        self.assertEqual(
+            force_str(input),
+            "0"
+        )
+
+    def test_failed_convert(self):
+        input = bytes.fromhex("a3f2b8c14e972d5a8fb3c7291a64e0859c472bf63d18a0945e73b2c84f917ae2")
+        self.assertEqual(
+            force_str(input),
+            "Django Debug Toolbar was unable to parse value."
+        )