Update navigation.entries() and navigation.activation on prerender activation

We currently have plumbing for updating entries() on bfcache restore.
This reuses that plumbing for prerender activation.

This also ensures that we set `navigation.activation.from` correctly
in a prerendered document before activation (it should be the entry
that initiated the preload).

Bug: 1212819
Change-Id: I5835313b0d713e10802b3df442bb16e4d813f464
Reviewed-on: https://chromium-review.googlesource.com/c/chromium/src/+/5013956
Reviewed-by: Rakina Zata Amni <rakina@chromium.org>
Reviewed-by: Nasko Oskov <nasko@chromium.org>
Reviewed-by: Domenic Denicola <domenic@chromium.org>
Commit-Queue: Nate Chapin <japhet@chromium.org>
Cr-Commit-Position: refs/heads/main@{#1226462}
diff --git a/speculation-rules/prerender/navigation-api-location-replace.html b/speculation-rules/prerender/navigation-api-location-replace.html
new file mode 100644
index 0000000..e3ecf1b
--- /dev/null
+++ b/speculation-rules/prerender/navigation-api-location-replace.html
@@ -0,0 +1,61 @@
+<!DOCTYPE html>
+<meta name="timeout" content="long">
+<script src="/resources/testharness.js"></script>
+<script src="/resources/testharnessreport.js"></script>
+<script src="/common/utils.js"></script>
+<script src="/common/dispatcher/dispatcher.js"></script>
+<script src="/html/browsers/browsing-the-web/remote-context-helper/resources/remote-context-helper.js"></script>
+<script src="../resources/utils.js"></script>
+<script src="resources/utils.js"></script>
+
+<body>
+<script>
+setup(() => assertSpeculationRulesIsSupported());
+
+promise_test(async t => {
+  const rcHelper = new RemoteContextHelper();
+  const referrerRC = await rcHelper.addWindow(undefined, { features: 'noopener' });
+  assert_equals(await referrerRC.executeScript(() => navigation.entries().length), 1);
+  let referrerRCCurrentId = await referrerRC.executeScript(() => navigation.currentEntry.id);
+
+  const prerenderedRC = await addPrerenderRC(referrerRC);
+  let activationStateBeforeActivation = await prerenderedRC.executeScript(() => {
+    return {
+     entries: navigation.entries().map(e => ({ id: e.id, })),
+     activationEntryId: navigation.activation.entry?.id,
+     activationFromId: navigation.activation.from?.id,
+     activationNavigationType : navigation.activation.navigationType,
+    }
+  });
+  assert_equals(activationStateBeforeActivation.entries.length, 1);
+  assert_equals(activationStateBeforeActivation.activationFromId, referrerRCCurrentId);
+  assert_equals(activationStateBeforeActivation.activationEntryId, activationStateBeforeActivation.entries[0].id);
+  assert_equals(activationStateBeforeActivation.activationNavigationType, "push");
+
+  // Save the current entry before activation.
+  await prerenderedRC.executeScript(() => window.currentEntryBeforeActivation = navigation.currentEntry);
+
+  await activatePrerenderRC(referrerRC, prerenderedRC, url => {
+    location.replace(url);
+  });
+
+  let activationStateAfterActivation = await prerenderedRC.executeScript(() => {
+    return {
+     entries: navigation.entries().map(e => ({ id: e.id, })),
+     activationEntryId: navigation.activation.entry?.id,
+     activationFromId: navigation.activation.from?.id,
+     activationNavigationType : navigation.activation.navigationType,
+    }
+  });
+  assert_equals(activationStateAfterActivation.entries.length, 1);
+  assert_equals(activationStateAfterActivation.activationFromId, referrerRCCurrentId);
+  assert_equals(activationStateAfterActivation.activationEntryId, activationStateAfterActivation.entries[0].id);
+  assert_equals(activationStateAfterActivation.activationNavigationType, "replace");
+
+  let currentEntryIdentity = await prerenderedRC.executeScript(() => {
+    return window.currentEntryBeforeActivation === navigation.currentEntry &&
+           navigation.currentEntry === navigation.entries()[navigation.entries().length - 1];
+  });
+  assert_true(currentEntryIdentity);
+},`navigation.entries() and navigation.activation should be updated on activation and handle replacing correctly`);
+</script>
diff --git a/speculation-rules/prerender/navigation-api-multiple-entries.html b/speculation-rules/prerender/navigation-api-multiple-entries.html
new file mode 100644
index 0000000..e8e0f87
--- /dev/null
+++ b/speculation-rules/prerender/navigation-api-multiple-entries.html
@@ -0,0 +1,54 @@
+<!DOCTYPE html>
+<meta name="timeout" content="long">
+<script src="/resources/testharness.js"></script>
+<script src="/resources/testharnessreport.js"></script>
+<script src="/common/utils.js"></script>
+<script src="/common/dispatcher/dispatcher.js"></script>
+<script src="/html/browsers/browsing-the-web/remote-context-helper/resources/remote-context-helper.js"></script>
+<script src="../resources/utils.js"></script>
+<script src="resources/utils.js"></script>
+
+<body>
+<script>
+setup(() => assertSpeculationRulesIsSupported());
+
+promise_test(async t => {
+  const rcHelper = new RemoteContextHelper();
+
+  const referrerRC1 = await rcHelper.addWindow(undefined, { features: 'noopener' });
+  const referrerRC2 = await referrerRC1.navigateToNew();
+  const referrerRC3 = await referrerRC2.navigateToNew();
+  assert_equals(await referrerRC3.executeScript(() => navigation.entries().length), 3);
+
+  let referrerRC3CurrentId = await referrerRC3.executeScript(() => navigation.currentEntry.id);
+
+  const prerenderedRC = await addPrerenderRC(referrerRC3);
+  let activationStateBeforeActivation = await prerenderedRC.executeScript(() => {
+    return {
+     entries: navigation.entries().map(e => ({ id: e.id, })),
+     activationEntryId: navigation.activation.entry?.id,
+     activationFromId: navigation.activation.from?.id,
+     activationNavigationType : navigation.activation.navigationType,
+    }
+  });
+  assert_equals(activationStateBeforeActivation.entries.length, 1);
+  assert_equals(activationStateBeforeActivation.activationFromId, referrerRC3CurrentId);
+  assert_equals(activationStateBeforeActivation.activationEntryId, activationStateBeforeActivation.entries[0].id);
+  assert_equals(activationStateBeforeActivation.activationNavigationType, "push");
+
+  await activatePrerenderRC(referrerRC3, prerenderedRC);
+
+  let activationStateAfterActivation = await prerenderedRC.executeScript(() => {
+    return {
+     entries: navigation.entries().map(e => ({ id: e.id, })),
+     activationEntryId: navigation.activation.entry?.id,
+     activationFromId: navigation.activation.from?.id,
+     activationNavigationType : navigation.activation.navigationType,
+    }
+  });
+  assert_equals(activationStateAfterActivation.entries.length, 4);
+  assert_equals(activationStateAfterActivation.activationFromId, activationStateAfterActivation.entries[2].id);
+  assert_equals(activationStateAfterActivation.activationEntryId, activationStateAfterActivation.entries[3].id);
+  assert_equals(activationStateAfterActivation.activationNavigationType, "push");
+},`navigation.entries() and navigation.activation should be updated on activation - multiple entries`);
+</script>
diff --git a/speculation-rules/prerender/navigation-api.html b/speculation-rules/prerender/navigation-api.html
new file mode 100644
index 0000000..fd25e94
--- /dev/null
+++ b/speculation-rules/prerender/navigation-api.html
@@ -0,0 +1,60 @@
+<!DOCTYPE html>
+<meta name="timeout" content="long">
+<script src="/resources/testharness.js"></script>
+<script src="/resources/testharnessreport.js"></script>
+<script src="/common/utils.js"></script>
+<script src="/common/dispatcher/dispatcher.js"></script>
+<script src="/html/browsers/browsing-the-web/remote-context-helper/resources/remote-context-helper.js"></script>
+<script src="../resources/utils.js"></script>
+<script src="resources/utils.js"></script>
+
+<body>
+<script>
+setup(() => assertSpeculationRulesIsSupported());
+
+promise_test(async t => {
+  const rcHelper = new RemoteContextHelper();
+  const referrerRC = await rcHelper.addWindow(undefined, { features: 'noopener' });
+  assert_equals(await referrerRC.executeScript(() => navigation.entries().length), 1);
+  let referrerRCCurrentId = await referrerRC.executeScript(() => navigation.currentEntry.id);
+
+  const prerenderedRC = await addPrerenderRC(referrerRC);
+  let activationStateBeforeActivation = await prerenderedRC.executeScript(() => {
+    return {
+     entries: navigation.entries().map(e => ({ id: e.id, })),
+     activationEntryId: navigation.activation.entry?.id,
+     activationFromId: navigation.activation.from?.id,
+     activationNavigationType : navigation.activation.navigationType,
+    }
+  });
+  assert_equals(activationStateBeforeActivation.entries.length, 1);
+  assert_equals(activationStateBeforeActivation.activationFromId, referrerRCCurrentId);
+  assert_equals(activationStateBeforeActivation.activationEntryId, activationStateBeforeActivation.entries[0].id);
+  assert_equals(activationStateBeforeActivation.activationNavigationType, "push");
+
+  // Save the current entry before activation.
+  await prerenderedRC.executeScript(() => window.currentEntryBeforeActivation = navigation.currentEntry);
+
+  await activatePrerenderRC(referrerRC, prerenderedRC);
+
+  let activationStateAfterActivation = await prerenderedRC.executeScript(() => {
+    return {
+     entries: navigation.entries().map(e => ({ id: e.id, })),
+     activationEntryId: navigation.activation.entry?.id,
+     activationFromId: navigation.activation.from?.id,
+     activationNavigationType : navigation.activation.navigationType,
+    }
+  });
+  assert_equals(activationStateAfterActivation.entries.length, 2);
+  assert_equals(activationStateAfterActivation.activationFromId, activationStateAfterActivation.entries[0].id);
+  assert_equals(activationStateAfterActivation.activationFromId, referrerRCCurrentId);
+  assert_equals(activationStateAfterActivation.activationEntryId, activationStateAfterActivation.entries[1].id);
+  assert_equals(activationStateAfterActivation.activationNavigationType, "push");
+
+  let currentEntryIdentity = await prerenderedRC.executeScript(() => {
+    return window.currentEntryBeforeActivation === navigation.currentEntry &&
+           navigation.currentEntry === navigation.entries()[navigation.entries().length - 1];
+  });
+  assert_true(currentEntryIdentity);
+},`navigation.entries() and navigation.activation should be updated on activation`);
+</script>