feat: improve test

Author: NguyenHoangSon96

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

@@ -25,7 +25,6 @@
 import java.util.List;
 import java.util.Objects;
 import java.util.concurrent.Executor;
-import java.util.concurrent.TimeUnit;
 import javax.annotation.Nonnull;
 import javax.annotation.Nullable;
 
@@ -40,7 +39,7 @@
  */
 public final class GrpcCallOption {
 
-    private final Deadline deadLineAfter;
+    private final Deadline deadline;
     private final Executor executor;
     private final String compressorName;
     private final Boolean waitForReady;
@@ -49,7 +48,7 @@ public final class GrpcCallOption {
     private final CallOption[] callOptionCallback;
 
     private GrpcCallOption(@Nonnull final Builder builder) {
-        this.deadLineAfter = builder.deadLineAfter;
+        this.deadline = builder.deadline;
         this.executor = builder.executor;
         this.compressorName = builder.compressorName;
         this.waitForReady = builder.waitForReady;
@@ -59,13 +58,13 @@ private GrpcCallOption(@Nonnull final Builder builder) {
     }
 
     /**
-     * Returns the deadline that is after the given duration from now.
+     * Returns the absolute deadline for a call.
      *
      * @return the Deadline object
      */
     @Nullable
-    public Deadline getDeadlineAfter() {
-        return deadLineAfter;
+    public Deadline getDeadline() {
+        return deadline;
     }
 
     /**
@@ -135,7 +134,7 @@ public boolean equals(final Object o) {
             return false;
         }
         GrpcCallOption that = (GrpcCallOption) o;
-        return Objects.equals(deadLineAfter, that.deadLineAfter)
+        return Objects.equals(deadline, that.deadline)
                 && Objects.equals(executor, that.executor)
                 && Objects.equals(compressorName, that.compressorName)
                 && Objects.equals(waitForReady, that.waitForReady)
@@ -145,7 +144,7 @@ public boolean equals(final Object o) {
 
     @Override
     public int hashCode() {
-        return Objects.hash(deadLineAfter,
+        return Objects.hash(deadline,
                 executor,
                 compressorName,
                 waitForReady,
@@ -157,7 +156,7 @@ public int hashCode() {
     @Override
     public String toString() {
         return "GrpcCallOption{"
-                + "deadLineAfter=" + deadLineAfter
+                + "deadline=" + deadline
                 + ", executor=" + executor
                 + ", compressorName='" + compressorName
                 + '\''
@@ -168,7 +167,7 @@ public String toString() {
     }
 
     public static final class Builder {
-        private Deadline deadLineAfter;
+        private Deadline deadline;
         private Executor executor;
         private String compressorName;
         private Boolean waitForReady;
@@ -177,20 +176,18 @@ public static final class Builder {
         private final List<CallOption> callOptions = new ArrayList<>();
 
         /**
-         * Sets a deadline that is after the given {@code duration} from
-         * now.
-         * @param duration The duration
-         * @param timeUnit The time unit
+         * Sets the absolute deadline for a rpc call.
+         * @param deadline The deadline
          * @return this
          */
-        public Builder withDeadlineAfter(final long duration, @Nonnull final TimeUnit timeUnit) {
+        public Builder withDeadline(final @Nonnull Deadline deadline) {
             var callOption = new org.apache.arrow.flight.CallOptions.GrpcCallOption() {
                 @Override
                 public <T extends AbstractStub<T>> T wrapStub(final T stub) {
-                    return stub.withDeadlineAfter(duration, timeUnit);
+                    return stub.withDeadline(deadline);
                 }
             };
-            this.deadLineAfter = Deadline.after(duration, timeUnit);
+            this.deadline = deadline;
             callOptions.add(callOption);
             return this;
         }

src/test/java/com/influxdb/v3/client/query/QueryOptionsTest.java

@@ -25,9 +25,13 @@
 import java.nio.charset.StandardCharsets;
 import java.util.ArrayList;
 import java.util.List;
+import java.util.concurrent.Executor;
+import java.util.concurrent.Executors;
+import java.util.concurrent.TimeUnit;
 import java.util.stream.Stream;
 import javax.annotation.Nonnull;
 
+import io.grpc.Deadline;
 import io.grpc.ManagedChannel;
 import io.grpc.ManagedChannelBuilder;
 import org.apache.arrow.flight.CallOption;
@@ -178,11 +182,15 @@ void setInboundMessageSizeLarge() throws Exception {
 
     @Test
     void grpcCallOption() {
+        Executor executor = Executors.newSingleThreadExecutor();
+        Deadline deadline = Deadline.after(2, TimeUnit.SECONDS);
         GrpcCallOption grpcCallOption = new GrpcCallOption.Builder()
                 .withMaxInboundMessageSize(1024)
                 .withMaxOutboundMessageSize(1024)
                 .withCompressorName("my-compressor")
                 .withWaitForReady()
+                .withExecutor(executor)
+                .withDeadline(deadline)
                 .build();
         ManagedChannel channel = ManagedChannelBuilder.forAddress("localhost", 3333)
                 .usePlaintext()
@@ -193,10 +201,14 @@ void grpcCallOption() {
         }
 
         io.grpc.CallOptions stubCallOptions = stub.getCallOptions();
-        Assertions.assertThat(stubCallOptions.getMaxInboundMessageSize()).isEqualTo(grpcCallOption.getMaxInboundMessageSize());
-        Assertions.assertThat(stubCallOptions.getMaxOutboundMessageSize()).isEqualTo(grpcCallOption.getMaxOutboundMessageSize());
+        Assertions.assertThat(stubCallOptions.getMaxInboundMessageSize())
+                .isEqualTo(grpcCallOption.getMaxInboundMessageSize());
+        Assertions.assertThat(stubCallOptions.getMaxOutboundMessageSize())
+                .isEqualTo(grpcCallOption.getMaxOutboundMessageSize());
         Assertions.assertThat(stubCallOptions.getCompressor()).isEqualTo(grpcCallOption.getCompressorName());
         Assertions.assertThat(stubCallOptions.isWaitForReady()).isEqualTo(grpcCallOption.getWaitForReady());
+        Assertions.assertThat(stubCallOptions.getExecutor()).isEqualTo(grpcCallOption.getExecutor());
+        Assertions.assertThat(stubCallOptions.getDeadline()).isEqualTo(grpcCallOption.getDeadline());
     }
 
     private FlightServer simpleFlightServer(@Nonnull final URI uri,