1 files changed, 8 insertions, 6 deletions
diff --git a/uriloader/exthandler/tests/mochitest/browser_pdf_save_as.js b/uriloader/exthandler/tests/mochitest/browser_pdf_save_as.js
index a08fe342cc..5d4a2080a7 100644
--- a/uriloader/exthandler/tests/mochitest/browser_pdf_save_as.js
+++ b/uriloader/exthandler/tests/mochitest/browser_pdf_save_as.js
@@ -16,9 +16,9 @@ const TEST_PATH = getRootDirectory(gTestPath).replace(
 const { saveToDisk, alwaysAsk, handleInternally, useSystemDefault } =
   Ci.nsIHandlerInfo;
 const MockFilePicker = SpecialPowers.MockFilePicker;
-MockFilePicker.init(window);
+MockFilePicker.init(window.browsingContext);
 
-async function testPdfFilePicker(mimeInfo) {
+async function testPdfFilePicker() {
   await BrowserTestUtils.withNewTab(
     `data:text/html,<a id="test-link" href="${TEST_PATH}/file_pdf_application_pdf.pdf">Test PDF Link</a>`,
     async browser => {
@@ -37,12 +37,14 @@ async function testPdfFilePicker(mimeInfo) {
       let filePickerPromise = new Promise(resolve => {
         MockFilePicker.showCallback = fp => {
           ok(true, "filepicker should be visible");
-          ok(
-            fp.defaultExtension === "pdf",
+          Assert.strictEqual(
+            fp.defaultExtension,
+            "pdf",
             "Default extension in filepicker should be pdf"
           );
-          ok(
-            fp.defaultString === "file_pdf_application_pdf.pdf",
+          Assert.strictEqual(
+            fp.defaultString,
+            "file_pdf_application_pdf.pdf",
             "Default string name in filepicker should have the correct pdf file name"
           );
           setTimeout(resolve, 0);