From 709ccb176ae405c4e90fe318aabcdc10386c087a Mon Sep 17 00:00:00 2001 From: george Date: Thu, 3 Feb 2022 19:30:10 +0000 Subject: [PATCH] Test the dashboard POST request instead of directly testing the API. --- tests/test_api_keys.py | 14 +++++++++----- 1 file changed, 9 insertions(+), 5 deletions(-) diff --git a/tests/test_api_keys.py b/tests/test_api_keys.py index fd0f8611..20ecaded 100644 --- a/tests/test_api_keys.py +++ b/tests/test_api_keys.py @@ -1,3 +1,5 @@ +from flask import url_for + from app.db import Session from app.models import User, ApiKey from tests.utils import login @@ -20,9 +22,7 @@ def test_create_delete_api_key(flask_client): def test_delete_all_api_key(flask_client): # create two test users - user_1 = User.create( - email="a1@b.c", password="password", name="Test User 1", activated=True - ) + user_1 = login(flask_client) user_2 = User.create( email="a2@b.c", password="password", name="Test User 2", activated=True ) @@ -42,8 +42,12 @@ def test_delete_all_api_key(flask_client): assert ApiKey.filter(ApiKey.user_id == user_2.id).count() == 1 # delete all of user 1's API keys - ApiKey.delete_all(user_1.id) - Session.commit() + r = flask_client.post( + url_for("dashboard.api_key"), + data={"form-name": "delete-all"}, + follow_redirects=True, + ) + assert r.status_code == 200 assert ( ApiKey.count() == 1 ) # assert that the total number of API keys for all users is now 1.