blob: e8ab9d7656e2dd32670b0e418b34bc93ec4d35f3 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
|
package com.softwaremill.sttp.akkahttp
import akka.NotUsed
import akka.actor.ActorSystem
import akka.stream.{ActorMaterializer, Materializer}
import akka.stream.scaladsl.Source
import akka.util.ByteString
import com.softwaremill.sttp.SttpBackend
import com.softwaremill.sttp.testing.ConvertToFuture
import com.softwaremill.sttp.testing.streaming.{StreamingTest, TestStreamingBackend}
import scala.concurrent.Future
class AkkaHttpStreamingTest extends StreamingTest[Future, Source[ByteString, Any]] {
private implicit val actorSystem: ActorSystem = ActorSystem("sttp-test")
private implicit val materializer: ActorMaterializer = ActorMaterializer()
override val testStreamingBackend: TestStreamingBackend[Future, Source[ByteString, Any]] =
new AkkaHttpTestStreamingBackend(actorSystem)
}
class AkkaHttpTestStreamingBackend(
actorSystem: ActorSystem
)(implicit materializer: Materializer)
extends TestStreamingBackend[Future, Source[ByteString, Any]] {
override implicit val backend: SttpBackend[Future, Source[ByteString, Any]] =
AkkaHttpBackend.usingActorSystem(actorSystem)
override implicit val convertToFuture: ConvertToFuture[Future] =
ConvertToFuture.future
override def bodyProducer(body: String): Source[ByteString, NotUsed] =
Source.single(ByteString(body))
override def bodyConsumer(stream: Source[ByteString, Any]): Future[String] =
stream.map(_.utf8String).runReduce(_ + _)
}
|