1 files changed, 100 insertions, 0 deletions
diff --git a/comm/suite/components/tests/browser/browser_350525.js b/comm/suite/components/tests/browser/browser_350525.js
new file mode 100644
index 0000000000..c5fb0e11e7
--- /dev/null
+++ b/comm/suite/components/tests/browser/browser_350525.js
@@ -0,0 +1,100 @@
+function test() {
+  /** Test for Bug 350525 **/
+
+  function test(aLambda) {
+    try {
+      return aLambda() || true;
+    }
+    catch (ex) { }
+    return false;
+  }
+
+  waitForExplicitFinish();
+
+  ////////////////////////////
+  // setWindowValue, et al. //
+  ////////////////////////////
+  let key = "Unique name: " + Date.now();
+  let value = "Unique value: " + Math.random();
+
+  // test adding
+  ok(test(() => ss.setWindowValue(window, key, value)), "set a window value");
+
+  // test retrieving
+  is(ss.getWindowValue(window, key), value, "stored window value matches original");
+
+  // test deleting
+  ok(test(() => ss.deleteWindowValue(window, key)), "delete the window value");
+
+  // value should not exist post-delete
+  is(ss.getWindowValue(window, key), "", "window value was deleted");
+
+  // test deleting a non-existent value
+  ok(test(() => ss.deleteWindowValue(window, key)), "delete non-existent window value");
+
+  /////////////////////////
+  // setTabValue, et al. //
+  /////////////////////////
+  key = "Unique name: " + Math.random();
+  value = "Unique value: " + Date.now();
+  let tab = getBrowser().addTab();
+  tab.linkedBrowser.stop();
+
+  // test adding
+  ok(test(() => ss.setTabValue(tab, key, value)), "store a tab value");
+
+  // test retrieving
+  is(ss.getTabValue(tab, key), value, "stored tab value match original");
+
+  // test deleting
+  ok(test(() => ss.deleteTabValue(tab, key)), "delete the tab value");
+  // value should not exist post-delete
+  is(ss.getTabValue(tab, key), "", "tab value was deleted");
+
+  // test deleting a non-existent value
+  ok(test(() => ss.deleteTabValue(tab, key)), "delete non-existent tab value");
+
+  // clean up
+  getBrowser().removeTab(tab);
+
+  /////////////////////////////////////
+  // getClosedTabCount, undoCloseTab //
+  /////////////////////////////////////
+
+  // get closed tab count
+  let count = ss.getClosedTabCount(window);
+  let max_tabs_undo = Services.prefs.getIntPref("browser.sessionstore.max_tabs_undo");
+  ok(0 <= count && count <= max_tabs_undo,
+     "getClosedTabCount returns zero or at most max_tabs_undo");
+
+  // create a new tab
+  let testURL = "about:";
+  tab = getBrowser().addTab(testURL);
+  tab.linkedBrowser.addEventListener("load", function testTabLBLoad(aEvent) {
+    this.removeEventListener("load", testTabLBLoad, true);
+    // make sure that the next closed tab will increase getClosedTabCount
+    Services.prefs.setIntPref("browser.sessionstore.max_tabs_undo", max_tabs_undo + 1);
+
+    // remove tab
+    getBrowser().removeTab(tab);
+
+    // getClosedTabCount
+    var newcount = ss.getClosedTabCount(window);
+    ok(newcount > count, "after closing a tab, getClosedTabCount has been incremented");
+
+    // undoCloseTab
+    tab = test(() => ss.undoCloseTab(window, 0));
+    ok(tab, "undoCloseTab doesn't throw")
+
+    tab.linkedBrowser.addEventListener("load", function testTabLBLoad2(aEvent) {
+      this.removeEventListener("load", testTabLBLoad2, true);
+      is(this.currentURI.spec, testURL, "correct tab was reopened");
+
+      // clean up
+      if (Services.prefs.prefHasUserValue("browser.sessionstore.max_tabs_undo"))
+        Services.prefs.clearUserPref("browser.sessionstore.max_tabs_undo");
+      getBrowser().removeTab(tab);
+      finish();
+    }, true);
+  }, true);
+}