add test to token

id.saladeaula.digital/tests/routes/test_token.py

@@ -1,4 +1,6 @@
 import json
+import pprint
+from base64 import b64encode
 from http import HTTPMethod, HTTPStatus
 from urllib.parse import urlencode
 
@@ -14,7 +16,7 @@ def test_token(
     http_api_proxy: HttpApiProxy,
     lambda_context: LambdaContext,
 ):
-    client_id = 'd72d4005-1fa7-4430-9754-80d5e2487bb6'
+    client_id = '8c5e92b0-9ed4-451e-8935-66084d8544b1'
 
     r = app.lambda_handler(
         http_api_proxy(
@@ -39,7 +41,7 @@ def test_token(
     assert r['statusCode'] == HTTPStatus.OK
 
     r = json.loads(r['body'])
-    assert r['expires_in'] == 600
+    assert r['expires_in'] == 180
 
     tokens = dynamodb_persistence_layer.query(
         key_cond_expr='#pk = :pk',
@@ -50,7 +52,7 @@ def test_token(
             ':pk': 'OAUTH2#TOKEN',
         },
     )
-    assert len(tokens['items']) == 2
+    assert len(tokens['items']) == 3
 
     r = app.lambda_handler(
         http_api_proxy(
@@ -79,4 +81,46 @@ def test_token(
             ':pk': 'OAUTH2#TOKEN',
         },
     )
-    assert len(r['items']) == 3
+    assert len(r['items']) == 4
+
+
+def test_refresh_token(
+    app,
+    seeds,
+    dynamodb_persistence_layer: DynamoDBPersistenceLayer,
+    http_api_proxy: HttpApiProxy,
+    lambda_context: LambdaContext,
+):
+    client_id = 'd72d4005-1fa7-4430-9754-80d5e2487bb6'
+    client_secret = '1nFD8alDbGHgc3g1RLY960xyRJVee0SlMoIB0MUlSuiJy28W'
+    auth = b64encode(f'{client_id}:{client_secret}'.encode()).decode()
+    r = app.lambda_handler(
+        http_api_proxy(
+            raw_path='/token',
+            method=HTTPMethod.POST,
+            headers={
+                'Authorization': f'Basic {auth}',
+                'Content-Type': 'application/x-www-form-urlencoded',
+            },
+            body=urlencode(
+                {
+                    'grant_type': 'refresh_token',
+                    'refresh_token': 'CyF3Ik3b9hMIo3REVv27gZAHd7dvwZq6QrkhWr7qHEen4UVy',
+                    'client_id': client_id,
+                }
+            ),
+        ),
+        lambda_context,
+    )
+
+    r = dynamodb_persistence_layer.query(
+        key_cond_expr='#pk = :pk',
+        expr_attr_name={
+            '#pk': 'id',
+        },
+        expr_attr_values={
+            ':pk': 'OAUTH2#TOKEN',
+        },
+    )
+
+    assert len(r['items']) == 2