Move WebExtensions enabled Add-ons Manager

1 files changed, 216 insertions, 0 deletions

diff --git a/toolkit/mozapps/webextensions/test/xpcshell/test_uninstall.js b/toolkit/mozapps/webextensions/test/xpcshell/test_uninstall.js
new file mode 100644
index 000000000..6b12489f2
--- /dev/null
+++ b/toolkit/mozapps/webextensions/test/xpcshell/test_uninstall.js
@@ -0,0 +1,216 @@
+/* Any copyright is dedicated to the Public Domain.
+ * http://creativecommons.org/publicdomain/zero/1.0/
+ */
+
+// This verifies that add-ons can be uninstalled.
+
+var addon1 = {
+  id: "addon1@tests.mozilla.org",
+  version: "1.0",
+  name: "Test 1",
+  targetApplications: [{
+    id: "xpcshell@tests.mozilla.org",
+    minVersion: "1",
+    maxVersion: "1"
+  }]
+};
+
+const profileDir = gProfD.clone();
+profileDir.append("extensions");
+
+// Sets up the profile by installing an add-on.
+function run_test() {
+  createAppInfo("xpcshell@tests.mozilla.org", "XPCShell", "1", "1.9.2");
+
+  do_test_pending();
+  startupManager();
+
+  AddonManager.getAddonByID("addon1@tests.mozilla.org", callback_soon(function(olda1) {
+    do_check_eq(olda1, null);
+
+    writeInstallRDFForExtension(addon1, profileDir);
+
+    restartManager();
+
+    AddonManager.getAddonByID("addon1@tests.mozilla.org", function(a1) {
+      do_check_neq(a1, null);
+      do_check_true(a1.isActive);
+      do_check_false(a1.userDisabled);
+      do_check_true(isExtensionInAddonsList(profileDir, a1.id));
+      do_check_eq(a1.pendingOperations, 0);
+      do_check_in_crash_annotation(addon1.id, addon1.version);
+
+      do_execute_soon(run_test_1);
+    });
+  }));
+}
+
+function end_test() {
+  do_execute_soon(do_test_finished);
+}
+
+// Uninstalling an add-on should work.
+function run_test_1() {
+  prepare_test({
+    "addon1@tests.mozilla.org": [
+      "onUninstalling"
+    ]
+  });
+  AddonManager.getAddonByID("addon1@tests.mozilla.org", function(a1) {
+    do_check_eq(a1.pendingOperations, 0);
+    do_check_neq(a1.operationsRequiringRestart &
+                 AddonManager.OP_NEEDS_RESTART_UNINSTALL, 0);
+    a1.uninstall();
+    do_check_true(hasFlag(a1.pendingOperations, AddonManager.PENDING_UNINSTALL));
+    do_check_in_crash_annotation(addon1.id, addon1.version);
+
+    ensure_test_completed();
+
+    AddonManager.getAddonsWithOperationsByTypes(null, function(list) {
+      do_check_eq(list.length, 1);
+      do_check_eq(list[0].id, "addon1@tests.mozilla.org");
+
+      do_execute_soon(check_test_1);
+    });
+  });
+}
+
+function check_test_1() {
+  restartManager();
+
+  AddonManager.getAddonByID("addon1@tests.mozilla.org", function(a1) {
+    do_check_eq(a1, null);
+    do_check_false(isExtensionInAddonsList(profileDir, "addon1@tests.mozilla.org"));
+    do_check_not_in_crash_annotation(addon1.id, addon1.version);
+
+    var dest = profileDir.clone();
+    dest.append(do_get_expected_addon_name("addon1@tests.mozilla.org"));
+    do_check_false(dest.exists());
+    writeInstallRDFForExtension(addon1, profileDir);
+    do_execute_soon(run_test_2);
+  });
+}
+
+// Cancelling the uninstall should send onOperationCancelled
+function run_test_2() {
+  restartManager();
+
+  prepare_test({
+    "addon1@tests.mozilla.org": [
+      "onUninstalling"
+    ]
+  });
+
+  AddonManager.getAddonByID("addon1@tests.mozilla.org", function(a1) {
+    do_check_neq(a1, null);
+    do_check_true(a1.isActive);
+    do_check_false(a1.userDisabled);
+    do_check_true(isExtensionInAddonsList(profileDir, a1.id));
+    do_check_eq(a1.pendingOperations, 0);
+    a1.uninstall();
+    do_check_true(hasFlag(a1.pendingOperations, AddonManager.PENDING_UNINSTALL));
+
+    ensure_test_completed();
+
+    prepare_test({
+      "addon1@tests.mozilla.org": [
+        "onOperationCancelled"
+      ]
+    });
+    a1.cancelUninstall();
+    do_check_eq(a1.pendingOperations, 0);
+
+    ensure_test_completed();
+
+    do_execute_soon(check_test_2);
+  });
+}
+
+function check_test_2() {
+  restartManager();
+
+  AddonManager.getAddonByID("addon1@tests.mozilla.org", function(a1) {
+    do_check_neq(a1, null);
+    do_check_true(a1.isActive);
+    do_check_false(a1.userDisabled);
+    do_check_true(isExtensionInAddonsList(profileDir, a1.id));
+
+    run_test_3();
+  });
+}
+
+// Uninstalling an item pending disable should still require a restart
+function run_test_3() {
+  AddonManager.getAddonByID("addon1@tests.mozilla.org", function(a1) {
+    prepare_test({
+      "addon1@tests.mozilla.org": [
+        "onDisabling"
+      ]
+    });
+    a1.userDisabled = true;
+    ensure_test_completed();
+
+    do_check_true(hasFlag(AddonManager.PENDING_DISABLE, a1.pendingOperations));
+    do_check_true(a1.isActive);
+
+    prepare_test({
+      "addon1@tests.mozilla.org": [
+        "onUninstalling"
+      ]
+    });
+    a1.uninstall();
+
+    check_test_3();
+  });
+}
+
+function check_test_3() {
+  ensure_test_completed();
+
+  AddonManager.getAddonByID("addon1@tests.mozilla.org", function(a1) {
+    do_check_neq(a1, null);
+    do_check_true(hasFlag(AddonManager.PENDING_UNINSTALL, a1.pendingOperations));
+
+    prepare_test({
+      "addon1@tests.mozilla.org": [
+        "onOperationCancelled"
+      ]
+    });
+    a1.cancelUninstall();
+    ensure_test_completed();
+    do_check_true(hasFlag(AddonManager.PENDING_DISABLE, a1.pendingOperations));
+
+    do_execute_soon(run_test_4);
+  });
+}
+
+// Test that uninstalling an inactive item should happen without a restart
+function run_test_4() {
+  restartManager();
+
+  AddonManager.getAddonByID("addon1@tests.mozilla.org", function(a1) {
+    do_check_neq(a1, null);
+    do_check_false(a1.isActive);
+    do_check_true(a1.userDisabled);
+    do_check_false(isExtensionInAddonsList(profileDir, a1.id));
+
+    prepare_test({
+      "addon1@tests.mozilla.org": [
+        ["onUninstalling", false],
+        "onUninstalled"
+      ]
+    });
+    a1.uninstall();
+    ensure_test_completed();
+
+    check_test_4();
+  });
+}
+
+function check_test_4() {
+  AddonManager.getAddonByID("addon1@tests.mozilla.org", function(a1) {
+    do_check_eq(a1, null);
+
+    end_test();
+  });
+}