wpt: add redirect-to-dataurl.any.js

Author: KhafraDev

test/wpt/server/routes/redirect.mjs

@@ -0,0 +1,102 @@
+import { setTimeout } from 'timers/promises'
+
+const stash = new Map()
+
+/**
+ * @see https://github.com/web-platform-tests/wpt/blob/master/fetch/connection-pool/resources/network-partition-key.py
+ * @param {Parameters<import('http').RequestListener>[0]} req
+ * @param {Parameters<import('http').RequestListener>[1]} res
+ * @param {URL} url
+ */
+export async function route (req, res, { searchParams }) {
+  let stashedData = { count: 0, preflight: 0 }
+  let status = 302
+  res.setHeader('Content-Type', 'text/plain')
+  res.setHeader('Cache-Control', 'no-cache')
+  res.setHeader('Pragma', 'no-cache')
+
+  if (Object.hasOwn(req.headers, 'origin')) {
+    res.setHeader('Access-Control-Allow-Origin', req.headers.origin ?? '')
+    res.setHeader('Access-Control-Allow-Credentials', 'true')
+  } else {
+    res.setHeader('Access-Control-Allow-Origin', '*')
+  }
+
+  let token = null
+  if (searchParams.has('token')) {
+    token = searchParams.get('token')
+    const data = stash.get(token)
+    stash.delete(token)
+    if (data) {
+      stashedData = data
+    }
+  }
+
+  if (req.method === 'OPTIONS') {
+    if (searchParams.has('allow_headers')) {
+      res.setHeader('Access-Control-Allow-Headers', searchParams.get('allow_headers'))
+    }
+
+    stashedData.preflight = '1'
+
+    if (!searchParams.has('redirect_preflight')) {
+      if (token) {
+        stash.set(searchParams.get('token'), stashedData)
+      }
+
+      res.statusCode = 200
+      res.end('')
+      return
+    }
+  }
+
+  if (searchParams.has('redirect_status')) {
+    status = parseInt(searchParams.get('redirect_status'))
+  }
+
+  stashedData.count += 1
+
+  if (searchParams.has('location')) {
+    let url = decodeURIComponent(searchParams.get('location'))
+
+    if (!searchParams.has('simple')) {
+      const scheme = new URL(url).protocol
+
+      if (scheme === 'http:' || scheme === 'https:') {
+        url += url.includes('?') ? '&' : '?'
+
+        for (const [key, value] of searchParams) {
+          url += '&' + encodeURIComponent(key) + '=' + encodeURIComponent(value)
+        }
+
+        url += '&count=' + stashedData.count
+      }
+    }
+
+    res.setHeader('location', url)
+  }
+
+  if (searchParams.has('redirect_referrerpolicy')) {
+    res.setHeader('Referrer-Policy', searchParams.get('redirect_referrerpolicy'))
+  }
+
+  if (searchParams.has('delay')) {
+    await setTimeout(parseFloat(searchParams.get('delay') ?? 0))
+  }
+
+  if (token) {
+    stash.set(searchParams.get('token'), stashedData)
+
+    if (searchParams.has('max_count')) {
+      const maxCount = parseInt(searchParams.get('max_count'))
+
+      if (stashedData.count > maxCount) {
+        res.end((stashedData.count - 1).toString())
+        return
+      }
+    }
+  }
+
+  res.statusCode = status
+  res.end('')
+}

test/wpt/server/server.mjs

@@ -5,7 +5,8 @@ import process from 'node:process'
 import { fileURLToPath } from 'node:url'
 import { createReadStream, readFileSync } from 'node:fs'
 import { setTimeout as sleep } from 'node:timers/promises'
-import { route } from './routes/network-partition-key.mjs'
+import { route as networkPartitionRoute } from './routes/network-partition-key.mjs'
+import { route as redirectRoute } from './routes/redirect.mjs'
 
 const tests = fileURLToPath(join(import.meta.url, '../../tests'))
 
@@ -221,13 +222,16 @@ const server = createServer(async (req, res) => {
       break
     }
     case '/fetch/connection-pool/resources/network-partition-key.py': {
-      return route(req, res, fullUrl)
+      return networkPartitionRoute(req, res, fullUrl)
     }
     case '/resources/top.txt': {
       return createReadStream(join(tests, 'fetch/api/', fullUrl.pathname))
         .on('end', () => res.end())
         .pipe(res)
     }
+    case '/fetch/api/resources/redirect.py': {
+      return redirectRoute(req, res, fullUrl)
+    }
     default: {
       res.statusCode = 200
       res.end('body')

test/wpt/tests/fetch/api/redirect/redirect-to-dataurl.any.js

@@ -0,0 +1,28 @@
+// META: global=window,worker
+// META: script=/common/get-host-info.sub.js
+
+var dataURL = "data:text/plain;base64,cmVzcG9uc2UncyBib2R5";
+var body = "response's body";
+var contentType = "text/plain";
+
+function redirectDataURL(desc, redirectUrl, mode) {
+    var url = redirectUrl +  "?cors&location=" + encodeURIComponent(dataURL);
+
+    var requestInit = {"mode": mode};
+
+    promise_test(function(test) {
+        return promise_rejects_js(test, TypeError, fetch(url, requestInit));
+    }, desc);
+}
+
+var redirUrl = get_host_info().HTTP_ORIGIN + "/fetch/api/resources/redirect.py";
+var corsRedirUrl = get_host_info().HTTP_REMOTE_ORIGIN + "/fetch/api/resources/redirect.py";
+
+redirectDataURL("Testing data URL loading after same-origin redirection (cors mode)", redirUrl, "cors");
+redirectDataURL("Testing data URL loading after same-origin redirection (no-cors mode)", redirUrl, "no-cors");
+redirectDataURL("Testing data URL loading after same-origin redirection (same-origin mode)", redirUrl, "same-origin");
+
+redirectDataURL("Testing data URL loading after cross-origin redirection (cors mode)", corsRedirUrl, "cors");
+redirectDataURL("Testing data URL loading after cross-origin redirection (no-cors mode)", corsRedirUrl, "no-cors");
+
+done();