describe("PDF Recognition", function() { var win; before(function* () { this.timeout(60000); // Load Zotero pane and install PDF tools yield Zotero.Promise.all([ loadZoteroPane().then(w => win = w) ]); }); beforeEach(function* () { yield selectLibrary(win); }); afterEach(function() { for(let win of getWindows("chrome://zotero/content/progressQueueDialog.xhtml")) { win.close(); } Zotero.ProgressQueues.get('recognize').cancel(); Zotero.RecognizePDF.recognizeStub = null; Zotero.Prefs.clear('autoRenameFiles.linked'); }); after(function() { if (win) { win.close(); } }); it("should recognize a PDF by DOI and rename the file", async function () { if (Zotero.automatedTest) this.skip(); // TODO: Mock services this.timeout(30000); // Import the PDF var testdir = getTestDataDirectory(); testdir.append("recognizePDF_test_DOI.pdf"); var collection = await createDataObject('collection'); var attachment = await Zotero.Attachments.importFromFile({ file: testdir, collections: [collection.id] }); win.ZoteroPane.recognizeSelected(); var addedIDs = await waitForItemEvent("add"); var modifiedIDs = await waitForItemEvent("modify"); assert.lengthOf(addedIDs, 1); var item = Zotero.Items.get(addedIDs[0]); assert.equal(item.getField("title"), "Shaping the Research Agenda"); assert.equal(item.getField("libraryCatalog"), "DOI.org (Crossref)"); assert.lengthOf(modifiedIDs, 2); // Wait for status to show as complete var progressWindow = getWindows("chrome://zotero/content/progressQueueDialog.xhtml")[0]; var completeStr = Zotero.getString("general.finished"); while (progressWindow.document.getElementById("label").value != completeStr) { await Zotero.Promise.delay(20); } // The file should have been renamed assert.equal( attachment.attachmentFilename, Zotero.Attachments.getFileBaseNameFromItem(item) + '.pdf' ); }); it("should recognize a PDF by arXiv ID", async function () { if (Zotero.automatedTest) this.skip(); // TODO: Mock services this.timeout(30000); // Import the PDF var testdir = getTestDataDirectory(); testdir.append("recognizePDF_test_arXiv.pdf"); var attachment = await Zotero.Attachments.importFromFile({ file: testdir }); // Recognize the PDF win.ZoteroPane.recognizeSelected(); var addedIDs = await waitForItemEvent("add"); var modifiedIDs = await waitForItemEvent("modify"); // Item and note assert.lengthOf(addedIDs, 2); var item = Zotero.Items.get(addedIDs[0]); assert.equal(item.getField("title"), "Scaling study of an improved fermion action on quenched lattices"); assert.lengthOf(modifiedIDs, 1); // Wait for status to show as complete var progressWindow = getWindows("chrome://zotero/content/progressQueueDialog.xhtml")[0]; var completeStr = Zotero.getString("general.finished"); while (progressWindow.document.getElementById("label").value != completeStr) { await Zotero.Promise.delay(20); } }); it("should put new item in same collection", async function () { if (Zotero.automatedTest) this.skip(); // TODO: Mock services this.timeout(30000); // Import the PDF var testdir = getTestDataDirectory(); testdir.append("recognizePDF_test_arXiv.pdf"); var collection = await createDataObject('collection'); var attachment = await Zotero.Attachments.importFromFile({ file: testdir, collections: [collection.id] }); win.ZoteroPane.recognizeSelected(); var addedIDs = await waitForItemEvent("add"); var modifiedIDs = await waitForItemEvent("modify"); // Item and note assert.lengthOf(addedIDs, 2); var item = Zotero.Items.get(addedIDs[0]); assert.lengthOf(modifiedIDs, 1); // Wait for status to show as complete var progressWindow = getWindows("chrome://zotero/content/progressQueueDialog.xhtml")[0]; var completeStr = Zotero.getString("general.finished"); while (progressWindow.document.getElementById("label").value != completeStr) { await Zotero.Promise.delay(20); } assert.isTrue(collection.hasItem(item.id)); }); it("should recognize PDF by arXiv ID and put new item in same collection in group library", async function () { if (Zotero.automatedTest) this.skip(); // TODO: Mock services this.timeout(30000); var testdir = getTestDataDirectory(); testdir.append("recognizePDF_test_arXiv.pdf"); var group = await getGroup(); var collection = await createDataObject('collection', { libraryID: group.libraryID }); var attachment = await Zotero.Attachments.importFromFile({ libraryID: group.libraryID, file: testdir, collections: [collection.id], }); win.ZoteroPane.recognizeSelected(); var addedIDs = await waitForItemEvent("add"); var modifiedIDs = await waitForItemEvent("modify"); // Item and note assert.lengthOf(addedIDs, 2); var item = Zotero.Items.get(addedIDs[0]); assert.lengthOf(modifiedIDs, 1); // Wait for status to show as complete var progressWindow = getWindows("chrome://zotero/content/progressQueueDialog.xhtml")[0]; var completeStr = Zotero.getString("general.finished"); while (progressWindow.document.getElementById("label").value != completeStr) { await Zotero.Promise.delay(20); } assert.isTrue(collection.hasItem(item.id)); }); it.skip("should recognize PDF by ISBN and put new item in same collection in group library", async function () { if (Zotero.automatedTest) this.skip(); // TODO: Mock services this.timeout(30000); var testdir = getTestDataDirectory(); testdir.append("recognizePDF_test_ISBN.pdf"); var group = await getGroup(); var collection = await createDataObject('collection', { libraryID: group.libraryID }); var attachment = await Zotero.Attachments.importFromFile({ libraryID: group.libraryID, file: testdir, collections: [collection.id], }); win.ZoteroPane.recognizeSelected(); var addedIDs = await waitForItemEvent("add"); var modifiedIDs = await waitForItemEvent("modify"); assert.lengthOf(addedIDs, 1); var item = Zotero.Items.get(addedIDs[0]); assert.lengthOf(modifiedIDs, 2); // Wait for status to show as complete var progressWindow = getWindows("chrome://zotero/content/progressQueueDialog.xhtml")[0]; var completeStr = Zotero.getString("general.finished"); while (progressWindow.document.getElementById("label").value != completeStr) { await Zotero.Promise.delay(20); } assert.isTrue(collection.hasItem(item.id)); }); it("should recognize PDF by title and put new item in same collection in group library", async function () { if (Zotero.automatedTest) this.skip(); // TODO: Mock services this.timeout(30000); var testdir = getTestDataDirectory(); testdir.append("recognizePDF_test_title.pdf"); var group = await getGroup(); var collection = await createDataObject('collection', { libraryID: group.libraryID }); var attachment = await Zotero.Attachments.importFromFile({ libraryID: group.libraryID, file: testdir, collections: [collection.id], }); win.ZoteroPane.recognizeSelected(); var addedIDs = await waitForItemEvent("add"); var modifiedIDs = await waitForItemEvent("modify"); assert.lengthOf(addedIDs, 1); var item = Zotero.Items.get(addedIDs[0]); assert.lengthOf(modifiedIDs, 2); // Wait for status to show as complete var progressWindow = getWindows("chrome://zotero/content/progressQueueDialog.xhtml")[0]; var completeStr = Zotero.getString("general.finished"); while (progressWindow.document.getElementById("label").value != completeStr) { await Zotero.Promise.delay(20); } assert.isTrue(collection.hasItem(item.id)); }); it("should rename a linked file attachment using parent metadata if no existing file attachments and pref enabled", async function () { Zotero.Prefs.set('autoRenameFiles.linked', true); var itemTitle = Zotero.Utilities.randomString();; Zotero.RecognizePDF.recognizeStub = async function () { return createDataObject('item', { title: itemTitle }); }; // Link to the PDF var tempDir = await getTempDirectory(); var tempFile = OS.Path.join(tempDir, 'test.pdf'); await OS.File.copy(OS.Path.join(getTestDataDirectory().path, 'test.pdf'), tempFile); var attachment = await Zotero.Attachments.linkFromFile({ file: tempFile }); win.ZoteroPane.recognizeSelected(); var addedIDs = await waitForItemEvent("add"); var modifiedIDs = await waitForItemEvent("modify"); assert.lengthOf(addedIDs, 1); var item = Zotero.Items.get(addedIDs[0]); assert.equal(item.getField("title"), itemTitle); assert.lengthOf(modifiedIDs, 2); // Wait for status to show as complete var progressWindow = getWindows("chrome://zotero/content/progressQueueDialog.xhtml")[0]; var completeStr = Zotero.getString("general.finished"); while (progressWindow.document.getElementById("label").value != completeStr) { await Zotero.Promise.delay(20); } // The file should have been renamed assert.equal( attachment.attachmentFilename, Zotero.Attachments.getFileBaseNameFromItem(item) + '.pdf' ); }); it("shouldn't rename a linked file attachment using parent metadata if pref disabled", async function () { Zotero.Prefs.set('autoRenameFiles.linked', false); var itemTitle = Zotero.Utilities.randomString();; Zotero.RecognizePDF.recognizeStub = async function () { return createDataObject('item', { title: itemTitle }); }; // Link to the PDF var tempDir = await getTempDirectory(); var tempFile = OS.Path.join(tempDir, 'test.pdf'); await OS.File.copy(OS.Path.join(getTestDataDirectory().path, 'test.pdf'), tempFile); var attachment = await Zotero.Attachments.linkFromFile({ file: tempFile }); win.ZoteroPane.recognizeSelected(); var addedIDs = await waitForItemEvent("add"); var modifiedIDs = await waitForItemEvent("modify"); assert.lengthOf(addedIDs, 1); var item = Zotero.Items.get(addedIDs[0]); assert.equal(item.getField("title"), itemTitle); assert.lengthOf(modifiedIDs, 2); // Wait for status to show as complete var progressWindow = getWindows("chrome://zotero/content/progressQueueDialog.xhtml")[0]; var completeStr = Zotero.getString("general.finished"); while (progressWindow.document.getElementById("label").value != completeStr) { await Zotero.Promise.delay(20); } // The file should not have been renamed assert.equal(attachment.attachmentFilename, 'test.pdf'); }); });