Ensure starting mock Fluentd

fluent · May 2, 2016 · 10d57db · 10d57db
1 parent 871948d
commit 10d57db
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/test/java/org/fluentd/logger/sender/TestAsyncRawSocketSender.java b/src/test/java/org/fluentd/logger/sender/TestAsyncRawSocketSender.java
@@ -49,6 +49,7 @@ public void process(MessagePack msgpack, Socket socket) throws IOException {
             }
         });
         fluentd.start();
+        fluentd.waitUntilReady();
 
         // start asyncSenders
         Sender asyncSender = new AsyncRawSocketSender("localhost", port);
@@ -111,6 +112,7 @@ public void process(MessagePack msgpack, Socket socket) throws IOException {
             }
         });
         fluentd.start();
+        fluentd.waitUntilReady();
 
         // start asyncSenders
         Sender asyncSender = new AsyncRawSocketSender("localhost", port);
@@ -162,6 +164,7 @@ public void process(MessagePack msgpack, Socket socket) throws IOException {
             }
         });
         fluentds[0].start();
+        fluentds[0].waitUntilReady();
         ports[1] = MockFluentd.randomPort();
         elists[1] = new ArrayList<Event>();
         fluentds[1] = new MockFluentd(ports[1], new MockFluentd.MockProcess() {
@@ -180,6 +183,7 @@ public void process(MessagePack msgpack, Socket socket) throws IOException {
             }
         });
         fluentds[1].start();
+        fluentds[1].waitUntilReady();
 
         // start AsyncSenders
         Sender[] asyncSenders = new Sender[2];
@@ -272,6 +276,7 @@ public void process(MessagePack msgpack, Socket socket) throws IOException {
 
         MockFluentd fluentd = new MockFluentd(port, mockProcess);
         fluentd.start();
+        fluentd.waitUntilReady();
 
         Sender asyncSender = new AsyncRawSocketSender("localhost", port);
         assertFalse(asyncSender.isConnected());
@@ -369,6 +374,7 @@ public void run() {
                 try {
                     bufferFull.await(20, TimeUnit.SECONDS);
                     fluentd.start();
+                    fluentd.waitUntilReady();
                 } catch (InterruptedException e) {
                     e.printStackTrace();
                 }