Move test utils into a package object, rather than base class.

1 files changed, 36 insertions, 0 deletions

diff --git a/src/test/scala/scala/async/TestLatch.scala b/src/test/scala/scala/async/TestLatch.scala
new file mode 100644
index 0000000..676ea63
--- /dev/null
+++ b/src/test/scala/scala/async/TestLatch.scala
@@ -0,0 +1,36 @@
+package scala.async
+
+import concurrent.{CanAwait, Awaitable}
+import concurrent.duration.Duration
+import java.util.concurrent.{TimeoutException, CountDownLatch, TimeUnit}
+
+object TestLatch {
+  val DefaultTimeout = Duration(5, TimeUnit.SECONDS)
+
+  def apply(count: Int = 1) = new TestLatch(count)
+}
+
+
+class TestLatch(count: Int = 1) extends Awaitable[Unit] {
+  private var latch = new CountDownLatch(count)
+
+  def countDown() = latch.countDown()
+
+  def isOpen: Boolean = latch.getCount == 0
+
+  def open() = while (!isOpen) countDown()
+
+  def reset() = latch = new CountDownLatch(count)
+
+  @throws(classOf[TimeoutException])
+  def ready(atMost: Duration)(implicit permit: CanAwait) = {
+    val opened = latch.await(atMost.toNanos, TimeUnit.NANOSECONDS)
+    if (!opened) throw new TimeoutException(s"Timeout of ${(atMost.toString)}.")
+    this
+  }
+
+  @throws(classOf[Exception])
+  def result(atMost: Duration)(implicit permit: CanAwait): Unit = {
+    ready(atMost)
+  }
+}