wip

Author: NguyenHoangSon96

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

@@ -26,29 +26,27 @@
 import java.io.IOException;
 import java.net.InetAddress;
 import java.net.URISyntaxException;
-import java.net.http.HttpHeaders;
 import java.security.KeyStoreException;
 import java.security.NoSuchAlgorithmException;
 import java.security.UnrecoverableKeyException;
 import java.security.cert.CertificateException;
 import java.time.Duration;
 import java.time.Instant;
 import java.time.temporal.ChronoUnit;
-import java.util.List;
 import java.util.Map;
 import java.util.Optional;
 import java.util.concurrent.ExecutionException;
 import java.util.function.Supplier;
 import com.influxdb.v3.client.AbstractMockServerTest;
 import com.influxdb.v3.client.InfluxDBApiException;
-import com.influxdb.v3.client.InfluxDBApiHttpException;
 import com.influxdb.v3.client.InfluxDBApiNettyException;
 import com.influxdb.v3.client.InfluxDBClient;
 import com.influxdb.v3.client.TestUtils;
 import com.influxdb.v3.client.config.ClientConfig;
 import com.influxdb.v3.client.config.NettyHttpClientConfig;
 import com.influxdb.v3.client.write.WriteOptions;
-import static org.assertj.core.api.AssertionsForClassTypes.catchThrowable;
+import io.netty.handler.codec.http.DefaultHttpHeaders;
+import io.netty.handler.codec.http.HttpHeaders;
 import io.netty.handler.codec.http.HttpMethod;
 import io.netty.handler.proxy.HttpProxyHandler;
 import io.netty.handler.ssl.JdkSslContext;
@@ -63,6 +61,8 @@
 import org.junit.jupiter.api.AfterEach;
 import org.junit.jupiter.api.Test;
 
+import static org.assertj.core.api.AssertionsForClassTypes.catchThrowable;
+
 public class RestClientTest extends AbstractMockServerTest {
 
     private RestClient restClient;
@@ -419,12 +419,11 @@ public void errorFromBodyText() throws URISyntaxException, IOException {
 
     @Test
     public void generateHttpException() {
-        HttpHeaders headers = HttpHeaders.of(Map.of(
-                        "content-type", List.of("application/json"),
-                        "retry-after", List.of("300")),
-                (key, value) -> true);
+        HttpHeaders headers = new DefaultHttpHeaders();
+        headers.add("content-type", "application/json");
+        headers.add("retry-after", "300");
 
-        InfluxDBApiHttpException exception = new InfluxDBApiHttpException(
+        InfluxDBApiNettyException exception = new InfluxDBApiNettyException(
                 new InfluxDBApiException("test exception"), headers, 418);
 
         Assertions.assertThat(exception.headers()).isEqualTo(headers);