Also check for trailers only response

Signed-off-by: Sri Krishna Paritala <skrishna@buf.build>
connectrpc · Sep 3, 2024 · 61f7cff · 61f7cff
1 parent cfe0305
commit 61f7cff
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/packages/connect/src/protocol-grpc/validate-trailer.ts b/packages/connect/src/protocol-grpc/validate-trailer.ts
@@ -31,7 +31,12 @@ export function validateTrailer(trailer: Headers, header: Headers): void {
     });
     throw err;
   }
-  if (!trailer.has(headerGrpcStatus) && !trailer.has(headerStatusDetailsBin)) {
+  if (
+    !header.has(headerGrpcStatus) &&
+    !header.has(headerStatusDetailsBin) &&
+    !trailer.has(headerGrpcStatus) &&
+    !trailer.has(headerStatusDetailsBin)
+  ) {
     throw new ConnectError("protocol error: missing status", Code.Internal);
   }
 }