Skip to content

Commit c7f1fed

Browse files
feat: update test case
1 parent d604a2a commit c7f1fed

File tree

2 files changed

+21
-1
lines changed

2 files changed

+21
-1
lines changed

src/main/java/com/influxdb/v3/client/internal/FlightSqlClient.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -272,7 +272,7 @@ private void setChannelTypeAndEventLoop(@Nonnull final NettyChannelBuilder netty
272272
}
273273
}
274274

275-
private ProxyDetector createProxyDetector(@Nonnull final String hostUrl, @Nonnull final String proxyUrl) {
275+
ProxyDetector createProxyDetector(@Nonnull final String hostUrl, @Nonnull final String proxyUrl) {
276276
URI proxyUri = URI.create(proxyUrl);
277277
URI hostUri = URI.create(hostUrl);
278278
return (targetServerAddress) -> {

src/test/java/com/influxdb/v3/client/internal/FlightSqlClientTest.java

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,9 +21,12 @@
2121
*/
2222
package com.influxdb.v3.client.internal;
2323

24+
import java.io.IOException;
25+
import java.net.InetSocketAddress;
2426
import java.net.URISyntaxException;
2527
import java.util.Map;
2628

29+
import io.grpc.ProxyDetector;
2730
import io.grpc.internal.GrpcUtil;
2831
import org.apache.arrow.flight.CallHeaders;
2932
import org.apache.arrow.flight.CallInfo;
@@ -272,6 +275,23 @@ public void useParamsFromQueryConfig() throws Exception {
272275
}
273276
}
274277

278+
@Test
279+
void createProxyDetector() {
280+
ClientConfig clientConfig = new ClientConfig.Builder()
281+
.host("https://localhost:80")
282+
.build();
283+
try (FlightSqlClient flightSqlClient = new FlightSqlClient(clientConfig)) {
284+
String hostUrl = "https://youtube.com";
285+
String proxyUrl = "https://facebook.com";
286+
ProxyDetector proxyDetector = flightSqlClient.createProxyDetector(hostUrl, proxyUrl);
287+
Assertions.assertThat(proxyDetector.proxyFor(
288+
new InetSocketAddress("142.250.198.142", 80)
289+
)).isNull();
290+
} catch (Exception e) {
291+
throw new RuntimeException(e);
292+
}
293+
}
294+
275295
static class CallHeadersMiddleware implements FlightClientMiddleware.Factory {
276296
CallHeaders headers;
277297

0 commit comments

Comments
 (0)