--- a/browser/components/migration/tests/marionette/test_refresh_firefox.py
+++ b/browser/components/migration/tests/marionette/test_refresh_firefox.py
@@ -40,55 +40,51 @@ class TestFirefoxRefresh(MarionetteTestC
arguments[1],
"username",
"password"
);
Services.logins.addLogin(myLogin)
""", script_args=(self._username, self._password))
def createBookmarkInMenu(self):
- self.marionette.execute_script("""
+ error = self.runAsyncCode("""
let url = arguments[0];
let title = arguments[1];
- PlacesUtils.bookmarks.insertBookmark(PlacesUtils.bookmarks.bookmarksMenuFolder,
- makeURI(url), 0, title);
+ PlacesUtils.bookmarks.insert({
+ parentGuid: PlacesUtils.bookmarks.menuGuid, url, title
+ }).then(() => marionetteScriptFinished(false), marionetteScriptFinished);
""", script_args=(self._bookmarkURL, self._bookmarkText))
+ if error:
+ print error
def createBookmarksOnToolbar(self):
- self.marionette.execute_script("""
+ error = self.runAsyncCode("""
+ let children = [];
for (let i = 1; i <= 5; i++) {
- PlacesUtils.bookmarks.insertBookmark(PlacesUtils.toolbarFolderId,
- makeURI(`about:rights?p=${i}`), 0, `Bookmark ${i}`);
+ children.push({url: `about:rights?p=${i}`, title: `Bookmark ${i}`});
}
+ PlacesUtils.bookmarks.insertTree({
+ guid: PlacesUtils.bookmarks.toolbarGuid,
+ children
+ }).then(() => marionetteScriptFinished(false), marionetteScriptFinished);
""")
+ if error:
+ print error
def createHistory(self):
error = self.runAsyncCode("""
- // Copied from PlacesTestUtils, which isn't available in Marionette tests.
- let didReturn;
- PlacesUtils.asyncHistory.updatePlaces(
- [{title: arguments[1], uri: makeURI(arguments[0]), visits: [{
- transitionType: Ci.nsINavHistoryService.TRANSITION_LINK,
- visitDate: (Date.now() - 5000) * 1000,
- referrerURI: makeURI("about:mozilla"),
- }]
- }],
- {
- handleError(resultCode, place) {
- didReturn = true;
- marionetteScriptFinished("Unexpected error in adding visit: " + resultCode);
- },
- handleResult() {},
- handleCompletion() {
- if (!didReturn) {
- marionetteScriptFinished(false);
- }
- },
- }
- );
+ PlacesUtils.history.insert({
+ url: arguments[0],
+ title: arguments[1],
+ visits: [{
+ date: new Date(Date.now() - 5000),
+ referrer: "about:mozilla"
+ }]
+ }).then(() => marionetteScriptFinished(false),
+ ex => marionetteScriptFinished("Unexpected error in adding visit: " + ex));
""", script_args=(self._historyURL, self._historyTitle))
if error:
print error
def createFormHistory(self):
error = self.runAsyncCode("""
let updateDefinition = {
op: "add",
@@ -204,20 +200,22 @@ class TestFirefoxRefresh(MarionetteTestC
loginCount = self.marionette.execute_script("""
return Services.logins.getAllLogins().length;
""")
# Note that we expect 2 logins - one from us, one from sync.
self.assertEqual(loginCount, 2, "No other logins are present")
def checkBookmarkInMenu(self):
- titleInBookmarks = self.marionette.execute_script("""
+ titleInBookmarks = self.runAsyncCode("""
let url = arguments[0];
- let bookmarkIds = PlacesUtils.bookmarks.getBookmarkIdsForURI(makeURI(url), {}, {});
- return bookmarkIds.length == 1 ? PlacesUtils.bookmarks.getItemTitle(bookmarkIds[0]) : "";
+ PlacesUtils.bookmarks.fetch({url}).then(
+ bookmark => marionetteScriptFinished(bookmark ? bookmark.title : ""),
+ ex => marionetteScriptFinished(ex)
+ );
""", script_args=(self._bookmarkURL,))
self.assertEqual(titleInBookmarks, self._bookmarkText)
def checkBookmarkToolbarVisibility(self):
toolbarVisible = self.marionette.execute_script("""
const BROWSER_DOCURL = "chrome://browser/content/browser.xul";
let xulStore = Cc["@mozilla.org/xul/xulstore;1"].getService(Ci.nsIXULStore);
return xulStore.getValue(BROWSER_DOCURL, "PersonalToolbar", "collapsed")