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: // Checks that we handle a locked database when there are extension changes
michael@0: // in progress
michael@0: 
michael@0: Components.utils.import("resource://gre/modules/osfile.jsm");
michael@0: 
michael@0: // Will be left alone
michael@0: var addon1 = {
michael@0:   id: "addon1@tests.mozilla.org",
michael@0:   version: "1.0",
michael@0:   name: "Test 1",
michael@0:   targetApplications: [{
michael@0:     id: "xpcshell@tests.mozilla.org",
michael@0:     minVersion: "2",
michael@0:     maxVersion: "2"
michael@0:   }]
michael@0: };
michael@0: 
michael@0: // Will be enabled
michael@0: var addon2 = {
michael@0:   id: "addon2@tests.mozilla.org",
michael@0:   version: "1.0",
michael@0:   name: "Test 2",
michael@0:   targetApplications: [{
michael@0:     id: "xpcshell@tests.mozilla.org",
michael@0:     minVersion: "2",
michael@0:     maxVersion: "2"
michael@0:   }]
michael@0: };
michael@0: 
michael@0: // Will be disabled
michael@0: var addon3 = {
michael@0:   id: "addon3@tests.mozilla.org",
michael@0:   version: "1.0",
michael@0:   name: "Test 3",
michael@0:   targetApplications: [{
michael@0:     id: "xpcshell@tests.mozilla.org",
michael@0:     minVersion: "2",
michael@0:     maxVersion: "2"
michael@0:   }]
michael@0: };
michael@0: 
michael@0: // Will be uninstalled
michael@0: var addon4 = {
michael@0:   id: "addon4@tests.mozilla.org",
michael@0:   version: "1.0",
michael@0:   name: "Test 4",
michael@0:   targetApplications: [{
michael@0:     id: "xpcshell@tests.mozilla.org",
michael@0:     minVersion: "2",
michael@0:     maxVersion: "2"
michael@0:   }]
michael@0: };
michael@0: 
michael@0: 
michael@0: // Will be updated
michael@0: var addon5 = {
michael@0:   id: "addon5@tests.mozilla.org",
michael@0:   version: "1.0",
michael@0:   name: "Test 5",
michael@0:   targetApplications: [{
michael@0:     id: "xpcshell@tests.mozilla.org",
michael@0:     minVersion: "2",
michael@0:     maxVersion: "2"
michael@0:   }]
michael@0: };
michael@0: 
michael@0: const profileDir = gProfD.clone();
michael@0: profileDir.append("extensions");
michael@0: 
michael@0: add_task(function() {
michael@0: 
michael@0:   createAppInfo("xpcshell@tests.mozilla.org", "XPCShell", "2", "2");
michael@0: 
michael@0:   writeInstallRDFForExtension(addon1, profileDir);
michael@0:   writeInstallRDFForExtension(addon2, profileDir);
michael@0:   writeInstallRDFForExtension(addon3, profileDir);
michael@0:   writeInstallRDFForExtension(addon4, profileDir);
michael@0:   writeInstallRDFForExtension(addon5, profileDir);
michael@0: 
michael@0:   // Make it look like add-on 5 was installed some time in the past so the update is
michael@0:   // detected
michael@0:   let path = getFileForAddon(profileDir, addon5.id).path;
michael@0:   yield promiseSetExtensionModifiedTime(path, Date.now() - (60000));
michael@0: 
michael@0:   // Startup the profile and setup the initial state
michael@0:   startupManager();
michael@0: 
michael@0:   check_startup_changes(AddonManager.STARTUP_CHANGE_INSTALLED, []);
michael@0:   check_startup_changes(AddonManager.STARTUP_CHANGE_UNINSTALLED, []);
michael@0: 
michael@0:   let a1, a2, a3, a4, a5, a6;
michael@0: 
michael@0:   [a2] = yield promiseAddonsByIDs(["addon2@tests.mozilla.org"]);
michael@0:   a2.userDisabled = true;
michael@0: 
michael@0:   restartManager();
michael@0: 
michael@0:   [a1, a2, a3, a4, a5] =
michael@0:     yield promiseAddonsByIDs(["addon1@tests.mozilla.org",
michael@0:                               "addon2@tests.mozilla.org",
michael@0:                               "addon3@tests.mozilla.org",
michael@0:                               "addon4@tests.mozilla.org",
michael@0:                               "addon5@tests.mozilla.org"]);
michael@0: 
michael@0:   a2.userDisabled = false;
michael@0:   a3.userDisabled = true;
michael@0:   a4.uninstall();
michael@0: 
michael@0:   yield promiseInstallAllFiles([do_get_addon("test_locked2_5"),
michael@0:                                 do_get_addon("test_locked2_6")]);
michael@0:   do_check_neq(a1, null);
michael@0:   do_check_true(a1.isActive);
michael@0:   do_check_false(a1.userDisabled);
michael@0:   do_check_false(a1.appDisabled);
michael@0:   do_check_eq(a1.pendingOperations, AddonManager.PENDING_NONE);
michael@0:   do_check_true(isExtensionInAddonsList(profileDir, a1.id));
michael@0: 
michael@0:   do_check_neq(a2, null);
michael@0:   do_check_false(a2.isActive);
michael@0:   do_check_false(a2.userDisabled);
michael@0:   do_check_false(a2.appDisabled);
michael@0:   do_check_eq(a2.pendingOperations, AddonManager.PENDING_ENABLE);
michael@0:   do_check_false(isExtensionInAddonsList(profileDir, a2.id));
michael@0: 
michael@0:   do_check_neq(a3, null);
michael@0:   do_check_true(a3.isActive);
michael@0:   do_check_true(a3.userDisabled);
michael@0:   do_check_false(a3.appDisabled);
michael@0:   do_check_eq(a3.pendingOperations, AddonManager.PENDING_DISABLE);
michael@0:   do_check_true(isExtensionInAddonsList(profileDir, a3.id));
michael@0: 
michael@0:   do_check_neq(a4, null);
michael@0:   do_check_true(a4.isActive);
michael@0:   do_check_false(a4.userDisabled);
michael@0:   do_check_false(a4.appDisabled);
michael@0:   do_check_eq(a4.pendingOperations, AddonManager.PENDING_UNINSTALL);
michael@0:   do_check_true(isExtensionInAddonsList(profileDir, a4.id));
michael@0: 
michael@0:   do_check_neq(a5, null);
michael@0:   do_check_eq(a5.version, "1.0");
michael@0:   do_check_true(a5.isActive);
michael@0:   do_check_false(a5.userDisabled);
michael@0:   do_check_false(a5.appDisabled);
michael@0:   do_check_eq(a5.pendingOperations, AddonManager.PENDING_UPGRADE);
michael@0:   do_check_true(isExtensionInAddonsList(profileDir, a5.id));
michael@0: 
michael@0:   // Open another handle on the JSON DB with as much Unix and Windows locking
michael@0:   // as we can to simulate some other process interfering with it
michael@0:   shutdownManager();
michael@0:   do_print("Locking " + gExtensionsJSON.path);
michael@0:   let options = {
michael@0:     winShare: 0
michael@0:   };
michael@0:   if (OS.Constants.libc.O_EXLOCK)
michael@0:     options.unixFlags = OS.Constants.libc.O_EXLOCK;
michael@0: 
michael@0:   let file = yield OS.File.open(gExtensionsJSON.path, {read:true, write:true, existing:true}, options);
michael@0: 
michael@0:   let filePermissions = gExtensionsJSON.permissions;
michael@0:   if (!OS.Constants.Win) {
michael@0:     gExtensionsJSON.permissions = 0;
michael@0:   }
michael@0:   startupManager(false);
michael@0: 
michael@0:   check_startup_changes(AddonManager.STARTUP_CHANGE_INSTALLED, []);
michael@0:   check_startup_changes(AddonManager.STARTUP_CHANGE_UNINSTALLED, []);
michael@0: 
michael@0:   [a1, a2, a3, a4, a5, a6] =
michael@0:     yield promiseAddonsByIDs(["addon1@tests.mozilla.org",
michael@0:                               "addon2@tests.mozilla.org",
michael@0:                               "addon3@tests.mozilla.org",
michael@0:                               "addon4@tests.mozilla.org",
michael@0:                               "addon5@tests.mozilla.org",
michael@0:                               "addon6@tests.mozilla.org"]);
michael@0: 
michael@0:   do_check_neq(a1, null);
michael@0:   do_check_true(a1.isActive);
michael@0:   do_check_false(a1.userDisabled);
michael@0:   do_check_false(a1.appDisabled);
michael@0:   do_check_eq(a1.pendingOperations, AddonManager.PENDING_NONE);
michael@0:   do_check_true(isExtensionInAddonsList(profileDir, a1.id));
michael@0: 
michael@0:   do_check_neq(a2, null);
michael@0:   do_check_true(a2.isActive);
michael@0:   do_check_false(a2.userDisabled);
michael@0:   do_check_false(a2.appDisabled);
michael@0:   do_check_eq(a2.pendingOperations, AddonManager.PENDING_NONE);
michael@0:   do_check_true(isExtensionInAddonsList(profileDir, a2.id));
michael@0: 
michael@0:   do_check_neq(a3, null);
michael@0:   do_check_false(a3.isActive);
michael@0:   do_check_true(a3.userDisabled);
michael@0:   do_check_false(a3.appDisabled);
michael@0:   do_check_eq(a3.pendingOperations, AddonManager.PENDING_NONE);
michael@0:   do_check_false(isExtensionInAddonsList(profileDir, a3.id));
michael@0: 
michael@0:   do_check_eq(a4, null);
michael@0: 
michael@0:   do_check_neq(a5, null);
michael@0:   do_check_eq(a5.version, "2.0");
michael@0:   do_check_true(a5.isActive);
michael@0:   do_check_false(a5.userDisabled);
michael@0:   do_check_false(a5.appDisabled);
michael@0:   do_check_eq(a5.pendingOperations, AddonManager.PENDING_NONE);
michael@0:   do_check_true(isExtensionInAddonsList(profileDir, a5.id));
michael@0: 
michael@0:   do_check_neq(a6, null);
michael@0:   do_check_true(a6.isActive);
michael@0:   do_check_false(a6.userDisabled);
michael@0:   do_check_false(a6.appDisabled);
michael@0:   do_check_eq(a6.pendingOperations, AddonManager.PENDING_NONE);
michael@0:   do_check_true(isExtensionInAddonsList(profileDir, a6.id));
michael@0: 
michael@0:   // After allowing access to the original DB things should still be
michael@0:   // back how they were before the lock
michael@0:   shutdownManager();
michael@0:   yield file.close();
michael@0:   gExtensionsJSON.permissions = filePermissions;
michael@0:   startupManager();
michael@0: 
michael@0:   // On Unix, we can save the DB even when the original file wasn't
michael@0:   // readable, so our changes were saved. On Windows,
michael@0:   // these things happened when we had no access to the database so
michael@0:   // they are seen as external changes when we get the database back
michael@0:   if (gXPISaveError) {
michael@0:     do_print("Previous XPI save failed");
michael@0:     check_startup_changes(AddonManager.STARTUP_CHANGE_INSTALLED,
michael@0:         ["addon6@tests.mozilla.org"]);
michael@0:     check_startup_changes(AddonManager.STARTUP_CHANGE_UNINSTALLED,
michael@0:         ["addon4@tests.mozilla.org"]);
michael@0:   }
michael@0:   else {
michael@0:     do_print("Previous XPI save succeeded");
michael@0:     check_startup_changes(AddonManager.STARTUP_CHANGE_INSTALLED, []);
michael@0:     check_startup_changes(AddonManager.STARTUP_CHANGE_UNINSTALLED, []);
michael@0:   }
michael@0: 
michael@0:   [a1, a2, a3, a4, a5, a6] =
michael@0:     yield promiseAddonsByIDs(["addon1@tests.mozilla.org",
michael@0:                               "addon2@tests.mozilla.org",
michael@0:                               "addon3@tests.mozilla.org",
michael@0:                               "addon4@tests.mozilla.org",
michael@0:                               "addon5@tests.mozilla.org",
michael@0:                               "addon6@tests.mozilla.org"]);
michael@0: 
michael@0:   do_check_neq(a1, null);
michael@0:   do_check_true(a1.isActive);
michael@0:   do_check_false(a1.userDisabled);
michael@0:   do_check_false(a1.appDisabled);
michael@0:   do_check_eq(a1.pendingOperations, AddonManager.PENDING_NONE);
michael@0:   do_check_true(isExtensionInAddonsList(profileDir, a1.id));
michael@0: 
michael@0:   do_check_neq(a2, null);
michael@0:   do_check_true(a2.isActive);
michael@0:   do_check_false(a2.userDisabled);
michael@0:   do_check_false(a2.appDisabled);
michael@0:   do_check_eq(a2.pendingOperations, AddonManager.PENDING_NONE);
michael@0:   do_check_true(isExtensionInAddonsList(profileDir, a2.id));
michael@0: 
michael@0:   do_check_neq(a3, null);
michael@0:   do_check_false(a3.isActive);
michael@0:   do_check_true(a3.userDisabled);
michael@0:   do_check_false(a3.appDisabled);
michael@0:   do_check_eq(a3.pendingOperations, AddonManager.PENDING_NONE);
michael@0:   do_check_false(isExtensionInAddonsList(profileDir, a3.id));
michael@0: 
michael@0:   do_check_eq(a4, null);
michael@0: 
michael@0:   do_check_neq(a5, null);
michael@0:   do_check_eq(a5.version, "2.0");
michael@0:   do_check_true(a5.isActive);
michael@0:   do_check_false(a5.userDisabled);
michael@0:   do_check_false(a5.appDisabled);
michael@0:   do_check_eq(a5.pendingOperations, AddonManager.PENDING_NONE);
michael@0:   do_check_true(isExtensionInAddonsList(profileDir, a5.id));
michael@0: 
michael@0:   do_check_neq(a6, null);
michael@0:   do_check_true(a6.isActive);
michael@0:   do_check_false(a6.userDisabled);
michael@0:   do_check_false(a6.appDisabled);
michael@0:   do_check_eq(a6.pendingOperations, AddonManager.PENDING_NONE);
michael@0:   do_check_true(isExtensionInAddonsList(profileDir, a6.id));
michael@0: });
michael@0: 
michael@0: function run_test() {
michael@0:   run_next_test();
michael@0: }
michael@0: