From 58f536d48acaf57576f067af25a8708696232745 Mon Sep 17 00:00:00 2001
From: Hiroki Nakagawa <nhiroki@chromium.org>
Date: Sun, 17 Oct 2021 21:47:00 -0700
Subject: [PATCH] Prerender: Upstream tests for storage APIs in same-origin
 prerendered pages (2)

This CL upstreams tests for storage APIs (Storage Foundation,
Web Database) in same-origin prerendered pages to the WPT. The behavior
of these APIs is not defined in the spec yet, but we already reached a
consensus that the storage APIs in same-origin prerendered pages are
just allowed.

See the GitHub issue for details:
https://github.com/jeremyroman/alternate-loading-modes/issues/7#issuecomment-944358336

Bug: 1253158
Change-Id: If34e4f325341a95f036beff43af19dde3504a8ac
Reviewed-on: https://chromium-review.googlesource.com/c/chromium/src/+/3226531
Reviewed-by: Kouhei Ueno <kouhei@chromium.org>
Reviewed-by: Lingqi Chi <lingqi@chromium.org>
Commit-Queue: Hiroki Nakagawa <nhiroki@chromium.org>
Cr-Commit-Position: refs/heads/main@{#932416}
---
 .../storage-foundation-access.https.html      | 42 +++++++++++
 .../resources/web-database-access.html        | 21 ++++++
 .../prerender/storage-foundation.https.html   | 67 ++++++++++++++++++
 speculation-rules/prerender/web-database.html | 70 +++++++++++++++++++
 4 files changed, 200 insertions(+)
 create mode 100644 speculation-rules/prerender/resources/storage-foundation-access.https.html
 create mode 100644 speculation-rules/prerender/resources/web-database-access.html
 create mode 100644 speculation-rules/prerender/storage-foundation.https.html
 create mode 100644 speculation-rules/prerender/web-database.html
diff --git a/speculation-rules/prerender/resources/storage-foundation-access.https.html b/speculation-rules/prerender/resources/storage-foundation-access.https.html
new file mode 100644
index 00000000000000..4ab8b062c93f84
--- /dev/null
+++ b/speculation-rules/prerender/resources/storage-foundation-access.https.html
@@ -0,0 +1,42 @@
+<!DOCTYPE html>
+<script src="/resources/testharness.js"></script>
+<script src="/resources/testharnessreport.js"></script>
+<script>
+
+async function readFromFile(f) {
+  const {buffer, readBytes} = await f.read(new Uint8Array(4), 0);
+  return buffer;
+}
+
+async function writeToFile(f) {
+  const writeBuffer = new Uint8Array(4);
+  writeBuffer.set([75, 76, 77, 78]);
+
+  await storageFoundation.requestCapacity(4);
+
+  const {writtenBytes} = await f.write(writeBuffer, 0);
+  await f.flush();
+  return writtenBytes == 4;
+}
+
+async function readWriteTest() {
+  const bc = new BroadcastChannel('prerender-channel');
+  assert_true(document.prerendering);
+
+  let f = await storageFoundation.open('test_file');
+  const readBuffer = await readFromFile(f);
+  const writeResult = await writeToFile(f);
+  assert_true(writeResult);
+
+  const writtenBuffer = await readFromFile(f);
+  await f.close();
+
+  bc.postMessage({
+    readBuffer : readBuffer,
+    writtenBuffer : writtenBuffer
+  });
+  bc.close();
+}
+
+readWriteTest();
+</script>
diff --git a/speculation-rules/prerender/resources/web-database-access.html b/speculation-rules/prerender/resources/web-database-access.html
new file mode 100644
index 00000000000000..99741906ed3b70
--- /dev/null
+++ b/speculation-rules/prerender/resources/web-database-access.html
@@ -0,0 +1,21 @@
+<!DOCTYPE html>
+<script src="/resources/testharness.js"></script>
+<script src="/resources/testharnessreport.js"></script>
+<script>
+
+const bc = new BroadcastChannel('prerender-channel');
+assert_true(document.prerendering);
+
+let result = "Success";
+const db = openDatabase("test", "1.0", "test database", 1024);
+db.transaction(function (tx) {
+  tx.executeSql('CREATE TABLE IF NOT EXISTS foo (text)');
+  tx.executeSql('INSERT INTO foo (text) VALUES ("bar")');
+}, function(error) {
+  result = error;
+}, function() {
+  bc.postMessage(result);
+  bc.close();
+});
+
+</script>
diff --git a/speculation-rules/prerender/storage-foundation.https.html b/speculation-rules/prerender/storage-foundation.https.html
new file mode 100644
index 00000000000000..ca1b9363f4b243
--- /dev/null
+++ b/speculation-rules/prerender/storage-foundation.https.html
@@ -0,0 +1,67 @@
+<!DOCTYPE html>
+<title>Same-origin prerendering can access Storage Foundation API</title>
+<meta name="timeout" content="long">
+<script src="/resources/testharness.js"></script>
+<script src="/resources/testharnessreport.js"></script>
+<script src="resources/utils.js"></script>
+<body>
+<script>
+
+setup(() => assertSpeculationRulesIsSupported());
+
+const expectedReadBuffer = [65, 66, 67, 68];
+
+async function writeToFile() {
+  let f = await storageFoundation.open('test_file');
+  const writeBuffer = new Uint8Array(4);
+  writeBuffer.set(expectedReadBuffer);
+
+  await storageFoundation.requestCapacity(4);
+
+  const {writtenBytes} = await f.write(writeBuffer, 0);
+  await f.flush();
+  await f.close();
+  return writtenBytes == 4;
+}
+
+async function readFromFile() {
+  let f = await storageFoundation.open('test_file');
+  const {buffer, readBytes} = await f.read(new Uint8Array(4), 0);
+  await f.close();
+  return buffer;
+}
+
+promise_test(async t => {
+  const bc = new BroadcastChannel('prerender-channel');
+
+  const gotMessage = new Promise(resolve => {
+    bc.addEventListener('message', e => {
+      resolve(e.data);
+    }, {
+      once: true
+    });
+  });
+
+  const writeResult = await writeToFile();
+  assert_true(
+    writeResult,
+    'primary page should be able to write to storage foundation API.');
+
+  // Start prerendering a page that attempts to access storage foundation API.
+  startPrerendering(`resources/storage-foundation-access.https.html`);
+  const result = await gotMessage;
+
+  assert_array_equals(
+    result.readBuffer, expectedReadBuffer,
+    'prerendering page should be able to read from storage foundation API');
+
+  const readBufferWrittenByPrerender = await readFromFile();
+  assert_array_equals(
+    result.writtenBuffer, readBufferWrittenByPrerender,
+    'prerendering page should be able to write to storage foundation API');
+
+  await storageFoundation.delete('test_file');
+}, 'prerendering page should be able to access storage foundation API');
+
+</script>
+</body>
diff --git a/speculation-rules/prerender/web-database.html b/speculation-rules/prerender/web-database.html
new file mode 100644
index 00000000000000..3e59c3c2959777
--- /dev/null
+++ b/speculation-rules/prerender/web-database.html
@@ -0,0 +1,70 @@
+<!DOCTYPE html>
+<title>Same-origin prerendering can access Web Database</title>
+<meta name="timeout" content="long">
+<script src="/resources/testharness.js"></script>
+<script src="/resources/testharnessreport.js"></script>
+<script src="resources/utils.js"></script>
+<body>
+<script>
+
+setup(() => assertSpeculationRulesIsSupported());
+
+async function insertQuery() {
+  return new Promise(resolve => {
+    const db = openDatabase("test", "1.0", "test database", 1024);
+    db.transaction(function (tx) {
+      tx.executeSql('CREATE TABLE IF NOT EXISTS foo (text)');
+      tx.executeSql('INSERT INTO foo (text) VALUES ("bar")');
+    }, function(error) {
+      resolve(error);
+    }, function() {
+      resolve("Success");
+    });
+  });
+}
+
+async function selectQuery() {
+  return new Promise(resolve => {
+    const db = openDatabase("test", "1.0", "test database", 1024);
+    db.transaction(function (tx) {
+        tx.executeSql('CREATE TABLE IF NOT EXISTS foo (text)');
+        tx.executeSql('SELECT * FROM foo', [], function (tx, results) {
+          resolve(results.rows.length);
+        });
+      }, function(tx, error) {
+        resolve(error);
+      });
+  });
+}
+
+promise_test(async t => {
+  const bc = new BroadcastChannel('prerender-channel');
+
+  const gotMessage = new Promise(resolve => {
+    bc.addEventListener('message', e => {
+      resolve(e.data);
+    }, {
+      once: true
+    });
+  });
+
+  const insertResult = await insertQuery();
+  assert_equals(insertResult, "Success",
+        'primary page should be able to execute statements from Web Database.');
+
+  // Start prerendering a page that attempts to access Web Database.
+  startPrerendering('resources/web-database-access.html');
+  const result = await gotMessage;
+
+  assert_equals(
+    result, "Success",
+    'prerendering page should be able to read from Web Database');
+
+  const selectResult = await selectQuery();
+  assert_equals(
+    selectResult, 2,
+    'prerendering page should be able to write to Web Database');
+}, 'prerendering page should be able to access Web Database');
+
+</script>
+</body>
