From 40dea87639800f23f17b1f7107fc71a145a3a0c1 Mon Sep 17 00:00:00 2001 From: Quan Pham Date: Mon, 12 Jan 2026 14:07:21 -0500 Subject: [PATCH] `new_project` method creates `ProjectUser` entry for the PI This reflects the actual application behavior when PIs crete new projects through the web UI. --- src/coldfront_plugin_cloud/tests/base.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/src/coldfront_plugin_cloud/tests/base.py b/src/coldfront_plugin_cloud/tests/base.py index 7cccb1bd..3040b1e6 100644 --- a/src/coldfront_plugin_cloud/tests/base.py +++ b/src/coldfront_plugin_cloud/tests/base.py @@ -109,7 +109,14 @@ def new_project(self, title=None, pi=None) -> Project: pi = pi or self.new_user() status = ProjectStatusChoice.objects.get(name="New") - Project.objects.create(title=title, pi=pi, status=status) + project = Project.objects.create(title=title, pi=pi, status=status) + ProjectUser.objects.get_or_create( + user=pi, + project=project, + role=ProjectUserRoleChoice.objects.get(name="Manager"), + status=ProjectUserStatusChoice.objects.get(name="Active"), + ) + return Project.objects.get(title=title) def new_project_user(self, user, project, role="Manager", status="Active"):