1 files changed, 147 insertions, 0 deletions
diff --git a/remote/test/browser/emulation/browser_setUserAgentOverride.js b/remote/test/browser/emulation/browser_setUserAgentOverride.js
new file mode 100644
index 0000000000..9275675452
--- /dev/null
+++ b/remote/test/browser/emulation/browser_setUserAgentOverride.js
@@ -0,0 +1,147 @@
+/* Any copyright is dedicated to the Public Domain.
+ * http://creativecommons.org/publicdomain/zero/1.0/ */
+
+"use strict";
+
+const DOC = toDataURL(`<script>document.write(navigator.userAgent);</script>`);
+
+add_task(async function invalidPlatform({ client }) {
+  const { Emulation } = client;
+  const userAgent = "Mozilla/5.0 (rv: 23) Romanesco/42.0\n";
+
+  for (const platform of [null, true, 1, [], {}]) {
+    let errorThrown = "";
+    try {
+      await Emulation.setUserAgentOverride({ userAgent, platform });
+    } catch (e) {
+      errorThrown = e.message;
+    }
+    ok(
+      errorThrown.match(/platform: string value expected/),
+      `Fails with invalid type: ${platform}`
+    );
+  }
+});
+
+add_task(async function setAndResetUserAgent({ client }) {
+  const { Emulation } = client;
+  const userAgent = "Mozilla/5.0 (rv: 23) Romanesco/42.0";
+
+  await loadURL(DOC);
+  const originalUserAgent = await getNavigatorProperty("userAgent");
+
+  isnot(originalUserAgent, userAgent, "Custom user agent hasn't been set");
+
+  await Emulation.setUserAgentOverride({ userAgent });
+  await loadURL(DOC);
+  is(
+    await getNavigatorProperty("userAgent"),
+    userAgent,
+    "Custom user agent has been set"
+  );
+
+  await Emulation.setUserAgentOverride({ userAgent: "" });
+  await loadURL(DOC);
+  is(
+    await getNavigatorProperty("userAgent"),
+    originalUserAgent,
+    "Custom user agent has been reset"
+  );
+});
+
+add_task(async function invalidUserAgent({ Emulation }) {
+  const userAgent = "Mozilla/5.0 (rv: 23) Romanesco/42.0\n";
+
+  await loadURL(DOC);
+  isnot(
+    await getNavigatorProperty("userAgent"),
+    userAgent,
+    "Custom user agent hasn't been set"
+  );
+
+  let errorThrown = false;
+  try {
+    await Emulation.setUserAgentOverride({ userAgent });
+  } catch (e) {
+    errorThrown = true;
+  }
+  ok(errorThrown, "Invalid user agent format raised error");
+});
+
+add_task(async function setAndResetPlatform({ client }) {
+  const { Emulation } = client;
+  const userAgent = "Mozilla/5.0 (rv: 23) Romanesco/42.0";
+  const platform = "foobar";
+
+  await loadURL(DOC);
+  const originalUserAgent = await getNavigatorProperty("userAgent");
+  const originalPlatform = await getNavigatorProperty("platform");
+
+  isnot(userAgent, originalUserAgent, "Custom user agent hasn't been set");
+  isnot(platform, originalPlatform, "Custom platform hasn't been set");
+
+  await Emulation.setUserAgentOverride({ userAgent, platform });
+  await loadURL(DOC);
+  is(
+    await getNavigatorProperty("userAgent"),
+    userAgent,
+    "Custom user agent has been set"
+  );
+  is(
+    await getNavigatorProperty("platform"),
+    platform,
+    "Custom platform has been set"
+  );
+
+  await Emulation.setUserAgentOverride({ userAgent: "", platform: "" });
+  await loadURL(DOC);
+  is(
+    await getNavigatorProperty("userAgent"),
+    originalUserAgent,
+    "Custom user agent has been reset"
+  );
+  is(
+    await getNavigatorProperty("platform"),
+    originalPlatform,
+    "Custom platform has been reset"
+  );
+});
+
+add_task(async function notSetForNewContext({ client }) {
+  const { Emulation, Target } = client;
+  const userAgent = "Mozilla/5.0 (rv: 23) Romanesco/42.0";
+  const platform = "foobar";
+
+  await Emulation.setUserAgentOverride({ userAgent, platform });
+  await loadURL(DOC);
+  is(
+    await getNavigatorProperty("userAgent"),
+    userAgent,
+    "Custom user agent has been set"
+  );
+  is(
+    await getNavigatorProperty("platform"),
+    platform,
+    "Custom platform has been set"
+  );
+
+  const { targetInfo } = await openTab(Target);
+  await Target.activateTarget({ targetId: targetInfo.targetId });
+
+  isnot(
+    await getNavigatorProperty("userAgent"),
+    userAgent,
+    "Custom user agent has not been set"
+  );
+  isnot(
+    await getNavigatorProperty("platform"),
+    platform,
+    "Custom platform has not been set"
+  );
+});
+
+async function getNavigatorProperty(prop) {
+  return SpecialPowers.spawn(gBrowser.selectedBrowser, [prop], _prop => {
+    return content.navigator[_prop];
+  });
+}