1 files changed, 94 insertions, 0 deletions
diff --git a/browser/components/contextualidentity/test/browser/browser_relatedTab.js b/browser/components/contextualidentity/test/browser/browser_relatedTab.js
new file mode 100644
index 0000000000..3451d3d16e
--- /dev/null
+++ b/browser/components/contextualidentity/test/browser/browser_relatedTab.js
@@ -0,0 +1,94 @@
+"use strict";
+
+/*
+ * Bug 1325014 - Adding tab related to current tab inherits current tab's container usercontextid unless otherwise specified
+ */
+
+add_task(async function () {
+  let tab = BrowserTestUtils.addTab(gBrowser, "about:blank", {
+    userContextId: 1,
+  });
+  let ReferrerInfo = Components.Constructor(
+    "@mozilla.org/referrer-info;1",
+    "nsIReferrerInfo",
+    "init"
+  );
+
+  gBrowser.selectedTab = tab;
+  let relatedTab = BrowserTestUtils.addTab(gBrowser, "about:blank", {
+    relatedToCurrent: true,
+  });
+  is(
+    relatedTab.getAttribute("usercontextid"),
+    "1",
+    "Related tab (relatedToCurrent) inherits current tab's usercontextid"
+  );
+  is(
+    relatedTab.linkedBrowser.contentPrincipal.userContextId,
+    1,
+    "Related tab's browser actually inherits the current tab's usercontextid"
+  );
+  BrowserTestUtils.removeTab(relatedTab);
+
+  gBrowser.selectedTab = tab;
+  relatedTab = BrowserTestUtils.addTab(gBrowser, "about:blank", {
+    relatedToCurrent: true,
+    userContextId: 2,
+  });
+  is(
+    relatedTab.getAttribute("usercontextid"),
+    "2",
+    "Related tab (relatedToCurrent) with overridden usercontextid"
+  );
+  is(
+    relatedTab.linkedBrowser.contentPrincipal.userContextId,
+    2,
+    "Related tab's browser actually gets overridden usercontextid"
+  );
+  BrowserTestUtils.removeTab(relatedTab);
+
+  gBrowser.selectedTab = tab;
+  let referrerInfo = new ReferrerInfo(
+    Ci.nsIReferrerInfo.EMPTY,
+    true,
+    gBrowser.currentURI
+  );
+  relatedTab = BrowserTestUtils.addTab(gBrowser, "about:blank", {
+    referrerInfo,
+  });
+  is(
+    relatedTab.getAttribute("usercontextid"),
+    "1",
+    "Related tab (referrer) inherits current tab's usercontextid"
+  );
+  is(
+    relatedTab.linkedBrowser.contentPrincipal.userContextId,
+    1,
+    "Related tab's browser (referrer) actually inherits the current tab's usercontextid"
+  );
+  BrowserTestUtils.removeTab(relatedTab);
+
+  gBrowser.selectedTab = tab;
+  referrerInfo = new ReferrerInfo(
+    Ci.nsIReferrerInfo.EMPTY,
+    true,
+    gBrowser.currentURI
+  );
+  relatedTab = BrowserTestUtils.addTab(gBrowser, "about:blank", {
+    referrerInfo,
+    userContextId: 2,
+  });
+  is(
+    relatedTab.getAttribute("usercontextid"),
+    "2",
+    "Related tab (referrer) with overridden usercontextid"
+  );
+  is(
+    relatedTab.linkedBrowser.contentPrincipal.userContextId,
+    2,
+    "Related tab's browser (referrer) actually gets overridden usercontextid"
+  );
+  BrowserTestUtils.removeTab(relatedTab);
+
+  BrowserTestUtils.removeTab(tab);
+});