@lightupai/polaris 0.0.5 → 0.0.6

package/tests/helpers.ts

@@ -0,0 +1,71 @@
+import type { Sql } from "../src/service/db";
+
+/**
+ * Reset test data: drop and recreate all tables using the canonical schema.
+ * This ensures tests always match the current schema in db.ts.
+ */
+export async function resetTestData(sql: Sql): Promise<void> {
+  await sql`DROP TABLE IF EXISTS events`;
+  await sql`DROP TABLE IF EXISTS sessions`;
+  await sql`DROP TABLE IF EXISTS projects`;
+  await sql`DROP TABLE IF EXISTS users`;
+  await sql`DROP TABLE IF EXISTS orgs`;
+  await sql`
+    CREATE TABLE orgs (
+      id TEXT PRIMARY KEY,
+      name TEXT NOT NULL,
+      slug TEXT UNIQUE,
+      domain TEXT,
+      slack_team_id TEXT,
+      slack_bot_token TEXT,
+      slack_system_channel_id TEXT,
+      created_at TIMESTAMPTZ NOT NULL DEFAULT now()
+    )
+  `;
+  await sql`
+    CREATE TABLE users (
+      id TEXT PRIMARY KEY,
+      email TEXT NOT NULL UNIQUE,
+      name TEXT NOT NULL,
+      org_id TEXT NOT NULL REFERENCES orgs(id),
+      participant_id TEXT NOT NULL,
+      created_at TIMESTAMPTZ NOT NULL DEFAULT now()
+    )
+  `;
+  await sql`
+    CREATE TABLE projects (
+      id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
+      org_id TEXT NOT NULL REFERENCES orgs(id),
+      name TEXT NOT NULL,
+      slack_channel_id TEXT,
+      slack_channel_name TEXT,
+      created_at TIMESTAMPTZ NOT NULL DEFAULT now(),
+      UNIQUE (org_id, name)
+    )
+  `;
+  await sql`
+    CREATE TABLE sessions (
+      name TEXT NOT NULL,
+      project_id UUID NOT NULL REFERENCES projects(id),
+      org_id TEXT NOT NULL,
+      driver TEXT,
+      created_at TIMESTAMPTZ NOT NULL DEFAULT now(),
+      PRIMARY KEY (project_id, name)
+    )
+  `;
+  await sql`
+    CREATE TABLE events (
+      id UUID PRIMARY KEY,
+      org_id TEXT NOT NULL,
+      project_id UUID NOT NULL REFERENCES projects(id),
+      session TEXT NOT NULL,
+      timestamp TIMESTAMPTZ NOT NULL,
+      source TEXT NOT NULL,
+      sender TEXT NOT NULL,
+      payload JSONB NOT NULL
+    )
+  `;
+  await sql`CREATE INDEX idx_events_project ON events(project_id, timestamp)`;
+  await sql`CREATE INDEX idx_events_session ON events(project_id, session, timestamp)`;
+  await sql`INSERT INTO orgs (id, name) VALUES ('default', 'Default') ON CONFLICT DO NOTHING`;
+}

package/tests/service.test.ts

@@ -2,6 +2,7 @@ import { describe, expect, test, beforeAll, afterAll, beforeEach } from "bun:tes
 import { startServer } from "../src/service/server";
 import type { Sql } from "../src/service/db";
 import type { Server } from "bun";
+import { resetTestData } from "./helpers";
 
 const DATABASE_URL = process.env.DATABASE_URL ?? "postgres://polaris:polaris@localhost:5432/polaris_test";
 
@@ -21,19 +22,7 @@ afterAll(async () => {
 });
 
 beforeEach(async () => {
-  await sql`DROP TABLE IF EXISTS events`;
-  await sql`DROP TABLE IF EXISTS sessions`;
-  await sql`DROP TABLE IF EXISTS projects`;
-  await sql`DROP TABLE IF EXISTS users`;
-  await sql`DROP TABLE IF EXISTS orgs`;
-  await sql`CREATE TABLE IF NOT EXISTS orgs (id TEXT PRIMARY KEY, name TEXT NOT NULL, slug TEXT UNIQUE, domain TEXT, slack_team_id TEXT, slack_bot_token TEXT, slack_system_channel_id TEXT, created_at TIMESTAMPTZ NOT NULL DEFAULT now())`;
-  await sql`CREATE TABLE IF NOT EXISTS users (id TEXT PRIMARY KEY, email TEXT NOT NULL UNIQUE, name TEXT NOT NULL, org_id TEXT NOT NULL REFERENCES orgs(id), participant_id TEXT NOT NULL, created_at TIMESTAMPTZ NOT NULL DEFAULT now())`;
-  await sql`CREATE TABLE IF NOT EXISTS projects (name TEXT NOT NULL, org_id TEXT NOT NULL REFERENCES orgs(id), created_at TIMESTAMPTZ NOT NULL DEFAULT now(), PRIMARY KEY (org_id, name))`;
-  await sql`CREATE TABLE IF NOT EXISTS sessions (name TEXT NOT NULL, project TEXT NOT NULL, org_id TEXT NOT NULL, driver TEXT, created_at TIMESTAMPTZ NOT NULL DEFAULT now(), PRIMARY KEY (org_id, project, name), FOREIGN KEY (org_id, project) REFERENCES projects(org_id, name))`;
-  await sql`CREATE TABLE IF NOT EXISTS events (id UUID PRIMARY KEY, org_id TEXT NOT NULL, project TEXT NOT NULL, session TEXT NOT NULL, timestamp TIMESTAMPTZ NOT NULL, source TEXT NOT NULL, sender TEXT NOT NULL, payload JSONB NOT NULL)`;
-  await sql`CREATE INDEX IF NOT EXISTS idx_events_project ON events(org_id, project, timestamp)`;
-  await sql`CREATE INDEX IF NOT EXISTS idx_events_session ON events(org_id, project, session, timestamp)`;
-  await sql`INSERT INTO orgs (id, name) VALUES ('default', 'Default') ON CONFLICT DO NOTHING`;
+  await resetTestData(sql);
 });
 
 async function post(path: string, body: unknown) {

package/tests/types.test.ts

@@ -208,12 +208,12 @@ describe("PolarisEvent", () => {
 
 describe("Project", () => {
   test("parses valid project", () => {
-    const result = Project.parse({ name: "pj", created_at: "2026-06-05T10:00:00.000Z" });
+    const result = Project.parse({ id: "a0eebc99-9c0b-4ef8-bb6d-6bb9bd380a11", name: "pj", created_at: "2026-06-05T10:00:00.000Z" });
     expect(result.name).toBe("pj");
   });
 
   test("rejects empty name", () => {
-    expect(() => Project.parse({ name: "", created_at: "2026-06-05T10:00:00.000Z" })).toThrow();
+    expect(() => Project.parse({ id: "a0eebc99-9c0b-4ef8-bb6d-6bb9bd380a11", name: "", created_at: "2026-06-05T10:00:00.000Z" })).toThrow();
   });
 });
 

package/tests/web.test.ts

@@ -12,16 +12,17 @@ import {
 import { createApp } from "../src/web/app";
 import { createDb, createOrg, createUser, type Sql } from "../src/service/db";
 import { createToken } from "../src/service/auth";
+import { resetTestData } from "./helpers";
 
 const DATABASE_URL = process.env.DATABASE_URL ?? "postgres://polaris:polaris@localhost:5432/polaris_test";
 
 // --- View context helpers ---
 
 const base = { token: "test-token", userName: "Manu Bansal", orgName: "Lightup", orgSlug: "lightup-data" as string | null, email: "manu@lightup.ai" };
-const fresh = { ...base, orgSlug: null, slackConnected: false, cliInstalled: false, hasConnectedSession: false };
-const slackDone = { ...base, slackConnected: true, cliInstalled: false, hasConnectedSession: false };
-const cliDone = { ...base, slackConnected: true, cliInstalled: true, hasConnectedSession: false };
-const allDone = { ...base, slackConnected: true, cliInstalled: true, hasConnectedSession: true };
+const fresh = { ...base, orgSlug: null, slackConnected: false, cliInstalled: false, hasConnectedSession: false, totalPrompts: 0 };
+const slackDone = { ...base, slackConnected: true, cliInstalled: false, hasConnectedSession: false, totalPrompts: 0 };
+const cliDone = { ...base, slackConnected: true, cliInstalled: true, hasConnectedSession: false, totalPrompts: 0 };
+const allDone = { ...base, slackConnected: true, cliInstalled: true, hasConnectedSession: true, totalPrompts: 42 };
 
 // --- Setup view ---
 
@@ -35,7 +36,7 @@ describe("renderSetupView", () => {
     // Connect Slack button is present
     expect(html).toContain("Connect Slack");
     // Install CLI command is present
-    expect(html).toContain("npx @lightupai/polaris login");
+    expect(html).toContain("npx @lightupai/polaris");
     // Connect session command is present
     expect(html).toContain("/polaris join my-project my-session");
   });
@@ -52,7 +53,7 @@ describe("renderSetupView", () => {
     const highlightIdx = html.indexOf("border-polaris-300");
     expect(highlightIdx).toBeGreaterThan(devicesIdx);
     // Install CLI command present
-    expect(html).toContain("npx @lightupai/polaris login");
+    expect(html).toContain("npx @lightupai/polaris");
   });
 
   test("cli done: floor and devices done, sessions is highlighted", () => {
@@ -80,29 +81,23 @@ describe("renderSetupView", () => {
 describe("renderActiveView", () => {
   test("shows compact floor bar", () => {
     const html = renderActiveView(allDone, mockActiveSessions, mockProjects, mockDevices);
-    expect(html).toContain("Live");
+    expect(html).toContain("Connected");
     // No Connect Slack button
     expect(html).not.toContain("Connect Slack");
   });
 
-  test("shows session cards with roles", () => {
-    const html = renderActiveView(allDone, mockActiveSessions, mockProjects, mockDevices);
-    expect(html).toContain("polaris/auth");
-    expect(html).toContain("polaris/slack-bridge");
-    // Role badges are rendered (Advisor shown when participant ID derivation
-    // from display name doesn't match fixture — expected for mock data)
-    expect(html).toContain("Advisor");
-  });
-
-  test("shows session descriptions and event counts", () => {
+  test("shows sessions nested under projects", () => {
     const html = renderActiveView(allDone, mockActiveSessions, mockProjects, mockDevices);
-    expect(html).toContain("Google SSO + JWT auth");
-    expect(html).toContain("42 events");
+    // Session names shown within project cards
+    expect(html).toContain("auth");
+    expect(html).toContain("slack-bridge");
+    // Project names as card headers
+    expect(html).toContain("polaris");
+    expect(html).toContain("data-pipeline");
   });
 
-  test("shows other participants in sessions", () => {
+  test("shows driver info in session rows", () => {
     const html = renderActiveView(allDone, mockActiveSessions, mockProjects, mockDevices);
-    expect(html).toContain("agent:security-reviewer");
     expect(html).toContain("user:krishna");
   });
 
@@ -263,16 +258,7 @@ describe("routes", () => {
   });
 
   beforeEach(async () => {
-    await sql`DROP TABLE IF EXISTS events`;
-    await sql`DROP TABLE IF EXISTS sessions`;
-    await sql`DROP TABLE IF EXISTS projects`;
-    await sql`DROP TABLE IF EXISTS users`;
-    await sql`DROP TABLE IF EXISTS orgs`;
-    await sql`CREATE TABLE IF NOT EXISTS orgs (id TEXT PRIMARY KEY, name TEXT NOT NULL, slug TEXT UNIQUE, domain TEXT, slack_team_id TEXT, slack_bot_token TEXT, slack_system_channel_id TEXT, created_at TIMESTAMPTZ NOT NULL DEFAULT now())`;
-    await sql`CREATE TABLE IF NOT EXISTS users (id TEXT PRIMARY KEY, email TEXT NOT NULL UNIQUE, name TEXT NOT NULL, org_id TEXT NOT NULL REFERENCES orgs(id), participant_id TEXT NOT NULL, created_at TIMESTAMPTZ NOT NULL DEFAULT now())`;
-    await sql`CREATE TABLE IF NOT EXISTS projects (name TEXT NOT NULL, org_id TEXT NOT NULL REFERENCES orgs(id), created_at TIMESTAMPTZ NOT NULL DEFAULT now(), PRIMARY KEY (org_id, name))`;
-    await sql`CREATE TABLE IF NOT EXISTS sessions (name TEXT NOT NULL, project TEXT NOT NULL, org_id TEXT NOT NULL, driver TEXT, created_at TIMESTAMPTZ NOT NULL DEFAULT now(), PRIMARY KEY (org_id, project, name), FOREIGN KEY (org_id, project) REFERENCES projects(org_id, name))`;
-    await sql`CREATE TABLE IF NOT EXISTS events (id UUID PRIMARY KEY, org_id TEXT NOT NULL, project TEXT NOT NULL, session TEXT NOT NULL, timestamp TIMESTAMPTZ NOT NULL, source TEXT NOT NULL, sender TEXT NOT NULL, payload JSONB NOT NULL)`;
+    await resetTestData(sql);
 
     await createOrg(sql, "test-org", "Test Org", "test.com");
     await createUser(sql, "user-1", "test@test.com", "Test User", "test-org", "user:test");