fune/browser/base/content/test/general/browser_restore_isAppTab.js
Gijs Kruitbosch 6b70b349d9 Bug 1379447 - fix intermittent failure in browser_alltabslistener caused by tab hitting about:blank when waiting for page stops, r=dao
MozReview-Commit-ID: Sn4uj4jMDc

--HG--
extra : rebase_source : c2ef1651e84ba54ca1e7584ed0dc533c2cbe1cbd
2017-07-19 23:17:01 +01:00

110 lines
3.6 KiB
JavaScript

/* Any copyright is dedicated to the Public Domain.
* http://creativecommons.org/publicdomain/zero/1.0/ */
/* eslint-env mozilla/frame-script */
const {TabStateFlusher} = Cu.import("resource:///modules/sessionstore/TabStateFlusher.jsm", {});
const DUMMY = "http://example.com/browser/browser/base/content/test/general/dummy_page.html";
function frameScript() {
addMessageListener("Test:GetIsAppTab", function() {
sendAsyncMessage("Test:IsAppTab", { isAppTab: docShell.isAppTab });
});
}
function loadFrameScript(browser) {
browser.messageManager.loadFrameScript("data:,(" + frameScript.toString() + ")();", true);
}
function isBrowserAppTab(browser) {
return new Promise(resolve => {
function listener({ data }) {
browser.messageManager.removeMessageListener("Test:IsAppTab", listener);
resolve(data.isAppTab);
}
// It looks like same-process messages may be reordered by the message
// manager, so we need to wait one tick before sending the message.
executeSoon(function() {
browser.messageManager.addMessageListener("Test:IsAppTab", listener);
browser.messageManager.sendAsyncMessage("Test:GetIsAppTab");
});
});
}
// Restarts the child process by crashing it then reloading the tab
var restart = async function(browser) {
// If the tab isn't remote this would crash the main process so skip it
if (!browser.isRemoteBrowser)
return;
// Make sure the main process has all of the current tab state before crashing
await TabStateFlusher.flush(browser);
await BrowserTestUtils.crashBrowser(browser);
let tab = gBrowser.getTabForBrowser(browser);
SessionStore.reviveCrashedTab(tab);
await promiseTabLoaded(tab);
};
add_task(async function navigate() {
let tab = BrowserTestUtils.addTab(gBrowser, "about:robots");
let browser = tab.linkedBrowser;
gBrowser.selectedTab = tab;
await BrowserTestUtils.browserStopped(gBrowser);
loadFrameScript(browser);
let isAppTab = await isBrowserAppTab(browser);
ok(!isAppTab, "Docshell shouldn't think it is an app tab");
gBrowser.pinTab(tab);
isAppTab = await isBrowserAppTab(browser);
ok(isAppTab, "Docshell should think it is an app tab");
gBrowser.loadURI(DUMMY);
await BrowserTestUtils.browserStopped(gBrowser);
loadFrameScript(browser);
isAppTab = await isBrowserAppTab(browser);
ok(isAppTab, "Docshell should think it is an app tab");
gBrowser.unpinTab(tab);
isAppTab = await isBrowserAppTab(browser);
ok(!isAppTab, "Docshell shouldn't think it is an app tab");
gBrowser.pinTab(tab);
isAppTab = await isBrowserAppTab(browser);
ok(isAppTab, "Docshell should think it is an app tab");
gBrowser.loadURI("about:robots");
await BrowserTestUtils.browserStopped(gBrowser);
loadFrameScript(browser);
isAppTab = await isBrowserAppTab(browser);
ok(isAppTab, "Docshell should think it is an app tab");
gBrowser.removeCurrentTab();
});
add_task(async function crash() {
if (!gMultiProcessBrowser || !("nsICrashReporter" in Ci))
return;
let tab = BrowserTestUtils.addTab(gBrowser, DUMMY);
let browser = tab.linkedBrowser;
gBrowser.selectedTab = tab;
await BrowserTestUtils.browserStopped(gBrowser);
loadFrameScript(browser);
let isAppTab = await isBrowserAppTab(browser);
ok(!isAppTab, "Docshell shouldn't think it is an app tab");
gBrowser.pinTab(tab);
isAppTab = await isBrowserAppTab(browser);
ok(isAppTab, "Docshell should think it is an app tab");
await restart(browser);
loadFrameScript(browser);
isAppTab = await isBrowserAppTab(browser);
ok(isAppTab, "Docshell should think it is an app tab");
gBrowser.removeCurrentTab();
});