test(tokens-repo): break out assert_original() in rest

This commit is contained in:
Houkime 2022-12-28 14:51:25 +00:00 committed by Inex Code
parent e125f3a4b1
commit f542c1e6c7

View file

@ -37,6 +37,10 @@ DATE_FORMATS = [
] ]
def assert_original(filename):
assert read_json(filename) == TOKENS_FILE_CONTETS
def test_get_tokens_info(authorized_client, tokens_file): def test_get_tokens_info(authorized_client, tokens_file):
response = authorized_client.get("/auth/tokens") response = authorized_client.get("/auth/tokens")
assert response.status_code == 200 assert response.status_code == 200
@ -58,7 +62,7 @@ def test_get_tokens_unauthorized(client, tokens_file):
def test_delete_token_unauthorized(client, tokens_file): def test_delete_token_unauthorized(client, tokens_file):
response = client.delete("/auth/tokens") response = client.delete("/auth/tokens")
assert response.status_code == 401 assert response.status_code == 401
assert read_json(tokens_file) == TOKENS_FILE_CONTETS assert_original(tokens_file)
def test_delete_token(authorized_client, tokens_file): def test_delete_token(authorized_client, tokens_file):
@ -82,7 +86,7 @@ def test_delete_self_token(authorized_client, tokens_file):
"/auth/tokens", json={"token_name": "test_token"} "/auth/tokens", json={"token_name": "test_token"}
) )
assert response.status_code == 400 assert response.status_code == 400
assert read_json(tokens_file) == TOKENS_FILE_CONTETS assert_original(tokens_file)
def test_delete_nonexistent_token(authorized_client, tokens_file): def test_delete_nonexistent_token(authorized_client, tokens_file):
@ -90,13 +94,13 @@ def test_delete_nonexistent_token(authorized_client, tokens_file):
"/auth/tokens", json={"token_name": "test_token3"} "/auth/tokens", json={"token_name": "test_token3"}
) )
assert response.status_code == 404 assert response.status_code == 404
assert read_json(tokens_file) == TOKENS_FILE_CONTETS assert_original(tokens_file)
def test_refresh_token_unauthorized(client, tokens_file): def test_refresh_token_unauthorized(client, tokens_file):
response = client.post("/auth/tokens") response = client.post("/auth/tokens")
assert response.status_code == 401 assert response.status_code == 401
assert read_json(tokens_file) == TOKENS_FILE_CONTETS assert_original(tokens_file)
def test_refresh_token(authorized_client, tokens_file): def test_refresh_token(authorized_client, tokens_file):
@ -112,7 +116,7 @@ def test_refresh_token(authorized_client, tokens_file):
def test_get_new_device_auth_token_unauthorized(client, tokens_file): def test_get_new_device_auth_token_unauthorized(client, tokens_file):
response = client.post("/auth/new_device") response = client.post("/auth/new_device")
assert response.status_code == 401 assert response.status_code == 401
assert read_json(tokens_file) == TOKENS_FILE_CONTETS assert_original(tokens_file)
def test_get_new_device_auth_token(authorized_client, tokens_file): def test_get_new_device_auth_token(authorized_client, tokens_file):
@ -133,13 +137,13 @@ def test_get_and_delete_new_device_token(authorized_client, tokens_file):
"/auth/new_device", json={"token": response.json()["token"]} "/auth/new_device", json={"token": response.json()["token"]}
) )
assert response.status_code == 200 assert response.status_code == 200
assert read_json(tokens_file) == TOKENS_FILE_CONTETS assert_original(tokens_file)
def test_delete_token_unauthenticated(client, tokens_file): def test_delete_token_unauthenticated(client, tokens_file):
response = client.delete("/auth/new_device") response = client.delete("/auth/new_device")
assert response.status_code == 401 assert response.status_code == 401
assert read_json(tokens_file) == TOKENS_FILE_CONTETS assert_original(tokens_file)
def test_get_and_authorize_new_device(client, authorized_client, tokens_file): def test_get_and_authorize_new_device(client, authorized_client, tokens_file):
@ -163,7 +167,7 @@ def test_authorize_new_device_with_invalid_token(client, tokens_file):
json={"token": "invalid_token", "device": "new_device"}, json={"token": "invalid_token", "device": "new_device"},
) )
assert response.status_code == 404 assert response.status_code == 404
assert read_json(tokens_file) == TOKENS_FILE_CONTETS assert_original(tokens_file)
def test_get_and_authorize_used_token(client, authorized_client, tokens_file): def test_get_and_authorize_used_token(client, authorized_client, tokens_file):
@ -214,7 +218,7 @@ def test_authorize_without_token(client, tokens_file):
json={"device": "new_device"}, json={"device": "new_device"},
) )
assert response.status_code == 422 assert response.status_code == 422
assert read_json(tokens_file) == TOKENS_FILE_CONTETS assert_original(tokens_file)
# Recovery tokens # Recovery tokens
@ -243,7 +247,7 @@ def test_authorize_without_token(client, tokens_file):
def test_get_recovery_token_status_unauthorized(client, tokens_file): def test_get_recovery_token_status_unauthorized(client, tokens_file):
response = client.get("/auth/recovery_token") response = client.get("/auth/recovery_token")
assert response.status_code == 401 assert response.status_code == 401
assert read_json(tokens_file) == TOKENS_FILE_CONTETS assert_original(tokens_file)
def test_get_recovery_token_when_none_exists(authorized_client, tokens_file): def test_get_recovery_token_when_none_exists(authorized_client, tokens_file):
@ -256,7 +260,7 @@ def test_get_recovery_token_when_none_exists(authorized_client, tokens_file):
"expiration": None, "expiration": None,
"uses_left": None, "uses_left": None,
} }
assert read_json(tokens_file) == TOKENS_FILE_CONTETS assert_original(tokens_file)
def test_generate_recovery_token(authorized_client, client, tokens_file): def test_generate_recovery_token(authorized_client, client, tokens_file):