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_notebook_jobs(self): project1 = ProjectFactory() NotebookJobFactory(project=project1) project1.archive() project2 = ProjectFactory() job2 = NotebookJobFactory(project=project2) job2.archive() assert NotebookJob.all.count() == 2 CleaningIntervals.ARCHIVED = -10 delete_archived_notebook_jobs() # Although the other entity is archived it's not deleted because of project1 assert NotebookJob.all.count() == 1
def test_delete_notebook_jobs(self): project1 = ProjectFactory() NotebookJobFactory(project=project1) project1.archive() project2 = ProjectFactory() job2 = NotebookJobFactory(project=project2) job2.archive() assert NotebookJob.all.count() == 2 conf.set(CLEANING_INTERVALS_ARCHIVES, -10) delete_archived_notebook_jobs() # Although the other entity is archived it's not deleted because of project1 assert NotebookJob.all.count() == 1