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)