Fixed to use received flag instead of Thread.sleep(1000)

Signed-off-by: Tsuyoshi Ozawa <[email protected]>

Author: oza

src/test/java/org/fluentd/logger/TestFluentLogger.java

@@ -18,6 +18,7 @@
 import java.util.concurrent.ExecutorService;
 import java.util.concurrent.Executors;
 import java.util.concurrent.TimeUnit;
+import java.util.concurrent.atomic.AtomicBoolean;
 import java.util.concurrent.atomic.AtomicReference;
 
 import static org.junit.Assert.*;
@@ -455,6 +456,7 @@ public void testFlushOnClose() throws Exception {
         int port = MockFluentd.randomPort();
         String host = "localhost";
         final List<Event> elist = new ArrayList<Event>();
+        final AtomicBoolean received = new AtomicBoolean(false);
         MockFluentd fluentd = new MockFluentd(port, new MockFluentd.MockProcess() {
             public void process(MessagePack msgpack, Socket socket) throws IOException {
                 BufferedInputStream in = new BufferedInputStream(socket.getInputStream());
@@ -463,6 +465,7 @@ public void process(MessagePack msgpack, Socket socket) throws IOException {
                     while (true) {
                         Event e = unpacker.read(Event.class);
                         elist.add(e);
+                        received.set(true);
                     }
                     //socket.close();
                 } catch (EOFException e) {
@@ -488,7 +491,10 @@ public void process(MessagePack msgpack, Socket socket) throws IOException {
         // close loggers and it should flush the buffer
         logger.close();
 
-        Thread.sleep(1000);
+        // wait for fluentd's getting at least one kv pair
+        while(!received.get()) {
+            Thread.sleep(100);
+        }
 
         // close mock fluentd
         fluentd.close();