diff --git a/integration-tests/src/main/java/org/apache/druid/quidem/BrokerBasedTestConnection.java b/integration-tests/src/main/java/org/apache/druid/quidem/ExposedAsBrokerQueryComponentSupplierWrapper.java similarity index 97% rename from integration-tests/src/main/java/org/apache/druid/quidem/BrokerBasedTestConnection.java rename to integration-tests/src/main/java/org/apache/druid/quidem/ExposedAsBrokerQueryComponentSupplierWrapper.java index 339b2f66ed7..329c169aad7 100644 --- a/integration-tests/src/main/java/org/apache/druid/quidem/BrokerBasedTestConnection.java +++ b/integration-tests/src/main/java/org/apache/druid/quidem/ExposedAsBrokerQueryComponentSupplierWrapper.java @@ -129,11 +129,14 @@ import java.io.IOException; import java.util.List; import java.util.Properties; -class BrokerBasedTestConnection implements QueryComponentSupplier +/** + * A wrapper class to expose a {@link QueryComponentSupplier} as a Broker service. + */ +public class ExposedAsBrokerQueryComponentSupplierWrapper implements QueryComponentSupplier { private QueryComponentSupplier delegate; - public BrokerBasedTestConnection(QueryComponentSupplier delegate) + public ExposedAsBrokerQueryComponentSupplierWrapper(QueryComponentSupplier delegate) { this.delegate = delegate; } @@ -180,7 +183,7 @@ class BrokerBasedTestConnection implements QueryComponentSupplier installForServerModules(builder); - overrideModules.addAll(BrokerBasedTestConnection.brokerModules()); + overrideModules.addAll(ExposedAsBrokerQueryComponentSupplierWrapper.brokerModules()); overrideModules.add(new DiscovertModule()); } diff --git a/integration-tests/src/main/java/org/apache/druid/quidem/Launcher.java b/integration-tests/src/main/java/org/apache/druid/quidem/Launcher.java index 53a656d4a11..f63f2bfe9e2 100644 --- a/integration-tests/src/main/java/org/apache/druid/quidem/Launcher.java +++ b/integration-tests/src/main/java/org/apache/druid/quidem/Launcher.java @@ -83,7 +83,7 @@ public class Launcher ConfigurationInstance ci = CONFIG_STORE.getConfigurationInstance( config, - x -> new BrokerBasedTestConnection(x) + x -> new ExposedAsBrokerQueryComponentSupplierWrapper(x) ); return ci; }