raboof/pekko
1
0
Fork 0
Code Issues Pull requests Projects Releases 6 Packages Wiki Activity Actions 11

#20123 Added recoverWithRetries which adds a parameter to limit the (#20137)

Browse source 
number of attempts. Existing recoverWith method behavior remains unchanged.
This commit is contained in:
David Knapp 2016-04-20 06:24:12 -07:00 • committed by Patrik Nordwall
parent e0af113f17
commit d7fe98e1c4
8 changed files with 157 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

22
akka-stream-tests/src/test/scala/akka/stream/scaladsl/FlowRecoverWithSpec.scala
View file

@ -53,7 +53,7 @@ class FlowRecoverWithSpec extends AkkaSpec {
.expectError(ex)
}
"be able to recover with th same unmaterialized source if configured" in assertAllStagesStopped {
"be able to recover with the same unmaterialized source if configured" in assertAllStagesStopped {
val src = Source(1 to 3).map { a ⇒ if (a == 3) throw ex else a }
src.recoverWith { case t: Throwable ⇒ src }
.runWith(TestSink.probe[Int])
@ -99,7 +99,7 @@ class FlowRecoverWithSpec extends AkkaSpec {
.expectComplete()
}
"terminate with exception if altrnative source failed" in assertAllStagesStopped {
"terminate with exception if partial function fails to match after an alternative source failure" in assertAllStagesStopped {
Source(1 to 3).map { a ⇒ if (a == 3) throw new IndexOutOfBoundsException() else a }
.recoverWith {
case t: IndexOutOfBoundsException ⇒
@ -112,5 +112,23 @@ class FlowRecoverWithSpec extends AkkaSpec {
.request(1)
.expectError(ex)
}
"terminate with exception after set number of retries" in assertAllStagesStopped {
Source(1 to 3).map { a ⇒ if (a == 3) throw new IndexOutOfBoundsException() else a }
.recoverWithRetries(3, {
case t: Throwable ⇒
Source(List(11, 22)).concat(Source.failed(ex))
}).runWith(TestSink.probe[Int])
.request(2)
.expectNextN(List(1, 2))
.request(2)
.expectNextN(List(11, 22))
.request(2)
.expectNextN(List(11, 22))
.request(2)
.expectNextN(List(11, 22))
.request(1)
.expectError(ex)
}
}
}