test(tokens-repo): use assert recent

This commit is contained in:
Houkime 2023-01-06 10:59:59 +00:00 committed by Inex Code
parent 851d90b30c
commit 6cb9cc6d03
3 changed files with 17 additions and 15 deletions

View file

@ -37,3 +37,11 @@ def generate_users_query(query_array):
def mnemonic_to_hex(mnemonic):
return Mnemonic(language="english").to_entropy(mnemonic).hex()
def assert_recovery_recent(time_generated):
assert (
datetime.datetime.strptime(time_generated, "%Y-%m-%dT%H:%M:%S.%f")
- datetime.timedelta(seconds=5)
< datetime.datetime.now()
)

View file

@ -3,7 +3,13 @@
# pylint: disable=missing-function-docstring
import datetime
from tests.common import generate_api_query, mnemonic_to_hex, read_json, write_json
from tests.common import (
generate_api_query,
mnemonic_to_hex,
read_json,
write_json,
assert_recovery_recent,
)
from tests.test_graphql.common import assert_empty, assert_data, assert_ok
API_RECOVERY_QUERY = """
@ -90,12 +96,7 @@ def test_graphql_generate_recovery_key(client, authorized_client, tokens_file):
assert read_json(tokens_file)["recovery_token"] is not None
time_generated = read_json(tokens_file)["recovery_token"]["date"]
assert time_generated is not None
assert (
datetime.datetime.strptime(time_generated, "%Y-%m-%dT%H:%M:%S.%f")
- datetime.timedelta(seconds=5)
< datetime.datetime.now()
)
assert_recovery_recent(time_generated)
# Try to get token status
response = authorized_client.post(
"/graphql",

View file

@ -9,6 +9,7 @@ from tests.common import (
RECOVERY_KEY_VALIDATION_DATETIME,
DEVICE_KEY_VALIDATION_DATETIME,
NearFuture,
assert_recovery_recent,
)
DATE_FORMATS = [
@ -90,14 +91,6 @@ def rest_get_recovery_date(client):
return status["date"]
def assert_recovery_recent(time_generated):
assert (
datetime.datetime.strptime(time_generated, "%Y-%m-%dT%H:%M:%S.%f")
- datetime.timedelta(seconds=5)
< datetime.datetime.now()
)
def assert_no_recovery(client):
assert not rest_get_recovery_status(client)["exists"]