Merge pull request #780 from FozzieHi/api-key-tests

Add unit tests for API keys.

tests/dashboard/test_api_keys.py

@@ -0,0 +1,69 @@
+from flask import url_for
+
+from app.db import Session
+from app.models import User, ApiKey
+from tests.utils import login
+
+
+def test_create_delete_api_key(flask_client):
+    user = login(flask_client)
+    Session.commit()
+
+    # create api_key
+    create_r = flask_client.post(
+        url_for("dashboard.api_key"),
+        data={"form-name": "create", "name": "for test"},
+        follow_redirects=True,
+    )
+    assert create_r.status_code == 200
+    api_key = ApiKey.get_by(user_id=user.id)
+    assert ApiKey.count() == 1
+    assert api_key.name == "for test"
+
+    # delete api_key
+    delete_r = flask_client.post(
+        url_for("dashboard.api_key"),
+        data={"form-name": "delete", "api-key-id": api_key.id},
+        follow_redirects=True,
+    )
+    assert delete_r.status_code == 200
+    assert ApiKey.count() == 0
+
+
+def test_delete_all_api_keys(flask_client):
+    # create two test users
+    user_1 = login(flask_client)
+    user_2 = User.create(
+        email="a2@b.c", password="password", name="Test User 2", activated=True
+    )
+    Session.commit()
+
+    # create api_key for both users
+    ApiKey.create(user_1.id, "for test")
+    ApiKey.create(user_1.id, "for test 2")
+    ApiKey.create(user_2.id, "for test")
+    Session.commit()
+
+    assert (
+        ApiKey.count() == 3
+    )  # assert that the total number of API keys for all users is 3.
+    # assert that each user has the API keys created
+    assert ApiKey.filter(ApiKey.user_id == user_1.id).count() == 2
+    assert ApiKey.filter(ApiKey.user_id == user_2.id).count() == 1
+
+    # delete all of user 1's API keys
+    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.
+    assert (
+        ApiKey.filter(ApiKey.user_id == user_1.id).count() == 0
+    )  # assert that user 1 now has 0 API keys
+    assert (
+        ApiKey.filter(ApiKey.user_id == user_2.id).count() == 1
+    )  # assert that user 2 still has 1 API key