should send 100 Continue before 101 Switching Protocols when both Upg…

…rade and Expect headers are present
varshith257 · Sep 24, 2024 · 414e317 · 414e317
1 parent 60bcf9b
commit 414e317
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/zio-http/jvm/src/test/scala/zio/http/ServerSpec.scala b/zio-http/jvm/src/test/scala/zio/http/ServerSpec.scala
@@ -520,17 +520,22 @@ object ServerSpec extends RoutesRunnableSpec {
       } +
       test("should send 100 Continue before 101 Switching Protocols when both Upgrade and Expect headers are present") {
         val upgradeRoute = Method.GET / "upgrade" -> Handler.fromZIO {
-          ZIO.succeed(Response.switchingProtocols.withUpgrade("https"))
+          ZIO.succeed(
+            Response
+              .status(Status.SwitchingProtocols)
+              .addHeader(Header.Upgrade.withValue("https"))
+              .addHeader(Header.Connection.Upgrade),
+          )
         }
 
         val app = Routes(upgradeRoute)
 
         // Build request with both Expect: 100-continue and Upgrade headers
         val request = Request
           .get("/upgrade")
-          .addHeader(Header.Expect(Header.Expect.Continue))
-          .addHeader(Header.Connection("upgrade"))
-          .addHeader(Header.Upgrade("https"))
+          .addHeader(Header.Expect.Continue)
+          .addHeader(Header.Connection.Upgrade)
+          .addHeader(Header.Upgrade.withValue("https"))
 
         for {
           response <- app.runZIO(request)