// Useful "constants" var sqlDateTimeRe = /^\d{4}-\d{2}-\d{2} \d{2}:\d{2}:\d{2}$/; var isoDateTimeRe = /^\d{4}-\d{2}-\d{2}T\d{2}:\d{2}:\d{2}Z$/; var zoteroObjectKeyRe = /^[23456789ABCDEFGHIJKLMNPQRSTUVWXYZ]{8}$/; // based on Zotero.Utilities::generateObjectKey() /** * Waits for a DOM event on the specified node. Returns a promise * resolved with the event. */ function waitForDOMEvent(target, event, capture) { var deferred = Zotero.Promise.defer(); var func = function(ev) { target.removeEventListener("event", func, capture); deferred.resolve(ev); } target.addEventListener(event, func, capture); return deferred.promise; } /** * Open a chrome window and return a promise for the window * * @return {Promise} */ function loadWindow(winurl, argument) { var win = window.openDialog(winurl, "_blank", "chrome", argument); return waitForDOMEvent(win, "load").then(function() { return win; }); } /** * Open a browser window and return a promise for the window * * @return {Promise} */ function loadBrowserWindow() { var win = window.openDialog("chrome://browser/content/browser.xul", "", "all,height=400,width=1000"); return waitForDOMEvent(win, "load").then(function() { return win; }); } /** * Loads a Zotero pane in a new window and selects My Library. Returns the containing window. */ var loadZoteroPane = Zotero.Promise.coroutine(function* () { var win = yield loadBrowserWindow(); Zotero.Prefs.clear('lastViewedFolder'); win.ZoteroOverlay.toggleDisplay(true); // Hack to wait for pane load to finish. This is the same hack // we use in ZoteroPane.js, so either it's not good enough // there or it should be good enough here. yield Zotero.Promise.delay(52); yield waitForItemsLoad(win, 0); return win; }); /** * Waits for a window with a specific URL to open. Returns a promise for the window, and * optionally passes the window to a callback immediately for use with modal dialogs, * which prevent async code from continuing */ function waitForWindow(uri, callback) { var deferred = Zotero.Promise.defer(); Components.utils.import("resource://gre/modules/Services.jsm"); var loadobserver = function(ev) { ev.originalTarget.removeEventListener("load", loadobserver, false); if(ev.target.location.href == uri) { Services.ww.unregisterNotification(winobserver); var win = ev.target.docShell .QueryInterface(Components.interfaces.nsIInterfaceRequestor) .getInterface(Components.interfaces.nsIDOMWindow); if (callback) { callback(win); } deferred.resolve(win); } }; var winobserver = {"observe":function(subject, topic, data) { if(topic != "domwindowopened") return; var win = subject.QueryInterface(Components.interfaces.nsIDOMWindow); win.addEventListener("load", loadobserver, false); }}; Services.ww.registerNotification(winobserver); return deferred.promise; } var selectLibrary = Zotero.Promise.coroutine(function* (win) { yield win.ZoteroPane.collectionsView.selectLibrary(Zotero.Libraries.userLibraryID); yield waitForItemsLoad(win); }); var waitForItemsLoad = function (win, collectionRowToSelect) { var resolve; var promise = new Zotero.Promise(() => resolve = arguments[0]); var zp = win.ZoteroPane; var cv = zp.collectionsView; cv.addEventListener('load', function () { if (collectionRowToSelect !== undefined) { cv.selection.select(collectionRowToSelect); } zp.addEventListener('itemsLoaded', function () { resolve(); }); }); return promise; } /** * Waits for a single item event. Returns a promise for the item ID(s). */ function waitForItemEvent(event) { var deferred = Zotero.Promise.defer(); var notifierID = Zotero.Notifier.registerObserver({notify:function(ev, type, ids, extraData) { if(ev == event) { Zotero.Notifier.unregisterObserver(notifierID); deferred.resolve(ids); } }}, ["item"]); return deferred.promise; } /** * Looks for windows with a specific URL. */ function getWindows(uri) { Components.utils.import("resource://gre/modules/Services.jsm"); var enumerator = Services.wm.getEnumerator(null); var wins = []; while(enumerator.hasMoreElements()) { var win = enumerator.getNext(); if(win.location == uri) { wins.push(win); } } return wins; } /** * Resolve a promise when a specified callback returns true. interval * specifies the interval between checks. timeout specifies when we * should assume failure. */ function waitForCallback(cb, interval, timeout) { var deferred = Zotero.Promise.defer(); if(interval === undefined) interval = 100; if(timeout === undefined) timeout = 10000; var start = Date.now(); var id = setInterval(function() { var success = cb(); if(success) { clearInterval(id); deferred.resolve(success); } else if(Date.now() - start > timeout*1000) { clearInterval(id); deferred.reject(new Error("Promise timed out")); } }, interval); return deferred.promise; } // // Data objects // function createUnsavedDataObject(objectType, params) { params = params || {}; if (objectType == 'item') { var param = params.itemType || 'book'; } var obj = new Zotero[Zotero.Utilities.capitalize(objectType)](param); switch (objectType) { case 'collection': case 'search': obj.name = params.name !== undefined ? params.name : "Test"; break; } var allowedParams = ['parentID', 'parentKey', 'synced', 'version']; allowedParams.forEach(function (param) { if (params[param] !== undefined) { obj[param] = params[param]; } }) return obj; } var createDataObject = Zotero.Promise.coroutine(function* (objectType, params, saveOptions) { var obj = createUnsavedDataObject(objectType, params); yield obj.saveTx(saveOptions); return obj; }); /** * Return a promise for the error thrown by a promise, or false if none */ function getPromiseError(promise) { return promise.thenReturn(false).catch(e => e); } /** * Ensures that the PDF tools are installed, or installs them if not. * * @return {Promise} */ var installPDFTools = Zotero.Promise.coroutine(function* () { if(Zotero.Fulltext.pdfConverterIsRegistered() && Zotero.Fulltext.pdfInfoIsRegistered()) { return; } var version = yield Zotero.Fulltext.getLatestPDFToolsVersion(); yield Zotero.Fulltext.downloadPDFTool('info', version); yield Zotero.Fulltext.downloadPDFTool('converter', version); }); /** * @return {Promise} */ function uninstallPDFTools() { return Zotero.Fulltext.removePDFTools(); } /** * Returns the nsIFile corresponding to the test data directory * (i.e., test/tests/data) */ function getTestDataDirectory() { Components.utils.import("resource://gre/modules/Services.jsm"); var resource = Services.io.getProtocolHandler("resource"). QueryInterface(Components.interfaces.nsIResProtocolHandler), resURI = Services.io.newURI("resource://zotero-unit-tests/data", null, null); return Services.io.newURI(resource.resolveURI(resURI), null, null). QueryInterface(Components.interfaces.nsIFileURL).file; } /** * Returns an absolute path to an empty temporary directory * (i.e., test/tests/data) */ var getTempDirectory = Zotero.Promise.coroutine(function* getTempDirectory() { Components.utils.import("resource://gre/modules/osfile.jsm"); let path, attempts = 3, zoteroTmpDirPath = Zotero.getTempDirectory().path; while (attempts--) { path = OS.Path.join(zoteroTmpDirPath, Zotero.Utilities.randomString()); try { yield OS.File.makeDir(path, { ignoreExisting: false }); break; } catch (e) { if (!attempts) throw e; // Throw on last attempt } } return path; }); /** * Resets the Zotero DB and restarts Zotero. Returns a promise resolved * when this finishes. */ function resetDB() { var db = Zotero.getZoteroDatabase(); return Zotero.reinit(function() { db.remove(false); }).then(function() { return Zotero.Schema.schemaUpdatePromise; }); } /** * Equivalent to JSON.stringify, except that object properties are stringified * in a sorted order. */ function stableStringify(obj) { return JSON.stringify(obj, function(k, v) { if (v && typeof v == "object" && !Array.isArray(v)) { let o = {}, keys = Object.keys(v).sort(); for (let i = 0; i < keys.length; i++) { o[keys[i]] = v[keys[i]]; } return o; } return v; }, "\t"); } /** * Loads specified sample data from file */ function loadSampleData(dataName) { let data = Zotero.File.getContentsFromURL('resource://zotero-unit-tests/data/' + dataName + '.js'); return JSON.parse(data); } /** * Generates sample item data that is stored in data/sampleItemData.js */ function generateAllTypesAndFieldsData() { let data = {}; let itemTypes = Zotero.ItemTypes.getTypes(); // For most fields, use the field name as the value, but this doesn't // work well for some fields that expect values in certain formats let specialValues = { date: '1999-12-31', filingDate: '2000-01-02', accessDate: '1997-06-13 23:59:58', number: 3, numPages: 4, issue: 5, volume: 6, numberOfVolumes: 7, edition: 8, seriesNumber: 9, ISBN: '978-1-234-56789-7', ISSN: '1234-5679', url: 'http://www.example.com', pages: '1-10', DOI: '10.1234/example.doi', runningTime: '1:22:33', language: 'en-US' }; // Item types that should not be included in sample data let excludeItemTypes = ['note', 'attachment']; for (let i = 0; i < itemTypes.length; i++) { if (excludeItemTypes.indexOf(itemTypes[i].name) != -1) continue; let itemFields = data[itemTypes[i].name] = { itemType: itemTypes[i].name }; let fields = Zotero.ItemFields.getItemTypeFields(itemTypes[i].id); for (let j = 0; j < fields.length; j++) { let field = fields[j]; field = Zotero.ItemFields.getBaseIDFromTypeAndField(itemTypes[i].id, field) || field; let name = Zotero.ItemFields.getName(field), value; // Use field name as field value if (specialValues[name]) { value = specialValues[name]; } else { value = name.charAt(0).toUpperCase() + name.substr(1); // Make it look nice (sentence case) value = value.replace(/([a-z])([A-Z])/g, '$1 $2') .replace(/ [A-Z](?![A-Z])/g, m => m.toLowerCase()); // not all-caps words } itemFields[name] = value; } let creatorTypes = Zotero.CreatorTypes.getTypesForItemType(itemTypes[i].id), creators = itemFields.creators = []; for (let j = 0; j < creatorTypes.length; j++) { let typeName = creatorTypes[j].name; creators.push({ creatorType: typeName, firstName: typeName + 'First', lastName: typeName + 'Last' }); } } return data; } /** * Populates the database with sample items * The field values should be in the form exactly as they would appear in Zotero */ function populateDBWithSampleData(data) { return Zotero.DB.executeTransaction(function* () { for (let itemName in data) { let item = data[itemName]; let zItem = new Zotero.Item(item.itemType); for (let itemField in item) { if (itemField == 'itemType') continue; if (itemField == 'creators') { zItem.setCreators(item[itemField]); continue; } if (itemField == 'tags') { // Must save item first continue; } zItem.setField(itemField, item[itemField]); } if (item.tags && item.tags.length) { for (let i=0; i} */ function importFileAttachment(filename) { let testfile = getTestDataDirectory(); filename.split('/').forEach((part) => testfile.append(part)); return Zotero.Attachments.importFromFile({file: testfile}); }