Updated test

keyshade-xyz · Sep 14, 2024 · 7e23c29 · 7e23c29
1 parent 9abe7e3
commit 7e23c29
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 12 deletions.
diff --git a/apps/api/src/common/slug-generator.spec.ts b/apps/api/src/common/slug-generator.spec.ts
@@ -39,7 +39,7 @@ describe('generateEntitySlug', () => {
 
   describe('generateEntitySlug for each entity type', () => {
     it('should generate a unique slug for WORKSPACE_ROLE', async () => {
-      prisma.workspaceRole.findMany.mockResolvedValue([
+      prisma.$queryRaw.mockResolvedValue([
         {
           slug: 'workspace-role-0'
         }
@@ -54,35 +54,35 @@ describe('generateEntitySlug', () => {
     })
 
     it('should generate a unique slug for WORKSPACE', async () => {
-      prisma.workspace.findMany.mockResolvedValue([])
+      prisma.$queryRaw.mockResolvedValue([])
 
       const slug = await generateEntitySlug('Workspace', 'WORKSPACE', prisma)
       expect(slug).toBe('workspace-0')
     })
 
     it('should generate a unique slug for PROJECT', async () => {
-      prisma.project.findMany.mockResolvedValue([{ slug: 'project-z' }])
+      prisma.$queryRaw.mockResolvedValue([{ slug: 'project-z' }])
 
       const slug = await generateEntitySlug('Project', 'PROJECT', prisma)
       expect(slug).toBe('project-00')
     })
 
     it('should generate a unique slug for VARIABLE', async () => {
-      prisma.variable.findMany.mockResolvedValue([{ slug: 'variable-az' }])
+      prisma.$queryRaw.mockResolvedValue([{ slug: 'variable-az' }])
 
       const slug = await generateEntitySlug('Variable', 'VARIABLE', prisma)
       expect(slug).toBe('variable-b0')
     })
 
     it('should generate a unique slug for SECRET', async () => {
-      prisma.secret.findMany.mockResolvedValue([{ slug: 'secret-9' }])
+      prisma.$queryRaw.mockResolvedValue([{ slug: 'secret-9' }])
 
       const slug = await generateEntitySlug('Secret', 'SECRET', prisma)
       expect(slug).toBe('secret-a')
     })
 
     it('should generate a unique slug for INTEGRATION', async () => {
-      prisma.integration.findMany.mockResolvedValue([{ slug: 'integration-b' }])
+      prisma.$queryRaw.mockResolvedValue([{ slug: 'integration-b' }])
 
       const slug = await generateEntitySlug(
         'Integration',
@@ -93,9 +93,7 @@ describe('generateEntitySlug', () => {
     })
 
     it('should generate a unique slug for ENVIRONMENT', async () => {
-      prisma.environment.findMany.mockResolvedValue([
-        { slug: 'environment-zz' }
-      ])
+      prisma.$queryRaw.mockResolvedValue([{ slug: 'environment-zz' }])
 
       const slug = await generateEntitySlug(
         'Environment',
@@ -106,7 +104,7 @@ describe('generateEntitySlug', () => {
     })
 
     it('should generate a unique slug for API_KEY', async () => {
-      prisma.apiKey.findMany.mockResolvedValue([{ slug: 'api-key-09' }])
+      prisma.$queryRaw.mockResolvedValue([{ slug: 'api-key-09' }])
 
       const slug = await generateEntitySlug('Api @Key', 'API_KEY', prisma)
       expect(slug).toBe('api-key-0a')

diff --git a/apps/api/src/common/slug-generator.ts b/apps/api/src/common/slug-generator.ts
@@ -1,7 +1,7 @@
 import { PrismaService } from '@/prisma/prisma.service'
 import { Workspace } from '@prisma/client'
 
-const incrementSlugSuffix = (
+export const incrementSlugSuffix = (
   existingSlug: string,
   baseSlug: string
 ): string => {
@@ -71,7 +71,9 @@ const getWorkspaceRoleIfSlugExists = async (
   prisma: PrismaService
 ): Promise<string> => {
   const search = `${slug}-[a-z0-9]*`
-  const existingSlug = await prisma.$queryRaw<{ slug: string }[]>`
+  const existingSlug: { slug: string }[] = await prisma.$queryRaw<
+    { slug: string }[]
+  >`
     SELECT slug
     FROM "WorkspaceRole"
     WHERE slug ~ ${search}