summaryrefslogtreecommitdiffstats
path: root/docshell/test/browser/browser_ua_emulation.js
blob: f1e9657cd3c9609642dbcefc4fec01b27cbe273b (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
/* Any copyright is dedicated to the Public Domain.
   http://creativecommons.org/publicdomain/zero/1.0/ */

"use strict";

const URL = "data:text/html;charset=utf-8,<iframe id='test-iframe'></iframe>";

// Test that the docShell UA emulation works
async function contentTaskNoOverride() {
  let docshell = docShell;
  is(
    docshell.browsingContext.customUserAgent,
    "",
    "There should initially be no customUserAgent"
  );
}

async function contentTaskOverride() {
  let docshell = docShell;
  is(
    docshell.browsingContext.customUserAgent,
    "foo",
    "The user agent should be changed to foo"
  );

  is(
    content.navigator.userAgent,
    "foo",
    "The user agent should be changed to foo"
  );

  let frameWin = content.document.querySelector("#test-iframe").contentWindow;
  is(
    frameWin.navigator.userAgent,
    "foo",
    "The UA should be passed on to frames."
  );

  let newFrame = content.document.createElement("iframe");
  content.document.body.appendChild(newFrame);

  let newFrameWin = newFrame.contentWindow;
  is(
    newFrameWin.navigator.userAgent,
    "foo",
    "Newly created frames should use the new UA"
  );

  newFrameWin.location.reload();
  await ContentTaskUtils.waitForEvent(newFrame, "load");

  is(
    newFrameWin.navigator.userAgent,
    "foo",
    "New UA should persist across reloads"
  );
}

add_task(async function () {
  await BrowserTestUtils.withNewTab(
    { gBrowser, url: URL },
    async function (browser) {
      await SpecialPowers.spawn(browser, [], contentTaskNoOverride);

      let browsingContext = BrowserTestUtils.getBrowsingContextFrom(browser);
      browsingContext.customUserAgent = "foo";

      await SpecialPowers.spawn(browser, [], contentTaskOverride);
    }
  );
});