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 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, } )