Skip to content

Commit 0c4d0ac

Browse files
feat: add test case
1 parent f859de9 commit 0c4d0ac

File tree

1 file changed

+23
-0
lines changed

1 file changed

+23
-0
lines changed

src/test/java/com/influxdb/v3/client/InfluxDBClientTest.java

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,14 +25,18 @@
2525
import java.net.InetSocketAddress;
2626
import java.net.ProxySelector;
2727
import java.net.URI;
28+
import java.security.NoSuchAlgorithmException;
2829
import java.time.Instant;
2930
import java.util.Map;
3031
import java.util.Properties;
3132
import java.util.UUID;
3233
import java.util.stream.Stream;
34+
import javax.net.ssl.SSLContext;
35+
import javax.net.ssl.SSLException;
3336

3437
import io.grpc.HttpConnectProxiedSocketAddress;
3538
import io.grpc.ProxyDetector;
39+
import io.netty.handler.ssl.SslContext;
3640
import org.assertj.core.api.Assertions;
3741
import org.junit.jupiter.api.Test;
3842
import org.junit.jupiter.api.condition.EnabledIfEnvironmentVariable;
@@ -43,6 +47,25 @@
4347

4448
public class InfluxDBClientTest {
4549

50+
@Test
51+
void testCustomSslContext() throws NoSuchAlgorithmException, SSLException {
52+
// Test for java.net SslContext
53+
ClientConfig.Builder builder = new ClientConfig.Builder();
54+
ClientConfig clientConfig = builder.sslContext(null).build();
55+
Assertions.assertThat(clientConfig.getSslContext()).isNull();
56+
57+
clientConfig = builder.sslContext(SSLContext.getDefault()).build();
58+
Assertions.assertThat(clientConfig.getSslContext()).isNotNull();
59+
60+
// Test for grpc SslContext
61+
ClientConfig.Builder builder1 = new ClientConfig.Builder();
62+
ClientConfig clientConfig1 = builder1.grpcSslContext(null).build();
63+
Assertions.assertThat(clientConfig1.getGrpcSslContext()).isNull();
64+
65+
clientConfig1 = builder1.grpcSslContext(SslContext.newClientContext()).build();
66+
Assertions.assertThat(clientConfig1.getGrpcSslContext()).isNotNull();
67+
}
68+
4669
@EnabledIfEnvironmentVariable(named = "TESTING_INFLUXDB_URL", matches = ".*")
4770
@EnabledIfEnvironmentVariable(named = "TESTING_INFLUXDB_TOKEN", matches = ".*")
4871
@EnabledIfEnvironmentVariable(named = "TESTING_INFLUXDB_DATABASE", matches = ".*")

0 commit comments

Comments
 (0)