Skip to content

Commit 3900eef

Browse files
committed
fix(drizzle): handle single-value input in encryptQuery test mock
The mock now checks for both array and non-array inputs, matching the real ProtectClient API overloads.
1 parent 5838202 commit 3900eef

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

packages/drizzle/__tests__/test-utils.ts

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,8 +6,10 @@ import { createProtectOperators } from '../src/pg'
66
export const ENCRYPTED_VALUE = '{"v":"encrypted-value"}'
77

88
export function createMockProtectClient() {
9-
const encryptQuery = vi.fn(async (terms: unknown[]) => ({
10-
data: terms.map(() => ENCRYPTED_VALUE),
9+
const encryptQuery = vi.fn(async (terms: unknown[] | unknown) => ({
10+
data: Array.isArray(terms)
11+
? terms.map(() => ENCRYPTED_VALUE)
12+
: [ENCRYPTED_VALUE],
1113
}))
1214

1315
return {

0 commit comments

Comments
 (0)