Files
Janka UrygaandGitHub 7fc1a200cc fix: request APIs in promises passed to after() in actions/handlers (#94964)
Fixes a bug where Route Handlers and Server Actions were prevented from
calling `headers()` (and other request APIs) if
- the call happened after the response is finished
- the call *did not happen* inside an after-callback
generally, this happens when you pass a promise to after.
```ts
const longRunning = () => {
  // assume this resolves after the response
  await new Promise((resolve) => setTimeout(resolve, 100))
  // response is done
  after(() => {
    await headers() // ❌ should be allowed, but throws
  })
}

after(longRunning()) // or waitUntil, doesn't matter
```

The logic we used in the check was incorrect and assumed that an
`afterTaskStore` was present, but we can only wrap callbacks in this ALS
- we cannot change the async context of already-running promises.

One noteworthy change is that i had to exit `actionAsyncStorage` when we
do a render after the server action. Otherwise the checks in
`isRequestApiAllowedInCurrentPhase` might believe that they're still in
a server action when they aren't.
2026-06-19 19:23:06 +00:00

440 lines
16 KiB
TypeScript

/* eslint-env jest */
import { nextTestSetup } from 'e2e-utils'
import { retry } from 'next-test-utils'
import { REQUEST_API_NAMES } from './app/request-apis-in-promise/common'
describe('nextjs APIs in after()', () => {
const { next, skipped, isNextDev } = nextTestSetup({
files: __dirname,
skipStart: true,
skipDeployment: true, // reading runtime logs is not supported in deploy tests
})
if (skipped) return
let currentCliOutputIndex = 0
const ignorePreviousLogs = () => {
currentCliOutputIndex = next.cliOutput.length
}
const getLogs = () => {
return next.cliOutput.slice(currentCliOutputIndex)
}
beforeEach(() => {
ignorePreviousLogs()
})
let buildLogs: string
beforeAll(async () => {
if (!isNextDev) {
await next.build()
buildLogs = next.cliOutput
} else {
buildLogs = '(no build logs in dev)'
}
await next.start({ skipBuild: true })
})
describe('request APIs inside after()', () => {
it('cannot be called in a dynamic page', async () => {
const path = '/request-apis/page-dynamic'
await next.render(path)
await retry(() => {
const logs = getLogs()
expect(logs).not.toContain(`[${path}] headers(): ok`)
expect(logs).toContain(
`[${path}] headers(): error: Error: Route ${path} used \`headers()\` inside \`after()\` while rendering. This is not supported.`
)
expect(logs).not.toContain(`[${path}] cookies(): ok`)
expect(logs).toContain(
`[${path}] cookies(): error: Error: Route ${path} used \`cookies()\` inside \`after()\` while rendering. This is not supported.`
)
expect(logs).not.toContain(`[${path}] connection(): ok`)
expect(logs).toContain(
`[${path}] connection(): error: Error: Route ${path} used \`connection()\` inside \`after()\` while rendering.`
)
})
await retry(() => {
const logs = getLogs()
expect(logs).not.toContain(`[${path}] nested headers(): ok`)
expect(logs).toContain(
`[${path}] nested headers(): error: Error: Route ${path} used \`headers()\` inside \`after()\` while rendering. This is not supported.`
)
expect(logs).not.toContain(`[${path}] nested cookies(): ok`)
expect(logs).toContain(
`[${path}] nested cookies(): error: Error: Route ${path} used \`cookies()\` inside \`after()\` while rendering. This is not supported.`
)
expect(logs).not.toContain(`[${path}] nested connection(): ok`)
expect(logs).toContain(
`[${path}] nested connection(): error: Error: Route ${path} used \`connection()\` inside \`after()\` while rendering.`
)
})
})
describe('cannot be called in a prerendered page', () => {
it.each([
{
title: 'with `dynamic = "error"`',
path: '/request-apis/page-dynamic-error',
},
{
title: 'with `dynamic = "force-static"`',
path: '/request-apis/page-force-static',
},
])('$title', async ({ path }) => {
await next.render(path)
await retry(() => {
const logs = isNextDev ? getLogs() : buildLogs // in `next start` the error was logged at build time
expect(logs).not.toContain(`[${path}] headers(): ok`)
expect(logs).toContain(
`[${path}] headers(): error: Error: Route ${path} used \`headers()\` inside \`after()\` while rendering. This is not supported.`
)
expect(logs).not.toContain(`[${path}] cookies(): ok`)
expect(logs).toContain(
`[${path}] cookies(): error: Error: Route ${path} used \`cookies()\` inside \`after()\` while rendering. This is not supported.`
)
expect(logs).not.toContain(`[${path}] connection(): ok`)
expect(logs).toContain(
`[${path}] connection(): error: Error: Route ${path} used \`connection()\` inside \`after()\` while rendering.`
)
})
await retry(() => {
const logs = isNextDev ? getLogs() : buildLogs // in `next start` the error was logged at build time
expect(logs).not.toContain(`[${path}] nested headers(): ok`)
expect(logs).toContain(
`[${path}] nested headers(): error: Error: Route ${path} used \`headers()\` inside \`after()\` while rendering. This is not supported.`
)
expect(logs).not.toContain(`[${path}] nested cookies(): ok`)
expect(logs).toContain(
`[${path}] nested cookies(): error: Error: Route ${path} used \`cookies()\` inside \`after()\` while rendering. This is not supported.`
)
expect(logs).not.toContain(`[${path}] nested connection(): ok`)
expect(logs).toContain(
`[${path}] nested connection(): error: Error: Route ${path} used \`connection()\` inside \`after()\` while rendering.`
)
})
})
})
it('can be called in a server action', async () => {
const path = '/request-apis/server-action'
const browser = await next.browser(path)
await browser.elementByCss('button[type="submit"]').click()
await retry(() => {
const logs = getLogs()
expect(logs).toContain(`[${path}] headers(): ok`)
expect(logs).toContain(`[${path}] nested headers(): ok`)
expect(logs).toContain(`[${path}] cookies(): ok`)
expect(logs).toContain(`[${path}] nested cookies(): ok`)
expect(logs).toContain(`[${path}] connection(): ok`)
expect(logs).toContain(`[${path}] nested connection(): ok`)
})
})
it('can be called in a dynamic route handler', async () => {
const path = '/request-apis/route-handler-dynamic'
await next.render(path)
await retry(() => {
const logs = getLogs()
expect(logs).toContain(`[${path}] headers(): ok`)
expect(logs).toContain(`[${path}] nested headers(): ok`)
expect(logs).toContain(`[${path}] cookies(): ok`)
expect(logs).toContain(`[${path}] nested cookies(): ok`)
expect(logs).toContain(`[${path}] connection(): ok`)
expect(logs).toContain(`[${path}] nested connection(): ok`)
})
})
it('can be called in a prerendered route handler with `dynamic = "force-static"`', async () => {
const path = '/request-apis/route-handler-force-static'
await next.render(path)
await retry(() => {
const logs = isNextDev ? getLogs() : buildLogs // in `next start` the error was logged at build time
expect(logs).toContain(`[${path}] headers(): ok`)
expect(logs).toContain(`[${path}] nested headers(): ok`)
expect(logs).toContain(`[${path}] cookies(): ok`)
expect(logs).toContain(`[${path}] nested cookies(): ok`)
expect(logs).toContain(`[${path}] connection(): ok`)
expect(logs).toContain(`[${path}] nested connection(): ok`)
})
})
it('can be called in a prerendered route handler with `dynamic = "error" (but throw, because dynamic should error)`', async () => {
const path = '/request-apis/route-handler-dynamic-error'
await next.render(path)
await retry(() => {
const logs = isNextDev ? getLogs() : buildLogs // in `next start` the error was logged at build time
expect(logs).not.toContain(`[${path}] headers(): ok`)
expect(logs).toContain(
`[${path}] headers(): error: Error: Route ${path} with \`dynamic = "error"\` couldn't be rendered statically because it used \`headers()\`.`
)
expect(logs).not.toContain(`[${path}] nested headers(): ok`)
expect(logs).toContain(
`[${path}] nested headers(): error: Error: Route ${path} with \`dynamic = "error"\` couldn't be rendered statically because it used \`headers()\`.`
)
expect(logs).not.toContain(`[${path}] cookies(): ok`)
expect(logs).toContain(
`[${path}] cookies(): error: Error: Route ${path} with \`dynamic = "error"\` couldn't be rendered statically because it used \`cookies()\`.`
)
expect(logs).not.toContain(`[${path}] nested cookies(): ok`)
expect(logs).toContain(
`[${path}] nested cookies(): error: Error: Route ${path} with \`dynamic = "error"\` couldn't be rendered statically because it used \`cookies()\`.`
)
expect(logs).not.toContain(`[${path}] connection(): ok`)
expect(logs).toContain(
`[${path}] connection(): error: Error: Route ${path} with \`dynamic = "error"\` couldn't be rendered statically because it used \`connection()\`.`
)
expect(logs).not.toContain(`[${path}] nested connection(): ok`)
expect(logs).toContain(
`[${path}] nested connection(): error: Error: Route ${path} with \`dynamic = "error"\` couldn't be rendered statically because it used \`connection()\`.`
)
})
})
})
describe('draftMode status is readable, but cannot be changed', () => {
it.each([
{
title: 'dynamic page',
path: '/draft-mode/page-dynamic',
isDynamic: true,
},
{
title: 'static page',
path: '/draft-mode/page-static',
isDynamic: false,
},
{
title: 'dynamic route handler',
path: '/draft-mode/route-handler-dynamic',
isDynamic: true,
},
{
title: 'static route handler',
path: '/draft-mode/route-handler-static',
isDynamic: false,
},
])('$title', async ({ path, isDynamic }) => {
await next.render(path)
await retry(() => {
// in `next start`, static routes log the error at build time
const logs = isDynamic || isNextDev ? getLogs() : buildLogs
expect(logs).toContain(`[${path}] draft.isEnabled: false`)
expect(logs).toContain(
`Route ${path} used "draftMode().enable()" inside \`after()\``
)
expect(logs).toContain(
`Route ${path} used "draftMode().disable()" inside \`after()\``
)
})
})
it('server action', async () => {
const path = '/draft-mode/server-action'
const browser = await next.browser(path)
await browser.elementByCss('button[type="submit"]').click()
await retry(() => {
const logs = getLogs()
expect(logs).toContain(`[${path}] draft.isEnabled: false`)
expect(logs).toContain(
`Route ${path} used "draftMode().enable()" inside \`after()\``
)
expect(logs).toContain(
`Route ${path} used "draftMode().disable()" inside \`after()\``
)
})
})
})
describe('async APIs in promises passed to after', () => {
describe('in route handlers', () => {
it.each(REQUEST_API_NAMES)(
'does not error - %s',
async (apiName: string) => {
const cliOutputIndex = next.cliOutput.length
const getCliOutput = () => next.cliOutput.slice(cliOutputIndex)
await next.fetch(
`/request-apis-in-promise/route-handler?api=${apiName}`
)
const cliOutput = await retry(() => {
const cliOutput = getCliOutput()
expect(cliOutput).toContain(`route :: ${apiName} :: finished`)
return cliOutput
})
expect(cliOutput).toContain(`route :: ${apiName} :: promise :: ok`)
expect(cliOutput).not.toContain(
`route :: ${apiName} :: promise :: error`
)
expect(cliOutput).toContain(
`route :: ${apiName} :: nested after :: ok`
)
expect(cliOutput).not.toContain(
`route :: ${apiName} :: nested after :: error`
)
}
)
})
describe('in server actions', () => {
it.each(REQUEST_API_NAMES)(
'does not error - %s',
async (apiName: string) => {
const browser = await next.browser(
`/request-apis-in-promise/server-action`
)
const cliOutputIndex = next.cliOutput.length
const getCliOutput = () => next.cliOutput.slice(cliOutputIndex)
await browser
.elementByCss(
`form[data-api-name="${apiName}"] button[type="submit"]`
)
.click()
const cliOutput = await retry(() => {
const cliOutput = getCliOutput()
expect(cliOutput).toContain(`action :: ${apiName} :: finished`)
return cliOutput
})
expect(cliOutput).toContain(`action :: ${apiName} :: promise :: ok`)
expect(cliOutput).not.toContain(
`action :: ${apiName} :: promise :: error`
)
expect(cliOutput).toContain(
`action :: ${apiName} :: nested after :: ok`
)
expect(cliOutput).not.toContain(
`action :: ${apiName} :: nested after :: error`
)
}
)
})
describe('in renders', () => {
it.each(REQUEST_API_NAMES)(
'throws an error - %s',
async (apiName: string) => {
const route = `/request-apis-in-promise/render`
const path = `${route}?apiName=${apiName}`
const cliOutputIndex = next.cliOutput.length
const getCliOutput = () => next.cliOutput.slice(cliOutputIndex)
await next.fetch(path)
const cliOutput = await retry(() => {
const cliOutput = getCliOutput()
expect(cliOutput).toContain(`render :: ${apiName} :: finished`)
return cliOutput
})
const message = `Error: Route ${route} used \`${apiName}()\` inside \`after()\` while rendering.`
expect(cliOutput).toContain(
`render :: ${apiName} :: promise :: error: ${message}`
)
expect(cliOutput).not.toContain(
`render :: ${apiName} :: promise :: ok`
)
expect(cliOutput).toContain(
`render :: ${apiName} :: nested after :: error: ${message}`
)
expect(cliOutput).not.toContain(
`render :: ${apiName} :: nested after :: ok`
)
}
)
})
describe('in renders after server actions', () => {
it.each(REQUEST_API_NAMES)(
'throws an error - %s',
async (apiName: string) => {
const route = '/request-apis-in-promise/render-after-action'
const initialCliOutputIndex = next.cliOutput.length
const browser = await next.browser(route)
// Clear cookies after every run
await using _ = defer(() => browser.deleteCookies())
// Sanity check: we should not be running any of the APIs now,
// only when a server action causes a rerender
expect(next.cliOutput.slice(initialCliOutputIndex)).not.toContain(
`render after action :: ${apiName} :: starting`
)
const cliOutputIndex = next.cliOutput.length
const getCliOutput = () => next.cliOutput.slice(cliOutputIndex)
await browser
.elementByCss(
`form[data-api-name="${apiName}"] button[type="submit"]`
)
.click()
const cliOutput = await retry(() => {
const cliOutput = getCliOutput()
expect(cliOutput).toContain(
`render after action :: ${apiName} :: finished`
)
return cliOutput
})
const message = `Error: Route ${route} used \`${apiName}()\` inside \`after()\` while rendering.`
expect(cliOutput).toContain(
`render after action :: ${apiName} :: promise :: error: ${message}`
)
expect(cliOutput).not.toContain(
`render after action :: ${apiName} :: promise :: ok`
)
expect(cliOutput).toContain(
`render after action :: ${apiName} :: nested after :: error: ${message}`
)
expect(cliOutput).not.toContain(
`render after action :: ${apiName} :: nested after :: ok`
)
}
)
})
})
})
function defer(callback: () => void | Promise<void>): AsyncDisposable {
return {
[Symbol.asyncDispose]: async () => {
return callback()
},
}
}