Add headers to faked response in client tests

antonsviridov-clearscore · Oct 25, 2022 · e86371c · e86371c
1 parent dd74c0c
commit e86371c
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/modules/compliance-tests/src/smithy4s/compliancetests/ClientHttpComplianceTestCase.scala b/modules/compliance-tests/src/smithy4s/compliancetests/ClientHttpComplianceTestCase.scala
@@ -42,6 +42,7 @@ import smithy4s.tests.DefaultSchemaVisitor
 import scala.concurrent.duration._
 import smithy4s.http.HttpMediaType
 import org.http4s.MediaType
+import org.http4s.Header
 
 abstract class ClientHttpComplianceTestCase[
     P,
@@ -207,9 +208,15 @@ abstract class ClientHttpComplianceTestCase[
                       .through(utf8Encode)
                   }
                   .getOrElse(fs2.Stream.empty)
+              val headers: Seq[Header.ToRaw] =
+                testCase.headers.toList.flatMap(_.toList).map {
+                  case (key, value) =>
+                    Header.Raw(CIString(key), value)
+                }
               req.body.compile.drain.as(
                 Response[IO](status)
                   .withBodyStream(body)
+                  .putHeaders(headers: _*)
                   .putHeaders(
                     `Content-Type`(MediaType.unsafeParse(mediaType.value))
                   )