SelfPrivacy
/
selfprivacy-rest-api
7
2
Fork 1
Code Issues 15 Pull Requests 3 Projects Releases 1 Wiki Activity

test(tokens-repo): untie refresh token testing from token repo

redis/connection-pool
Houkime 2023-01-04 15:37:48 +00:00 committed by Inex Code
parent 469f9d292d
commit 6eb5800e4e
1 changed files with 23 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
tests/test_graphql/test_api_devices.py
View File

@ -2,13 +2,8 @@
# pylint: disable=unused-argument # pylint: disable=unused-argument
# pylint: disable=missing-function-docstring # pylint: disable=missing-function-docstring
import datetime import datetime
import pytest
from mnemonic import Mnemonic from mnemonic import Mnemonic
from selfprivacy_api.repositories.tokens.json_tokens_repository import (
JsonTokensRepository,
)
from selfprivacy_api.models.tokens.token import Token
from tests.common import generate_api_query, read_json, write_json from tests.common import generate_api_query, read_json, write_json
from tests.conftest import DEVICE_WE_AUTH_TESTS_WITH, TOKENS_FILE_CONTENTS from tests.conftest import DEVICE_WE_AUTH_TESTS_WITH, TOKENS_FILE_CONTENTS
@ -24,11 +19,6 @@ devices {
""" """
@pytest.fixture
def token_repo():
return JsonTokensRepository()
def graphql_get_devices(client): def graphql_get_devices(client):
response = client.post( response = client.post(
"/graphql", "/graphql",
@ -40,6 +30,13 @@ def graphql_get_devices(client):
return devices return devices
def graphql_get_caller_token_info(client):
devices = graphql_get_devices(client)
for device in devices:
if device["isCaller"] is True:
return device
def assert_same(graphql_devices, abstract_devices): def assert_same(graphql_devices, abstract_devices):
"""Orderless comparison""" """Orderless comparison"""
assert len(graphql_devices) == len(abstract_devices) assert len(graphql_devices) == len(abstract_devices)
@ -87,6 +84,15 @@ def assert_data(response):
return data return data
def set_client_token(client, token):
client.headers.update({"Authorization": "Bearer " + token})
def assert_token_valid(client, token):
set_client_token(client, token)
assert graphql_get_devices(client) is not None
def test_graphql_tokens_info(authorized_client, tokens_file): def test_graphql_tokens_info(authorized_client, tokens_file):
assert_original(authorized_client) assert_original(authorized_client)
@ -192,19 +198,19 @@ def test_graphql_refresh_token_unauthorized(client, tokens_file):
assert_empty(response) assert_empty(response)
def test_graphql_refresh_token(authorized_client, tokens_file, token_repo): def test_graphql_refresh_token(authorized_client, client, tokens_file):
caller_name_and_date = graphql_get_caller_token_info(authorized_client)
response = authorized_client.post( response = authorized_client.post(
"/graphql", "/graphql",
json={"query": REFRESH_TOKEN_MUTATION}, json={"query": REFRESH_TOKEN_MUTATION},
) )
assert_ok(response, "refreshDeviceApiToken") assert_ok(response, "refreshDeviceApiToken")
token = token_repo.get_token_by_name("test_token") new_token = response.json()["data"]["refreshDeviceApiToken"]["token"]
assert token == Token( assert_token_valid(client, new_token)
token=response.json()["data"]["refreshDeviceApiToken"]["token"],
device_name="test_token", set_client_token(client, new_token)
created_at=datetime.datetime(2022, 1, 14, 8, 31, 10, 789314), assert graphql_get_caller_token_info(client) == caller_name_and_date
)
NEW_DEVICE_KEY_MUTATION = """ NEW_DEVICE_KEY_MUTATION = """