1 files changed, 54 insertions, 0 deletions
diff --git a/toolkit/components/printing/tests/browser_print_frame.js b/toolkit/components/printing/tests/browser_print_frame.js
new file mode 100644
index 0000000000..6331a0dc4d
--- /dev/null
+++ b/toolkit/components/printing/tests/browser_print_frame.js
@@ -0,0 +1,54 @@
+/* Any copyright is dedicated to the Public Domain.
+ * http://creativecommons.org/publicdomain/zero/1.0/
+ */
+
+const frameSource = "<a href='about:mozilla'>some text</a>";
+const SOURCES = [
+  `Something else <iframe id="f" srcdoc="${frameSource}"></iframe>`,
+  `Something else <iframe id="f" src="https://example.com/document-builder.sjs?html=${frameSource}"></iframe>`,
+];
+
+async function getPreviewText(previewBrowser) {
+  return SpecialPowers.spawn(previewBrowser, [], function() {
+    return content.document.body.textContent;
+  });
+}
+
+add_task(async function print_frame() {
+  let i = 0;
+  for (const source of SOURCES) {
+    is(
+      document.querySelector(".printPreviewBrowser"),
+      null,
+      "There shouldn't be any print preview browser"
+    );
+
+    await BrowserTestUtils.withNewTab(
+      "data:text/html," + source,
+      async function(browser) {
+        let frameBC = browser.browsingContext.children[0];
+        let helper = new PrintHelper(browser);
+
+        // If you change this, change nsContextMenu.printFrame() too.
+        PrintUtils.startPrintWindow(frameBC, {
+          printFrameOnly: true,
+        });
+
+        // Wait for the dialog to be fully ready. The initial preview will be
+        // done at this point.
+        await helper.waitForDialog();
+
+        let textContent = await getPreviewText(
+          helper.currentPrintPreviewBrowser
+        );
+        is(textContent, "some text", "Correct content loaded");
+
+        let file = helper.mockFilePicker(`browser_print_frame-${i++}.pdf`);
+        await helper.assertPrintToFile(file, () => {
+          helper.click(helper.get("print-button"));
+        });
+        PrintHelper.resetPrintPrefs();
+      }
+    );
+  }
+});