add revoke

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

@@ -35,50 +35,50 @@ def test_token(
         ),
         lambda_context,
     )
-    auth_token = json.loads(r['body'])
-    print(auth_token)
 
-    # assert r['statusCode'] == HTTPStatus.OK
-    # assert auth_token['expires_in'] == 600
+    # print(r)
 
-    # 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
+    assert r['statusCode'] == HTTPStatus.OK
 
-    # r = app.lambda_handler(
-    #     http_api_proxy(
-    #         raw_path='/token',
-    #         method=HTTPMethod.POST,
-    #         headers={
-    #             'Content-Type': 'application/x-www-form-urlencoded',
-    #         },
-    #         body=urlencode(
-    #             {
-    #                 'grant_type': 'refresh_token',
-    #                 'refresh_token': auth_token['refresh_token'],
-    #                 'client_id': client_id,
-    #             }
-    #         ),
-    #     ),
-    #     lambda_context,
-    # )
+    r = json.loads(r['body'])
+    assert r['expires_in'] == 600
 
-    # assert r['statusCode'] == HTTPStatus.OK
+    tokens = dynamodb_persistence_layer.query(
+        key_cond_expr='#pk = :pk',
+        expr_attr_name={
+            '#pk': 'id',
+        },
+        expr_attr_values={
+            ':pk': 'OAUTH2#TOKEN',
+        },
+    )
+    assert len(tokens['items']) == 2
 
-    # 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']) == 3
+    r = app.lambda_handler(
+        http_api_proxy(
+            raw_path='/token',
+            method=HTTPMethod.POST,
+            headers={
+                'Content-Type': 'application/x-www-form-urlencoded',
+            },
+            body=urlencode(
+                {
+                    'grant_type': 'refresh_token',
+                    'refresh_token': r['refresh_token'],
+                    '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']) == 3