def test_archive(self):
project = ProjectFactory()
notebook_job = NotebookJobFactory(project=project)
tensorboard_job = TensorboardJobFactory(project=project)
assert notebook_job.deleted is False
assert tensorboard_job.deleted is False
assert NotebookJob.objects.count() == 1
assert TensorboardJob.all.count() == 1
notebook_job.archive()
tensorboard_job.archive()
assert notebook_job.deleted is True
assert tensorboard_job.deleted is True
assert NotebookJob.objects.count() == 0
assert TensorboardJob.objects.count() == 0
assert NotebookJob.all.count() == 1
assert TensorboardJob.all.count() == 1
notebook_job.restore()
tensorboard_job.restore()
assert notebook_job.deleted is False
assert tensorboard_job.deleted is False
assert NotebookJob.objects.count() == 1
assert TensorboardJob.objects.count() == 1
assert NotebookJob.all.count() == 1
assert TensorboardJob.all.count() == 1
def test_delete_tensorboard_jobs(self):
project1 = ProjectFactory()
TensorboardJobFactory(project=project1)
project1.archive()
project2 = ProjectFactory()
job2 = TensorboardJobFactory(project=project2)
job2.archive()
assert TensorboardJob.all.count() == 2
CleaningIntervals.ARCHIVED = -10
delete_archived_tensorboard_jobs()
# Although the other entity is archived it's not deleted because of project1
assert TensorboardJob.all.count() == 1
def test_delete_tensorboard_jobs(self):
project1 = ProjectFactory()
TensorboardJobFactory(project=project1)
project1.archive()
project2 = ProjectFactory()
job2 = TensorboardJobFactory(project=project2)
job2.archive()
assert TensorboardJob.all.count() == 2
conf.set(CLEANING_INTERVALS_ARCHIVES, -10)
delete_archived_tensorboard_jobs()
# Although the other entity is archived it's not deleted because of project1
assert TensorboardJob.all.count() == 1
