[incubator-pinot] branch tls-integration-test updated: patch BasicAuthRealtimeIntegrationTest with client transport changes

[apucher]

This is an automated email from the ASF dual-hosted git repository.

apucher pushed a commit to branch tls-integration-test
in repository https://gitbox.apache.org/repos/asf/incubator-pinot.git


The following commit(s) were added to refs/heads/tls-integration-test by this 
push:
     new 3037fbf  patch BasicAuthRealtimeIntegrationTest with client transport 
changes
3037fbf is described below

commit 3037fbfc75ba257362536edfdf604a9f9daa8b05
Author: Alexander Pucher <a...@alexpucher.com>
AuthorDate: Wed Jun 2 14:53:33 2021 -0700

    patch BasicAuthRealtimeIntegrationTest with client transport changes
---
 .../pinot/integration/tests/BasicAuthRealtimeIntegrationTest.java   | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git 
a/pinot-integration-tests/src/test/java/org/apache/pinot/integration/tests/BasicAuthRealtimeIntegrationTest.java
 
b/pinot-integration-tests/src/test/java/org/apache/pinot/integration/tests/BasicAuthRealtimeIntegrationTest.java
index f10b198..ad43e99 100644
--- 
a/pinot-integration-tests/src/test/java/org/apache/pinot/integration/tests/BasicAuthRealtimeIntegrationTest.java
+++ 
b/pinot-integration-tests/src/test/java/org/apache/pinot/integration/tests/BasicAuthRealtimeIntegrationTest.java
@@ -30,6 +30,7 @@ import org.apache.commons.httpclient.methods.PostMethod;
 import org.apache.commons.io.FileUtils;
 import org.apache.pinot.client.Connection;
 import org.apache.pinot.client.ConnectionFactory;
+import org.apache.pinot.client.JsonAsyncHttpPinotClientTransportFactory;
 import org.apache.pinot.client.Request;
 import org.apache.pinot.client.ResultSetGroup;
 import org.apache.pinot.core.common.MinionConstants;
@@ -141,8 +142,11 @@ public class BasicAuthRealtimeIntegrationTest extends 
BaseClusterIntegrationTest
   @Override
   protected Connection getPinotConnection() {
     if (_pinotConnection == null) {
+      JsonAsyncHttpPinotClientTransportFactory factory = new 
JsonAsyncHttpPinotClientTransportFactory();
+      factory.setHeaders(AUTH_HEADER);
+
       _pinotConnection =
-          ConnectionFactory.fromZookeeper(getZkUrl() + "/" + 
getHelixClusterName(), AUTH_HEADER);
+          ConnectionFactory.fromZookeeper(getZkUrl() + "/" + 
getHelixClusterName(), factory.buildTransport());
     }
     return _pinotConnection;
   }

---------------------------------------------------------------------
To unsubscribe, e-mail: commits-unsubscr...@pinot.apache.org
For additional commands, e-mail: commits-h...@pinot.apache.org