add tests for number_of_projects in projects and site_projects

rdmorganiser · Apr 5, 2023 · edd82f5 · edd82f5
1 parent d5e92af
commit edd82f5
Showing 1 changed file with 17 additions and 7 deletions.
diff --git a/rdmo/projects/tests/test_view_project.py b/rdmo/projects/tests/test_view_project.py
@@ -2,6 +2,7 @@
 
 import pytest
 from django.urls import reverse
+from pytest_django.asserts import assertTemplateUsed
 
 from rdmo.views.models import View
 
@@ -66,12 +67,16 @@ def test_list(db, client, username, password):
 
     if password:
         assert response.status_code == 200
+        assertTemplateUsed(response, 'projects/projects.html')
 
         if username == 'site':
             assert projects == []
+            assert response.context['number_of_projects'] == len([])
         else:
-            assert sorted(list(set([int(project_id) for project_id in projects]))) \
-                == view_project_permission_map.get(username, [])
+            # breakpoint()
+            user_projects_map = view_project_permission_map.get(username, [])
+            assert sorted(list(set(map(int, projects)))) == user_projects_map
+            assert response.context['number_of_projects'] == len(user_projects_map)
     else:
         assert response.status_code == 302
 
@@ -85,11 +90,16 @@ def test_site(db, client, username, password):
 
     projects = re.findall(r'/projects/(\d+)/update/', response.content.decode())
 
-    if username == 'site':
-        assert sorted([int(project_id) for project_id in projects]) \
-            == view_project_permission_map.get(username, [])
-    elif password:
-        assert response.status_code == 403
+    if password:
+
+        if username == 'site':
+            assert response.status_code == 200
+            assertTemplateUsed(response, 'projects/site_projects.html')
+            user_projects_map = view_project_permission_map.get(username, [])
+            assert sorted(list(set(map(int, projects)))) == user_projects_map
+            assert response.context['number_of_projects'] == len(user_projects_map)
+        else:
+            assert response.status_code == 403
     else:
         assert response.status_code == 302