Skip to content

Commit 24fcc00

Browse files
committed
fix OOM error in test
1 parent 102e0ab commit 24fcc00

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

analytics/src/test/java/com/segment/analytics/AnalyticsTest.java

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@
1919
import java.util.Collections;
2020
import java.util.concurrent.ExecutorService;
2121
import java.util.concurrent.Executors;
22+
import java.util.concurrent.TimeUnit;
2223
import java.util.concurrent.atomic.AtomicInteger;
2324
import org.junit.Before;
2425
import org.junit.Test;
@@ -143,7 +144,7 @@ public void threadSafeTest(MessageBuilderTest builder)
143144
} while (initialTime.until(now, ChronoUnit.MILLIS) < millisRunning);
144145

145146
service.shutdown();
146-
while (!service.isShutdown() || !service.isTerminated()) {}
147+
service.awaitTermination(5, TimeUnit.SECONDS);
147148

148149
verify(spy, times(counter.get())).enqueue(any(Message.class));
149150
}

0 commit comments

Comments
 (0)