Skip to content

Commit 6ae0893

Browse files
test: enhance mocks for vite and fs in prerender SSRF tests
1 parent 3719b02 commit 6ae0893

1 file changed

Lines changed: 26 additions & 1 deletion

File tree

packages/start-plugin-core/tests/prerender-ssrf.test.ts

Lines changed: 26 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,27 @@ vi.mock('../src/prerender', async () => {
1717
}
1818
})
1919

20+
// Mock vite to prevent actual server from starting
21+
vi.mock('vite', () => ({
22+
preview: vi.fn().mockResolvedValue({
23+
resolvedUrls: { local: ['http://localhost:5173/'] },
24+
close: vi.fn().mockResolvedValue(undefined),
25+
}),
26+
}))
27+
28+
// Mock fs to prevent actual file system operations
29+
vi.mock('node:fs', async () => {
30+
const actual = await vi.importActual<any>('node:fs')
31+
return {
32+
...actual,
33+
promises: {
34+
...actual.promises,
35+
mkdir: vi.fn().mockResolvedValue(undefined),
36+
writeFile: vi.fn().mockResolvedValue(undefined),
37+
},
38+
}
39+
})
40+
2041
const builder = {
2142
environments: {
2243
[VITE_ENVIRONMENT_NAMES.client]: {
@@ -29,7 +50,11 @@ const builder = {
2950
} as any
3051

3152
const fetchMock = vi.fn(
32-
async () => new Response('<html></html>', { status: 200 }),
53+
async () =>
54+
new Response('<html></html>', {
55+
status: 200,
56+
headers: { 'content-type': 'text/html' },
57+
}),
3358
)
3459

3560
vi.stubGlobal('fetch', fetchMock)

0 commit comments

Comments
 (0)