67 lines
2.1 KiB
Scala
67 lines
2.1 KiB
Scala
|
/**
|
||
|
|
* Copyright (C) 2015 Typesafe Inc. <http://www.typesafe.com>
|
||
|
|
*/
|
||
|
|
package akka.stream.io
|
||
|
|
|
||
|
|
import java.io.OutputStream
|
||
|
|
|
||
|
|
import akka.stream.scaladsl.Source
|
||
|
|
import akka.stream.testkit.StreamTestKit._
|
||
|
|
import akka.stream.testkit.{ AkkaSpec, StreamTestKit }
|
||
|
|
import akka.stream.{ ActorFlowMaterializer, ActorFlowMaterializerSettings }
|
||
|
|
import akka.testkit.TestProbe
|
||
|
|
import akka.util.ByteString
|
||
|
|
|
||
|
|
import scala.concurrent.Await
|
||
|
|
import scala.concurrent.duration._
|
||
|
|
|
||
|
|
class OutputStreamSinkSpec extends AkkaSpec(StreamTestKit.UnboundedMailboxConfig) {
|
||
|
|
|
||
|
|
val settings = ActorFlowMaterializerSettings(system).withDispatcher("akka.actor.default-dispatcher")
|
||
|
|
implicit val materializer = ActorFlowMaterializer(settings)
|
||
|
|
|
||
|
|
"OutputStreamSink" must {
|
||
|
|
"write bytes to void OutputStream" in checkThatAllStagesAreStopped {
|
||
|
|
val p = TestProbe()
|
||
|
|
val datas = List(ByteString("a"), ByteString("c"), ByteString("c"))
|
||
|
|
|
||
|
|
val completion = Source(datas)
|
||
|
|
.runWith(OutputStreamSink(() ⇒ new OutputStream {
|
||
|
|
override def write(i: Int): Unit = ()
|
||
|
|
override def write(bytes: Array[Byte]): Unit = p.ref ! ByteString(bytes).utf8String
|
||
|
|
}))
|
||
|
|
|
||
|
|
p.expectMsg(datas(0).utf8String)
|
||
|
|
p.expectMsg(datas(1).utf8String)
|
||
|
|
p.expectMsg(datas(2).utf8String)
|
||
|
|
Await.ready(completion, 3.seconds)
|
||
|
|
}
|
||
|
|
|
||
|
|
"close underlying stream when error received" in checkThatAllStagesAreStopped {
|
||
|
|
val p = TestProbe()
|
||
|
|
Source.failed(new TE("Boom!"))
|
||
|
|
.runWith(OutputStreamSink(() ⇒ new OutputStream {
|
||
|
|
override def write(i: Int): Unit = ()
|
||
|
|
override def close() = p.ref ! "closed"
|
||
|
|
}))
|
||
|
|
|
||
|
|
p.expectMsg("closed")
|
||
|
|
}
|
||
|
|
|
||
|
|
"close underlying stream when completion received" in checkThatAllStagesAreStopped {
|
||
|
|
val p = TestProbe()
|
||
|
|
Source.empty
|
||
|
|
.runWith(OutputStreamSink(() ⇒ new OutputStream {
|
||
|
|
override def write(i: Int): Unit = ()
|
||
|
|
override def write(bytes: Array[Byte]): Unit = p.ref ! ByteString(bytes).utf8String
|
||
|
|
override def close() = p.ref ! "closed"
|
||
|
|
}))
|
||
|
|
|
||
|
|
p.expectMsg("closed")
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|