Fix type error for web mock utils

pipe-cd · May 20, 2021 · 0aa0c8b · 0aa0c8b
1 parent 86e8767
commit 0aa0c8b
Showing 1 changed file with 14 additions and 11 deletions.
diff --git a/pkg/app/web/src/mocks/create-handler.ts b/pkg/app/web/src/mocks/create-handler.ts
@@ -14,15 +14,18 @@ export function createHandler<T extends { serializeBinary(): Uint8Array }>(
   serviceName: string,
   getResponseMessage: (requestData: Uint8Array) => T
 ): HandlerType {
-  return rest.post(createMask(serviceName), (req, res, ctx) => {
-    const arr: Uint8Array =
-      typeof req.body === "string" ? encoder.encode(req.body) : req.body;
-    const message = getResponseMessage(arr.slice(5));
-    const data = serialize(message.serializeBinary());
-    return res(
-      ctx.status(200),
-      ctx.set("Content-Type", "application/grpc-web+proto"),
-      ctx.body(data)
-    );
-  });
+  return rest.post<Uint8Array | string>(
+    createMask(serviceName),
+    (req, res, ctx) => {
+      const arr: Uint8Array =
+        typeof req.body === "string" ? encoder.encode(req.body) : req.body;
+      const message = getResponseMessage(arr.slice(5));
+      const data = serialize(message.serializeBinary());
+      return res(
+        ctx.status(200),
+        ctx.set("Content-Type", "application/grpc-web+proto"),
+        ctx.body(data)
+      );
+    }
+  );
 }