Merge pull request #10 from python-ellar/fixe_override_jwt_config

fixed jwt config override bug

Author: eadwinCode

ellar_jwt/services.py

@@ -1,3 +1,4 @@
+import functools
 import typing as t
 from datetime import timedelta
 
@@ -75,7 +76,10 @@ async def sign_async(
         headers: t.Optional[t.Dict[str, t.Any]] = None,
         **jwt_config: t.Any,
     ) -> str:
-        return await anyio.to_thread.run_sync(self.sign, payload, headers, **jwt_config)
+        func = self.sign
+        if jwt_config:
+            func = functools.partial(self.sign, **jwt_config)
+        return await anyio.to_thread.run_sync(func, payload, headers)
 
     def decode(
         self, token: str, verify: bool = True, **jwt_config: t.Any
@@ -109,4 +113,7 @@ def decode(
     async def decode_async(
         self, token: str, verify: bool = True, **jwt_config: t.Any
     ) -> t.Dict[str, t.Any]:
-        return await anyio.to_thread.run_sync(self.decode, token, verify, **jwt_config)
+        func = self.decode
+        if jwt_config:
+            func = functools.partial(self.decode, **jwt_config)
+        return await anyio.to_thread.run_sync(func, token, verify)

tests/test_jwt_service.py

@@ -471,3 +471,18 @@ async def test_sign_async_and_decode_async(self):
         token = await backend.sign_async(self.payload)
         decoded = await backend.decode_async(token)
         assert decoded["uuid"] == str(unique)
+
+    @pytest.mark.asyncio
+    async def test_sign_async_with_override_jwt_config(self):
+        backend = JWTService(
+            JWTConfiguration(
+                algorithm="HS256",
+                signing_secret_key=SECRET,
+                json_encoder=UUIDJSONEncoder,
+            )
+        )
+        unique = uuid.uuid4()
+        self.payload["uuid"] = unique
+        token = await backend.sign_async(self.payload, algorithm="HS384")
+        decoded = await backend.decode_async(token, algorithm="HS384")
+        assert decoded["uuid"] == str(unique)