def write_task_connection(cls, task_connection_entity):
"""Writes a TaskConnectionEntity to the database.
:param TaskConnectionEntity task_connection_entity:
"""
task_connection_entity.validate()
TaskConnectionRepository.write_record(
TaskConnectionMapper.to_record(task_connection_entity))
def test_write_record_new(self):
"""Test that we can create a new record."""
entity = TaskConnectionEntity.get_mock_object()
TaskConnectionRepository.write_record(
TaskConnectionMapper.to_record(entity)
)
connections = TaskConnectionRepository.fetch_connections_for_pipeline(entity.pipeline_id)
self.assertEquals(connections[0].id, entity.id)
def write_task_connection(cls, task_connection_entity):
"""Writes a TaskConnectionEntity to the database.
:param TaskConnectionEntity task_connection_entity:
"""
task_connection_entity.validate()
TaskConnectionRepository.write_record(
TaskConnectionMapper.to_record(task_connection_entity)
)
def test_fetch_connections_for_pipeline(self):
"""Test that only connections for a provided pipeline are returned."""
self.assertItemsEqual(
TaskConnectionRepository.fetch_connections_for_pipeline(self.pipeline1.id),
[
self.connection1,
],
)
def test_write_record_update(self):
"""Test that we can write an updated record."""
fake_uuid = uuid.uuid4()
# Assert connection's from_task_id is not expected
connection = TaskConnectionRepository.fetch_connections_for_pipeline(
self.connection1.pipeline_id
)[0]
self.assertNotEquals(connection.from_task_id, fake_uuid)
# Update connection's from_task_id
connection.from_task_id = fake_uuid
TaskConnectionRepository.write_record(connection)
# Assert connection's from_task_id is as expected
connection = TaskConnectionRepository.fetch_connections_for_pipeline(
self.connection1.pipeline_id
)[0]
self.assertEquals(connection.from_task_id, fake_uuid)
def fetch_task_connections_for_pipeline(cls, pipeline_id):
"""Returns list of TaskConnectionEntities for a pipeline.
:param str pipeline_id:
:returns list: TaskConnectionEntities
"""
return map(
TaskConnectionMapper.to_entity,
TaskConnectionRepository.fetch_connections_for_pipeline(pipeline_id),
)
def fetch_task_connections_for_pipeline(cls, pipeline_id):
"""Returns list of TaskConnectionEntities for a pipeline.
:param str pipeline_id:
:returns list: TaskConnectionEntities
"""
return map(
TaskConnectionMapper.to_entity,
TaskConnectionRepository.fetch_connections_for_pipeline(
pipeline_id),
)