1 files changed, 59 insertions, 0 deletions
diff --git a/browser/components/shell/test/browser_headless_screenshot_3.js b/browser/components/shell/test/browser_headless_screenshot_3.js
new file mode 100644
index 0000000000..b55635652d
--- /dev/null
+++ b/browser/components/shell/test/browser_headless_screenshot_3.js
@@ -0,0 +1,59 @@
+"use strict";
+
+add_task(async function () {
+  const cwdScreenshotPath = PathUtils.join(
+    Services.dirsvc.get("CurWorkD", Ci.nsIFile).path,
+    "screenshot.png"
+  );
+
+  // Test invalid URL arguments (either no argument or too many arguments).
+  await testFileCreationNegative(["-screenshot"], cwdScreenshotPath);
+  await testFileCreationNegative(
+    [
+      "http://mochi.test:8888/browser/browser/components/shell/test/headless.html",
+      "http://mochi.test:8888/headless.html",
+      "-screenshot",
+    ],
+    cwdScreenshotPath
+  );
+
+  // Test all four basic variations of the "window-size" argument.
+  await testFileCreationPositive(
+    [
+      "-url",
+      "http://mochi.test:8888/browser/browser/components/shell/test/headless.html",
+      "-screenshot",
+      "-window-size",
+      "800",
+    ],
+    cwdScreenshotPath
+  );
+  await testFileCreationPositive(
+    [
+      "-url",
+      "http://mochi.test:8888/browser/browser/components/shell/test/headless.html",
+      "-screenshot",
+      "-window-size=800",
+    ],
+    cwdScreenshotPath
+  );
+  await testFileCreationPositive(
+    [
+      "-url",
+      "http://mochi.test:8888/browser/browser/components/shell/test/headless.html",
+      "-screenshot",
+      "--window-size",
+      "800",
+    ],
+    cwdScreenshotPath
+  );
+  await testFileCreationPositive(
+    [
+      "-url",
+      "http://mochi.test:8888/browser/browser/components/shell/test/headless.html",
+      "-screenshot",
+      "--window-size=800",
+    ],
+    cwdScreenshotPath
+  );
+});