Fix mock type

xmtp · Jul 15, 2024 · dfb4446 · dfb4446
1 parent 6973ddf
commit dfb4446
Showing 1 changed file with 1 addition and 9 deletions.
diff --git a/packages/react-sdk/src/hooks/useCanMessage.test.ts b/packages/react-sdk/src/hooks/useCanMessage.test.ts
@@ -108,15 +108,7 @@ describe("useCanMessage", () => {
       }));
       const canMessageStaticSpy = vi
         .spyOn(Client, "canMessage")
-        .mockImplementation((peerAddress: string | string[]) =>
-          typeof peerAddress === "string"
-            ? Promise.resolve([true])
-            : Promise.resolve(
-                Array.from({ length: peerAddress.length }).fill(
-                  true,
-                ) as boolean[],
-              ),
-        );
+        .mockResolvedValue([true]);
 
       const { result } = renderHook(() => useCanMessage());