Bug 1304140 - Part 2: Add tests for the Large-Allocation header, r=smaug

MozReview-Commit-ID: B2UpbKd2ptp

dom/tests/browser/browser.ini

@@ -8,6 +8,8 @@ support-files =
   worker_bug1004814.js
   geo_leak_test.html
   dummy.html
+  test_largeAllocation.html
+  test_largeAllocation.html^headers^
   !/dom/tests/mochitest/geolocation/network_geolocation.sjs
 
 [browser_test_toolbars_visibility.js]
@@ -38,3 +40,8 @@ support-files =
 [browser_xhr_sandbox.js]
 skip-if= buildapp == 'mulet'
 [browser_bug1004814.js]
+[browser_largeAllocation.js]
+skip-if = !e10s # Large-Allocation requires e10s
+[browser_allocateGigabyte.js]
+disabled = Does not reliably pass on 32-bit systems - bug 1314098
+skip-if = !e10s

dom/tests/browser/browser_allocateGigabyte.js

@@ -0,0 +1,43 @@
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
+
+const TEST_URI = "http://example.com/browser/dom/tests/browser/test_largeAllocation.html";
+
+function expectProcessCreated() {
+  let os = Cc["@mozilla.org/observer-service;1"].getService(Ci.nsIObserverService);
+  return new Promise(resolve => {
+    let topic = "ipc:content-created";
+    function observer() {
+      os.removeObserver(observer, topic);
+      ok(true, "Expect process created");
+      resolve();
+    }
+    os.addObserver(observer, topic, /* weak = */ false);
+  });
+}
+
+add_task(function*() {
+  yield SpecialPowers.pushPrefEnv({
+    set: [
+      ["dom.largeAllocationHeader.enabled", true],
+    ]
+  });
+
+  // A toplevel tab should be able to navigate cross process!
+  yield BrowserTestUtils.withNewTab("about:blank", function*(aBrowser) {
+    let epc = expectProcessCreated();
+    yield ContentTask.spawn(aBrowser, TEST_URI, TEST_URI => {
+      content.document.location = TEST_URI;
+    });
+
+    // Wait for the new process to be created by the Large-Allocation header
+    yield epc;
+
+    // Allocate a gigabyte of memory in the content process
+    yield ContentTask.spawn(aBrowser, null, () => {
+      let arrayBuffer = new ArrayBuffer(1024*1024*1024);
+      ok(arrayBuffer, "Successfully allocated a gigabyte of memory in content process");
+    });
+  });
+});

dom/tests/browser/browser_largeAllocation.js

@@ -0,0 +1,214 @@
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
+
+const TEST_URI = "http://example.com/browser/dom/tests/browser/test_largeAllocation.html";
+
+function expectProcessCreated() {
+  let os = Cc["@mozilla.org/observer-service;1"].getService(Ci.nsIObserverService);
+  return new Promise(resolve => {
+    let topic = "ipc:content-created";
+    function observer() {
+      os.removeObserver(observer, topic);
+      ok(true, "Expect process created");
+      resolve();
+    }
+    os.addObserver(observer, topic, /* weak = */ false);
+  });
+}
+
+function expectNoProcess() {
+  let os = Cc["@mozilla.org/observer-service;1"].getService(Ci.nsIObserverService);
+  let topic = "ipc:content-created";
+  function observer() {
+    ok(false, "A process was created!");
+    os.removeObserver(observer, topic);
+  }
+  os.addObserver(observer, topic, /* weak = */ false);
+
+  return () => os.removeObserver(observer, topic);
+}
+
+function getPID(aBrowser) {
+  return ContentTask.spawn(aBrowser, null, () => {
+    const appinfo = Components.classes["@mozilla.org/xre/app-info;1"]
+            .getService(Components.interfaces.nsIXULRuntime);
+    return appinfo.processID;
+  });
+}
+
+add_task(function*() {
+  // I'm terrible and put this set of tests into a single file, so I need a longer timeout
+  requestLongerTimeout(2);
+
+  yield SpecialPowers.pushPrefEnv({
+    set: [
+      ["dom.largeAllocationHeader.enabled", true],
+    ]
+  });
+
+  // A toplevel tab should be able to navigate cross process!
+  yield BrowserTestUtils.withNewTab("about:blank", function*(aBrowser) {
+    ok(true, "Starting test 0");
+    let pid1 = yield getPID(aBrowser);
+
+    let epc = expectProcessCreated();
+    yield ContentTask.spawn(aBrowser, TEST_URI, TEST_URI => {
+      content.document.location = TEST_URI;
+    });
+
+    // Wait for the new process to be created
+    yield epc;
+
+    let pid2 = yield getPID(aBrowser);
+
+    isnot(pid1, pid2, "The pids should be different between the initial load and the new load");
+  });
+
+  // When a Large-Allocation document is loaded in an iframe, the header should
+  // be ignored, and the tab should stay in the current process.
+  yield BrowserTestUtils.withNewTab("about:blank", function*(aBrowser) {
+    ok(true, "Starting test 1");
+    let pid1 = yield getPID(aBrowser);
+
+    // Fail the test if we create a process
+    let stopExpectNoProcess = expectNoProcess();
+
+    yield ContentTask.spawn(aBrowser, TEST_URI, TEST_URI => {
+      content.document.body.innerHTML = `<iframe src='${TEST_URI}'></iframe>`;
+
+      return new Promise(resolve => {
+        content.document.body.querySelector('iframe').onload = () => {
+          ok(true, "Iframe finished loading");
+          resolve();
+        };
+      });
+    });
+
+    let pid2 = yield getPID(aBrowser);
+
+    is(pid1, pid2, "The PID should not have changed");
+
+    stopExpectNoProcess();
+  });
+
+  // If you have an opener cross process navigation shouldn't work
+  yield BrowserTestUtils.withNewTab("http://example.com", function*(aBrowser) {
+    ok(true, "Starting test 2");
+    let pid1 = yield getPID(aBrowser);
+
+    // Fail the test if we create a process
+    let stopExpectNoProcess = expectNoProcess();
+
+    let loaded = ContentTask.spawn(aBrowser, TEST_URI, TEST_URI => {
+      content.document.body.innerHTML = '<button>CLICK ME</button>';
+
+      return new Promise(resolve => {
+        content.document.querySelector('button').onclick = e => {
+          let w = content.window.open(TEST_URI, '_blank');
+          w.onload = () => {
+            ok(true, "Window finished loading");
+            w.close();
+            resolve();
+          };
+        };
+      });
+    });
+
+    yield BrowserTestUtils.synthesizeMouseAtCenter("button", {}, aBrowser);
+
+    yield loaded;
+
+    let pid2 = yield getPID(aBrowser);
+
+    is(pid1, pid2, "The PID should not have changed");
+
+    stopExpectNoProcess();
+  });
+
+  // Load Large-Allocation twice with about:blank load in between
+  yield BrowserTestUtils.withNewTab("about:blank", function*(aBrowser) {
+    ok(true, "Starting test 3");
+    let pid1 = yield getPID(aBrowser);
+
+    let epc = expectProcessCreated();
+
+    yield ContentTask.spawn(aBrowser, TEST_URI, TEST_URI => {
+      content.document.location = TEST_URI;
+    });
+
+    yield epc;
+
+    let pid2 = yield getPID(aBrowser);
+
+    isnot(pid1, pid2);
+
+    epc = expectProcessCreated();
+
+    yield BrowserTestUtils.browserLoaded(aBrowser);
+
+    yield ContentTask.spawn(aBrowser, null, () => content.document.location = "about:blank");
+
+    yield BrowserTestUtils.browserLoaded(aBrowser);
+
+    let pid3 = yield getPID(aBrowser);
+
+    is(pid2, pid3);
+
+    yield ContentTask.spawn(aBrowser, TEST_URI, TEST_URI => {
+      content.document.location = TEST_URI;
+    });
+
+    yield epc;
+
+    let pid4 = yield getPID(aBrowser);
+
+    isnot(pid1, pid4);
+    isnot(pid2, pid4);
+  });
+
+  // Load Large-Allocation then about:blank load, then back button press should load from bfcache.
+  yield BrowserTestUtils.withNewTab("about:blank", function*(aBrowser) {
+    ok(true, "Starting test 4");
+    let pid1 = yield getPID(aBrowser);
+
+    let epc = expectProcessCreated();
+
+    yield ContentTask.spawn(aBrowser, TEST_URI, TEST_URI => {
+      content.document.location = TEST_URI;
+    });
+
+    yield epc;
+
+    let pid2 = yield getPID(aBrowser);
+
+    isnot(pid1, pid2, "PIDs 1 and 2 should not match");
+
+    let stopExpectNoProcess = expectNoProcess();
+
+    yield BrowserTestUtils.browserLoaded(aBrowser);
+
+    // Switch to about:blank, so we can navigate back
+    yield ContentTask.spawn(aBrowser, null, () => {
+      content.document.location = "about:blank";
+    });
+
+    yield BrowserTestUtils.browserLoaded(aBrowser);
+
+    let pid3 = yield getPID(aBrowser);
+
+    is(pid2, pid3, "PIDs 2 and 3 should match");
+
+    // Navigate back to the previous page, loading it from bfcache
+    yield ContentTask.spawn(aBrowser, TEST_URI, TEST_URI => {
+      content.window.history.back();
+    });
+
+    let pid4 = yield getPID(aBrowser);
+
+    isnot(pid1, pid4, "PID 4 shouldn't match PID 1");
+    is(pid2, pid4, "PID 4 should match PID 2");
+
+    stopExpectNoProcess();
+  });
+});

dom/tests/browser/test_largeAllocation.html

@@ -0,0 +1,4 @@
+<!doctype html>
+<html>
+    <body>Loaded in a new process!</body>
+</html>

dom/tests/browser/test_largeAllocation.html^headers^

@@ -0,0 +1 @@
+Large-Allocation: 0