about summary refs log tree commit diff
path: root/tests/unit/auth/conftest.py
diff options
context:
space:
mode:
authorFrederick Muriuki Muriithi2024-06-18 12:38:29 -0500
committerFrederick Muriuki Muriithi2024-06-18 12:38:29 -0500
commit8e460b05da4d419aa1b53b1c639d3e370143de4f (patch)
treecbf9975e205f00912566334cf80ab3e9f6014e86 /tests/unit/auth/conftest.py
parentb515884aedac388a3899baaa0ca70af2f66773d4 (diff)
downloadgn-auth-8e460b05da4d419aa1b53b1c639d3e370143de4f.tar.gz
Update tests for new paradigm
* Create a jwt token generator in place of a static token
* Update some fixtures
* Skip some tests that will require more work to fix
Diffstat (limited to 'tests/unit/auth/conftest.py')
-rw-r--r--tests/unit/auth/conftest.py30
1 files changed, 26 insertions, 4 deletions
diff --git a/tests/unit/auth/conftest.py b/tests/unit/auth/conftest.py
index 7f9d42d..fa86a4c 100644
--- a/tests/unit/auth/conftest.py
+++ b/tests/unit/auth/conftest.py
@@ -4,19 +4,41 @@ import datetime
 from contextlib import contextmanager
 
 from gn_auth.auth.authentication.oauth2.models.oauth2token import OAuth2Token
+from gn_auth.auth.authentication.oauth2.grants.jwt_bearer_grant import JWTBearerTokenGenerator
 
 from .fixtures import * # pylint: disable=[wildcard-import,unused-wildcard-import]
 
-def get_tokeniser(user):
+SECRET_KEY = "this is the test secret key"
+SCOPE = "profile group role resource register-client"
+
+def _tokengenerator(user, client):
+    """Generate a JWT token for tests"""
+    _generator = JWTBearerTokenGenerator(
+        secret_key=SECRET_KEY,
+        alg="HS256")
+    return _generator(
+        grant_type="urn:ietf:params:oauth:grant-type:jwt-bearer",
+        client=client,
+        user=user,
+        scope=SCOPE,
+        expires_in=3600,
+        include_refresh_token=False)
+
+def get_tokeniser(user, client):
     """Get contextmanager for mocking token acquisition."""
     @contextmanager
     def __token__(*args, **kwargs):# pylint: disable=[unused-argument]
         yield {
             usr.user_id: OAuth2Token(
                 token_id=uuid.UUID("d32611e3-07fc-4564-b56c-786c6db6de2b"),
-                client=None, token_type="Bearer", access_token="123456ABCDE",
-                refresh_token=None, revoked=False, expires_in=864000,
-                user=usr, issued_at=int(datetime.datetime.now().timestamp()),
+                client=client,
+                token_type="Bearer",
+                access_token=_tokengenerator(user, client),
+                refresh_token=None,
+                revoked=False,
+                expires_in=864000,
+                user=usr,
+                issued_at=int(datetime.datetime.now().timestamp()),
                 scope="profile group role resource register-client")
         for usr in TEST_USERS
         }[user.user_id]