Add tests for StreamedResponse to make sure its header's map is case …

…insensitively
scala-steward · Jul 8, 2024 · f6bac89 · f6bac89
1 parent a21cb8e
commit f6bac89
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/play-ahc-ws-standalone/src/test/scala/play/api/libs/ws/ahc/AhcWSResponseSpec.scala b/play-ahc-ws-standalone/src/test/scala/play/api/libs/ws/ahc/AhcWSResponseSpec.scala
@@ -128,6 +128,14 @@ class AhcWSResponseSpec extends Specification with DefaultBodyReadables with Def
       headers.contains("Bar") must beTrue
     }
 
+    "get headers map which retrieves headers case insensitively (for streamed responses)" in {
+      val srcHeaders = Map("Foo" -> Seq("a"), "foo" -> Seq("b"), "FOO" -> Seq("b"), "Bar" -> Seq("baz"))
+      val response   = new StreamedResponse(null, 200, "", null, srcHeaders, null, true)
+      val headers    = response.headers
+      headers("foo") must_== Seq("a", "b", "b")
+      headers("BAR") must_== Seq("baz")
+    }
+
     "get a single header" in {
       val ahcResponse: AHCResponse = mock[AHCResponse]
       val ahcHeaders               = new DefaultHttpHeaders(true)

diff --git a/play-ahc-ws-standalone/src/test/scala/play/libs/ws/ahc/AhcWSResponseSpec.scala b/play-ahc-ws-standalone/src/test/scala/play/libs/ws/ahc/AhcWSResponseSpec.scala
@@ -46,6 +46,14 @@ class AhcWSResponseSpec extends Specification with DefaultBodyReadables with Def
       headers.get("BAR").asScala must_== Seq("baz")
     }
 
+    "get headers map which retrieves headers case insensitively (for streamed responses)" in {
+      val srcHeaders = Map("Foo" -> Seq("a"), "foo" -> Seq("b"), "FOO" -> Seq("b"), "Bar" -> Seq("baz"))
+      val response   = new StreamedResponse(null, 200, "", null, srcHeaders, null, true)
+      val headers    = response.getHeaders
+      headers.get("foo").asScala must_== Seq("a", "b", "b")
+      headers.get("BAR").asScala must_== Seq("baz")
+    }
+
     "get a single header" in {
       val srcResponse = mock[Response]
       val srcHeaders = new DefaultHttpHeaders()