From 5e92cbe91fd514e0e5b6853ed16a3d8de1b58eb9 Mon Sep 17 00:00:00 2001
From: Daniel Kantor <kd@stacklok.com>
Date: Thu, 23 Jan 2025 18:58:50 +0100
Subject: [PATCH] implement type column values

---
 src/components/AlertsTable.tsx                | 15 ++++++++++-
 src/routes/__tests__/route-dashboard.test.tsx | 25 ++++++++-----------
 2 files changed, 24 insertions(+), 16 deletions(-)

diff --git a/src/components/AlertsTable.tsx b/src/components/AlertsTable.tsx
index d73faf92..a9256d72 100644
--- a/src/components/AlertsTable.tsx
+++ b/src/components/AlertsTable.tsx
@@ -37,6 +37,17 @@ const getTitle = (alert: AlertConversation) => {
   return title;
 };
 
+function TypeCellContent({ alert }: { alert: AlertConversation }) {
+  const conversationType = alert.conversation.type;
+  console.log({ conversationType });
+
+  if (conversationType === "chat") {
+    return "Chat";
+  }
+
+  return "Completion";
+}
+
 export function AlertsTable() {
   const {
     isMaliciousFilterActive,
@@ -151,7 +162,9 @@ export function AlertsTable() {
                       addSuffix: true,
                     })}
                   </Cell>
-                  <Cell className="truncate">{alert.trigger_type}</Cell>
+                  <Cell className="truncate">
+                    <TypeCellContent alert={alert} />
+                  </Cell>
                   <Cell className="truncate">{getTitle(alert)}</Cell>
                 </Row>
               ))}
diff --git a/src/routes/__tests__/route-dashboard.test.tsx b/src/routes/__tests__/route-dashboard.test.tsx
index 680d6ec2..6005e6b6 100644
--- a/src/routes/__tests__/route-dashboard.test.tsx
+++ b/src/routes/__tests__/route-dashboard.test.tsx
@@ -167,7 +167,7 @@ describe("Dashboard", () => {
       "row",
     )[1] as HTMLElement;
 
-    expect(within(firstRow).getByText(/codegate-secrets/i)).toBeVisible();
+    expect(within(firstRow).getByText(/chat/i)).toBeVisible();
     expect(within(firstRow).getByText(/[0-9]+.*ago/i)).toBeVisible();
   });
 
@@ -209,10 +209,10 @@ describe("Dashboard", () => {
 
     expect(screen.getByTestId(/alerts-count/i)).toHaveTextContent("2");
     expect(
-      screen.getByRole("row", {
-        name: /codegate-secrets/i,
-      }),
-    ).toBeVisible();
+      screen.getAllByRole("row", {
+        name: /chat/i,
+      }).length,
+    ).toBeGreaterThanOrEqual(1);
 
     userEvent.click(
       screen.getByRole("switch", {
@@ -244,15 +244,10 @@ describe("Dashboard", () => {
 
     expect(screen.getByTestId(/alerts-count/i)).toHaveTextContent("2");
     expect(
-      screen.getByRole("row", {
-        name: /codegate-secrets/i,
-      }),
-    ).toBeVisible();
-    expect(
-      screen.getByRole("row", {
-        name: /codegate-context-retriever/i,
-      }),
-    ).toBeVisible();
+      screen.getAllByRole("row", {
+        name: /chat/i,
+      }).length,
+    ).toBeGreaterThanOrEqual(1);
 
     await userEvent.type(screen.getByRole("searchbox"), "codegate-secrets");
 
@@ -262,7 +257,7 @@ describe("Dashboard", () => {
     const row = within(screen.getByTestId("alerts-table")).getAllByRole(
       "row",
     )[1] as HTMLElement;
-    expect(within(row).getByText(/codegate-secrets/i)).toBeVisible();
+    expect(within(row).getByText(/chat/i)).toBeVisible();
   });
 
   it("should sort alerts by date desc", async () => {