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
|
registerCleanupFunction(function* cleanup() {
while (gBrowser.tabs.length > 1) {
yield BrowserTestUtils.removeTab(gBrowser.tabs[gBrowser.tabs.length - 1]);
}
Services.search.currentEngine = originalEngine;
let engine = Services.search.getEngineByName("MozSearch");
Services.search.removeEngine(engine);
});
let originalEngine;
add_task(function* test_setup() {
// Stop search-engine loads from hitting the network
Services.search.addEngineWithDetails("MozSearch", "", "", "", "GET",
"http://example.com/?q={searchTerms}");
let engine = Services.search.getEngineByName("MozSearch");
originalEngine = Services.search.currentEngine;
Services.search.currentEngine = engine;
});
add_task(function*() { yield dropText("mochi.test/first", 1); });
add_task(function*() { yield dropText("javascript:'bad'"); });
add_task(function*() { yield dropText("jAvascript:'bad'"); });
add_task(function*() { yield dropText("search this", 1); });
add_task(function*() { yield dropText("mochi.test/second", 1); });
add_task(function*() { yield dropText("data:text/html,bad"); });
add_task(function*() { yield dropText("mochi.test/third", 1); });
// Single text/plain item, with multiple links.
add_task(function*() { yield dropText("mochi.test/1\nmochi.test/2", 2); });
add_task(function*() { yield dropText("javascript:'bad1'\nmochi.test/3", 0); });
add_task(function*() { yield dropText("mochi.test/4\ndata:text/html,bad1", 0); });
// Multiple text/plain items, with single and multiple links.
add_task(function*() {
yield drop([[{type: "text/plain",
data: "mochi.test/5"}],
[{type: "text/plain",
data: "mochi.test/6\nmochi.test/7"}]], 3);
});
// Single text/x-moz-url item, with multiple links.
// "text/x-moz-url" has titles in even-numbered lines.
add_task(function*() {
yield drop([[{type: "text/x-moz-url",
data: "mochi.test/8\nTITLE8\nmochi.test/9\nTITLE9"}]], 2);
});
// Single item with multiple types.
add_task(function*() {
yield drop([[{type: "text/plain",
data: "mochi.test/10"},
{type: "text/x-moz-url",
data: "mochi.test/11\nTITLE11"}]], 1);
});
function dropText(text, expectedTabOpenCount=0) {
return drop([[{type: "text/plain", data: text}]], expectedTabOpenCount);
}
function* drop(dragData, expectedTabOpenCount=0) {
let dragDataString = JSON.stringify(dragData);
info(`Starting test for datagData:${dragDataString}; expectedTabOpenCount:${expectedTabOpenCount}`);
let scriptLoader = Cc["@mozilla.org/moz/jssubscript-loader;1"].
getService(Ci.mozIJSSubScriptLoader);
let EventUtils = {};
scriptLoader.loadSubScript("chrome://mochikit/content/tests/SimpleTest/EventUtils.js", EventUtils);
let awaitDrop = BrowserTestUtils.waitForEvent(gBrowser.tabContainer, "drop");
let actualTabOpenCount = 0;
let openedTabs = [];
let checkCount = function(event) {
openedTabs.push(event.target);
actualTabOpenCount++;
return actualTabOpenCount == expectedTabOpenCount;
};
let awaitTabOpen = expectedTabOpenCount && BrowserTestUtils.waitForEvent(gBrowser.tabContainer, "TabOpen", false, checkCount);
// A drop type of "link" onto an existing tab would normally trigger a
// load in that same tab, but tabbrowser code in _getDragTargetTab treats
// drops on the outer edges of a tab differently (loading a new tab
// instead). Make events created by synthesizeDrop have all of their
// coordinates set to 0 (screenX/screenY), so they're treated as drops
// on the outer edge of the tab, thus they open new tabs.
var event = {
clientX: 0,
clientY: 0,
screenX: 0,
screenY: 0,
};
EventUtils.synthesizeDrop(gBrowser.selectedTab, gBrowser.selectedTab, dragData, "link", window, undefined, event);
let tabsOpened = false;
if (awaitTabOpen) {
yield awaitTabOpen;
info("Got TabOpen event");
tabsOpened = true;
for (let tab of openedTabs) {
yield BrowserTestUtils.removeTab(tab);
}
}
is(tabsOpened, !!expectedTabOpenCount, `Tabs for ${dragDataString} should only open if any of dropped items are valid`);
yield awaitDrop;
ok(true, "Got drop event");
}
|