def test_update_task_is_reference(meta: MetadataRepository, project: Project,
model: Model):
task_entity = Task("Test Task")
task_entity.project = meta.create_project(project)
task = meta.create_task(task_entity)
assert task is not None
id = task.id
model.task_id = task.id
model = meta.create_model(model)
task.name = "Test Task 2"
task.add_model(model)
task = meta.update_task(task)
assert id == task.id
assert "Test Task 2" == task.name
assert model == task.models.get(model.id)
task.name = "KEK"
actual_task = meta.get_task_by_id(task.id)
assert_objects_equal_except_fields(task,
actual_task,
excepted_fields=['name'])
def test_update_task_with_models(meta: MetadataRepository, project: Project,
task: Task, model: Model):
task.project = meta.create_project(project)
task = meta.create_task(task)
assert task is not None
id = task.id
model.task = task
model = meta.create_model(model)
task.add_model(model)
task = update_object_fields(task,
excepted_fields=['id', 'models', 'project_id'])
model = update_object_fields(model,
excepted_fields=[
'id', 'wrapper', 'artifact',
'output_meta', 'input_meta',
'requirements', 'transformer', 'task_id'
])
updated_task = meta.update_task(task)
assert id == task.id
assert updated_task is task
assert task == meta.get_task_by_id(task.id)
assert len(task.models) == 1
assert model.id in task.models
assert model == meta.get_model_by_id(model.id)
assert meta.get_model_by_id(model.id).name == 'Test Model2'
assert task.has_meta_repo
def test_delete_task(meta: MetadataRepository, project: Project, task: Task):
task.project = meta.create_project(project)
task = meta.create_task(task)
assert task is not None
meta.delete_task(task)
assert meta.get_task_by_id(task.id) is None
assert not task.has_meta_repo
assert task.id is None
def test_get_task_by_id(meta: MetadataRepository, project: Project,
task: Task):
project = meta.create_project(project)
task.project_id = project.id
task_new = meta.create_task(
task) # Assume that we do not change the task var in the create_task.
assert task_new is not None
assert task_new == meta.get_task_by_id(task_new.id)
assert task.has_meta_repo
def test_save_not_existing_task(meta: MetadataRepository, project: Project):
project = meta.create_project(project)
task = Task("Task")
task.project = project
saved_task = meta.save_task(task)
assert saved_task.name == task.name
assert saved_task.project_id == task.project_id
assert task.name == meta.get_task_by_id(saved_task.id).name
assert task.has_meta_repo
def test_save_task_is_reference(meta: MetadataRepository, project: Project):
project = meta.create_project(project)
task = Task("Task")
task.project = project
saved_task = meta.save_task(task)
saved_task.name = "KEK"
actual_task = meta.get_task_by_id(saved_task.id)
assert_objects_equal_except_fields(saved_task,
actual_task,
excepted_fields=['name'])
def test_create_task_is_reference(meta: MetadataRepository, project: Project,
task: Task):
task.project_id = meta.create_project(project).id
expected_task = meta.create_task(task)
assert expected_task is not None
task.name = "KEK"
actual_task = meta.get_task_by_id(expected_task.id)
assert_objects_equal_except_fields(expected_task,
actual_task,
excepted_fields=['name'])
def test_save_updated_existing_task(meta: MetadataRepository,
project: Project):
project = meta.create_project(project)
task = Task("Task")
task.project = project
task = meta.create_task(task)
task = update_object_fields(task,
excepted_fields=['id', 'models', 'project_id'])
saved_task = meta.save_task(task)
assert saved_task == task
assert task == meta.get_task_by_id(saved_task.id)
def test_update_task_source_is_changed(meta: MetadataRepository,
project: Project, task: Task,
model: Model):
task.project = meta.create_project(project)
saved_task = meta.create_task(task)
assert saved_task is task
id = saved_task.id
model.task = saved_task
model = meta.create_model(model)
saved_task = update_object_fields(
saved_task, excepted_fields=['id', 'models', 'project_id'])
saved_task.add_model(model)
saved_task = meta.update_task(saved_task)
assert id == saved_task.id
assert saved_task == meta.get_task_by_id(saved_task.id)
assert model == saved_task.models.get(model.id)
assert task is saved_task