1 files changed, 104 insertions, 0 deletions
diff --git a/dom/browser-element/mochitest/browserElement_BackForward.js b/dom/browser-element/mochitest/browserElement_BackForward.js
new file mode 100644
index 000000000..26b6344c4
--- /dev/null
+++ b/dom/browser-element/mochitest/browserElement_BackForward.js
@@ -0,0 +1,104 @@
+/* Any copyright is dedicated to the public domain.
+   http://creativecommons.org/publicdomain/zero/1.0/ */
+
+// Bug 741755 - Test that canGo{Back,Forward} and go{Forward,Back} work with
+// <iframe mozbrowser>.
+
+"use strict";
+SimpleTest.waitForExplicitFinish();
+browserElementTestHelpers.setEnabledPref(true);
+
+var iframe;
+function addOneShotIframeEventListener(event, fn) {
+  function wrapper(e) {
+    iframe.removeEventListener(event, wrapper);
+    fn(e);
+  };
+
+  iframe.addEventListener(event, wrapper);
+}
+
+function runTest() {
+  iframe = document.createElement('iframe');
+  iframe.setAttribute('mozbrowser', 'true');
+
+  addOneShotIframeEventListener('mozbrowserloadend', function() {
+    SimpleTest.executeSoon(test2);
+  });
+
+  iframe.src = browserElementTestHelpers.emptyPage1;
+  document.body.appendChild(iframe);
+}
+
+function checkCanGoBackAndForward(canGoBack, canGoForward, nextTest) {
+  var seenCanGoBackResult = false;
+  iframe.getCanGoBack().onsuccess = function(e) {
+    is(seenCanGoBackResult, false, "onsuccess handler shouldn't be called twice.");
+    seenCanGoBackResult = true;
+    is(e.target.result, canGoBack);
+    maybeRunNextTest();
+  };
+
+  var seenCanGoForwardResult = false;
+  iframe.getCanGoForward().onsuccess = function(e) {
+    is(seenCanGoForwardResult, false, "onsuccess handler shouldn't be called twice.");
+    seenCanGoForwardResult = true;
+    is(e.target.result, canGoForward);
+    maybeRunNextTest();
+  };
+
+  function maybeRunNextTest() {
+    if (seenCanGoBackResult && seenCanGoForwardResult) {
+      nextTest();
+    }
+  }
+}
+
+function test2() {
+  checkCanGoBackAndForward(false, false, test3);
+}
+
+function test3() {
+  addOneShotIframeEventListener('mozbrowserloadend', function() {
+    checkCanGoBackAndForward(true, false, test4);
+  });
+
+  SimpleTest.executeSoon(function() {
+    iframe.src = browserElementTestHelpers.emptyPage2;
+  });
+}
+
+function test4() {
+  addOneShotIframeEventListener('mozbrowserlocationchange', function(e) {
+    is(e.detail.url, browserElementTestHelpers.emptyPage3);
+    is(e.detail.canGoBack, true);
+    is(e.detail.canGoForward, false);
+    checkCanGoBackAndForward(true, false, test5);
+  });
+
+  SimpleTest.executeSoon(function() {
+    iframe.src = browserElementTestHelpers.emptyPage3;
+  });
+}
+
+function test5() {
+  addOneShotIframeEventListener('mozbrowserlocationchange', function(e) {
+    is(e.detail.url, browserElementTestHelpers.emptyPage2);
+    is(e.detail.canGoBack, true);
+    is(e.detail.canGoForward, true);
+    checkCanGoBackAndForward(true, true, test6);
+  });
+  iframe.goBack();
+}
+
+function test6() {
+  addOneShotIframeEventListener('mozbrowserlocationchange', function(e) {
+    is(e.detail.url, browserElementTestHelpers.emptyPage1);
+    is(e.detail.canGoBack, false);
+    is(e.detail.canGoForward, true);
+    checkCanGoBackAndForward(false, true, SimpleTest.finish);
+  });
+  iframe.goBack();
+}
+
+addEventListener('testready', runTest);