diff options
Diffstat (limited to 'implementations/monix/src/test/scala/com/softwaremill/sttp/impl/monix/MonixTestStreamingBackend.scala')
-rw-r--r-- | implementations/monix/src/test/scala/com/softwaremill/sttp/impl/monix/MonixTestStreamingBackend.scala | 22 |
1 files changed, 22 insertions, 0 deletions
diff --git a/implementations/monix/src/test/scala/com/softwaremill/sttp/impl/monix/MonixTestStreamingBackend.scala b/implementations/monix/src/test/scala/com/softwaremill/sttp/impl/monix/MonixTestStreamingBackend.scala new file mode 100644 index 0000000..3f84ec3 --- /dev/null +++ b/implementations/monix/src/test/scala/com/softwaremill/sttp/impl/monix/MonixTestStreamingBackend.scala @@ -0,0 +1,22 @@ +package com.softwaremill.sttp.impl.monix + +import java.nio.ByteBuffer + +import com.softwaremill.sttp.testing.streaming.{ConvertToFuture, TestStreamingBackend} +import monix.eval.Task +import monix.reactive.Observable + +trait MonixTestStreamingBackend extends TestStreamingBackend[Task, Observable[ByteBuffer]] { + + override implicit def convertToFuture: ConvertToFuture[Task] = convertMonixTaskToFuture + + override def bodyProducer(body: String): Observable[ByteBuffer] = + Observable.fromIterable(body.getBytes("utf-8").map(b => ByteBuffer.wrap(Array(b)))) + + override def bodyConsumer(stream: Observable[ByteBuffer]): Task[String] = + stream + .flatMap(bb => Observable.fromIterable(bb.array())) + .toListL + .map(bs => new String(bs.toArray, "utf8")) + +} |