michael@0: /* Any copyright is dedicated to the Public Domain.
michael@0:  *    http://creativecommons.org/publicdomain/zero/1.0/ */
michael@0: 
michael@0: /*
michael@0:  * Tests for mozIAsyncFavicons::replaceFaviconData()
michael@0:  */
michael@0: 
michael@0: let iconsvc = PlacesUtils.favicons;
michael@0: let histsvc = PlacesUtils.history;
michael@0: 
michael@0: let originalFavicon = {
michael@0:   file: do_get_file("favicon-normal16.png"),
michael@0:   uri: uri(do_get_file("favicon-normal16.png")),
michael@0:   data: readFileData(do_get_file("favicon-normal16.png")),
michael@0:   mimetype: "image/png"
michael@0: };
michael@0: 
michael@0: let uniqueFaviconId = 0;
michael@0: function createFavicon(fileName) {
michael@0:   let tempdir = Services.dirsvc.get("TmpD", Ci.nsILocalFile);
michael@0: 
michael@0:   // remove any existing file at the path we're about to copy to
michael@0:   let outfile = tempdir.clone();
michael@0:   outfile.append(fileName);
michael@0:   try { outfile.remove(false); } catch (e) {}
michael@0: 
michael@0:   originalFavicon.file.copyToFollowingLinks(tempdir, fileName);
michael@0: 
michael@0:   let stream = Cc["@mozilla.org/network/file-output-stream;1"]
michael@0:     .createInstance(Ci.nsIFileOutputStream);
michael@0:   stream.init(outfile, 0x02 | 0x08 | 0x10, 0600, 0);
michael@0: 
michael@0:   // append some data that sniffers/encoders will ignore that will distinguish
michael@0:   // the different favicons we'll create
michael@0:   uniqueFaviconId++;
michael@0:   let uniqueStr = "uid:" + uniqueFaviconId;
michael@0:   stream.write(uniqueStr, uniqueStr.length);
michael@0:   stream.close();
michael@0: 
michael@0:   do_check_eq(outfile.leafName.substr(0, fileName.length), fileName);
michael@0: 
michael@0:   return {
michael@0:     file: outfile,
michael@0:     uri: uri(outfile),
michael@0:     data: readFileData(outfile),
michael@0:     mimetype: "image/png"
michael@0:   };
michael@0: }
michael@0: 
michael@0: function checkCallbackSucceeded(callbackMimetype, callbackData, sourceMimetype, sourceData) {
michael@0:   do_check_eq(callbackMimetype, sourceMimetype);
michael@0:   do_check_true(compareArrays(callbackData, sourceData));
michael@0: }
michael@0: 
michael@0: function run_test() {
michael@0:   // check that the favicon loaded correctly
michael@0:   do_check_eq(originalFavicon.data.length, 286);
michael@0:   run_next_test();
michael@0: };
michael@0: 
michael@0: add_task(function test_replaceFaviconData_validHistoryURI() {
michael@0:   do_log_info("test replaceFaviconData for valid history uri");
michael@0: 
michael@0:   let pageURI = uri("http://test1.bar/");
michael@0:   yield promiseAddVisits(pageURI);
michael@0: 
michael@0:   let favicon = createFavicon("favicon1.png");
michael@0: 
michael@0:   iconsvc.replaceFaviconData(favicon.uri, favicon.data, favicon.data.length,
michael@0:     favicon.mimetype);
michael@0: 
michael@0:   let deferSetAndFetchFavicon = Promise.defer();
michael@0:   iconsvc.setAndFetchFaviconForPage(pageURI, favicon.uri, true,
michael@0:     PlacesUtils.favicons.FAVICON_LOAD_NON_PRIVATE,
michael@0:     function test_replaceFaviconData_validHistoryURI_check(aURI, aDataLen, aData, aMimeType) {
michael@0:       checkCallbackSucceeded(aMimeType, aData, favicon.mimetype, favicon.data);
michael@0:       checkFaviconDataForPage(
michael@0:         pageURI, favicon.mimetype, favicon.data,
michael@0:         function test_replaceFaviconData_validHistoryURI_callback() {
michael@0:           favicon.file.remove(false);
michael@0:           deferSetAndFetchFavicon.resolve();
michael@0:         });
michael@0:     });
michael@0:   yield deferSetAndFetchFavicon.promise;
michael@0: 
michael@0:   yield promiseClearHistory();
michael@0: });
michael@0: 
michael@0: add_task(function test_replaceFaviconData_overrideDefaultFavicon() {
michael@0:   do_log_info("test replaceFaviconData to override a later setAndFetchFaviconForPage");
michael@0: 
michael@0:   let pageURI = uri("http://test2.bar/");
michael@0:   yield promiseAddVisits(pageURI);
michael@0: 
michael@0:   let firstFavicon = createFavicon("favicon2.png");
michael@0:   let secondFavicon = createFavicon("favicon3.png");
michael@0: 
michael@0:   iconsvc.replaceFaviconData(
michael@0:     firstFavicon.uri, secondFavicon.data, secondFavicon.data.length,
michael@0:     secondFavicon.mimetype);
michael@0: 
michael@0:   let deferSetAndFetchFavicon = Promise.defer();
michael@0:   iconsvc.setAndFetchFaviconForPage(
michael@0:     pageURI, firstFavicon.uri, true,
michael@0:     PlacesUtils.favicons.FAVICON_LOAD_NON_PRIVATE,
michael@0:     function test_replaceFaviconData_overrideDefaultFavicon_check(aURI, aDataLen, aData, aMimeType) {
michael@0:       checkCallbackSucceeded(aMimeType, aData, secondFavicon.mimetype, secondFavicon.data);
michael@0:       checkFaviconDataForPage(
michael@0:         pageURI, secondFavicon.mimetype, secondFavicon.data,
michael@0:         function test_replaceFaviconData_overrideDefaultFavicon_callback() {
michael@0:           firstFavicon.file.remove(false);
michael@0:           secondFavicon.file.remove(false);
michael@0:           deferSetAndFetchFavicon.resolve();
michael@0:         });
michael@0:     });
michael@0:   yield deferSetAndFetchFavicon.promise;
michael@0: 
michael@0:   yield promiseClearHistory();
michael@0: });
michael@0: 
michael@0: add_task(function test_replaceFaviconData_replaceExisting() {
michael@0:   do_log_info("test replaceFaviconData to override a previous setAndFetchFaviconForPage");
michael@0: 
michael@0:   let pageURI = uri("http://test3.bar");
michael@0:   yield promiseAddVisits(pageURI);
michael@0: 
michael@0:   let firstFavicon = createFavicon("favicon4.png");
michael@0:   let secondFavicon = createFavicon("favicon5.png");
michael@0: 
michael@0:   let deferSetAndFetchFavicon = Promise.defer();
michael@0:   iconsvc.setAndFetchFaviconForPage(
michael@0:     pageURI, firstFavicon.uri, true,
michael@0:     PlacesUtils.favicons.FAVICON_LOAD_NON_PRIVATE,
michael@0:     function test_replaceFaviconData_replaceExisting_firstSet_check(aURI, aDataLen, aData, aMimeType) {
michael@0:       checkCallbackSucceeded(aMimeType, aData, firstFavicon.mimetype, firstFavicon.data);
michael@0:       checkFaviconDataForPage(
michael@0:         pageURI, firstFavicon.mimetype, firstFavicon.data,
michael@0:         function test_replaceFaviconData_overrideDefaultFavicon_firstCallback() {
michael@0:           iconsvc.replaceFaviconData(
michael@0:             firstFavicon.uri, secondFavicon.data, secondFavicon.data.length,
michael@0:             secondFavicon.mimetype);
michael@0:           promiseAsyncUpdates().then(function() {
michael@0:             checkFaviconDataForPage(
michael@0:               pageURI, secondFavicon.mimetype, secondFavicon.data,
michael@0:               function test_replaceFaviconData_overrideDefaultFavicon_secondCallback() {
michael@0:                 firstFavicon.file.remove(false);
michael@0:                 secondFavicon.file.remove(false);
michael@0:                 deferSetAndFetchFavicon.resolve();
michael@0:               });
michael@0:           });
michael@0:         });
michael@0:     });
michael@0:   yield deferSetAndFetchFavicon.promise;
michael@0: 
michael@0:   yield promiseClearHistory();
michael@0: });
michael@0: 
michael@0: add_task(function test_replaceFaviconData_unrelatedReplace() {
michael@0:   do_log_info("test replaceFaviconData to not make unrelated changes");
michael@0: 
michael@0:   let pageURI = uri("http://test4.bar/");
michael@0:   yield promiseAddVisits(pageURI);
michael@0: 
michael@0:   let favicon = createFavicon("favicon6.png");
michael@0:   let unrelatedFavicon = createFavicon("favicon7.png");
michael@0: 
michael@0:   iconsvc.replaceFaviconData(
michael@0:     unrelatedFavicon.uri, unrelatedFavicon.data, unrelatedFavicon.data.length,
michael@0:     unrelatedFavicon.mimetype);
michael@0: 
michael@0:   let deferSetAndFetchFavicon = Promise.defer();
michael@0:   iconsvc.setAndFetchFaviconForPage(
michael@0:     pageURI, favicon.uri, true,
michael@0:     PlacesUtils.favicons.FAVICON_LOAD_NON_PRIVATE,
michael@0:     function test_replaceFaviconData_unrelatedReplace_check(aURI, aDataLen, aData, aMimeType) {
michael@0:       checkCallbackSucceeded(aMimeType, aData, favicon.mimetype, favicon.data);
michael@0:       checkFaviconDataForPage(
michael@0:         pageURI, favicon.mimetype, favicon.data,
michael@0:         function test_replaceFaviconData_unrelatedReplace_callback() {
michael@0:           favicon.file.remove(false);
michael@0:           unrelatedFavicon.file.remove(false);
michael@0:           deferSetAndFetchFavicon.resolve();
michael@0:         });
michael@0:     });
michael@0:   yield deferSetAndFetchFavicon.promise;
michael@0: 
michael@0:   yield promiseClearHistory();
michael@0: });
michael@0: 
michael@0: add_task(function test_replaceFaviconData_badInputs() {
michael@0:   do_log_info("test replaceFaviconData to throw on bad inputs");
michael@0: 
michael@0:   let favicon = createFavicon("favicon8.png");
michael@0: 
michael@0:   let ex = null;
michael@0:   try {
michael@0:     iconsvc.replaceFaviconData(
michael@0:       favicon.uri, favicon.data, favicon.data.length, "");
michael@0:   } catch (e) {
michael@0:     ex = e;
michael@0:   } finally {
michael@0:     do_check_true(!!ex);
michael@0:   }
michael@0: 
michael@0:   ex = null;
michael@0:   try {
michael@0:     iconsvc.replaceFaviconData(
michael@0:       null, favicon.data, favicon.data.length, favicon.mimeType);
michael@0:   } catch (e) {
michael@0:     ex = e;
michael@0:   } finally {
michael@0:     do_check_true(!!ex);
michael@0:   }
michael@0: 
michael@0:   ex = null;
michael@0:   try {
michael@0:     iconsvc.replaceFaviconData(
michael@0:       favicon.uri, null, 0, favicon.mimeType);
michael@0:   } catch (e) {
michael@0:     ex = e;
michael@0:   } finally {
michael@0:     do_check_true(!!ex);
michael@0:   }
michael@0: 
michael@0:   favicon.file.remove(false);
michael@0: 
michael@0:   yield promiseClearHistory();
michael@0: });
michael@0: 
michael@0: add_task(function test_replaceFaviconData_twiceReplace() {
michael@0:   do_log_info("test replaceFaviconData on multiple replacements");
michael@0: 
michael@0:   let pageURI = uri("http://test5.bar/");
michael@0:   yield promiseAddVisits(pageURI);
michael@0: 
michael@0:   let firstFavicon = createFavicon("favicon9.png");
michael@0:   let secondFavicon = createFavicon("favicon10.png");
michael@0: 
michael@0:   iconsvc.replaceFaviconData(
michael@0:     firstFavicon.uri, firstFavicon.data, firstFavicon.data.length,
michael@0:     firstFavicon.mimetype);
michael@0:   iconsvc.replaceFaviconData(
michael@0:     firstFavicon.uri, secondFavicon.data, secondFavicon.data.length,
michael@0:     secondFavicon.mimetype);
michael@0: 
michael@0:   let deferSetAndFetchFavicon = Promise.defer();
michael@0:   iconsvc.setAndFetchFaviconForPage(
michael@0:     pageURI, firstFavicon.uri, true,
michael@0:     PlacesUtils.favicons.FAVICON_LOAD_NON_PRIVATE,
michael@0:     function test_replaceFaviconData_twiceReplace_check(aURI, aDataLen, aData, aMimeType) {
michael@0:       checkCallbackSucceeded(aMimeType, aData, secondFavicon.mimetype, secondFavicon.data);
michael@0:       checkFaviconDataForPage(
michael@0:         pageURI, secondFavicon.mimetype, secondFavicon.data,
michael@0:         function test_replaceFaviconData_twiceReplace_callback() {
michael@0:           firstFavicon.file.remove(false);
michael@0:           secondFavicon.file.remove(false);
michael@0:           deferSetAndFetchFavicon.resolve();
michael@0:         });
michael@0:     });
michael@0:   yield deferSetAndFetchFavicon.promise;
michael@0: 
michael@0:   yield promiseClearHistory();
michael@0: });