def test_to_record(self): """Test that an entity can be converted into a record.""" entity = TaskConnectionMapper.to_entity(self.url_to_log_connection) record = TaskConnectionMapper.to_record(entity) for c in record.__table__.columns: self.assertEquals( getattr(record, c.name), getattr(self.url_to_log_connection, c.name), )
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_to_entity(self): """Test that a record can be converted into an entity.""" self.assertEquals( TaskConnectionMapper.to_entity(self.url_to_log_connection).to_native(), { 'id': self.url_to_log_connection.id, 'from_task_id': self.url_to_log_connection.from_task_id, 'pipeline_id': self.url_to_log_connection.pipeline_id, 'to_task_id': self.url_to_log_connection.to_task_id, } )