37 lines
1.3 KiB
Python
37 lines
1.3 KiB
Python
|
from uuid import uuid4
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
from werkzeug.exceptions import Unauthorized
|
||
|
|
|
||
|
|
from ereuse_devicehub.client import UserClient, Client
|
||
|
|
from ereuse_devicehub.devicehub import Devicehub
|
||
|
|
from tests.conftest import create_user
|
||
|
|
|
||
|
|
|
||
|
|
def test_authenticate_success(app: Devicehub):
|
||
|
|
"""Checks the authenticate method."""
|
||
|
|
with app.app_context():
|
||
|
|
user = create_user()
|
||
|
|
response_user = app.auth.authenticate(token=str(user.token))
|
||
|
|
assert response_user == user
|
||
|
|
|
||
|
|
|
||
|
|
def test_authenticate_error(app: Devicehub):
|
||
|
|
"""Tests the authenticate method with wrong token values."""
|
||
|
|
with app.app_context():
|
||
|
|
MESSAGE = 'Provide a suitable token.'
|
||
|
|
create_user()
|
||
|
|
# Token doesn't exist
|
||
|
|
with pytest.raises(Unauthorized, message=MESSAGE):
|
||
|
|
app.auth.authenticate(token=str(uuid4()))
|
||
|
|
# Wrong token format
|
||
|
|
with pytest.raises(Unauthorized, message=MESSAGE):
|
||
|
|
app.auth.authenticate(token='this is a wrong uuid')
|
||
|
|
|
||
|
|
|
||
|
|
def test_auth_view(user: UserClient, client: Client):
|
||
|
|
"""Tests authentication at endpoint / view."""
|
||
|
|
user.get(res='User', item=user.user['id'], status=200)
|
||
|
|
client.get(res='User', item=user.user['id'], status=Unauthorized)
|
||
|
|
client.get(res='User', item=user.user['id'], token='wrong token', status=Unauthorized)
|