From fc9a6c7740231a5c429b0e7b05c04af948219de7 Mon Sep 17 00:00:00 2001
From: Zoltan Haindrich <kirk@rxd.hu>
Date: Thu, 16 May 2024 13:23:45 +0000
Subject: [PATCH] move/etc

---
 ...=> ExposedAsBrokerQueryComponentSupplierWrapper.java} | 9 ++++++---
 .../src/main/java/org/apache/druid/quidem/Launcher.java  | 2 +-
 2 files changed, 7 insertions(+), 4 deletions(-)
 rename integration-tests/src/main/java/org/apache/druid/quidem/{BrokerBasedTestConnection.java => ExposedAsBrokerQueryComponentSupplierWrapper.java} (97%)

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;
   }