def test_delete_task_ok(ebnt: Ebonite):
task = ebnt.get_or_create_task('Project', 'Task')
assert ebnt.meta_repo.get_task_by_id(task.id) is not None
ebnt.delete_task(task)
assert ebnt.meta_repo.get_task_by_id(task.id) is None
def test_delete_task_with_models(ebnt: Ebonite, model: Model):
task = ebnt.get_or_create_task('Project', 'Task')
model = ebnt.push_model(model, task)
assert ebnt.meta_repo.get_task_by_id(task.id) is not None
assert ebnt.meta_repo.get_model_by_id(model.id) is not None
with pytest.raises(TaskWithFKError):
ebnt.delete_task(task)
def test_delete_task_cascade_ok(ebnt: Ebonite, model: Model, mock_env,
image: Image, pipeline: Pipeline):
task = ebnt.get_or_create_task('Project', 'Task')
model = ebnt.push_model(model, task)
image.environment = ebnt.meta_repo.create_environment(mock_env)
task.add_pipeline(pipeline)
task.add_image(image)
task = ebnt.meta_repo.get_task_by_id(task.id).bind_as(task)
assert ebnt.meta_repo.get_task_by_id(task.id) is not None
assert ebnt.meta_repo.get_model_by_id(model.id) is not None
assert ebnt.meta_repo.get_pipeline_by_id(pipeline.id) is not None
assert ebnt.meta_repo.get_image_by_id(image.id) is not None
ebnt.delete_task(task, cascade=True)
assert ebnt.meta_repo.get_task_by_id(task.id) is None
assert ebnt.meta_repo.get_model_by_id(model.id) is None
assert ebnt.meta_repo.get_pipeline_by_id(pipeline.id) is None
assert ebnt.meta_repo.get_image_by_id(image.id) is None