Implement an IDP simulator for tests

Author: ArtyomVancyan

app.py

@@ -0,0 +1,50 @@
+import urllib.parse
+
+from fastapi import APIRouter
+from fastapi import FastAPI
+from fastapi import Request
+from oauthlib.oauth2 import Server
+from starlette.responses import RedirectResponse
+from starlette.responses import Response
+
+from .backend import TestOAuth2
+from .validator import TestValidator
+
+
+def get_idp():
+    application = FastAPI()
+    app_router = APIRouter()
+    oauth2_server = Server(TestValidator())
+
+    @app_router.get("/oauth/authorization")
+    async def authorization(request: Request):
+        uri = str(request.url)
+        http_method = request.method
+        headers = dict(request.headers)
+        body_bytes = await request.body()
+        body = body_bytes.decode("utf-8")
+
+        scopes, credentials = oauth2_server.validate_authorization_request(uri, http_method, body, headers)
+        uri = "http://idp/oauth/authorization?" + urllib.parse.urlencode({"scopes": ','.join(scopes), **credentials})
+        headers, body, status_code = oauth2_server.create_authorization_response(uri, http_method, body, headers)
+
+        if status_code == 302:
+            location = headers.get('Location', '')
+            return RedirectResponse(location, headers=headers, status_code=status_code)
+
+        return Response(content=body, status_code=status_code)
+
+    @app_router.post("/oauth/token")
+    async def token(request: Request):
+        uri = str(request.url)
+        http_method = request.method
+        headers = dict(request.headers)
+        body_bytes = await request.body()
+        body = body_bytes.decode("utf-8")
+
+        headers, body, status_code = oauth2_server.create_token_response(uri, http_method, body, headers, {})
+
+        return Response(content=body, headers=headers, status_code=status_code)
+
+    application.include_router(app_router)
+    return application

main.py

@@ -0,0 +1,11 @@
+from social_core.backends.oauth import BaseOAuth2
+
+
+class TestOAuth2(BaseOAuth2):
+    name = "test"
+    AUTHORIZATION_URL = "http://idp/oauth/authorization"
+    ACCESS_TOKEN_URL = "http://idp/oauth/token"
+    ACCESS_TOKEN_METHOD = "POST"
+    REDIRECT_STATE = False
+    STATE_PARAMETER = True
+    SEND_USER_AGENT = True

tests/idp/__init__.py

@@ -2,8 +2,7 @@
 from oauthlib.oauth2 import RequestValidator
 
 
-class MyRequestValidator(RequestValidator):
-
+class TestValidator(RequestValidator):
     def validate_client_id(self, client_id, request, *args, **kwargs):
         return True
 
@@ -17,7 +16,7 @@ def get_default_scopes(self, client_id, request, *args, **kwargs):
         return []
 
     def authenticate_client(self, request, *args, **kwargs):
-        request.client = Client(client_id="my_client", access_token="my_token")
+        request.client = Client(client_id="")
         return True
 
     def confirm_redirect_uri(self, client_id, code, redirect_uri, client, request, *args, **kwargs):