Learn from label removal patterns for newsletters

apps/web/app/api/google/webhook/process-history-item.ts

@@ -26,7 +26,7 @@ import { formatError } from "@/utils/error";
 import { createEmailProvider } from "@/utils/email/provider";
 import { enqueueDigestItem } from "@/utils/digest/index";
 import { HistoryEventType } from "@/app/api/google/webhook/types";
-import { handleLabelRemovedEvent } from "@/app/api/google/webhook/process-label-removed-event";
+import { processLabelRemoval } from "@/app/api/google/webhook/process-label-removed-event";
 
 export async function processHistoryItem(
   historyItem: {
@@ -67,7 +67,7 @@ export async function processHistoryItem(
 
   if (type === HistoryEventType.LABEL_REMOVED) {
     logger.info("Processing label removed event for learning", loggerOptions);
-    return handleLabelRemovedEvent(item, {
+    return processLabelRemoval(item, {
       emailAccount,
       provider,
     });

apps/web/app/api/google/webhook/process-label-removed-event.test.ts

@@ -1,7 +1,7 @@
 import { vi, describe, it, expect, beforeEach } from "vitest";
 import { ColdEmailStatus } from "@prisma/client";
 import { HistoryEventType } from "./types";
-import { handleLabelRemovedEvent } from "./process-label-removed-event";
+import { processLabelRemoval } from "./process-label-removed-event";
 import type { gmail_v1 } from "@googleapis/gmail";
 import { inboxZeroLabels } from "@/utils/label";
 import { saveLearnedPatterns } from "@/utils/rule/learned-patterns";
@@ -63,6 +63,24 @@ describe("process-label-removed-event", () => {
     } as gmail_v1.Schema$HistoryLabelRemoved,
   });
 
+  const mockGmail = {
+    users: {
+      labels: {
+        list: vi.fn().mockImplementation((_params) => {
+          return Promise.resolve({
+            data: {
+              labels: [
+                { id: "label-1", name: inboxZeroLabels.cold_email.name },
+                { id: "label-2", name: "Newsletter" },
+                { id: "label-3", name: "Marketing" },
+                { id: "label-4", name: "To Reply" },
+              ],
+            },
+          });
+        }),
+      },
+    },
+  } as any;
   const mockEmailAccount = {
     id: "email-account-id",
     email: "user@test.com",
@@ -82,24 +100,18 @@ describe("process-label-removed-event", () => {
   } as any;
 
   const defaultOptions = {
+    gmail: mockGmail,
     emailAccount: mockEmailAccount,
     provider: mockProvider,
   };
 
-  describe("handleLabelRemovedEvent", () => {
+  describe("processLabelRemoval", () => {
     it("should process Cold Email label removal and update ColdEmail status", async () => {
       prisma.coldEmail.upsert.mockResolvedValue({} as any);
 
       const historyItem = createLabelRemovedHistoryItem();
 
-      console.log("Test data:", JSON.stringify(historyItem.item, null, 2));
-
-      try {
-        await handleLabelRemovedEvent(historyItem.item, defaultOptions);
-      } catch (error) {
-        console.error("Function error:", error);
-        throw error;
-      }
+      await processLabelRemoval(historyItem.item, defaultOptions);
 
       expect(prisma.coldEmail.upsert).toHaveBeenCalledWith({
         where: {
@@ -126,7 +138,7 @@ describe("process-label-removed-event", () => {
         "label-2",
       ]);
 
-      await handleLabelRemovedEvent(historyItem.item, defaultOptions);
+      await processLabelRemoval(historyItem.item, defaultOptions);
 
       expect(saveLearnedPatterns).not.toHaveBeenCalled();
     });
@@ -136,7 +148,7 @@ describe("process-label-removed-event", () => {
         "label-4",
       ]);
 
-      await handleLabelRemovedEvent(historyItem.item, defaultOptions);
+      await processLabelRemoval(historyItem.item, defaultOptions);
 
       expect(saveLearnedPatterns).not.toHaveBeenCalled();
     });
@@ -146,7 +158,7 @@ describe("process-label-removed-event", () => {
         "label-2",
       ]);
 
-      await handleLabelRemovedEvent(historyItem.item, defaultOptions);
+      await processLabelRemoval(historyItem.item, defaultOptions);
 
       expect(saveLearnedPatterns).not.toHaveBeenCalled();
     });
@@ -156,7 +168,7 @@ describe("process-label-removed-event", () => {
         "label-2",
       ]);
 
-      await handleLabelRemovedEvent(historyItem.item, defaultOptions);
+      await processLabelRemoval(historyItem.item, defaultOptions);
 
       expect(saveLearnedPatterns).not.toHaveBeenCalled();
     });
@@ -166,7 +178,7 @@ describe("process-label-removed-event", () => {
         "label-3",
       ]);
 
-      await handleLabelRemovedEvent(historyItem.item, defaultOptions);
+      await processLabelRemoval(historyItem.item, defaultOptions);
 
       expect(saveLearnedPatterns).not.toHaveBeenCalled();
     });
@@ -177,7 +189,7 @@ describe("process-label-removed-event", () => {
         labelIds: ["label-1"],
       } as gmail_v1.Schema$HistoryLabelRemoved;
 
-      await handleLabelRemovedEvent(historyItem, defaultOptions);
+      await processLabelRemoval(historyItem, defaultOptions);
 
       expect(prisma.coldEmail.upsert).not.toHaveBeenCalled();
     });
@@ -188,7 +200,7 @@ describe("process-label-removed-event", () => {
         labelIds: ["label-1"],
       } as gmail_v1.Schema$HistoryLabelRemoved;
 
-      await handleLabelRemovedEvent(historyItem, defaultOptions);
+      await processLabelRemoval(historyItem, defaultOptions);
 
       expect(prisma.coldEmail.upsert).not.toHaveBeenCalled();
     });