Source code
Revision control
Copy as Markdown
Other Tools
Test Info: Warnings
- This test gets skipped with pattern: os == 'android'
- Manifest: browser/components/urlbar/tests/unit/xpcshell.toml
/* Any copyright is dedicated to the Public Domain.
/**
* These tests unit test the functionality of UrlbarController by stubbing out the
* model and providing stubs to be called.
*/
"use strict";
function getPostDataString(aIS) {
if (!aIS) {
return null;
}
let sis = Cc["@mozilla.org/scriptableinputstream;1"].createInstance(
Ci.nsIScriptableInputStream
);
sis.init(aIS);
let dataLines = sis.read(aIS.available()).split("\n");
// only want the last line
return dataLines[dataLines.length - 1];
}
function keywordResult(aURL, aPostData, aIsUnsafe) {
this.url = aURL;
this.postData = aPostData;
this.isUnsafe = aIsUnsafe;
}
function keyWordData() {}
keyWordData.prototype = {
init(aKeyWord, aURL, aPostData, aSearchWord) {
this.keyword = aKeyWord;
this.uri = Services.io.newURI(aURL);
this.postData = aPostData;
this.searchWord = aSearchWord;
this.method = this.postData ? "POST" : "GET";
},
};
function bmKeywordData(aKeyWord, aURL, aPostData, aSearchWord) {
this.init(aKeyWord, aURL, aPostData, aSearchWord);
}
bmKeywordData.prototype = new keyWordData();
function searchKeywordData(aKeyWord, aURL, aPostData, aSearchWord) {
this.init(aKeyWord, aURL, aPostData, aSearchWord);
}
searchKeywordData.prototype = new keyWordData();
var testData = [
[
],
[
],
[
new bmKeywordData(
"bmpostget",
"search2=%s",
"foo3"
),
],
[
],
[
new searchKeywordData(
"searchget",
null,
"foo4"
),
],
[
new searchKeywordData(
"searchpost",
"search={searchTerms}",
"foo5"
),
],
[
new searchKeywordData(
"searchpostget",
"search2={searchTerms}",
"foo6"
),
new keywordResult(
"search2=foo6",
true
),
],
// Bookmark keywords that don't take parameters should not be activated if a
[
new keywordResult(null, null, true),
],
[
new bmKeywordData(
"bmpost-noparam",
"not_a=param",
"foo8"
),
new keywordResult(null, null, true),
],
// Test escaping (%s = escaped, %S = raw)
// UTF-8 default
[
new bmKeywordData(
"bmget-escaping",
null,
"fo\xE9"
),
],
// Explicitly-defined ISO-8859-1
[
new bmKeywordData(
"bmget-escaping2",
null,
"fo\xE9"
),
],
// UTF-8 default
[
new bmKeywordData(
"bmget-escaping",
null,
"+/@"
),
],
// Explicitly-defined ISO-8859-1
[
new bmKeywordData(
"bmget-escaping2",
null,
"+/@"
),
],
// Test using a non-bmKeywordData object, to test the behavior of
// getShortcutOrURIAndPostData for non-keywords (setupKeywords only adds keywords for
// bmKeywordData objects)
];
add_task(async function test_getshortcutoruri() {
await setupKeywords();
for (let item of testData) {
let [data, result] = item;
let query = data.keyword;
if (data.searchWord) {
query += " " + data.searchWord;
}
let returnedData = await UrlbarUtils.getShortcutOrURIAndPostData(query);
// null result.url means we should expect the same query we sent in
let expected = result.url || query;
Assert.equal(
returnedData.url,
expected,
"got correct URL for " + data.keyword
);
Assert.equal(
getPostDataString(returnedData.postData),
result.postData,
"got correct postData for " + data.keyword
);
Assert.equal(
returnedData.mayInheritPrincipal,
!result.isUnsafe,
"got correct mayInheritPrincipal for " + data.keyword
);
}
await cleanupKeywords();
});
var folder = null;
async function setupKeywords() {
folder = await PlacesUtils.bookmarks.insert({
parentGuid: PlacesUtils.bookmarks.unfiledGuid,
type: PlacesUtils.bookmarks.TYPE_FOLDER,
title: "keyword-test",
});
for (let item of testData) {
let data = item[0];
if (data instanceof bmKeywordData) {
await PlacesUtils.bookmarks.insert({
url: data.uri,
parentGuid: folder.guid,
});
await PlacesUtils.keywords.insert({
keyword: data.keyword,
url: data.uri.spec,
postData: data.postData,
});
}
if (data instanceof searchKeywordData) {
await SearchTestUtils.installSearchExtension({
name: data.keyword,
keyword: data.keyword,
search_url: data.uri.spec,
search_url_get_params: "",
search_url_post_params: data.postData,
});
}
}
}
async function cleanupKeywords() {
await PlacesUtils.bookmarks.remove(folder);
}