summaryrefslogtreecommitdiff
path: root/system/docshell/test/browser/browser_bug422543.js
blob: 0655a17b7a67e5554d230adfcac8e1733c7e7177 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
/* Any copyright is dedicated to the Public Domain.
 * http://creativecommons.org/publicdomain/zero/1.0/ */

add_task(function* runTests() {
  yield setup();
  let browser = gBrowser.selectedBrowser;
  // Now that we're set up, initialize our frame script.
  yield checkListeners("initial", "listeners initialized");

  // Check if all history listeners are always notified.
  info("# part 1");
  yield whenPageShown(browser, () => browser.loadURI("http://www.example.com/"));
  yield checkListeners("newentry", "shistory has a new entry");
  ok(browser.canGoBack, "we can go back");

  yield whenPageShown(browser, () => browser.goBack());
  yield checkListeners("goback", "back to the first shentry");
  ok(browser.canGoForward, "we can go forward");

  yield whenPageShown(browser, () => browser.goForward());
  yield checkListeners("goforward", "forward to the second shentry");

  yield whenPageShown(browser, () => browser.reload());
  yield checkListeners("reload", "current shentry reloaded");

  yield whenPageShown(browser, () => browser.gotoIndex(0));
  yield checkListeners("gotoindex", "back to the first index");

  // Check nsISHistoryInternal.notifyOnHistoryReload
  info("# part 2");
  ok((yield notifyReload()), "reloading has not been canceled");
  yield checkListeners("reload", "saw the reload notification");

  // Let the first listener cancel the reload action.
  info("# part 3");
  yield resetListeners();
  yield setListenerRetval(0, false);
  ok(!(yield notifyReload()), "reloading has been canceled");
  yield checkListeners("reload", "saw the reload notification");

  // Let both listeners cancel the reload action.
  info("# part 4");
  yield resetListeners();
  yield setListenerRetval(1, false);
  ok(!(yield notifyReload()), "reloading has been canceled");
  yield checkListeners("reload", "saw the reload notification");

  // Let the second listener cancel the reload action.
  info("# part 5");
  yield resetListeners();
  yield setListenerRetval(0, true);
  ok(!(yield notifyReload()), "reloading has been canceled");
  yield checkListeners("reload", "saw the reload notification");
});

function listenOnce(message, arg = {}) {
  return new Promise(resolve => {
    let mm = gBrowser.selectedBrowser.messageManager;
    mm.addMessageListener(message + ":return", function listener(msg) {
      mm.removeMessageListener(message + ":return", listener);
      resolve(msg.data);
    });

    mm.sendAsyncMessage(message, arg);
  });
}

function checkListeners(aLast, aMessage) {
  return listenOnce("bug422543:getListenerStatus").then((listenerStatuses) => {
    is(listenerStatuses[0], aLast, aMessage);
    is(listenerStatuses[1], aLast, aMessage);
  });
}

function resetListeners() {
  return listenOnce("bug422543:resetListeners");
}

function notifyReload() {
  return listenOnce("bug422543:notifyReload").then(({ rval }) => {
    return rval;
  });
}

function setListenerRetval(num, val) {
  return listenOnce("bug422543:setRetval", { num, val });
}

function setup() {
  return BrowserTestUtils.openNewForegroundTab(gBrowser,
                                               "http://mochi.test:8888")
                         .then(function (tab) {
    let browser = tab.linkedBrowser;
    registerCleanupFunction(function* () {
      yield listenOnce("bug422543:cleanup");
      gBrowser.removeTab(tab);
    });

    browser.messageManager
           .loadFrameScript(getRootDirectory(gTestPath) + "file_bug422543_script.js", false);
  });
}

function whenPageShown(aBrowser, aNavigation) {
  let listener = ContentTask.spawn(aBrowser, null, function () {
    return new Promise(resolve => {
      addEventListener("pageshow", function onLoad() {
        removeEventListener("pageshow", onLoad, true);
        resolve();
      }, true);
    });
  });

  aNavigation();
  return listener;
}