Fixed race condition in periodic sink test.

Author: vjkoskela

src/test/java/com/arpnetworking/tsdcore/sinks/PeriodicStatisticsSinkTest.java

@@ -25,7 +25,6 @@
 import org.mockito.Mockito;
 
 import java.util.concurrent.ScheduledExecutorService;
-import java.util.concurrent.TimeUnit;
 
 /**
  * Tests for the <code>PeriodicStatisticsSink</code> class.
@@ -57,15 +56,15 @@ public void testFlushOnClose() {
     }
 
     @Test
-    public void testPeriodicFlush() throws InterruptedException {
+    public void testPeriodicFlush() {
         final ScheduledExecutorService executor = Mockito.mock(ScheduledExecutorService.class);
         final ArgumentCaptor<Runnable> runnableCaptor = ArgumentCaptor.forClass(Runnable.class);
         final Sink statisticsSink = new PeriodicStatisticsSink(_statisticsSinkBuilder, executor);
         Mockito.verify(executor).scheduleAtFixedRate(
                 runnableCaptor.capture(),
                 Mockito.anyLong(),
                 Mockito.anyLong(),
-                Mockito.<TimeUnit>any());
+                Mockito.any());
 
         final Runnable periodicRunnable = runnableCaptor.getValue();
 
@@ -91,7 +90,15 @@ public void testPeriodicFlush() throws InterruptedException {
 
     @Test
     public void testRecordProcessedAggregateData() {
-        final Sink statisticsSink = _statisticsSinkBuilder.build();
+        final ScheduledExecutorService executor = Mockito.mock(ScheduledExecutorService.class);
+        final ArgumentCaptor<Runnable> runnableCaptor = ArgumentCaptor.forClass(Runnable.class);
+        final Sink statisticsSink = new PeriodicStatisticsSink(_statisticsSinkBuilder, executor);
+        Mockito.verify(executor).scheduleAtFixedRate(
+                runnableCaptor.capture(),
+                Mockito.anyLong(),
+                Mockito.anyLong(),
+                Mockito.any());
+
         Mockito.verify(_mockMetricsFactory).create();
         Mockito.verify(_mockMetrics).resetCounter(COUNTER_NAME);