summaryrefslogtreecommitdiffstats
path: root/browser/components/extensions/test/browser/browser_ext_contextMenus_uninstall.js
blob: 0d0b7cac7b49839e85d03eb18b7459766ac64606 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
/* -*- Mode: indent-tabs-mode: nil; js-indent-level: 2 -*- */
/* vim: set sts=2 sw=2 et tw=80: */
"use strict";

add_task(async function () {
  let tab1 = await BrowserTestUtils.openNewForegroundTab(
    gBrowser,
    "http://mochi.test:8888/browser/browser/components/extensions/test/browser/context.html"
  );

  // Install an extension.
  let extension = ExtensionTestUtils.loadExtension({
    manifest: {
      permissions: ["contextMenus"],
    },

    background: function () {
      browser.contextMenus.create({ title: "a" });
      browser.contextMenus.create({ title: "b" });
      browser.test.notifyPass("contextmenus-icons");
    },
  });

  await extension.startup();
  await extension.awaitFinish("contextmenus-icons");

  // Open the context menu.
  let contextMenu = await openContextMenu("#img1");

  // Confirm that the extension menu item exists.
  let topLevelExtensionMenuItems = contextMenu.getElementsByAttribute(
    "ext-type",
    "top-level-menu"
  );
  is(
    topLevelExtensionMenuItems.length,
    1,
    "the top level extension menu item exists"
  );

  await closeContextMenu();

  // Uninstall the extension.
  await extension.unload();

  // Open the context menu.
  contextMenu = await openContextMenu("#img1");

  // Confirm that the extension menu item has been removed.
  topLevelExtensionMenuItems = contextMenu.getElementsByAttribute(
    "ext-type",
    "top-level-menu"
  );
  is(
    topLevelExtensionMenuItems.length,
    0,
    "no top level extension menu items should exist"
  );

  await closeContextMenu();

  // Install a new extension.
  extension = ExtensionTestUtils.loadExtension({
    manifest: {
      permissions: ["contextMenus"],
    },
    background: function () {
      browser.contextMenus.create({ title: "c" });
      browser.contextMenus.create({ title: "d" });
      browser.test.notifyPass("contextmenus-uninstall-second-extension");
    },
  });

  await extension.startup();
  await extension.awaitFinish("contextmenus-uninstall-second-extension");

  // Open the context menu.
  contextMenu = await openContextMenu("#img1");

  // Confirm that only the new extension menu item is in the context menu.
  topLevelExtensionMenuItems = contextMenu.getElementsByAttribute(
    "ext-type",
    "top-level-menu"
  );
  is(
    topLevelExtensionMenuItems.length,
    1,
    "only one top level extension menu item should exist"
  );

  // Close the context menu.
  await closeContextMenu();

  // Uninstall the extension.
  await extension.unload();

  // Open the context menu.
  contextMenu = await openContextMenu("#img1");

  // Confirm that no extension menu items exist.
  topLevelExtensionMenuItems = contextMenu.getElementsByAttribute(
    "ext-type",
    "top-level-menu"
  );
  is(
    topLevelExtensionMenuItems.length,
    0,
    "no top level extension menu items should exist"
  );

  await closeContextMenu();

  BrowserTestUtils.removeTab(tab1);
});