def test_notebook_creation_triggers_status_creation(self):
assert NotebookJobStatus.objects.count() == 0
project = ProjectFactory()
project.notebook = NotebookJobFactory()
project.save()
assert NotebookJobStatus.objects.count() == 1
assert project.notebook.last_status == JobLifeCycle.CREATED
def test_project_deletion_cascade_to_notebook_job(self):
assert NotebookJob.objects.count() == 0
project = ProjectFactory()
project.notebook = NotebookJobFactory()
project.save()
assert NotebookJob.objects.count() == 1
with patch('schedulers.tensorboard_scheduler.stop_tensorboard') as _:
with patch('schedulers.notebook_scheduler.stop_notebook') as _:
project.delete()
assert NotebookJob.objects.count() == 0
