+str #18501 improvements and fixes in StreamTestKit (adds toStrict)

2015-09-17 12:57:07 +02:00 · 2015-09-17 12:57:07 +02:00 · ab96ebfca0
commit ab96ebfca0
parent 8bb8dc4f68
5 changed files with 367 additions and 48 deletions
--- a/akka-stream-testkit/src/test/scala/akka/stream/testkit/StreamTestKitSpec.scala
+++ b/akka-stream-testkit/src/test/scala/akka/stream/testkit/StreamTestKitSpec.scala
@ -0,0 +1,125 @@
+/**
+ * Copyright (C) 2015 Typesafe Inc. <http://www.typesafe.com>
+ */
+package akka.stream.testkit
+
+import akka.stream._
+import akka.stream.scaladsl.Source
+import akka.stream.testkit.scaladsl.TestSink
+
+import scala.concurrent.duration._
+
+class StreamTestKitSpec extends AkkaSpec {
+
+  implicit val mat = ActorMaterializer()
+
+  val ex = new Exception("Boom!")
+
+  "A TestSink Probe" must {
+    "#toStrict" in {
+      Source(1 to 4).runWith(TestSink.probe())
+        .toStrict(300.millis) should ===(List(1, 2, 3, 4))
+    }
+
+    "#toStrict with failing source" in {
+      val msg = intercept[AssertionError] {
+        Source(() ⇒ new Iterator[Int] {
+          var i = 0
+          override def hasNext: Boolean = true
+          override def next(): Int = {
+            i += 1
+            i match {
+              case 3 ⇒ throw ex
+              case n ⇒ n
+            }
+          }
+        }).runWith(TestSink.probe())
+          .toStrict(300.millis)
+      }.getMessage
+
+      msg should include("Boom!")
+      msg should include("List(1, 2)")
+    }
+
+    "#toStrict when subscription was already obtained" in {
+      val p = Source(1 to 4).runWith(TestSink.probe())
+      p.expectSubscription()
+      p.toStrict(300.millis) should ===(List(1, 2, 3, 4))
+    }
+
+    "#expectNextOrError with right element" in {
+      Source(1 to 4).runWith(TestSink.probe())
+        .request(4)
+        .expectNextOrError(1, ex)
+    }
+
+    "#expectNextOrError with right exception" in {
+      Source.failed[Int](ex).runWith(TestSink.probe())
+        .request(4)
+        .expectNextOrError(1, ex)
+    }
+
+    "#expectNextOrError fail if the next element is not the expected one" in {
+      intercept[AssertionError] {
+        Source(1 to 4).runWith(TestSink.probe())
+          .request(4)
+          .expectNextOrError(100, ex)
+      }.getMessage should include("OnNext(1)")
+    }
+
+    "#expectError" in {
+      Source.failed[Int](ex).runWith(TestSink.probe())
+        .request(1)
+        .expectError() should ===(ex)
+    }
+
+    "#expectError fail if no error signalled" in {
+      intercept[AssertionError] {
+        Source(1 to 4).runWith(TestSink.probe())
+          .request(1)
+          .expectError()
+      }.getMessage should include("OnNext")
+    }
+
+    "#expectComplete should fail if error signalled" in {
+      intercept[AssertionError] {
+        Source.failed[Int](ex).runWith(TestSink.probe())
+          .request(1)
+          .expectComplete()
+      }.getMessage should include("OnError")
+    }
+
+    "#expectComplete should fail if next element signalled" in {
+      intercept[AssertionError] {
+        Source(1 to 4).runWith(TestSink.probe())
+          .request(1)
+          .expectComplete()
+      }.getMessage should include("OnNext")
+    }
+
+    "#expectNextOrComplete with right element" in {
+      Source(1 to 4).runWith(TestSink.probe())
+        .request(4)
+        .expectNextOrComplete(1)
+    }
+
+    "#expectNextOrComplete with completion" in {
+      Source.single(1).runWith(TestSink.probe())
+        .request(4)
+        .expectNextOrComplete(1)
+        .expectNextOrComplete(1337)
+    }
+
+    "#expectNextN given a number of elements" in {
+      Source(1 to 4).runWith(TestSink.probe())
+        .request(4)
+        .expectNextN(4) should ===(List(1, 2, 3, 4))
+    }
+
+    "#expectNextN given specific elements" in {
+      Source(1 to 4).runWith(TestSink.probe())
+        .request(4)
+        .expectNextN(4) should ===(List(1, 2, 3, 4))
+    }
+  }
+}