michael@0: /* Any copyright is dedicated to the Public Domain.
michael@0:    http://creativecommons.org/publicdomain/zero/1.0/ */
michael@0: 
michael@0: function promiseGetPlacesInfo(aPlacesIdentifiers) {
michael@0:   let deferred = Promise.defer();
michael@0:   PlacesUtils.asyncHistory.getPlacesInfo(aPlacesIdentifiers, {
michael@0:     _results: [],
michael@0:     _errors: [],
michael@0: 
michael@0:     handleResult: function handleResult(aPlaceInfo) {
michael@0:       this._results.push(aPlaceInfo);
michael@0:     },
michael@0:     handleError: function handleError(aResultCode, aPlaceInfo) {
michael@0:       this._errors.push({ resultCode: aResultCode, info: aPlaceInfo });
michael@0:     },
michael@0:     handleCompletion: function handleCompletion() {
michael@0:       deferred.resolve({ errors: this._errors, results: this._results });
michael@0:     }
michael@0:   });
michael@0: 
michael@0:   return deferred.promise;
michael@0: }
michael@0: 
michael@0: function ensurePlacesInfoObjectsAreEqual(a, b) {
michael@0:   do_check_true(a.uri.equals(b.uri));
michael@0:   do_check_eq(a.title, b.title);
michael@0:   do_check_eq(a.guid, b.guid);
michael@0:   do_check_eq(a.placeId, b.placeId);
michael@0: }
michael@0: 
michael@0: function test_getPlacesInfoExistentPlace() {
michael@0:   let testURI = NetUtil.newURI("http://www.example.tld");
michael@0:   yield promiseAddVisits(testURI);
michael@0: 
michael@0:   let getPlacesInfoResult = yield promiseGetPlacesInfo([testURI]);
michael@0:   do_check_eq(getPlacesInfoResult.results.length, 1);
michael@0:   do_check_eq(getPlacesInfoResult.errors.length, 0);
michael@0: 
michael@0:   let placeInfo = getPlacesInfoResult.results[0];
michael@0:   do_check_true(placeInfo instanceof Ci.mozIPlaceInfo);
michael@0: 
michael@0:   do_check_true(placeInfo.uri.equals(testURI));
michael@0:   do_check_eq(placeInfo.title, "test visit for " + testURI.spec);
michael@0:   do_check_true(placeInfo.guid.length > 0);
michael@0:   do_check_eq(placeInfo.visits, null);
michael@0: }
michael@0: add_task(test_getPlacesInfoExistentPlace);
michael@0: 
michael@0: function test_getPlacesInfoNonExistentPlace() {
michael@0:   let testURI = NetUtil.newURI("http://www.example_non_existent.tld");
michael@0:   let getPlacesInfoResult = yield promiseGetPlacesInfo(testURI);
michael@0:   do_check_eq(getPlacesInfoResult.results.length, 0);
michael@0:   do_check_eq(getPlacesInfoResult.errors.length, 1);
michael@0: }
michael@0: add_task(test_getPlacesInfoNonExistentPlace);
michael@0: 
michael@0: function test_promisedHelper() {
michael@0:   let (uri = NetUtil.newURI("http://www.helper_existent_example.tld")) {
michael@0:     yield promiseAddVisits(uri);
michael@0:     let placeInfo = yield PlacesUtils.promisePlaceInfo(uri);
michael@0:     do_check_true(placeInfo instanceof Ci.mozIPlaceInfo);
michael@0:   };
michael@0: 
michael@0:   let (uri = NetUtil.newURI("http://www.helper_non_existent_example.tld")) {
michael@0:     try {
michael@0:       let placeInfo = yield PlacesUtils.promisePlaceInfo(uri);
michael@0:       do_throw("PlacesUtils.promisePlaceInfo should have rejected the promise");
michael@0:     }
michael@0:     catch(ex) { }
michael@0:   };
michael@0: }
michael@0: add_task(test_promisedHelper);
michael@0: 
michael@0: function test_infoByGUID() {
michael@0:   let testURI = NetUtil.newURI("http://www.guid_example.tld");
michael@0:   yield promiseAddVisits(testURI);
michael@0: 
michael@0:   let placeInfoByURI = yield PlacesUtils.promisePlaceInfo(testURI);
michael@0:   let placeInfoByGUID = yield PlacesUtils.promisePlaceInfo(placeInfoByURI.guid);
michael@0:   ensurePlacesInfoObjectsAreEqual(placeInfoByURI, placeInfoByGUID);
michael@0: }
michael@0: add_task(test_infoByGUID);
michael@0: 
michael@0: function test_invalid_guid() {
michael@0:   try {
michael@0:     let placeInfoByGUID = yield PlacesUtils.promisePlaceInfo("###");
michael@0:     do_throw("getPlacesInfo should fail for invalid guids")
michael@0:   }
michael@0:   catch(ex) { }
michael@0: }
michael@0: add_task(test_invalid_guid);
michael@0: 
michael@0: function test_mixed_selection() {
michael@0:   let placeInfo1, placeInfo2;
michael@0:   let (uri = NetUtil.newURI("http://www.mixed_selection_test_1.tld")) {
michael@0:     yield promiseAddVisits(uri);
michael@0:     placeInfo1 = yield PlacesUtils.promisePlaceInfo(uri);
michael@0:   };
michael@0: 
michael@0:   let (uri = NetUtil.newURI("http://www.mixed_selection_test_2.tld")) {
michael@0:     yield promiseAddVisits(uri);
michael@0:     placeInfo2 = yield PlacesUtils.promisePlaceInfo(uri);
michael@0:   };
michael@0: 
michael@0:   let getPlacesInfoResult = yield promiseGetPlacesInfo([placeInfo1.uri, placeInfo2.guid]);
michael@0:   do_check_eq(getPlacesInfoResult.results.length, 2);
michael@0:   do_check_eq(getPlacesInfoResult.errors.length, 0);
michael@0: 
michael@0:   do_check_eq(getPlacesInfoResult.results[0].uri.spec, placeInfo1.uri.spec);
michael@0:   do_check_eq(getPlacesInfoResult.results[1].guid, placeInfo2.guid);
michael@0: }
michael@0: add_task(test_mixed_selection);
michael@0: 
michael@0: function run_test() {
michael@0:   run_next_test();
michael@0: }