Make MultiSenderTest more robust

komamitsu · Oct 11, 2015 · f660e39 · f660e39
1 parent 1ca7ea1
commit f660e39
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/src/test/java/org/komamitsu/fluency/sender/MultiSenderTest.java b/src/test/java/org/komamitsu/fluency/sender/MultiSenderTest.java
@@ -57,6 +57,7 @@ public void testSend()
         server0.start();
         final MockMultiTCPServerWithMetrics server1 = new MockMultiTCPServerWithMetrics();
         server1.start();
+        TimeUnit.MILLISECONDS.sleep(1000);
 
         int concurency = 20;
         final int reqNum = 5000;
@@ -92,7 +93,13 @@ public void run()
             });
         }
 
-        assertTrue(latch.await(6, TimeUnit.SECONDS));
+        for (int i = 0; i < 60; i++) {
+            if (latch.await(1, TimeUnit.SECONDS)) {
+                break;
+            }
+        }
+        assertEquals(0, latch.getCount());
+
         sender.close();
         TimeUnit.MILLISECONDS.sleep(1000);
         server1.stop();