add unit test: testBuilderCallback

apache · Mar 12, 2024 · deb218a · deb218a
1 parent ae67f8b
commit deb218a
Showing 1 changed file with 27 additions and 0 deletions.
diff --git a/hugegraph-common/src/test/java/org/apache/hugegraph/unit/rest/RestClientTest.java b/hugegraph-common/src/test/java/org/apache/hugegraph/unit/rest/RestClientTest.java
@@ -21,6 +21,7 @@
 import java.util.HashMap;
 import java.util.Map;
 import java.util.UUID;
+import java.util.concurrent.TimeUnit;
 import java.util.function.BiFunction;
 
 import javax.net.ssl.SSLContext;
@@ -320,6 +321,28 @@ public void testAuthContext() {
         Assert.assertNull(client.getAuthContext());
     }
 
+    @SneakyThrows
+    @Test
+    public void testBuilderCallback() {
+        //default config
+        MockRestClientImpl restClient = new MockRestClientImpl(TEST_URL,
+                                                               RestClientConfig.builder().build());
+        OkHttpClient okHttpClient = Whitebox.getInternalState(restClient, "client");
+        Assert.assertEquals(okHttpClient.connectTimeoutMillis(), 10000);
+        Assert.assertEquals(okHttpClient.readTimeoutMillis(), 10000);
+
+        //set config by builderCallback
+        RestClientConfig config = RestClientConfig.builder().builderCallback(
+                builder -> builder.connectTimeout(5, TimeUnit.SECONDS)
+                                  .readTimeout(30, TimeUnit.SECONDS)).build();
+
+        restClient = new MockRestClientImpl(TEST_URL, config);
+        okHttpClient = Whitebox.getInternalState(restClient, "client");
+
+        Assert.assertEquals(okHttpClient.connectTimeoutMillis(), 5000);
+        Assert.assertEquals(okHttpClient.readTimeoutMillis(), 30000);
+    }
+
     @SneakyThrows
     @Test
     public void testRequest() {
@@ -485,6 +508,10 @@ public MockRestClientImpl(String url, int timeout) {
             super(url, timeout);
         }
 
+        public MockRestClientImpl(String url, RestClientConfig config) {
+            super(url, config);
+        }
+
         @Override
         protected void checkStatus(Response response, int... statuses) {
             // pass