tor-browser

test_Tools_SearchBrowsingHistory.js (6541B)

---

/**
* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/.
*/

const { searchBrowsingHistory, stripSearchBrowsingHistoryFields } =
 ChromeUtils.importESModule(
   "moz-src:///browser/components/aiwindow/models/Tools.sys.mjs"
 );

/**
* searchBrowsingHistory tests
*
* Wrapper test: ensures Tools.searchBrowsingHistory() returns a valid JSON
* structure and surfaces the underlying error when the semantic DB is not
* initialized. Real search behavior is tested elsewhere.
*/

add_task(async function test_searchBrowsingHistory_wrapper() {
 const outputStr = await searchBrowsingHistory({
   searchTerm: "",
   startTs: null,
   endTs: null,
   historyLimit: 15,
 });

 const output = JSON.parse(outputStr);

 Assert.equal(output.searchTerm, "", "searchTerm match");
 Assert.ok("results" in output, "results field present");
 Assert.ok(Array.isArray(output.results), "results is an array");

 // Error expected
 Assert.ok("error" in output, "error field present");
});

/**
* stripSearchBrowsingHistoryFields tests
*/

add_task(function test_stripSearchBrowsingHistoryFields_omits_large_fields() {
 const input = JSON.stringify({
   searchTerm: "firefox",
   count: 6,
   results: [
     {
       title: "Firefox Privacy Notice — Mozilla",
       url: "https://www.mozilla.org/en-US/privacy/firefox/",
       visitDate: "2025-11-28T05:32:33.096Z",
       visitCount: 2,
       relevanceScore: 0.7472805678844452,
       favicon: "page-icon:https://www.mozilla.org/en-US/privacy/firefox/",
       thumbnail: "https://www.mozilla.org/media/img/m24/og.3a69dffad83e.png",
     },
     // no thumbnail
     {
       title: "Planet Mozilla",
       url: "https://planet.mozilla.org/",
       visitDate: "2025-11-28T05:32:37.796Z",
       visitCount: 1,
       relevanceScore: 0.5503441691398621,
       favicon: "page-icon:https://planet.mozilla.org/",
     },
     // fake record: no favicon
     {
       title: "Firefox Privacy Notice — Mozilla 2",
       url: "https://www.mozilla.org/en-US/privacy/firefox/2/",
       visitDate: "2025-11-28T05:32:33.096Z",
       visitCount: 1,
       relevanceScore: 0.54,
       thumbnail: "https://www.mozilla.org/media/img/m24/og.3a69dffad83e.png",
     },
     // fake record: no favicon and no thumbnail
     {
       title: "Planet Mozilla 2",
       url: "https://planet.mozilla.org/2/",
       visitDate: "2025-11-28T05:32:37.796Z",
       visitCount: 1,
       relevanceScore: 0.53,
     },
     {
       title: "Bugzilla Main Page",
       url: "https://bugzilla.mozilla.org/home",
       visitDate: "2025-11-28T05:32:49.807Z",
       visitCount: 1,
       relevanceScore: 0.5060983002185822,
       favicon: "page-icon:https://bugzilla.mozilla.org/home",
       thumbnail:
         "https://bugzilla.mozilla.org/extensions/OpenGraph/web/moz-social-bw-rgb-1200x1200.png",
     },
     {
       title: "Bugzilla Main Page",
       url: "https://bugzilla.mozilla.org/",
       visitDate: "2025-11-28T05:32:49.272Z",
       visitCount: 1,
       relevanceScore: 0.5060983002185822,
       favicon: "page-icon:https://bugzilla.mozilla.org/",
     },
   ],
 });

 const outputStr = stripSearchBrowsingHistoryFields(input);
 const output = JSON.parse(outputStr);

 Assert.equal(
   output.searchTerm,
   "firefox",
   "searchTerm preserved after stripping"
 );
 Assert.equal(output.count, 6, "count preserved");
 Assert.equal(output.results.length, 6, "results length preserved");

 Assert.equal(
   output.results[0].title,
   "Firefox Privacy Notice — Mozilla",
   "title same"
 );
 Assert.equal(
   output.results[0].url,
   "https://www.mozilla.org/en-US/privacy/firefox/",
   "url same"
 );
 Assert.equal(
   output.results[0].visitDate,
   "2025-11-28T05:32:33.096Z",
   "visitDate same"
 );
 Assert.equal(output.results[0].visitCount, 2, "visitCount same");
 Assert.equal(
   output.results[0].relevanceScore,
   0.7472805678844452,
   "relevanceScore same"
 );
 Assert.equal(output.results[0].favicon, undefined, "favicon removed");
 Assert.equal(output.results[0].thumbnail, undefined, "thumbnail removed");

 for (const [idx, r] of output.results.entries()) {
   Assert.ok(!("favicon" in r), `favicon removed for result[${idx}]`);
   Assert.ok(!("thumbnail" in r), `thumbnail removed for result[${idx}]`);
 }
});

add_task(
 function test_stripSearchBrowsingHistoryFields_passthrough_on_empty_results() {
   // empty result
   const noResults = JSON.stringify({
     searchTerm: "test",
     count: 0,
     results: [],
   });

   const outputStr = stripSearchBrowsingHistoryFields(noResults);
   Assert.equal(
     outputStr,
     noResults,
     "When results is empty, function should return original string"
   );

   const withError = JSON.stringify({
     searchTerm: "test",
     error: "something went wrong",
     results: [],
   });

   // no result with error
   const outputErr = stripSearchBrowsingHistoryFields(withError);
   Assert.equal(
     outputErr,
     withError,
     "When error is present, function should return original string"
   );

   const notJSON = "this is not json";
   const outputErrNonJSON = stripSearchBrowsingHistoryFields(notJSON);
   Assert.equal(
     outputErrNonJSON,
     notJSON,
     "Invalid JSON input should be returned unchanged"
   );
 }
);

add_task(
 function test_stripSearchBrowsingHistoryFields_passthrough_on_unexpected_results_structure() {
   // result = null
   const unexpectedResults = null;

   const outputStr = stripSearchBrowsingHistoryFields(unexpectedResults);
   Assert.equal(
     outputStr,
     unexpectedResults,
     "When any other data structure besides JSON (null), return original data"
   );

   // reulst = ""
   const emptyStringResults = "";

   const outputStrEmpty = stripSearchBrowsingHistoryFields(emptyStringResults);
   Assert.equal(
     outputStrEmpty,
     emptyStringResults,
     "When any other data structure besides JSON (''), return original data"
   );

   // result = "abc"
   const nonJSONStringResults = "abc";

   const outputStrNonJSON =
     stripSearchBrowsingHistoryFields(nonJSONStringResults);
   Assert.equal(
     outputStrNonJSON,
     nonJSONStringResults,
     "When any other data structure besides JSON ('abc'), return original data"
   );
 }
);