Extend the usability of get_app fixture

Author: ArtyomVancyan

tests/conftest.py

@@ -9,11 +9,11 @@
 from fastapi import Request
 from social_core.backends.github import GithubOAuth2
 from social_core.backends.oauth import BaseOAuth2
-from starlette.responses import RedirectResponse
 from starlette.responses import Response
 
 from fastapi_oauth2.client import OAuth2Client
 from fastapi_oauth2.middleware import OAuth2Middleware
+from fastapi_oauth2.router import router as oauth2_router
 from fastapi_oauth2.security import OAuth2
 from tests.idp import TestOAuth2
 from tests.idp import get_idp
@@ -41,28 +41,18 @@ def backends():
 
 @pytest.fixture
 def get_app():
-    def fixture_wrapper(authentication: OAuth2 = None):
+    def fixture_wrapper(
+            authentication: OAuth2 = None,  # type of security
+            with_idp=False,  # used to test oauth2 flow
+            with_ssr=True,  # used to test oauth2 flow
+    ):
         if not authentication:
             authentication = OAuth2()
 
         oauth2 = authentication
         application = FastAPI()
         app_router = APIRouter()
 
-        @app_router.get("/oauth2/{provider}/auth")
-        async def login(request: Request, provider: str):
-            return await request.auth.clients[provider].login_redirect(request)
-
-        @app_router.get("/oauth2/{provider}/token")
-        async def token(request: Request, provider: str):
-            return await request.auth.clients[provider].token_redirect(request, app=get_idp())
-
-        @app_router.get("/oauth2/logout")
-        def logout(request: Request):
-            response = RedirectResponse(request.base_url)
-            response.delete_cookie("Authorization")
-            return response
-
         @app_router.get("/user")
         def user(request: Request, _: str = Depends(oauth2)):
             return request.user
@@ -88,9 +78,21 @@ def auth(request: Request):
             )
             return response
 
+        if with_idp:
+            @app_router.get("/oauth2/{provider}/token")
+            async def token(request: Request, provider: str):
+                if request.auth.ssr:
+                    return await request.auth.clients[provider].token_redirect(request, app=get_idp())
+                return await request.auth.clients[provider].token_data(request)
+
         application.include_router(app_router)
-        application.mount("", get_idp())
+        application.include_router(oauth2_router)
+
+        if with_idp:
+            application.mount("", get_idp())
+
         application.add_middleware(OAuth2Middleware, config={
+            "enable_ssr": with_ssr,
             "allow_http": True,
             "clients": [
                 OAuth2Client(