Add projects.validate_delete

apiserver/tests/automated/test_project_delete.py

@@ -0,0 +1,54 @@
+from apiserver.apierrors import errors
+from apiserver.database.model import EntityVisibility
+from apiserver.tests.automated import TestService
+from apiserver.database.utils import id as db_id
+
+
+class TestProjectsDelete(TestService):
+    def setUp(self, version="2.14"):
+        super().setUp(version=version)
+
+    def new_task(self, **kwargs):
+        return self.create_temp(
+            "tasks", type="testing", name=db_id(), input=dict(view=dict()), **kwargs
+        )
+
+    def new_model(self, **kwargs):
+        return self.create_temp("models", uri="file:///a/b", name=db_id(), labels={}, **kwargs)
+
+    def new_project(self, **kwargs):
+        return self.create_temp("projects", name=db_id(), description="", **kwargs)
+
+    def test_delete_fails_with_active_task(self):
+        project = self.new_project()
+        self.new_task(project=project)
+        res = self.api.projects.validate_delete(project=project)
+        self.assertEqual(res.tasks, 1)
+        self.assertEqual(res.non_archived_tasks, 1)
+        with self.api.raises(errors.bad_request.ProjectHasTasks):
+            self.api.projects.delete(project=project)
+
+    def test_delete_with_archived_task(self):
+        project = self.new_project()
+        self.new_task(project=project, system_tags=[EntityVisibility.archived.value])
+        res = self.api.projects.validate_delete(project=project)
+        self.assertEqual(res.tasks, 1)
+        self.assertEqual(res.non_archived_tasks, 0)
+        self.api.projects.delete(project=project)
+
+    def test_delete_fails_with_active_model(self):
+        project = self.new_project()
+        self.new_model(project=project)
+        res = self.api.projects.validate_delete(project=project)
+        self.assertEqual(res.models, 1)
+        self.assertEqual(res.non_archived_models, 1)
+        with self.api.raises(errors.bad_request.ProjectHasModels):
+            self.api.projects.delete(project=project)
+
+    def test_delete_with_archived_model(self):
+        project = self.new_project()
+        self.new_model(project=project, system_tags=[EntityVisibility.archived.value])
+        res = self.api.projects.validate_delete(project=project)
+        self.assertEqual(res.models, 1)
+        self.assertEqual(res.non_archived_models, 0)
+        self.api.projects.delete(project=project)