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
|
/* Any copyright is dedicated to the Public Domain.
* http://creativecommons.org/publicdomain/zero/1.0/ */
/**
* Test that if Nimbus provides invalid values for extraParams, the search
* service can still initialize and run.
* This is separate to test_getSubmission_params_prefNimbus so that initialization
* can be properly tested.
*/
"use strict";
const { NimbusFeatures } = ChromeUtils.importESModule(
"resource://nimbus/ExperimentAPI.sys.mjs"
);
const baseURL = "https://www.google.com/search?q=foo";
const baseURLSearchConfigV2 = "https://www.google.com/search?";
let getVariableStub;
let updateStub;
add_setup(async function () {
consoleAllowList.push("Failed to load nimbus variables for extraParams");
updateStub = sinon.stub(NimbusFeatures.search, "onUpdate");
getVariableStub = sinon.stub(NimbusFeatures.search, "getVariable");
sinon.stub(NimbusFeatures.search, "ready").resolves();
// The test engines used in this test need to be recognized as 'default'
// engines, or their MozParams will be ignored.
await SearchTestUtils.useTestEngines();
});
add_task(async function test_bad_nimbus_setting_on_init() {
getVariableStub.withArgs("extraParams").returns({ foo: "bar" });
await AddonTestUtils.promiseStartupManager();
await Services.search.init();
Assert.ok(
updateStub.called,
"Should have called onUpdate to listen for future updates"
);
const engine = Services.search.getEngineByName("engine-pref");
Assert.equal(
engine.getSubmission("foo").uri.spec,
baseURL,
"Should have been able to get a submission URL"
);
});
add_task(async function test_switch_to_good_nimbus_setting() {
// Switching to a good structure should provide the parameter.
getVariableStub.withArgs("extraParams").returns([
{
key: "code",
// The & and = in this parameter are to check that they are correctly
// encoded, and not treated as a separate parameter.
value: "supergood&id=unique123456",
},
]);
updateStub.firstCall.args[0]();
const engine = Services.search.getEngineByName("engine-pref");
Assert.equal(
engine.getSubmission("foo").uri.spec,
SearchUtils.newSearchConfigEnabled
? baseURLSearchConfigV2 + "code=supergood%26id%3Dunique123456&q=foo"
: baseURL + "&code=supergood%26id%3Dunique123456",
"Should have got the submission URL with the updated code"
);
});
add_task(async function test_switch_back_to_bad_nimbus_setting() {
// Switching from a valid Nimbus setting to an invalid setting should fallback
// to a valid submission URL.
getVariableStub.withArgs("extraParams").returns({ bar: "foo" });
updateStub.firstCall.args[0]();
const engine = Services.search.getEngineByName("engine-pref");
Assert.equal(
engine.getSubmission("foo").uri.spec,
baseURL,
"Should have not have the extra parameters on the submission URL after a bad update"
);
});
|
