@Test public void testFindColumn() throws Exception { try (Cursor cursor = new SimpleCursor( QueryResult.newBuilder() .addFields(Field.newBuilder().setName("col0").build()) .addFields(Field.newBuilder().setName("col1").build()) .addFields(Field.newBuilder().setName("col2").build()) .build())) { Assert.assertEquals(0, cursor.findColumn("col0")); Assert.assertEquals(1, cursor.findColumn("col1")); Assert.assertEquals(2, cursor.findColumn("col2")); } }
@Test public void testFindColumn() throws Exception { try (Cursor cursor = new SimpleCursor( QueryResult.newBuilder() .addFields(Field.newBuilder().setName("col1").build()) .addFields(Field.newBuilder().setName("COL2").build()) // case-insensitive .addFields(Field.newBuilder().setName("col1").build()) // duplicate .addFields(Field.newBuilder().setName("col4").build()) // skip duplicate .build())) { Assert.assertEquals(1, cursor.findColumn("col1")); // should return first col1 Assert.assertEquals(2, cursor.findColumn("Col2")); // should be case-insensitive Assert.assertEquals(4, cursor.findColumn("col4")); // index should skip over duplicate } }