summaryrefslogtreecommitdiffstats
path: root/toolkit/components/viewsource/test/browser/browser_contextmenu.js
blob: afc8ab8c84e2dacb6f10ee90694555c8d622e09b (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
115
/* Any copyright is dedicated to the Public Domain.
 * http://creativecommons.org/publicdomain/zero/1.0/
 */

var source =
  "data:text/html,text<link%20href='http://example.com/'%20/>more%20text<a%20href='mailto:abc@def.ghi'>email</a>";
var gViewSourceWindow, gContextMenu, gCopyLinkMenuItem, gCopyEmailMenuItem;

var expectedData = [];

add_task(async function () {
  // Full source in view source tab
  let newTab = await openDocument(source);
  await onViewSourceWindowOpen(window);

  let contextMenu = document.getElementById("contentAreaContextMenu");

  for (let test of expectedData) {
    await checkMenuItems(contextMenu, test[0], test[1], test[2], test[3]);
  }

  gBrowser.removeTab(newTab);

  // Selection source in view source tab
  expectedData = [];
  newTab = await openDocumentSelect(source, "body");
  await onViewSourceWindowOpen(window);

  contextMenu = document.getElementById("contentAreaContextMenu");

  for (let test of expectedData) {
    await checkMenuItems(contextMenu, test[0], test[1], test[2], test[3]);
  }

  gBrowser.removeTab(newTab);
});

async function onViewSourceWindowOpen(aWindow) {
  gViewSourceWindow = aWindow;

  gCopyLinkMenuItem = aWindow.document.getElementById("context-copylink");
  gCopyEmailMenuItem = aWindow.document.getElementById("context-copyemail");

  let browser = gBrowser.selectedBrowser;
  await SpecialPowers.spawn(browser, [], async function (arg) {
    let tags = content.document.querySelectorAll("a[href]");
    Assert.equal(
      tags[0].href,
      "view-source:http://example.com/",
      "Link has correct href"
    );
    Assert.equal(tags[1].href, "mailto:abc@def.ghi", "Link has correct href");
  });

  expectedData.push(["a[href]", true, false, "http://example.com/"]);
  expectedData.push(["a[href^=mailto]", false, true, "abc@def.ghi"]);
  expectedData.push(["span", false, false, null]);
}

async function checkMenuItems(
  contextMenu,
  selector,
  copyLinkExpected,
  copyEmailExpected,
  expectedClipboardContent
) {
  let browser = gBrowser.selectedBrowser;
  await SpecialPowers.spawn(browser, [{ selector }], async function (arg) {
    content.document.querySelector(arg.selector).scrollIntoView();
  });

  let popupShownPromise = BrowserTestUtils.waitForEvent(
    contextMenu,
    "popupshown"
  );
  await BrowserTestUtils.synthesizeMouseAtCenter(
    selector,
    { type: "contextmenu", button: 2 },
    browser
  );
  await popupShownPromise;

  is(
    gCopyLinkMenuItem.hidden,
    !copyLinkExpected,
    "Copy link menuitem is " + (copyLinkExpected ? "not hidden" : "hidden")
  );
  is(
    gCopyEmailMenuItem.hidden,
    !copyEmailExpected,
    "Copy email menuitem is " + (copyEmailExpected ? "not hidden" : "hidden")
  );

  if (copyLinkExpected || copyEmailExpected) {
    await new Promise((resolve, reject) => {
      waitForClipboard(
        expectedClipboardContent,
        function () {
          contextMenu.activateItem(
            copyLinkExpected ? gCopyLinkMenuItem : gCopyEmailMenuItem
          );
        },
        resolve,
        reject
      );
    });
  } else {
    let popupHiddenPromise = BrowserTestUtils.waitForEvent(
      contextMenu,
      "popuphidden"
    );
    contextMenu.hidePopup();
    await popupHiddenPromise;
  }
}