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_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)
