2 files changed, 37 insertions, 0 deletions
diff --git a/gn2/tests/unit/wqflask/oauth2/__init__.py b/gn2/tests/unit/wqflask/oauth2/__init__.py
new file mode 100644
index 00000000..e69de29b
--- /dev/null
+++ b/gn2/tests/unit/wqflask/oauth2/__init__.py
diff --git a/gn2/tests/unit/wqflask/oauth2/test_tokens.py b/gn2/tests/unit/wqflask/oauth2/test_tokens.py
new file mode 100644
index 00000000..ee527f51
--- /dev/null
+++ b/gn2/tests/unit/wqflask/oauth2/test_tokens.py
@@ -0,0 +1,37 @@
+"""Test oauth2 jwt tokens"""
+from gn2.wqflask.oauth2.tokens import JWTToken
+
+
+JWT_BEARER_TOKEN = b"eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCIs\
+ImN0eSI6Impzb24ifQ.eyJpc3MiOiJHTjIiLCJ\
+zdWIiOiIxMjM0IiwiYXVkIjoiR04yIiwiZXhwI\
+joiMTIzNDUifQ.ETSr_7O4ZWLac5l4pinO9Xeb\
+mzTO7xp_LvbgxjnskDc"
+
+
+def test_encode_token():
+    """Test encoding a jwt token."""
+    token = JWTToken(
+        key="secret",
+        registered_claims={
+            "iss": "GN2",
+            "sub": "1234",
+            "aud": "GN2",
+            "exp": "12345",
+        }
+    )
+    assert token.encode() == JWT_BEARER_TOKEN
+    assert token.bearer_token == {
+        "Authorization": f"Bearer {JWT_BEARER_TOKEN}"
+    }
+
+
+def test_decode_token():
+    """Test decoding a jwt token."""
+    claims = JWTToken.decode(JWT_BEARER_TOKEN, "secret")
+    assert claims == {
+        'iss': 'GN2',
+        'sub': '1234',
+        'aud': 'GN2',
+        'exp': '12345'
+    }