Improve testing

Author: jonathanmelitski

src/auth.py

@@ -13,7 +13,6 @@ def get_jwks():
             if key not in settings.JWKS_CACHE:
                 missing = True
         if not missing:
-            print("Can used cached keys")
             return settings.JWKS_CACHE
     # Make a request to get the JWKS
     for key in settings.JWKS_URL:

tests/test_load.py

@@ -17,10 +17,8 @@
 THREADS = 16
 
 
-def make_request():
-    access_token, _ = get_tokens()
-
-    url = "http://localhost:80/analytics/"
+def make_request(token: str):
+    url = "http://localhost:8000/analytics/"
     payload = json.dumps(
         {
             "product": random.randint(1, 10),
@@ -42,7 +40,7 @@ def make_request():
     )
     headers = {
         "Content-Type": "application/json",
-        "Authorization": f"Bearer {access_token}",
+        "Authorization": f"Bearer {token}",
     }
 
     try:
@@ -56,8 +54,10 @@ def make_request():
 
 def run_threads():
     with ThreadPoolExecutor(max_workers=THREADS) as executor:
+        # Fetch token once to improve performance
+        access_token, _ = get_tokens()
         for _ in range(NUMBER_OF_REQUESTS):
-            executor.submit(make_request)
+            executor.submit(make_request, access_token)
 
 
 def test_load():