def test_tensorboard_creation_triggers_status_creation(self):
assert TensorboardJobStatus.objects.count() == 0
project = ProjectFactory()
project.tensorboard = TensorboardJobFactory()
project.save()
assert TensorboardJobStatus.objects.count() == 1
assert project.tensorboard.last_status == JobLifeCycle.CREATED
def test_project_deletion_cascade_to_tensorboard_job(self):
assert TensorboardJob.objects.count() == 0
project = ProjectFactory()
project.tensorboard = TensorboardJobFactory()
project.save()
assert TensorboardJob.objects.count() == 1
with patch('schedulers.tensorboard_scheduler.stop_tensorboard') as _:
with patch('schedulers.notebook_scheduler.stop_notebook') as _:
project.delete()
assert TensorboardJob.objects.count() == 0