test(tokens-repo): split between abstract api and backend-specific tests
parent
16e0bd56ce
commit
4cfe0515ea
|
@ -0,0 +1,154 @@
|
||||||
|
# pylint: disable=redefined-outer-name
|
||||||
|
# pylint: disable=unused-argument
|
||||||
|
# pylint: disable=missing-function-docstring
|
||||||
|
"""
|
||||||
|
tests that restrict json token repository implementation
|
||||||
|
"""
|
||||||
|
|
||||||
|
import pytest
|
||||||
|
|
||||||
|
|
||||||
|
from datetime import datetime
|
||||||
|
|
||||||
|
from selfprivacy_api.models.tokens.token import Token
|
||||||
|
from selfprivacy_api.repositories.tokens.exceptions import (
|
||||||
|
TokenNotFound,
|
||||||
|
)
|
||||||
|
from selfprivacy_api.repositories.tokens.json_tokens_repository import (
|
||||||
|
JsonTokensRepository,
|
||||||
|
)
|
||||||
|
from tests.common import read_json
|
||||||
|
|
||||||
|
from test_tokens_repository import ORIGINAL_TOKEN_CONTENT
|
||||||
|
from test_tokens_repository import (
|
||||||
|
tokens,
|
||||||
|
mock_recovery_key_generate,
|
||||||
|
mock_generate_token,
|
||||||
|
mock_new_device_key_generate,
|
||||||
|
empty_keys,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def test_delete_token(tokens):
|
||||||
|
repo = JsonTokensRepository()
|
||||||
|
input_token = Token(
|
||||||
|
token="KG9ni-B-CMPk327Zv1qC7YBQaUGaBUcgdkvMvQ2atFI",
|
||||||
|
device_name="primary_token",
|
||||||
|
created_at=datetime(2022, 7, 15, 17, 41, 31, 675698),
|
||||||
|
)
|
||||||
|
|
||||||
|
repo.delete_token(input_token)
|
||||||
|
assert read_json(tokens / "tokens.json")["tokens"] == [
|
||||||
|
{
|
||||||
|
"token": "3JKgLOtFu6ZHgE4OU-R-VdW47IKpg-YQL0c6n7bol68",
|
||||||
|
"name": "second_token",
|
||||||
|
"date": "2022-07-15 17:41:31.675698Z",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"token": "LYiwFDekvALKTQSjk7vtMQuNP_6wqKuV-9AyMKytI_8",
|
||||||
|
"name": "third_token",
|
||||||
|
"date": "2022-07-15T17:41:31.675698Z",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"token": "dD3CFPcEZvapscgzWb7JZTLog7OMkP7NzJeu2fAazXM",
|
||||||
|
"name": "forth_token",
|
||||||
|
"date": "2022-07-15T17:41:31.675698",
|
||||||
|
},
|
||||||
|
]
|
||||||
|
|
||||||
|
|
||||||
|
def test_delete_not_found_token(tokens):
|
||||||
|
repo = JsonTokensRepository()
|
||||||
|
input_token = Token(
|
||||||
|
token="imbadtoken",
|
||||||
|
device_name="primary_token",
|
||||||
|
created_at=datetime(2022, 7, 15, 17, 41, 31, 675698),
|
||||||
|
)
|
||||||
|
with pytest.raises(TokenNotFound):
|
||||||
|
assert repo.delete_token(input_token) is None
|
||||||
|
|
||||||
|
assert read_json(tokens / "tokens.json")["tokens"] == ORIGINAL_TOKEN_CONTENT
|
||||||
|
|
||||||
|
|
||||||
|
def test_create_recovery_key(tokens, mock_recovery_key_generate):
|
||||||
|
repo = JsonTokensRepository()
|
||||||
|
|
||||||
|
assert repo.create_recovery_key(uses_left=1, expiration=None) is not None
|
||||||
|
assert read_json(tokens / "tokens.json")["recovery_token"] == {
|
||||||
|
"token": "889bf49c1d3199d71a2e704718772bd53a422020334db051",
|
||||||
|
"date": "2022-07-15T17:41:31.675698",
|
||||||
|
"expiration": None,
|
||||||
|
"uses_left": 1,
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
def test_use_mnemonic_recovery_key(tokens, mock_generate_token):
|
||||||
|
repo = JsonTokensRepository()
|
||||||
|
|
||||||
|
assert repo.use_mnemonic_recovery_key(
|
||||||
|
mnemonic_phrase="uniform clarify napkin bid dress search input armor police cross salon because myself uphold slice bamboo hungry park",
|
||||||
|
device_name="newdevice",
|
||||||
|
) == Token(
|
||||||
|
token="ur71mC4aiI6FIYAN--cTL-38rPHS5D6NuB1bgN_qKF4",
|
||||||
|
device_name="newdevice",
|
||||||
|
created_at=datetime(2022, 11, 14, 6, 6, 32, 777123),
|
||||||
|
)
|
||||||
|
|
||||||
|
assert read_json(tokens / "tokens.json")["tokens"] == [
|
||||||
|
{
|
||||||
|
"date": "2022-07-15 17:41:31.675698",
|
||||||
|
"name": "primary_token",
|
||||||
|
"token": "KG9ni-B-CMPk327Zv1qC7YBQaUGaBUcgdkvMvQ2atFI",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"token": "3JKgLOtFu6ZHgE4OU-R-VdW47IKpg-YQL0c6n7bol68",
|
||||||
|
"name": "second_token",
|
||||||
|
"date": "2022-07-15 17:41:31.675698Z",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"token": "LYiwFDekvALKTQSjk7vtMQuNP_6wqKuV-9AyMKytI_8",
|
||||||
|
"name": "third_token",
|
||||||
|
"date": "2022-07-15T17:41:31.675698Z",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"token": "dD3CFPcEZvapscgzWb7JZTLog7OMkP7NzJeu2fAazXM",
|
||||||
|
"name": "forth_token",
|
||||||
|
"date": "2022-07-15T17:41:31.675698",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"date": "2022-11-14T06:06:32.777123",
|
||||||
|
"name": "newdevice",
|
||||||
|
"token": "ur71mC4aiI6FIYAN--cTL-38rPHS5D6NuB1bgN_qKF4",
|
||||||
|
},
|
||||||
|
]
|
||||||
|
assert read_json(tokens / "tokens.json")["recovery_token"] == {
|
||||||
|
"date": "2022-11-11T11:48:54.228038",
|
||||||
|
"expiration": None,
|
||||||
|
"token": "ed653e4b8b042b841d285fa7a682fa09e925ddb2d8906f54",
|
||||||
|
"uses_left": 1,
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
def test_get_new_device_key(tokens, mock_new_device_key_generate):
|
||||||
|
repo = JsonTokensRepository()
|
||||||
|
|
||||||
|
assert repo.get_new_device_key() is not None
|
||||||
|
assert read_json(tokens / "tokens.json")["new_device"] == {
|
||||||
|
"date": "2022-07-15T17:41:31.675698",
|
||||||
|
"expiration": "2022-07-15T17:41:31.675698",
|
||||||
|
"token": "43478d05b35e4781598acd76e33832bb",
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
def test_delete_new_device_key(tokens):
|
||||||
|
repo = JsonTokensRepository()
|
||||||
|
|
||||||
|
assert repo.delete_new_device_key() is None
|
||||||
|
assert "new_device" not in read_json(tokens / "tokens.json")
|
||||||
|
|
||||||
|
|
||||||
|
def test_delete_new_device_key_when_empty(empty_keys):
|
||||||
|
repo = JsonTokensRepository()
|
||||||
|
|
||||||
|
repo.delete_new_device_key()
|
||||||
|
assert "new_device" not in read_json(empty_keys / "empty_keys.json")
|
|
@ -0,0 +1,9 @@
|
||||||
|
{
|
||||||
|
"tokens": [
|
||||||
|
{
|
||||||
|
"token": "KG9ni-B-CMPk327Zv1qC7YBQaUGaBUcgdkvMvQ2atFI",
|
||||||
|
"name": "primary_token",
|
||||||
|
"date": "2022-07-15 17:41:31.675698"
|
||||||
|
}
|
||||||
|
]
|
||||||
|
}
|
|
@ -0,0 +1,26 @@
|
||||||
|
{
|
||||||
|
"tokens": [
|
||||||
|
{
|
||||||
|
"token": "KG9ni-B-CMPk327Zv1qC7YBQaUGaBUcgdkvMvQ2atFI",
|
||||||
|
"name": "primary_token",
|
||||||
|
"date": "2022-07-15 17:41:31.675698"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"token": "3JKgLOtFu6ZHgE4OU-R-VdW47IKpg-YQL0c6n7bol68",
|
||||||
|
"name": "second_token",
|
||||||
|
"date": "2022-07-15 17:41:31.675698Z"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"token": "LYiwFDekvALKTQSjk7vtMQuNP_6wqKuV-9AyMKytI_8",
|
||||||
|
"name": "third_token",
|
||||||
|
"date": "2022-07-15T17:41:31.675698Z"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"token": "dD3CFPcEZvapscgzWb7JZTLog7OMkP7NzJeu2fAazXM",
|
||||||
|
"name": "forth_token",
|
||||||
|
"date": "2022-07-15T17:41:31.675698"
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"recovery_token": null,
|
||||||
|
"new_device": null
|
||||||
|
}
|
|
@ -0,0 +1,35 @@
|
||||||
|
{
|
||||||
|
"tokens": [
|
||||||
|
{
|
||||||
|
"token": "KG9ni-B-CMPk327Zv1qC7YBQaUGaBUcgdkvMvQ2atFI",
|
||||||
|
"name": "primary_token",
|
||||||
|
"date": "2022-07-15 17:41:31.675698"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"token": "3JKgLOtFu6ZHgE4OU-R-VdW47IKpg-YQL0c6n7bol68",
|
||||||
|
"name": "second_token",
|
||||||
|
"date": "2022-07-15 17:41:31.675698Z"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"token": "LYiwFDekvALKTQSjk7vtMQuNP_6wqKuV-9AyMKytI_8",
|
||||||
|
"name": "third_token",
|
||||||
|
"date": "2022-07-15T17:41:31.675698Z"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"token": "dD3CFPcEZvapscgzWb7JZTLog7OMkP7NzJeu2fAazXM",
|
||||||
|
"name": "forth_token",
|
||||||
|
"date": "2022-07-15T17:41:31.675698"
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"recovery_token": {
|
||||||
|
"token": "ed653e4b8b042b841d285fa7a682fa09e925ddb2d8906f54",
|
||||||
|
"date": "2022-11-11T11:48:54.228038",
|
||||||
|
"expiration": null,
|
||||||
|
"uses_left": 2
|
||||||
|
},
|
||||||
|
"new_device": {
|
||||||
|
"token": "2237238de23dc71ab558e317bdb8ff8e",
|
||||||
|
"date": "2022-10-26 20:50:47.973212",
|
||||||
|
"expiration": "2022-10-26 21:00:47.974153"
|
||||||
|
}
|
||||||
|
}
|
|
@ -236,34 +236,6 @@ def test_create_token(tokens, mock_token_generate):
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
def test_delete_token(tokens):
|
|
||||||
repo = JsonTokensRepository()
|
|
||||||
input_token = Token(
|
|
||||||
token="KG9ni-B-CMPk327Zv1qC7YBQaUGaBUcgdkvMvQ2atFI",
|
|
||||||
device_name="primary_token",
|
|
||||||
created_at=datetime(2022, 7, 15, 17, 41, 31, 675698),
|
|
||||||
)
|
|
||||||
|
|
||||||
repo.delete_token(input_token)
|
|
||||||
assert read_json(tokens / "tokens.json")["tokens"] == [
|
|
||||||
{
|
|
||||||
"token": "3JKgLOtFu6ZHgE4OU-R-VdW47IKpg-YQL0c6n7bol68",
|
|
||||||
"name": "second_token",
|
|
||||||
"date": "2022-07-15 17:41:31.675698Z",
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"token": "LYiwFDekvALKTQSjk7vtMQuNP_6wqKuV-9AyMKytI_8",
|
|
||||||
"name": "third_token",
|
|
||||||
"date": "2022-07-15T17:41:31.675698Z",
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"token": "dD3CFPcEZvapscgzWb7JZTLog7OMkP7NzJeu2fAazXM",
|
|
||||||
"name": "forth_token",
|
|
||||||
"date": "2022-07-15T17:41:31.675698",
|
|
||||||
},
|
|
||||||
]
|
|
||||||
|
|
||||||
|
|
||||||
def test_delete_not_found_token(tokens):
|
def test_delete_not_found_token(tokens):
|
||||||
repo = JsonTokensRepository()
|
repo = JsonTokensRepository()
|
||||||
input_token = Token(
|
input_token = Token(
|
||||||
|
@ -274,7 +246,7 @@ def test_delete_not_found_token(tokens):
|
||||||
with pytest.raises(TokenNotFound):
|
with pytest.raises(TokenNotFound):
|
||||||
assert repo.delete_token(input_token) is None
|
assert repo.delete_token(input_token) is None
|
||||||
|
|
||||||
assert read_json(tokens / "tokens.json")["tokens"] == ORIGINAL_TOKEN_CONTENT
|
# assert read_json(tokens / "tokens.json")["tokens"] == ORIGINAL_TOKEN_CONTENT
|
||||||
|
|
||||||
|
|
||||||
def test_refresh_token(tokens, mock_token_generate):
|
def test_refresh_token(tokens, mock_token_generate):
|
||||||
|
@ -330,12 +302,12 @@ def test_create_recovery_key(tokens, mock_recovery_key_generate):
|
||||||
repo = JsonTokensRepository()
|
repo = JsonTokensRepository()
|
||||||
|
|
||||||
assert repo.create_recovery_key(uses_left=1, expiration=None) is not None
|
assert repo.create_recovery_key(uses_left=1, expiration=None) is not None
|
||||||
assert read_json(tokens / "tokens.json")["recovery_token"] == {
|
# assert read_json(tokens / "tokens.json")["recovery_token"] == {
|
||||||
"token": "889bf49c1d3199d71a2e704718772bd53a422020334db051",
|
# "token": "889bf49c1d3199d71a2e704718772bd53a422020334db051",
|
||||||
"date": "2022-07-15T17:41:31.675698",
|
# "date": "2022-07-15T17:41:31.675698",
|
||||||
"expiration": None,
|
# "expiration": None,
|
||||||
"uses_left": 1,
|
# "uses_left": 1,
|
||||||
}
|
# }
|
||||||
|
|
||||||
|
|
||||||
def test_use_mnemonic_recovery_key_when_empty(
|
def test_use_mnemonic_recovery_key_when_empty(
|
||||||
|
@ -433,6 +405,7 @@ def test_use_menemonic_recovery_key_when_null(null_keys):
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
# agnostic test mixed with an implementation test
|
||||||
def test_use_mnemonic_recovery_key(tokens, mock_generate_token):
|
def test_use_mnemonic_recovery_key(tokens, mock_generate_token):
|
||||||
repo = JsonTokensRepository()
|
repo = JsonTokensRepository()
|
||||||
|
|
||||||
|
@ -445,40 +418,40 @@ def test_use_mnemonic_recovery_key(tokens, mock_generate_token):
|
||||||
created_at=datetime(2022, 11, 14, 6, 6, 32, 777123),
|
created_at=datetime(2022, 11, 14, 6, 6, 32, 777123),
|
||||||
)
|
)
|
||||||
|
|
||||||
assert read_json(tokens / "tokens.json")["tokens"] == [
|
# assert read_json(tokens / "tokens.json")["tokens"] == [
|
||||||
{
|
# {
|
||||||
"date": "2022-07-15 17:41:31.675698",
|
# "date": "2022-07-15 17:41:31.675698",
|
||||||
"name": "primary_token",
|
# "name": "primary_token",
|
||||||
"token": "KG9ni-B-CMPk327Zv1qC7YBQaUGaBUcgdkvMvQ2atFI",
|
# "token": "KG9ni-B-CMPk327Zv1qC7YBQaUGaBUcgdkvMvQ2atFI",
|
||||||
},
|
# },
|
||||||
{
|
# {
|
||||||
"token": "3JKgLOtFu6ZHgE4OU-R-VdW47IKpg-YQL0c6n7bol68",
|
# "token": "3JKgLOtFu6ZHgE4OU-R-VdW47IKpg-YQL0c6n7bol68",
|
||||||
"name": "second_token",
|
# "name": "second_token",
|
||||||
"date": "2022-07-15 17:41:31.675698Z",
|
# "date": "2022-07-15 17:41:31.675698Z",
|
||||||
},
|
# },
|
||||||
{
|
# {
|
||||||
"token": "LYiwFDekvALKTQSjk7vtMQuNP_6wqKuV-9AyMKytI_8",
|
# "token": "LYiwFDekvALKTQSjk7vtMQuNP_6wqKuV-9AyMKytI_8",
|
||||||
"name": "third_token",
|
# "name": "third_token",
|
||||||
"date": "2022-07-15T17:41:31.675698Z",
|
# "date": "2022-07-15T17:41:31.675698Z",
|
||||||
},
|
# },
|
||||||
{
|
# {
|
||||||
"token": "dD3CFPcEZvapscgzWb7JZTLog7OMkP7NzJeu2fAazXM",
|
# "token": "dD3CFPcEZvapscgzWb7JZTLog7OMkP7NzJeu2fAazXM",
|
||||||
"name": "forth_token",
|
# "name": "forth_token",
|
||||||
"date": "2022-07-15T17:41:31.675698",
|
# "date": "2022-07-15T17:41:31.675698",
|
||||||
},
|
# },
|
||||||
{
|
# {
|
||||||
"date": "2022-11-14T06:06:32.777123",
|
# "date": "2022-11-14T06:06:32.777123",
|
||||||
"name": "newdevice",
|
# "name": "newdevice",
|
||||||
"token": "ur71mC4aiI6FIYAN--cTL-38rPHS5D6NuB1bgN_qKF4",
|
# "token": "ur71mC4aiI6FIYAN--cTL-38rPHS5D6NuB1bgN_qKF4",
|
||||||
},
|
# },
|
||||||
]
|
# ]
|
||||||
|
|
||||||
assert read_json(tokens / "tokens.json")["recovery_token"] == {
|
# assert read_json(tokens / "tokens.json")["recovery_token"] == {
|
||||||
"date": "2022-11-11T11:48:54.228038",
|
# "date": "2022-11-11T11:48:54.228038",
|
||||||
"expiration": None,
|
# "expiration": None,
|
||||||
"token": "ed653e4b8b042b841d285fa7a682fa09e925ddb2d8906f54",
|
# "token": "ed653e4b8b042b841d285fa7a682fa09e925ddb2d8906f54",
|
||||||
"uses_left": 1,
|
# "uses_left": 1,
|
||||||
}
|
# }
|
||||||
|
|
||||||
|
|
||||||
##################
|
##################
|
||||||
|
@ -490,25 +463,25 @@ def test_get_new_device_key(tokens, mock_new_device_key_generate):
|
||||||
repo = JsonTokensRepository()
|
repo = JsonTokensRepository()
|
||||||
|
|
||||||
assert repo.get_new_device_key() is not None
|
assert repo.get_new_device_key() is not None
|
||||||
assert read_json(tokens / "tokens.json")["new_device"] == {
|
# assert read_json(tokens / "tokens.json")["new_device"] == {
|
||||||
"date": "2022-07-15T17:41:31.675698",
|
# "date": "2022-07-15T17:41:31.675698",
|
||||||
"expiration": "2022-07-15T17:41:31.675698",
|
# "expiration": "2022-07-15T17:41:31.675698",
|
||||||
"token": "43478d05b35e4781598acd76e33832bb",
|
# "token": "43478d05b35e4781598acd76e33832bb",
|
||||||
}
|
# }
|
||||||
|
|
||||||
|
|
||||||
def test_delete_new_device_key(tokens):
|
def test_delete_new_device_key(tokens):
|
||||||
repo = JsonTokensRepository()
|
repo = JsonTokensRepository()
|
||||||
|
|
||||||
assert repo.delete_new_device_key() is None
|
assert repo.delete_new_device_key() is None
|
||||||
assert "new_device" not in read_json(tokens / "tokens.json")
|
# assert "new_device" not in read_json(tokens / "tokens.json")
|
||||||
|
|
||||||
|
|
||||||
def test_delete_new_device_key_when_empty(empty_keys):
|
def test_delete_new_device_key_when_empty(empty_keys):
|
||||||
repo = JsonTokensRepository()
|
repo = JsonTokensRepository()
|
||||||
|
|
||||||
repo.delete_new_device_key()
|
repo.delete_new_device_key()
|
||||||
assert "new_device" not in read_json(empty_keys / "empty_keys.json")
|
# assert "new_device" not in read_json(empty_keys / "empty_keys.json")
|
||||||
|
|
||||||
|
|
||||||
def test_use_invalid_mnemonic_new_device_key(
|
def test_use_invalid_mnemonic_new_device_key(
|
||||||
|
|
Loading…
Reference in New Issue