fix: do not include deleted projects in findAllActiveDocumentsWithAgr…

…eggatedData
latitude-dev · Sep 21, 2024 · 71d1811 · 71d1811
1 parent 29c1a90
commit 71d1811
Show file tree

Hide file tree

Showing 3 changed files with 28 additions and 0 deletions.
diff --git a/packages/core/src/repositories/projectsRepository.test.ts b/packages/core/src/repositories/projectsRepository.test.ts
@@ -86,5 +86,25 @@ describe('ProjectsRepository', async () => {
 
       expect(projects).toHaveLength(1)
     })
+
+    it('should not include deleted projects', async () => {
+      const { project: deletedProject } = await createProject({
+        workspace,
+        deletedAt: new Date(),
+      })
+
+      const { project: activeProject } = await createProject({
+        workspace,
+      })
+
+      const result = await repository.findAllActiveDocumentsWithAgreggatedData()
+
+      expect(result.ok).toBe(true)
+      const projects = result.unwrap()
+
+      expect(projects).toHaveLength(1)
+      expect(projects[0]?.id).toBe(activeProject.id)
+      expect(projects.find((p) => p.id === deletedProject.id)).toBeUndefined()
+    })
   })
 })
diff --git a/packages/core/src/repositories/projectsRepository.ts b/packages/core/src/repositories/projectsRepository.ts
@@ -105,6 +105,7 @@ export class ProjectsRepository extends Repository<typeof tt, Project> {
       })
       .from(this.scope)
       .leftJoin(aggredatedData, eq(aggredatedData.id, this.scope.id))
+      .where(isNull(this.scope.deletedAt))
 
     return Result.ok(result)
   }

diff --git a/packages/core/src/tests/factories/projects.ts b/packages/core/src/tests/factories/projects.ts
@@ -4,6 +4,7 @@ import { DocumentVersion, Providers, User, Workspace } from '../../browser'
 import { unsafelyGetUser } from '../../data-access'
 import { mergeCommit } from '../../services/commits'
 import { createNewDocument, updateDocument } from '../../services/documents'
+import { destroyProject, updateProject } from '../../services/projects'
 import { createProject as createProjectFn } from '../../services/projects/create'
 import { createApiKey } from './apiKeys'
 import { createDraft } from './commits'
@@ -40,6 +41,7 @@ export async function flattenDocumentStructure({
 
 export type ICreateProject = {
   name?: string
+  deletedAt?: Date
   workspace?: Workspace | ICreateWorkspace
   providers?: { type: Providers; name: string }[]
   evaluations?: Omit<IEvaluationData, 'workspace'>[]
@@ -74,6 +76,11 @@ export async function createProject(projectData: Partial<ICreateProject> = {}) {
   })
   let { project, commit } = result.unwrap()
 
+  if (projectData.deletedAt)
+    await updateProject(project, { deletedAt: projectData.deletedAt }).then(
+      (r) => r.unwrap(),
+    )
+
   const providersToCreate =
     projectData.providers == undefined
       ? [{ type: Providers.OpenAI, name: faker.internet.domainName() }]