improve: apply suggestions from code review

grpc content type could be application/grpc+{proto, json, etc.}
http4s · Nov 13, 2024 · cb7d8d7 · cb7d8d7
1 parent f59040e
commit cb7d8d7
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/codegen/testing/src/test/scala/hello/world/TestServiceSuite.scala b/codegen/testing/src/test/scala/hello/world/TestServiceSuite.scala
@@ -75,11 +75,18 @@ class TestServiceSuite extends CatsEffectSuite with ScalaCheckEffectSuite {
     val client = Client.fromHttpApp(TestService.toRoutes(impl).orNotFound)
     implicit val methodExceptPost: Arbitrary[Method] =
       Arbitrary(Gen.oneOf(Method.all.filterNot(_ === Method.POST)))
-    forAllF { (meth: Method) =>
+    implicit val wellKnownGRPCHeaders: Arbitrary[Header.ToRaw] = Arbitrary(
+      Gen.oneOf(
+        "Content-Type" -> "application/grpc",
+        "Content-Type" -> "application/grpc+proto",
+        "Content-Type" -> "application/grpc+json",
+      )
+    )
+    forAllF { (meth: Method, grpcHeader: Header.ToRaw) =>
       client
         .run(
           Request[IO](meth, uri"/hello.world.TestService/any/url")
-            .withHeaders("Content-Type" -> "application/grpc")
+            .withHeaders(grpcHeader)
         )
         .use { resp =>
           val headers = resp.headers