@Test
public void testSingleColumn() throws Exception {
InMemoryRecordSet records =
new InMemoryRecordSet(
ImmutableList.of(VARCHAR),
ImmutableList.copyOf(
new List<?>[] {
ImmutableList.of("abc"), ImmutableList.of("def"), ImmutableList.of("g")
}));
OperatorContext operatorContext =
driverContext.addOperatorContext(0, RecordProjectOperator.class.getSimpleName());
Operator operator = new RecordProjectOperator(operatorContext, records);
MaterializedResult expected =
resultBuilder(driverContext.getSession(), VARCHAR).row("abc").row("def").row("g").build();
OperatorAssertion.assertOperatorEquals(operator, expected);
}
@Test
public void testMultiColumn() throws Exception {
InMemoryRecordSet records =
new InMemoryRecordSet(
ImmutableList.of(VARCHAR, BIGINT),
ImmutableList.of(
ImmutableList.of("abc", 1L),
ImmutableList.of("def", 2L),
ImmutableList.of("g", 0L)));
OperatorContext operatorContext =
driverContext.addOperatorContext(0, RecordProjectOperator.class.getSimpleName());
Operator operator = new RecordProjectOperator(operatorContext, records);
MaterializedResult expected =
resultBuilder(driverContext.getSession(), VARCHAR, BIGINT)
.row("abc", 1)
.row("def", 2)
.row("g", 0)
.build();
OperatorAssertion.assertOperatorEquals(operator, expected);
}