summaryrefslogtreecommitdiffstats
path: root/comm/mail/base/test/browser/browser_webSearchTelemetry.js
blob: cd420e3b8313c7f8975c51495091d7d0fbc44d4a (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
/* Any copyright is dedicated to the Public Domain.
 * http://creativecommons.org/publicdomain/zero/1.0/ */

/* globals openWebSearch */

/**
 * Test telemetry related to web search usage.
 */

let { TelemetryTestUtils } = ChromeUtils.importESModule(
  "resource://testing-common/TelemetryTestUtils.sys.mjs"
);
let { MockRegistrar } = ChromeUtils.importESModule(
  "resource://testing-common/MockRegistrar.sys.mjs"
);

/** @implements {nsIExternalProtocolService} */
let mockExternalProtocolService = {
  loadURI(aURI, aWindowContext) {},
  QueryInterface: ChromeUtils.generateQI(["nsIExternalProtocolService"]),
};

let mockExternalProtocolServiceCID = MockRegistrar.register(
  "@mozilla.org/uriloader/external-protocol-service;1",
  mockExternalProtocolService
);

registerCleanupFunction(() => {
  MockRegistrar.unregister(mockExternalProtocolServiceCID);
});

/**
 * Test that we're counting how many times search on web was used.
 */
add_task(async function test_web_search_usage() {
  Services.telemetry.clearScalars();

  const NUM_SEARCH = 5;
  let engine = await Services.search.getDefault();
  await Promise.all(
    Array.from({ length: NUM_SEARCH }).map(() => openWebSearch("thunderbird"))
  );

  let scalars = TelemetryTestUtils.getProcessScalars("parent", true);
  Assert.equal(
    scalars["tb.websearch.usage"][engine.name.toLowerCase()],
    NUM_SEARCH,
    "Count of search on web times must be correct."
  );
});