Cleanup, Refactor and add more tests

Author: npalaska

lib/pbench/server/api/__init__.py

@@ -127,6 +127,7 @@ def create_app():
     # JWT specific configuration
     app.config["JWT_SECRET_KEY"] = os.getenv("SECRET_KEY", "my_precious")
     app.config["BCRYPT_LOG_ROUNDS"] = int(app.config_server.get("bycrypt_log_rounds"))
+    app.config["TOKEN_EXPIRATION_DURATION"] = app.config_server.get("token_expiration_duration")
 
     app.bcrypt = Bcrypt(app)
 

lib/pbench/server/api/resources/models.py

@@ -37,12 +37,23 @@ def encode_auth_token(self, app, user_id):
         Generates the Auth Token
         :return: string
         """
-        try:
+        expire = app.config.get("TOKEN_EXPIRATION_DURATION")
+        if not expire:
+            # Expiration is not defined, token is set to expire in a second
+            payload = {
+                "iat": datetime.datetime.utcnow(),
+                "exp": datetime.datetime.utcnow() + datetime.timedelta(
+                    milliseconds=100),
+                "sub": user_id,
+            }
+        else:
             payload = {
                 "iat": datetime.datetime.utcnow(),
-                "exp": datetime.datetime.utcnow() + datetime.timedelta(minutes=5),
+                "exp": datetime.datetime.utcnow() + datetime.timedelta(
+                    minutes=int(expire)),
                 "sub": user_id,
             }
+        try:
             return jwt.encode(
                 payload, app.config.get("JWT_SECRET_KEY"), algorithm="HS256"
             )