Fix test cases

apache · Jul 25, 2024 · 8f4cfd0 · 8f4cfd0
1 parent ddf4338
commit 8f4cfd0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 6 deletions.
diff --git a/...ry-runtime/src/test/java/org/apache/pinot/query/service/dispatch/QueryDispatcherTest.java b/...ry-runtime/src/test/java/org/apache/pinot/query/service/dispatch/QueryDispatcherTest.java
@@ -27,7 +27,6 @@
 import java.util.concurrent.CountDownLatch;
 import java.util.concurrent.TimeoutException;
 import java.util.concurrent.atomic.AtomicLong;
-import org.apache.pinot.common.config.TlsConfig;
 import org.apache.pinot.common.proto.Worker;
 import org.apache.pinot.query.QueryEnvironment;
 import org.apache.pinot.query.QueryEnvironmentTestBase;
@@ -61,8 +60,7 @@ public void setUp()
     for (int i = 0; i < QUERY_SERVER_COUNT; i++) {
       int availablePort = QueryTestUtils.getAvailablePort();
       QueryRunner queryRunner = Mockito.mock(QueryRunner.class);
-      TlsConfig tlsConfig = Mockito.mock(TlsConfig.class);
-      QueryServer queryServer = Mockito.spy(new QueryServer(availablePort, queryRunner, tlsConfig));
+      QueryServer queryServer = Mockito.spy(new QueryServer(availablePort, queryRunner, null));
       queryServer.start();
       _queryServerMap.put(availablePort, queryServer);
     }

diff --git a/pinot-query-runtime/src/test/java/org/apache/pinot/query/service/server/QueryServerTest.java b/pinot-query-runtime/src/test/java/org/apache/pinot/query/service/server/QueryServerTest.java
@@ -29,7 +29,6 @@
 import java.util.Objects;
 import java.util.Random;
 import java.util.concurrent.TimeUnit;
-import org.apache.pinot.common.config.TlsConfig;
 import org.apache.pinot.common.proto.PinotQueryWorkerGrpc;
 import org.apache.pinot.common.proto.Plan;
 import org.apache.pinot.common.proto.Worker;
@@ -77,8 +76,7 @@ public void setUp()
     for (int i = 0; i < QUERY_SERVER_COUNT; i++) {
       int availablePort = QueryTestUtils.getAvailablePort();
       QueryRunner queryRunner = mock(QueryRunner.class);
-      TlsConfig tlsConfig = mock(TlsConfig.class);
-      QueryServer queryServer = new QueryServer(availablePort, queryRunner, tlsConfig);
+      QueryServer queryServer = new QueryServer(availablePort, queryRunner, null);
       queryServer.start();
       _queryServerMap.put(availablePort, queryServer);
       _queryRunnerMap.put(availablePort, queryRunner);