138 lines
4.3 KiB
Python
138 lines
4.3 KiB
Python
from time import time
|
|
from typing import Dict
|
|
from unittest.mock import patch
|
|
|
|
from flask import Flask
|
|
|
|
from fittrackee.oauth2.client import create_oauth_client
|
|
from fittrackee.oauth2.models import OAuth2Client
|
|
from fittrackee.users.models import User
|
|
|
|
from ..utils import random_domain, random_string
|
|
|
|
TEST_METADATA = {
|
|
'client_name': random_string(),
|
|
'client_uri': random_string(),
|
|
'redirect_uris': [random_domain()],
|
|
'scope': 'read write',
|
|
}
|
|
|
|
|
|
class TestCreateOAuth2Client:
|
|
def test_it_creates_oauth_client(self, app: Flask, user_1: User) -> None:
|
|
oauth_client = create_oauth_client(TEST_METADATA, user_1)
|
|
|
|
assert isinstance(oauth_client, OAuth2Client)
|
|
|
|
def test_oauth_client_id_is_generated_with_gen_salt(
|
|
self, app: Flask, user_1: User
|
|
) -> None:
|
|
client_id = random_string()
|
|
with patch(
|
|
'fittrackee.oauth2.client.gen_salt', return_value=client_id
|
|
):
|
|
|
|
oauth_client = create_oauth_client(TEST_METADATA, user_1)
|
|
|
|
assert oauth_client.client_id == client_id
|
|
|
|
def test_oauth_client_client_id_issued_at_is_initialized(
|
|
self, app: Flask, user_1: User
|
|
) -> None:
|
|
client_id_issued_at = int(time())
|
|
with patch(
|
|
'fittrackee.oauth2.client.time', return_value=client_id_issued_at
|
|
):
|
|
oauth_client = create_oauth_client(TEST_METADATA, user_1)
|
|
|
|
assert oauth_client.client_id_issued_at == client_id_issued_at
|
|
|
|
def test_oauth_client_has_expected_name(
|
|
self, app: Flask, user_1: User
|
|
) -> None:
|
|
client_name = random_string()
|
|
client_metadata: Dict = {**TEST_METADATA, 'client_name': client_name}
|
|
|
|
oauth_client = create_oauth_client(client_metadata, user_1)
|
|
|
|
assert oauth_client.client_name == client_name
|
|
|
|
def test_oauth_client_has_expected_client_uri(
|
|
self, app: Flask, user_1: User
|
|
) -> None:
|
|
client_uri = random_domain()
|
|
client_metadata: Dict = {**TEST_METADATA, 'client_uri': client_uri}
|
|
|
|
oauth_client = create_oauth_client(client_metadata, user_1)
|
|
|
|
assert oauth_client.client_uri == client_uri
|
|
|
|
def test_oauth_client_has_expected_grant_types(
|
|
self, app: Flask, user_1: User
|
|
) -> None:
|
|
oauth_client = create_oauth_client(TEST_METADATA, user_1)
|
|
|
|
assert oauth_client.grant_types == ['authorization_code']
|
|
|
|
def test_oauth_client_has_expected_redirect_uris(
|
|
self, app: Flask, user_1: User
|
|
) -> None:
|
|
redirect_uris = [random_domain()]
|
|
client_metadata: Dict = {
|
|
**TEST_METADATA,
|
|
'redirect_uris': redirect_uris,
|
|
}
|
|
|
|
oauth_client = create_oauth_client(client_metadata, user_1)
|
|
|
|
assert oauth_client.redirect_uris == redirect_uris
|
|
|
|
def test_oauth_client_has_expected_response_types(
|
|
self, app: Flask, user_1: User
|
|
) -> None:
|
|
response_types = ['code']
|
|
client_metadata: Dict = {
|
|
**TEST_METADATA,
|
|
'response_types': response_types,
|
|
}
|
|
|
|
oauth_client = create_oauth_client(client_metadata, user_1)
|
|
|
|
assert oauth_client.response_types == ['code']
|
|
|
|
def test_oauth_client_has_expected_scope(
|
|
self, app: Flask, user_1: User
|
|
) -> None:
|
|
scope = 'profile'
|
|
client_metadata: Dict = {**TEST_METADATA, 'scope': scope}
|
|
|
|
oauth_client = create_oauth_client(client_metadata, user_1)
|
|
|
|
assert oauth_client.scope == scope
|
|
|
|
def test_oauth_client_has_expected_token_endpoint_auth_method(
|
|
self, app: Flask, user_1: User
|
|
) -> None:
|
|
oauth_client = create_oauth_client(TEST_METADATA, user_1)
|
|
|
|
assert oauth_client.token_endpoint_auth_method == 'client_secret_post'
|
|
|
|
def test_when_auth_method_is_not_none_oauth_client_secret_is_generated(
|
|
self, app: Flask, user_1: User
|
|
) -> None:
|
|
client_secret = random_string()
|
|
with patch(
|
|
'fittrackee.oauth2.client.gen_salt', return_value=client_secret
|
|
):
|
|
|
|
oauth_client = create_oauth_client(TEST_METADATA, user_1)
|
|
|
|
assert oauth_client.client_secret == client_secret
|
|
|
|
def test_it_creates_oauth_client_for_given_user(
|
|
self, app: Flask, user_1: User
|
|
) -> None:
|
|
oauth_client = create_oauth_client(TEST_METADATA, user_1)
|
|
|
|
assert oauth_client.user_id == user_1.id
|