validateUniquess optional prismaClient parameter

packages/api/src/validations/__tests__/validations.test.js

@@ -1144,4 +1144,29 @@ describe('validateUniqueness', () => {
     }
     expect.assertions(1)
   })
+
+  it('uses the given prisma client', async () => {
+    const mockFindFirstOther = jest.fn()
+    const mockPrismaClient = {
+      $transaction: async (func) =>
+        func({
+          user: {
+            findFirst: mockFindFirstOther,
+          },
+        }),
+    }
+    mockFindFirstOther.mockImplementation(() => null)
+
+    expect(mockFindFirstOther).not.toBeCalled()
+    await validateUniqueness(
+      'user',
+      { email: '[email protected]' },
+      { prismaClient: mockPrismaClient },
+      () => {
+        expect(true).toEqual(true)
+      }
+    )
+    expect(mockFindFirstOther).toBeCalled()
+    expect(mockFindFirst).not.toBeCalled()
+  })
 })

packages/api/src/validations/validations.ts

@@ -143,7 +143,9 @@ interface PresenceValidatorOptions extends WithOptionalMessage {
   allowEmptyString?: boolean
 }
 
-interface UniquenessValidatorOptions extends WithRequiredMessage {}
+interface UniquenessValidatorOptions extends WithOptionalMessage {
+  prismaClient?: PrismaClient
+}
 type UniquenessWhere = Record<'AND' | 'NOT', Array<Record<string, unknown>>>
 
 interface ValidationRecipe {
@@ -632,10 +634,10 @@ export async function validateUniqueness(
     | ((tx: PrismaClient) => Promise<any>),
   callback?: (tx: PrismaClient) => Promise<any>
 ): Promise<any> {
-  const db = new PrismaClient()
   const { $self, $scope, ...rest } = fields
-  let options = {}
+  let options: UniquenessValidatorOptions = {}
   let validCallback: (tx: PrismaClient) => Promise<any>
+  let db = null
 
   if (typeof optionsOrCallback === 'function') {
     validCallback = optionsOrCallback
@@ -644,6 +646,14 @@ export async function validateUniqueness(
     validCallback = callback as (tx: PrismaClient) => Promise<any>
   }
 
+  if (options.prismaClient) {
+    const { prismaClient, ...restOptions } = options
+    options = restOptions
+    db = prismaClient
+  } else {
+    db = new PrismaClient()
+  }
+
   const where: UniquenessWhere = {
     AND: [rest],
     NOT: [],