fix: use explicit type for json_b64encode and json_b64decode

Author: lepture

src/joserfc/util.py

@@ -56,11 +56,10 @@ def int_to_base64(num: int) -> str:
     return urlsafe_b64encode(s).decode("utf-8", "strict")
 
 
-def json_b64encode(text: Any) -> bytes:
-    if isinstance(text, dict):
-        text = json.dumps(text, ensure_ascii=True, separators=(",", ":"))
+def json_b64encode(data: dict[str, Any]) -> bytes:
+    text = json.dumps(data, ensure_ascii=True, separators=(",", ":"))
     return urlsafe_b64encode(to_bytes(text, "ascii"))
 
 
-def json_b64decode(text: Any) -> Any:
-    return json.loads(urlsafe_b64decode(to_bytes(text, "ascii")))
+def json_b64decode(text: bytes) -> Any:
+    return json.loads(urlsafe_b64decode(text))

tests/test_util.py

@@ -17,9 +17,6 @@ def test_to_unicode(self):
     def test_int_to_base64(self):
         self.assertRaises(ValueError, util.int_to_base64, -1)
 
-    def test_json_b64encode(self):
-        self.assertEqual(util.json_b64encode("{}"), b"e30")
-
     def test_urlsafe_b64decode(self):
         self.assertEqual(util.urlsafe_b64decode(b"_foo123-"), b"\xfd\xfa(\xd7m\xfe")
         self.assertRaises(binascii.Error, util.urlsafe_b64decode, b"+foo123/")