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

test(tokens-repo): allow ading uses in a helper recovery function

redis/connection-pool
Houkime 2022-12-30 12:52:12 +00:00 committed by Inex Code
parent 3aa3d197e2
commit 42fa5fe524
1 changed files with 13 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
tests/test_rest_endpoints/test_auth.py
View File

@ -251,15 +251,23 @@ def test_get_recovery_token_when_none_exists(authorized_client, tokens_file):
assert_original(tokens_file)
def rest_make_recovery_token(client, expires_at=None, timeformat=None):
if expires_at is None:
response = client.post("/auth/recovery_token")
else:
def rest_make_recovery_token(client, expires_at=None, timeformat=None, uses=None):
json = {}
if expires_at is not None:
assert timeformat is not None
expires_at_str = expires_at.strftime(timeformat)
json["expiration"] = expires_at_str
if uses is not None:
json["uses"] = uses
if json == {}:
response = client.post("/auth/recovery_token")
else:
response = client.post(
"/auth/recovery_token",
json={"expiration": expires_at_str},
json=json,
)
assert response.status_code == 200