Examples with QueryRowImpl io.confluent.ksql.physical.common.QueryRowImpl used on opensource projects

Example 11 with QueryRowImpl

use of io.confluent.ksql.physical.common.QueryRowImpl in project ksql by confluentinc.

the class SelectOperatorTest method shouldSelectKeyWindowed.

@Test
public void shouldSelectKeyWindowed() {
    // Given:
    when(logicalNode.getAddAdditionalColumnsToIntermediateSchema()).thenReturn(true);
    when(logicalNode.getIntermediateSchema()).thenReturn(WINDOWED_INTERMEDIATE_SCHEMA_WITH_PSEUDO);
    when(predicateFactory.create(any(), any())).thenReturn(sqlPredicate);
    final SelectOperator selectOperator = new SelectOperator(logicalNode, logger, predicateFactory);
    selectOperator.addChild(child);
    final QueryRowImpl windowedRow = QueryRowImpl.of(WINDOWED_OUTPUT_SCHEMA, A_KEY, Optional.of(A_WINDOW), GenericRow.genericRow("a", "b"), A_ROWTIME);
    when(child.next()).thenReturn(windowedRow);
    when(sqlPredicate.getTransformer(logger)).thenReturn(transformer);
    final QueryRowImpl intermediateWindowedRow = QueryRowImpl.of(WINDOWED_INTERMEDIATE_SCHEMA_WITH_PSEUDO, A_KEY, Optional.of(A_WINDOW), GenericRow.genericRow("a", "b", A_ROWTIME, "k", A_WINDOW.start().toEpochMilli(), A_WINDOW.end().toEpochMilli()), A_ROWTIME);
    when(transformer.transform(A_KEY, intermediateWindowedRow.value(), new PullProcessingContext(12335L))).thenReturn(Optional.of(GenericRow.genericRow("a", "b", A_ROWTIME, "k", A_WINDOW.start().toEpochMilli(), A_WINDOW.end().toEpochMilli())));
    selectOperator.open();
    // When:
    Object result = selectOperator.next();
    // Then:
    assertThat(result, is(intermediateWindowedRow));
}