Skip to content

Commit ccb1f69

Browse files
Align mock type with actual usage (sync | async) for next/headers.headers
1 parent 3d04140 commit ccb1f69

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

packages/nextjs/src/__tests__/keyless-custom-headers.test.ts

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,8 @@ vi.mock('next/headers', () => ({
4545
})),
4646
}));
4747

48-
const mockHeaders = headers as unknown as MockedFunction<() => Promise<MockHeaders>>;
48+
type MockHeadersFn = () => MockHeaders | Promise<MockHeaders>;
49+
const mockHeaders = headers as unknown as MockedFunction<MockHeadersFn>;
4950

5051
// Type for mocking Next.js headers
5152
interface MockHeaders {
@@ -66,7 +67,7 @@ function createMockHeaders(customHeaders: Record<string, string | null> = {}): M
6667
const allHeaders = { ...defaultHeadersObj, ...customHeaders };
6768

6869
return {
69-
get: vi.fn((name: string) => allHeaders[name] || null),
70+
get: vi.fn((name: string) => allHeaders[name] ?? null),
7071
has: vi.fn((name: string) => Object.prototype.hasOwnProperty.call(allHeaders, name) && allHeaders[name] !== null),
7172
forEach: vi.fn((callback: (value: string, key: string) => void) => {
7273
Object.entries(allHeaders).forEach(([key, value]) => {

0 commit comments

Comments
 (0)