Alexander/FitTrackee
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

API - minor tests refacto

Browse Source
This commit is contained in:
Sam 2022-06-12 18:34:26 +02:00
parent ea4b6ed236
commit b678b6e76d
1 changed files with 17 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
fittrackee/tests/oauth2/test_oauth2_models.py
View File

@ -5,31 +5,15 @@ import pytest
from flask import Flask from flask import Flask
from fittrackee.oauth2.models import OAuth2Client, OAuth2Token from fittrackee.oauth2.models import OAuth2Client, OAuth2Token
from fittrackee.users.models import User
from ..mixins import RandomMixin from ..mixins import OAuth2Mixin
class OAuth2ModelTestCase(RandomMixin): class TestOAuth2ClientSerialize(OAuth2Mixin):
def create_oauth2_client(self) -> OAuth2Client: def test_it_returns_oauth_client(self, app: Flask, user_1: User) -> None:
oauth_client = OAuth2Client( oauth_client = self.create_oauth_client(user_1)
id=self.random_int(), oauth_client.client_id_issued_at = 1653738796
client_id=self.random_string(),
client_id_issued_at=1653738796,
)
oauth_client.set_client_metadata(
{
'client_name': self.random_string(),
'client_description': self.random_string(),
'redirect_uris': [self.random_string()],
'client_uri': self.random_domain(),
}
)
return oauth_client
class TestOAuth2ClientSerialize(OAuth2ModelTestCase):
def test_it_returns_oauth_client(self, app: Flask) -> None:
oauth_client = self.create_oauth2_client()
serialized_oauth_client = oauth_client.serialize() serialized_oauth_client = oauth_client.serialize()
@ -76,14 +60,18 @@ class TestOAuth2ClientSerialize(OAuth2ModelTestCase):
) )
class TestOAuth2Token(OAuth2ModelTestCase): class TestOAuth2Token(OAuth2Mixin):
@pytest.mark.parametrize( @pytest.mark.parametrize(
'input_expiration,expected_status', [(1000, True), (0, False)] 'input_expiration,expected_status', [(1000, True), (0, False)]
) )
def test_it_returns_refresh_token_status( def test_it_returns_refresh_token_status(
self, app: Flask, input_expiration: int, expected_status: bool self,
app: Flask,
user_1: User,
input_expiration: int,
expected_status: bool,
) -> None: ) -> None:
oauth_client = self.create_oauth2_client() oauth_client = self.create_oauth_client(user_1)
token = OAuth2Token( token = OAuth2Token(
client_id=oauth_client.client_id, client_id=oauth_client.client_id,
access_token=self.random_string(), access_token=self.random_string(),
@ -95,9 +83,9 @@ class TestOAuth2Token(OAuth2ModelTestCase):
assert token.is_refresh_token_active() is expected_status assert token.is_refresh_token_active() is expected_status
def test_it_returns_refresh_token_active_when_below_twice_expiration( def test_it_returns_refresh_token_active_when_below_twice_expiration(
self, app: Flask self, app: Flask, user_1: User
) -> None: ) -> None:
oauth_client = self.create_oauth2_client() oauth_client = self.create_oauth_client(user_1)
issued_at = int(time.time()) issued_at = int(time.time())
expires_in = self.random_int() expires_in = self.random_int()
token = OAuth2Token( token = OAuth2Token(
@ -115,9 +103,9 @@ class TestOAuth2Token(OAuth2ModelTestCase):
assert token.is_refresh_token_active() is True assert token.is_refresh_token_active() is True
def test_it_returns_refresh_token_inactive_when_token_revoked( def test_it_returns_refresh_token_inactive_when_token_revoked(
self, app: Flask self, app: Flask, user_1: User
) -> None: ) -> None:
oauth_client = self.create_oauth2_client() oauth_client = self.create_oauth_client(user_1)
token = OAuth2Token( token = OAuth2Token(
client_id=oauth_client.client_id, client_id=oauth_client.client_id,
access_token=self.random_string(), access_token=self.random_string(),