diff --git a/tests/e2e/edge-middleware.test.ts b/tests/e2e/edge-middleware.test.ts index e78333c81d..7bc2207467 100644 --- a/tests/e2e/edge-middleware.test.ts +++ b/tests/e2e/edge-middleware.test.ts @@ -373,7 +373,11 @@ test.describe('RSC cache poisoning', () => { test('Middleware rewrite', async ({ page, middleware }) => { const prefetchResponsePromise = new Promise((resolve) => { page.on('response', (response) => { - if (response.url().includes('/test/rewrite-to-cached-page')) { + if ( + (response.url().includes('/test/rewrite-to-cached-page') || + response.url().includes('/caching-rewrite-target')) && + response.status() === 200 + ) { resolve(response) } }) @@ -400,7 +404,7 @@ test.describe('RSC cache poisoning', () => { test('Middleware redirect', async ({ page, middleware }) => { const prefetchResponsePromise = new Promise((resolve) => { page.on('response', (response) => { - if (response.url().includes('/caching-redirect-target')) { + if (response.url().includes('/caching-redirect-target') && response.status() === 200) { resolve(response) } })