Update test

Author: avdata99

ckanext/activityinfo/tests/test_home.py

@@ -1,3 +1,4 @@
+from unittest import mock
 from types import SimpleNamespace
 import pytest
 from ckantoolkit.tests import factories as ckan_factories
@@ -22,13 +23,29 @@ def test_regular_user(self, app, setup_data):
 
         resp = app.get("/activity-info", headers=environ)
         assert resp.status_code == 200
-        # CKAN users without a registered ActivityInfo token, will be asked to add one
+        # This must be redirected to the /api-key URL
         assert "Add API key" in resp.body
 
     def test_activityinfo_user(self, app, setup_data):
         environ = {"Authorization": setup_data.activityinfo_user["token"]}
-
-        resp = app.get("/activity-info", headers=environ)
-        assert resp.status_code == 200
-        # Activity info users will see the ActivityInfo UI
-        assert "Update API key" in resp.body
+        fake_databases = [
+            {
+                "databaseId": "0001",
+                "label": "Database label 01",
+                "description": "",
+                "ownerId": "999999999",
+                "billingAccountId": 8888888888888888,
+                "suspended": False,
+                "publishedTemplate": False,
+                "languages": []
+            },
+        ]
+
+        with mock.patch(
+            "ckanext.activityinfo.data.base.ActivityInfoClient.get_databases",
+            return_value=fake_databases,
+        ):
+            resp = app.get("/activity-info", headers=environ)
+            assert resp.status_code == 200
+            # Activity info users will be redirected to see their databases
+            assert "Activity Info databases" in resp.body

ckanext/activityinfo/tests/test_home_api_key.py

@@ -0,0 +1,34 @@
+from types import SimpleNamespace
+import pytest
+from ckantoolkit.tests import factories as ckan_factories
+from ckanext.activityinfo.tests import factories
+
+
+@pytest.fixture
+def setup_data():
+    """test setup data"""
+    obj = SimpleNamespace()
+    # Create CKAN users
+    obj.activityinfo_user = factories.ActivityInfoUser()
+    obj.regular_user = ckan_factories.UserWithToken()
+
+    return obj
+
+
+@pytest.mark.usefixtures("clean_db")
+class TestActivityInfoUIApiKey:
+    def test_regular_user(self, app, setup_data):
+        environ = {"Authorization": setup_data.regular_user["token"]}
+
+        resp = app.get("/activity-info/api-key", headers=environ)
+        assert resp.status_code == 200
+        # CKAN users without a registered ActivityInfo token, will be asked to add one
+        assert "Add API key" in resp.body
+
+    def test_activityinfo_user(self, app, setup_data):
+        environ = {"Authorization": setup_data.activityinfo_user["token"]}
+
+        resp = app.get("/activity-info/api-key", headers=environ)
+        assert resp.status_code == 200
+        # Activity info users will be redirected to see their databases
+        assert "Update API key" in resp.body