Examples with QueryLogger io.confluent.ksql.logging.query.QueryLogger used on opensource projects

Example 1 with QueryLogger

use of io.confluent.ksql.logging.query.QueryLogger in project ksql by confluentinc.

the class KsqlResourceTest method queryLoggerShouldReceiveTerminateStatementsWhenHandleKsqlStatementWithTerminate.

@Test
public void queryLoggerShouldReceiveTerminateStatementsWhenHandleKsqlStatementWithTerminate() {
    // Given:
    final PersistentQueryMetadata queryMetadata = createQuery("CREATE STREAM test_explain AS SELECT * FROM test_stream;", emptyMap());
    final String terminateSql = "TERMINATE " + queryMetadata.getQueryId() + ";";
    // When:
    try (MockedStatic<QueryLogger> logger = Mockito.mockStatic(QueryLogger.class)) {
        ksqlResource.handleKsqlStatements(securityContext, new KsqlRequest(terminateSql, ImmutableMap.of(ConsumerConfig.AUTO_OFFSET_RESET_CONFIG, "earliest"), emptyMap(), 0L));
        // Then:
        logger.verify(() -> QueryLogger.info("Query terminated", terminateSql), times(1));
    }
}