Adds a delay before checking the metrics (#147)

higherkindness · Jan 18, 2018 · 48e08e3 · 48e08e3
1 parent 74802f6
commit 48e08e3
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 0 deletions.
diff --git a/modules/common/src/test/scala/RpcBaseTestSuite.scala b/modules/common/src/test/scala/RpcBaseTestSuite.scala
@@ -23,6 +23,8 @@ import org.scalatest.{Matchers, OneInstancePerTest, WordSpec}
 
 trait RpcBaseTestSuite extends WordSpec with Matchers with OneInstancePerTest with MockFactory {
 
+  val sleepM: ConcurrentMonad[Unit] = delayM(Thread.sleep(500))
+
   trait Helpers {
 
     def runK[F[_], A, B](kleisli: Kleisli[F, A, B], v: A): F[B] =

diff --git a/modules/common/src/test/scala/common.scala b/modules/common/src/test/scala/common.scala
@@ -22,4 +22,6 @@ package object common {
 
   type ConcurrentMonad[A] = IO[A]
 
+  def delayM[A](a: A): ConcurrentMonad[A] = IO(a)
+
 }
diff --git a/modules/prometheus/client/src/test/scala/BaseMonitorClientInterceptorTests.scala b/modules/prometheus/client/src/test/scala/BaseMonitorClientInterceptorTests.scala
@@ -130,8 +130,10 @@ abstract class BaseMonitorClientInterceptorTests extends RpcBaseTestSuite {
       (for {
         _ <- serverStart[ConcurrentMonad]
         _ <- clientProgram[ConcurrentMonad]
+        _ <- sleepM
         _ <- check1[ConcurrentMonad]
         _ <- clientProgram2[ConcurrentMonad]
+        _ <- sleepM
         _ <- check2[ConcurrentMonad]
         _ <- serverStop[ConcurrentMonad]
       } yield (): Unit).unsafeRunSync()
@@ -296,6 +298,7 @@ abstract class BaseMonitorClientInterceptorTests extends RpcBaseTestSuite {
         _ <- serverStart[ConcurrentMonad]
         _ <- unary[ConcurrentMonad]
         _ <- clientStreaming[ConcurrentMonad]
+        _ <- sleepM
         _ <- serverStop[ConcurrentMonad]
       } yield (): Unit).unsafeRunSync()