Test: Ensuring graceful DNS timeout handling (#79)

Author: PLNech

algoliasearch/src/test/java/com/algolia/search/saas/IndexTest.java

@@ -91,6 +91,10 @@ public void tearDown() throws Exception {
 
     @Test
     public void testSearchAsync() throws Exception {
+        testSearchAsync(Helpers.wait);
+    }
+
+    public void testSearchAsync(int waitTimeout) throws Exception {
         // Empty search.
         final CountDownLatch signal = new CountDownLatch(1);
         index.searchAsync(new Query(), new CompletionHandler() {
@@ -104,7 +108,7 @@ public void requestCompleted(JSONObject content, AlgoliaException error) {
                 signal.countDown();
             }
         });
-        assertTrue("No callback was called", signal.await(Helpers.wait, TimeUnit.SECONDS));
+        assertTrue("No callback was called", signal.await(waitTimeout, TimeUnit.SECONDS));
 
         // Search with query.
         final CountDownLatch signal2 = new CountDownLatch(1);
@@ -119,7 +123,7 @@ public void requestCompleted(JSONObject content, AlgoliaException error) {
                 signal2.countDown();
             }
         });
-        assertTrue("No callback was called", signal2.await(Helpers.wait, TimeUnit.SECONDS));
+        assertTrue("No callback was called", signal2.await(waitTimeout, TimeUnit.SECONDS));
     }
 
     @Test
@@ -433,6 +437,26 @@ public void testHostSwitch() throws Exception {
         testSearchAsync();
     }
 
+    @Test
+    public void testDNSTimeout() throws Exception {
+        // Given first host resulting in a DNS Timeout
+        String appId = (String) Whitebox.getInternalState(client, "applicationID");
+        List<String> hostsArray = (List<String>) Whitebox.getInternalState(client, "readHosts");
+        hostsArray.set(0, appId + "-dsn.algolia.biz");
+        Whitebox.setInternalState(client, "readHosts", hostsArray);
+
+        //And an index that does not cache search queries
+        index.disableSearchCache();
+
+
+        // Expect successful search within 5 seconds
+        long startTime = System.nanoTime();
+        testSearchAsync(5);
+        final long duration = (System.nanoTime() - startTime) / 1000000;
+
+        // Which should take at least 2 seconds, as per Client.connectTimeout
+        assertTrue("We should first timeout before successfully searching, but test took only " + duration + " ms.", duration > 2000);
+    }
 
     @Test
     public void testSNI() throws Exception {