diff --git a/ksqldb-engine/src/test/java/io/confluent/ksql/query/QueryBuilderTest.java b/ksqldb-engine/src/test/java/io/confluent/ksql/query/QueryBuilderTest.java index 4bd44fd8dbc8..742e3062d1b5 100644 --- a/ksqldb-engine/src/test/java/io/confluent/ksql/query/QueryBuilderTest.java +++ b/ksqldb-engine/src/test/java/io/confluent/ksql/query/QueryBuilderTest.java @@ -224,7 +224,7 @@ public void setup() { when(ksqlMaterializationFactory.create(any(), any(), any(), any())).thenReturn(materialization); when(processingLogContext.getLoggerFactory()).thenReturn(processingLoggerFactory); when(processingLoggerFactory.getLogger(any(), anyMap())).thenReturn(processingLogger); - when(ksqlConfig.getKsqlStreamConfigProps(anyString())).thenReturn(Collections.emptyMap()); + when(ksqlConfig.getKsqlStreamConfigProps(anyString(), anyBoolean())).thenReturn(Collections.emptyMap()); when(ksqlConfig.getString(KsqlConfig.KSQL_CUSTOM_METRICS_TAGS)).thenReturn(""); when(ksqlConfig.getString(KsqlConfig.KSQL_PERSISTENT_QUERY_NAME_PREFIX_CONFIG)) .thenReturn(PERSISTENT_PREFIX); @@ -664,7 +664,7 @@ private void shouldUseProvidedOptimizationConfig(final Object value) { // Given: final Map properties = Collections.singletonMap(StreamsConfig.TOPOLOGY_OPTIMIZATION_CONFIG, value); - when(ksqlConfig.getKsqlStreamConfigProps(anyString())).thenReturn(properties); + when(ksqlConfig.getKsqlStreamConfigProps(anyString(), anyBoolean())).thenReturn(properties); // When: final PersistentQueryMetadata queryMetadata = buildPersistentQuery( @@ -716,7 +716,7 @@ private void assertPropertiesContainDummyInterceptors() { @Test public void shouldAddMetricsInterceptorsToExistingList() { // Given: - when(ksqlConfig.getKsqlStreamConfigProps(anyString())).thenReturn(ImmutableMap.of( + when(ksqlConfig.getKsqlStreamConfigProps(anyString(), anyBoolean())).thenReturn(ImmutableMap.of( StreamsConfig.consumerPrefix(ConsumerConfig.INTERCEPTOR_CLASSES_CONFIG), ImmutableList.of(DummyConsumerInterceptor.class.getName()), StreamsConfig.producerPrefix(ConsumerConfig.INTERCEPTOR_CLASSES_CONFIG), @@ -737,7 +737,7 @@ public void shouldAddMetricsInterceptorsToExistingList() { @Test public void shouldAddMetricsInterceptorsToExistingString() { // When: - when(ksqlConfig.getKsqlStreamConfigProps(anyString())).thenReturn(ImmutableMap.of( + when(ksqlConfig.getKsqlStreamConfigProps(anyString(), anyBoolean())).thenReturn(ImmutableMap.of( StreamsConfig.consumerPrefix(ConsumerConfig.INTERCEPTOR_CLASSES_CONFIG), DummyConsumerInterceptor.class.getName(), StreamsConfig.producerPrefix(ConsumerConfig.INTERCEPTOR_CLASSES_CONFIG), @@ -759,7 +759,7 @@ public void shouldAddMetricsInterceptorsToExistingString() { @SuppressWarnings("unchecked") public void shouldAddMetricsInterceptorsToExistingStringList() { // When: - when(ksqlConfig.getKsqlStreamConfigProps(anyString())).thenReturn(ImmutableMap.of( + when(ksqlConfig.getKsqlStreamConfigProps(anyString(), anyBoolean())).thenReturn(ImmutableMap.of( StreamsConfig.consumerPrefix(ConsumerConfig.INTERCEPTOR_CLASSES_CONFIG), DummyConsumerInterceptor.class.getName() + ","