chore: fixed unit tests

ubiquity-os-marketplace · Dec 12, 2024 · 099ee70 · 099ee70
1 parent 8380526
commit 099ee70
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 12 deletions.
diff --git a/tests/__mocks__/handlers.ts b/tests/__mocks__/handlers.ts
@@ -8,12 +8,16 @@ export const handlers = [
   http.get(`${process.env.SUPABASE_URL}/rest/v1/users*`, ({ request }) => {
     const url = new URL(request.url);
     const id = url.searchParams.get("id");
+    const walletId = url.searchParams.get("wallet_id");
 
-    if (!id) {
-      return HttpResponse.text("", { status: 400 });
+    if (id) {
+      const idNumber = Number(id.match(/\d+/)?.[0]);
+      return HttpResponse.json(db.users.findFirst({ where: { id: { equals: idNumber } } }));
+    } else if (walletId) {
+      const idNumber = Number(walletId.match(/\d+/)?.[0]);
+      return HttpResponse.json(db.users.findFirst({ where: { wallet_id: { equals: idNumber } } }));
     }
-    const idNumber = Number(id.match(/\d+/)?.[0]);
-    return HttpResponse.json(db.users.findFirst({ where: { id: { equals: idNumber } } }));
+    return HttpResponse.text("", { status: 400 });
   }),
   http.patch(`${process.env.SUPABASE_URL}/rest/v1/users*`, async ({ request }) => {
     const url = new URL(request.url);
@@ -38,7 +42,7 @@ export const handlers = [
   }),
   http.patch(`${process.env.SUPABASE_URL}/rest/v1/wallets*`, async ({ request }) => {
     const url = new URL(request.url);
-    const id = url.searchParams.get("id");
+    const id = url.searchParams.get("id") ?? url.searchParams.get("wallet_id");
 
     if (!id) {
       return HttpResponse.text("", { status: 400 });

diff --git a/tests/main.test.ts b/tests/main.test.ts
@@ -1,4 +1,5 @@
 import { afterAll, afterEach, beforeAll, beforeEach, describe, expect, it, jest } from "@jest/globals";
+import { drop } from "@mswjs/data";
 import { ethers } from "ethers";
 import { plugin } from "../src/plugin";
 import { Context } from "../src/types";
@@ -11,12 +12,6 @@ import { Octokit } from "@octokit/rest";
 
 beforeAll(() => {
   server.listen();
-  for (const dbTable of Object.keys(dbSeed)) {
-    const tableName = dbTable as keyof typeof dbSeed;
-    for (const dbRow of dbSeed[tableName]) {
-      db[tableName].create(dbRow);
-    }
-  }
 });
 afterEach(() => {
   server.resetHandlers();
@@ -40,7 +35,15 @@ jest.mock("@ubiquity-os/plugin-sdk", () => ({
 }));
 
 describe("Wallet command tests", () => {
-  beforeEach(() => {});
+  beforeEach(() => {
+    drop(db);
+    for (const dbTable of Object.keys(dbSeed)) {
+      const tableName = dbTable as keyof typeof dbSeed;
+      for (const dbRow of dbSeed[tableName]) {
+        db[tableName].create(dbRow);
+      }
+    }
+  });
 
   it("Should handle /wallet comment", async () => {
     const spy = jest.spyOn(Logs.prototype, "ok");