From e5299b5ed4c9c9041217f196f5721d8b49bfbf00 Mon Sep 17 00:00:00 2001
From: Kevin Morris <kevr@0cost.org>
Date: Wed, 6 Oct 2021 23:17:08 -0700
Subject: [PATCH] fix(FastAPI): pkgbase/package tests

Signed-off-by: Kevin Morris <kevr@0cost.org>
---
 test/test_packages_routes.py | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)

diff --git a/test/test_packages_routes.py b/test/test_packages_routes.py
index 5afe011a..4118744a 100644
--- a/test/test_packages_routes.py
+++ b/test/test_packages_routes.py
@@ -335,6 +335,30 @@ def test_package_authenticated_maintainer(client: TestClient,
         resp = request.get(package_endpoint(package), cookies=cookies)
     assert resp.status_code == int(HTTPStatus.OK)
 
+    expected = [
+        "View PKGBUILD",
+        "View Changes",
+        "Download snapshot",
+        "Search wiki",
+        "Flag package out-of-date",
+        "Vote for this package",
+        "Enable notifications",
+        "Manage Co-Maintainers",
+        "Submit Request",
+        "Disown Package"
+    ]
+    for expected_text in expected:
+        assert expected_text in resp.text
+
+
+def test_package_authenticated_tu(client: TestClient,
+                                  tu_user: User,
+                                  package: Package):
+    cookies = {"AURSID": tu_user.login(Request(), "testPassword")}
+    with client as request:
+        resp = request.get(package_endpoint(package), cookies=cookies)
+    assert resp.status_code == int(HTTPStatus.OK)
+
     expected = [
         "View PKGBUILD",
         "View Changes",