From d3111d0c72940a9b761f64ef5254f3a2f6ca00c1 Mon Sep 17 00:00:00 2001 From: tim Date: Tue, 5 Mar 2024 11:37:49 +0000 Subject: [PATCH] format tests --- src/python-fastui/tests/test_auth_google.py | 86 ++++++++++----------- 1 file changed, 41 insertions(+), 45 deletions(-) diff --git a/src/python-fastui/tests/test_auth_google.py b/src/python-fastui/tests/test_auth_google.py index 7729d613..1f6d672c 100644 --- a/src/python-fastui/tests/test_auth_google.py +++ b/src/python-fastui/tests/test_auth_google.py @@ -1,20 +1,16 @@ -from datetime import datetime, timedelta -from typing import List, Optional +from datetime import timedelta import httpx import pytest -from fastapi import FastAPI -from pydantic import SecretStr - from fastui.auth.google import ( + EXCHANGE_CACHE, AuthError, GoogleAuthProvider, GoogleExchange, - GoogleExchangeError, GoogleUser, - EXCHANGE_CACHE, ) from httpx import Request, Response +from pydantic import SecretStr class MockTransport(httpx.AsyncBaseTransport): @@ -22,34 +18,34 @@ async def handle_async_request(self, request: Request) -> Response: url = str(request.url) method = request.method - if url == "https://oauth2.googleapis.com/token" and method == "POST": + if url == 'https://oauth2.googleapis.com/token' and method == 'POST': print(request.read()) - if b"code=bad_code" in request.read(): - return Response(200, json={"error": "bad code"}) + if b'code=bad_code' in request.read(): + return Response(200, json={'error': 'bad code'}) json_data = { - "access_token": "test_access_token", - "token_type": "Bearer", - "expires_in": 3600, - "refresh_token": "test_refresh_token", - "scope": "email profile", + 'access_token': 'test_access_token', + 'token_type': 'Bearer', + 'expires_in': 3600, + 'refresh_token': 'test_refresh_token', + 'scope': 'email profile', } return Response(200, json=json_data) - elif url == "https://www.googleapis.com/oauth2/v1/userinfo" and method == "GET": + elif url == 'https://www.googleapis.com/oauth2/v1/userinfo' and method == 'GET': json_data = { - "id": "12345", - "email": "user@example.com", - "verified_email": True, - "name": "Test User", - "given_name": "Test", - "family_name": "User", - "picture": "https://example.com/avatar.png", - "locale": "en", + 'id': '12345', + 'email': 'user@example.com', + 'verified_email': True, + 'name': 'Test User', + 'given_name': 'Test', + 'family_name': 'User', + 'picture': 'https://example.com/avatar.png', + 'locale': 'en', } return Response(200, json=json_data) - return Response(404, json={"error": "not found"}) + return Response(404, json={'error': 'not found'}) @pytest.fixture @@ -63,10 +59,10 @@ async def mock_httpx_client() -> httpx.AsyncClient: async def google_auth_provider(mock_httpx_client: httpx.AsyncClient): return GoogleAuthProvider( httpx_client=mock_httpx_client, - google_client_id="google_client_id", - google_client_secret=SecretStr("google_client_secret"), - redirect_uri="https://example.com/callback", - scopes=["email", "profile"], + google_client_id='google_client_id', + google_client_secret=SecretStr('google_client_secret'), + redirect_uri='https://example.com/callback', + scopes=['email', 'profile'], state_provider=False, exchange_cache_age=timedelta(minutes=5), ) @@ -74,41 +70,41 @@ async def google_auth_provider(mock_httpx_client: httpx.AsyncClient): async def test_authorization_url(google_auth_provider: GoogleAuthProvider): url = await google_auth_provider.authorization_url() - assert url.startswith("https://accounts.google.com/o/oauth2/v2/auth?") + assert url.startswith('https://accounts.google.com/o/oauth2/v2/auth?') async def test_exchange_code_success(google_auth_provider: GoogleAuthProvider): - exchange = await google_auth_provider.exchange_code("good_code") + exchange = await google_auth_provider.exchange_code('good_code') assert isinstance(exchange, GoogleExchange) - assert exchange.access_token == "test_access_token" - assert exchange.token_type == "Bearer" - assert exchange.scope == "email profile" - assert exchange.refresh_token == "test_refresh_token" + assert exchange.access_token == 'test_access_token' + assert exchange.token_type == 'Bearer' + assert exchange.scope == 'email profile' + assert exchange.refresh_token == 'test_refresh_token' async def test_exchange_code_error(google_auth_provider: GoogleAuthProvider): with pytest.raises(AuthError): - await google_auth_provider.exchange_code("bad_code") + await google_auth_provider.exchange_code('bad_code') async def test_refresh_access_token(google_auth_provider: GoogleAuthProvider): - new_token = await google_auth_provider.refresh_access_token("good_refresh_token") + new_token = await google_auth_provider.refresh_access_token('good_refresh_token') assert isinstance(new_token, GoogleExchange) - assert new_token.access_token == "test_access_token" + assert new_token.access_token == 'test_access_token' async def test_get_google_user(google_auth_provider: GoogleAuthProvider): exchange = GoogleExchange( - access_token="good_access_token", - token_type="Bearer", - scope="email profile", + access_token='good_access_token', + token_type='Bearer', + scope='email profile', expires_in=3600, - refresh_token="good_refresh_token", + refresh_token='good_refresh_token', ) user = await google_auth_provider.get_google_user(exchange) assert isinstance(user, GoogleUser) - assert user.id == "12345" - assert user.email == "user@example.com" + assert user.id == '12345' + assert user.email == 'user@example.com' async def test_exchange_cache( @@ -116,5 +112,5 @@ async def test_exchange_cache( ): EXCHANGE_CACHE._cache.clear() assert len(EXCHANGE_CACHE._cache) == 0 - await google_auth_provider.exchange_code("good_code") + await google_auth_provider.exchange_code('good_code') assert len(EXCHANGE_CACHE._cache) == 1