1 files changed, 102 insertions, 0 deletions
diff --git a/comm/mail/test/browser/im/browser_chatTabRestore.js b/comm/mail/test/browser/im/browser_chatTabRestore.js
new file mode 100644
index 0000000000..e80ecc0461
--- /dev/null
+++ b/comm/mail/test/browser/im/browser_chatTabRestore.js
@@ -0,0 +1,102 @@
+/* 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/. */
+
+"use strict";
+
+var utils = ChromeUtils.import("resource://testing-common/mozmill/utils.jsm");
+
+var { assert_tab_mode_name, mc } = ChromeUtils.import(
+  "resource://testing-common/mozmill/FolderDisplayHelpers.jsm"
+);
+
+/**
+ * Create a new chat tab, making that tab the current tab. We block until the
+ * message finishes loading. (Inspired by open_selected_message_in_new_tab)
+ */
+async function open_chat_tab() {
+  // Get the current tab count so we can make sure the tab actually opened.
+  let preCount =
+    mc.window.document.getElementById("tabmail").tabContainer.allTabs.length;
+
+  mc.window.document.getElementById("tabmail").openTab("chat", {});
+  await wait_for_chat_tab_to_open(mc);
+
+  if (
+    mc.window.document.getElementById("tabmail").tabContainer.allTabs.length !=
+    preCount + 1
+  ) {
+    throw new Error("The tab never actually got opened!");
+  }
+
+  let newTab = mc.window.document.getElementById("tabmail").tabInfo[preCount];
+  return newTab;
+}
+
+async function wait_for_chat_tab_to_open(aController) {
+  if (aController == null) {
+    aController = mc;
+  }
+
+  utils.waitFor(
+    function () {
+      let chatTabFound = false;
+      for (let tab of mc.window.document.getElementById("tabmail").tabInfo) {
+        if (tab.mode.type == "chat") {
+          chatTabFound = true;
+          break;
+        }
+      }
+      return chatTabFound;
+    },
+    "Timeout waiting for chat tab to open",
+    1000,
+    50
+  );
+
+  // The above may return immediately, meaning the event queue might not get a
+  // chance. Give it a chance now.
+  await new Promise(resolve => setTimeout(resolve));
+}
+
+/**
+ * This tests that the chat tab is restored properly after tabs are
+ * serialized. As for folder tabs, we can't test a restart (can we ?), so we
+ * just test the persist/restore cycle.
+ */
+add_task(async function test_chat_tab_restore() {
+  // Close everything but the first tab.
+  let closeTabs = function () {
+    while (mc.window.document.getElementById("tabmail").tabInfo.length > 1) {
+      mc.window.document.getElementById("tabmail").closeTab(1);
+    }
+  };
+
+  await open_chat_tab();
+  let state = mc.window.document.getElementById("tabmail").persistTabs();
+  closeTabs();
+  mc.window.document.getElementById("tabmail").restoreTabs(state);
+
+  if (
+    mc.window.document.getElementById("tabmail").tabContainer.allTabs.length < 2
+  ) {
+    throw new Error("The tab is not restored!");
+  }
+
+  let tabTypes = ["mail3PaneTab", "chat"];
+  for (let i in tabTypes) {
+    assert_tab_mode_name(
+      mc.window.document.getElementById("tabmail").tabInfo[i],
+      tabTypes[i]
+    );
+  }
+
+  closeTabs();
+
+  Assert.report(
+    false,
+    undefined,
+    undefined,
+    "Test ran to completion successfully"
+  );
+});