Add TokenHelper test codes

Author: teamhide

tests/core/helpers/test_token.py

@@ -0,0 +1,77 @@
+from typing import Any
+
+import jwt
+import pytest
+
+from core.config import config
+from core.helpers.token import TokenHelper, DecodeTokenException, ExpiredTokenException
+from tests.support.token import EXPIRED_TOKEN
+
+
+@pytest.mark.asyncio
+async def test_encode():
+    # Given
+    payload = {"user_id": 1}
+
+    # When
+    sut = TokenHelper.encode(payload=payload)
+
+    # Then
+    decoded_token: dict[str, Any] = jwt.decode(
+        sut, config.JWT_SECRET_KEY, config.JWT_ALGORITHM
+    )
+    assert decoded_token["user_id"] == 1
+
+
+@pytest.mark.asyncio
+async def test_decode():
+    # Given
+    token = jwt.encode({"user_id": 1}, config.JWT_SECRET_KEY, config.JWT_ALGORITHM)
+
+    # When
+    sut = TokenHelper.decode(token=token)
+
+    # Then
+    assert sut == {"user_id": 1}
+
+
+@pytest.mark.asyncio
+async def test_decode_expired_decode_error():
+    # Given
+    token = "invalid"
+
+    # When, Then
+    with pytest.raises(DecodeTokenException):
+        TokenHelper.decode(token=token)
+
+
+@pytest.mark.asyncio
+async def test_decode_expired_signature_error():
+    # Given
+    token = EXPIRED_TOKEN
+
+    # When, Then
+    with pytest.raises(ExpiredTokenException):
+        TokenHelper.decode(token=token)
+
+
+@pytest.mark.asyncio
+async def test_decode_expired_token():
+    # Given
+    token = EXPIRED_TOKEN
+
+    # When
+    sut: dict[str, Any] = TokenHelper.decode_expired_token(token=token)
+
+    # Then
+    assert sut["user_id"] == 1
+
+
+@pytest.mark.asyncio
+async def test_decode_expired_token_decode_error():
+    # Given
+    token = "invalid"
+
+    # When, Then
+    with pytest.raises(DecodeTokenException):
+        TokenHelper.decode_expired_token(token=token)

tests/support/token.py

@@ -1,3 +1,5 @@
 INVALID_REFRESH_TOKEN = "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJhIjoxfQ.VF_eJEtfxZ9PEOqWunr26x0jF2n0o2dyKpmMwisIISY"
 
 USER_ID_1_TOKEN = "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJzdWIiOiJyZWZyZXNoIiwidXNlcl9pZCI6MX0.p3PLbILjOq4-i2OmkH2uARu6UQun01dxPArSribeQ8w"
+
+EXPIRED_TOKEN = "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJ1c2VyX2lkIjoxLCJleHAiOjEzNDU0NDYwNzJ9.3GaF7JKtagEsnUXlujCF_3KFJvNXPpP18G5quV4C0Go"