fix(tests): seed Project membership in new staging fixtures

Author: mzxchandra

apps/sim/lib/logs/public-queries.integration.ts

@@ -12,6 +12,7 @@ import {
   workflowExecutionSnapshots,
   workspace,
 } from '@sim/db/schema'
+import { deleteWorkspaceFixture, insertWorkspaceFixture } from '@sim/db/testing/workspace-fixtures'
 import { generateId } from '@sim/utils/id'
 import { eq, sql } from 'drizzle-orm'
 import { afterAll, beforeAll, describe, expect, it } from 'vitest'
@@ -84,7 +85,7 @@ beforeAll(async () => {
     createdAt: now,
     updatedAt: now,
   })
-  await db.insert(workspace).values({
+  await insertWorkspaceFixture(db, {
     id: ids.workspace,
     name: 'Public Log Paging',
     ownerId: ids.owner,
@@ -136,7 +137,7 @@ afterAll(async () => {
   await db.delete(jobExecutionLogs).where(eq(jobExecutionLogs.workspaceId, ids.workspace))
   await db.delete(workflowExecutionSnapshots).where(eq(workflowExecutionSnapshots.id, ids.snapshot))
   await db.delete(workflow).where(eq(workflow.id, ids.workflow))
-  await db.delete(workspace).where(eq(workspace.id, ids.workspace))
+  await deleteWorkspaceFixture(db, eq(workspace.id, ids.workspace))
   await db.delete(user).where(eq(user.id, ids.owner))
 })
 

apps/sim/lib/mothership/chat/application/read-stream-status.integration.ts

@@ -1,6 +1,7 @@
 /** Exercises the transcript-free stream status read against real PostgreSQL rows. */
 import { db } from '@sim/db'
 import { copilotChats, permissions, user, workspace } from '@sim/db/schema'
+import { deleteWorkspaceFixture, insertWorkspaceFixture } from '@sim/db/testing/workspace-fixtures'
 import { createSessionPrincipal } from '@sim/testing/factories/principal.factory'
 import { generateId } from '@sim/utils/id'
 import { eq, inArray } from 'drizzle-orm'
@@ -33,7 +34,7 @@ beforeAll(async () => {
       updatedAt: now,
     }))
   )
-  await db.insert(workspace).values({
+  await insertWorkspaceFixture(db, {
     id: workspaceId,
     name: 'Chat stream status fixture',
     ownerId,
@@ -52,7 +53,7 @@ beforeAll(async () => {
 
 afterAll(async () => {
   await db.delete(copilotChats).where(eq(copilotChats.workspaceId, workspaceId))
-  await db.delete(workspace).where(eq(workspace.id, workspaceId))
+  await deleteWorkspaceFixture(db, eq(workspace.id, workspaceId))
   await db.delete(user).where(inArray(user.id, [ownerId, memberId]))
   await db.$client.end()
 })

apps/sim/lib/workspaces/__integration__/recently-deleted.integration.ts

@@ -9,6 +9,7 @@ import {
   workspace,
   workspaceFiles,
 } from '@sim/db/schema'
+import { deleteWorkspaceFixture, insertWorkspaceFixture } from '@sim/db/testing/workspace-fixtures'
 import { generateId } from '@sim/utils/id'
 import { inArray, sql } from 'drizzle-orm'
 import { afterAll, beforeAll, describe, expect, it } from 'vitest'
@@ -103,7 +104,8 @@ describe('Recently Deleted listing against PostgreSQL', () => {
         updatedAt: now,
       }))
     )
-    await db.insert(workspace).values(
+    await insertWorkspaceFixture(
+      db,
       [workspaceId, otherWorkspaceId].map((id) => ({
         id,
         name: 'Recently deleted fixture',
@@ -216,7 +218,7 @@ describe('Recently Deleted listing against PostgreSQL', () => {
   })
 
   afterAll(async () => {
-    await db.delete(workspace).where(inArray(workspace.id, [workspaceId, otherWorkspaceId]))
+    await deleteWorkspaceFixture(db, inArray(workspace.id, [workspaceId, otherWorkspaceId]))
     await db.delete(user).where(inArray(user.id, [aliceId, bobId]))
     await db.$client.end()
   })