Added aother test

connectrpc · Oct 26, 2023 · b01b18e · b01b18e
1 parent 3a35a26
commit b01b18e
Showing 1 changed file with 50 additions and 13 deletions.
diff --git a/packages/connect/src/router-transport.spec.ts b/packages/connect/src/router-transport.spec.ts
@@ -25,6 +25,7 @@ import { createAsyncIterable } from "./protocol/async-iterable.js";
 import { createRouterTransport } from "./router-transport.js";
 import { ConnectError } from "./connect-error.js";
 import { Code } from "./code.js";
+import { MethodType } from "./method-type.js";
 
 describe("createRoutesTransport", function () {
   const testService = {
@@ -226,22 +227,58 @@ describe("createRoutesTransport", function () {
     });
   });
   describe("supports self describing method type", function () {
-    const methodDefinition = {
-      name: "Unary",
-      kind: MethodKind.Unary,
-      I: Int32Value,
-      O: StringValue,
-      service: testService,
-      idempotency: MethodIdempotency.NoSideEffects,
-    } as const;
-    const transport = createRouterTransport(({ rpc }) => {
-      rpc(methodDefinition, (request) => {
-        return { value: `${request.value}-RESPONSE` };
+    it("should work for unary", async function () {
+      const methodDefinition = {
+        name: "Unary",
+        kind: MethodKind.Unary,
+        I: Int32Value,
+        O: StringValue,
+        service: testService,
+        idempotency: MethodIdempotency.NoSideEffects,
+      } as const;
+      const transport = createRouterTransport(({ rpc }) => {
+        rpc(methodDefinition, (request) => {
+          return { value: `${request.value}-RESPONSE` };
+        });
       });
+
+      const client = createPromiseClient(testService, transport);
+
+      const res = await client.unary({ value: 13 });
+      expect(res.value).toBe("13-RESPONSE");
     });
 
-    const client = createPromiseClient(testService, transport);
-    it("should work for unary", async function () {
+    it("should work with method built from service", async function () {
+      const methodDefinition = {
+        ...testService.methods.unary,
+        service: testService,
+      } as const;
+      const transport = createRouterTransport(({ rpc }) => {
+        rpc(methodDefinition, (request) => {
+          return { value: `${request.value}-RESPONSE` };
+        });
+      });
+
+      const client = createPromiseClient(testService, transport);
+
+      const res = await client.unary({ value: 13 });
+      expect(res.value).toBe("13-RESPONSE");
+    });
+
+    it("should work with method built with explicit type", async function () {
+      const methodDefinition: MethodType<Int32Value, StringValue> = {
+        ...testService.methods.unary,
+        service: testService,
+      };
+
+      const transport = createRouterTransport(({ rpc }) => {
+        rpc(methodDefinition, (request) => {
+          return { value: `${request.value}-RESPONSE` };
+        });
+      });
+
+      const client = createPromiseClient(testService, transport);
+
       const res = await client.unary({ value: 13 });
       expect(res.value).toBe("13-RESPONSE");
     });