mirror of
https://github.com/clearml/clearml-server
synced 2025-01-31 19:06:55 +00:00
55 lines
2.2 KiB
Python
55 lines
2.2 KiB
Python
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)
|