test_Tools_GetUserMemories.js

Enable keyboard shortcuts

/* 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/. */

do_get_profile();

const { getUserMemories } = ChromeUtils.importESModule(

  "moz-src:///browser/components/aiwindow/models/Tools.sys.mjs"

);

const { SecurityProperties } = ChromeUtils.importESModule(

  "moz-src:///browser/components/aiwindow/models/SecurityProperties.sys.mjs"

);

add_task(async function test_getUserMemories_sets_security_flags() {

  const securityProperties = new SecurityProperties();

  await getUserMemories({}, securityProperties);

  securityProperties.commit();

  Assert.equal(securityProperties.privateData, true, "private_data set");

  Assert.equal(

    securityProperties.untrustedInput,

    false,

    "untrusted_input not set"

);

});

add_task(async function test_getUserMemories_allowed_when_flags_set() {

  const securityProperties = new SecurityProperties();

  securityProperties.setPrivateData();

  securityProperties.setUntrustedInput();

  securityProperties.commit();

  const result = await getUserMemories({}, securityProperties);

  Assert.ok(Array.isArray(result), "returns array, not a refusal");

});