Merge pull request #1532 from vectos/fix-dx

DX: fix some issues
softwaremill · Aug 26, 2022 · 44dc113 · 44dc113
2 parents cc54908 + b478576
commit 44dc113
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/effects/zio/src/main/scalajvm/sttp/client3/httpclient/zio/package.scala b/effects/zio/src/main/scalajvm/sttp/client3/httpclient/zio/package.scala
@@ -0,0 +1,9 @@
+package sttp.client3.httpclient
+
+import sttp.capabilities.WebSockets
+import sttp.capabilities.zio.ZioStreams
+import sttp.client3.SttpBackend
+
+package object zio {
+  type SttpClient = SttpBackend[_root_.zio.Task, ZioStreams with WebSockets]
+}
diff --git a/effects/zio1/src/main/scalajvm/sttp/client3/httpclient/zio/HttpClientZioBackend.scala b/effects/zio1/src/main/scalajvm/sttp/client3/httpclient/zio/HttpClientZioBackend.scala
@@ -175,7 +175,7 @@ object HttpClientZioBackend {
     *
     * See [[SttpBackendStub]] for details on how to configure stub responses.
     */
-  def stub: SttpBackendStub[Task, ZioStreams] =
+  def stub: SttpBackendStub[Task, ZioStreams with WebSockets] =
     SttpBackendStub(new RIOMonadAsyncError[Any])
 
   val stubLayer: ZLayer[Any, Nothing, SttpClientStubbing with SttpClient] = SttpClientStubbing.layer