Source code
Revision control
Copy as Markdown
Other Tools
Test Info: Warnings
-
CI results over the last 21 days
- This test runs only with pattern: os != 'android'
- Manifest: browser/components/aiwindow/models/tests/xpcshell/xpcshell.toml
/* 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
const { buildBrowserContextPrompt } = ChromeUtils.importESModule(
"moz-src:///browser/components/aiwindow/models/PromptLoader.sys.mjs"
);
const { sanitizeUntrustedContent } = ChromeUtils.importESModule(
"moz-src:///browser/components/aiwindow/models/ChatUtils.sys.mjs"
);
const { _setRemoteClientForTesting, _clearRemoteClientForTesting } =
ChromeUtils.importESModule(
"moz-src:///browser/components/aiwindow/models/Utils.sys.mjs"
);
registerCleanupFunction(() => _clearRemoteClientForTesting());
function tabModule(module, version, prompts) {
return {
kind: "module",
feature: "browser-context",
module,
model: "generic",
version,
prompts,
};
}
const LEGACY_TAB = tabModule("tab", "1.0", "legacy tab: {url}");
const TAB_V1 = tabModule(
"tab-conditional-fields",
"1.0",
"v1 tab: {url}\n{#pageSlug}Page slug: {pageSlug}{/pageSlug}\nTitle: {title}"
);
const TAB_V2 = tabModule("tab-conditional-fields", "2.0", "v2 tab: {url}");
const PARAMS_V1 = {
kind: "params",
feature: "browser-context",
model: "generic",
version: "1.0",
is_default: true,
modules: [
{ name: "tab-conditional-fields", version: "1.0" },
{ name: "mentions", version: "1.0" },
],
};
function buildWith(records, mapping) {
_setRemoteClientForTesting({ get: async () => records });
return buildBrowserContextPrompt("generic", {
getRealTimeMapping: async () => ({
url: "about:smartpage?page=my-page",
title: "My page",
description: "",
pageSlug: "my-page",
hasTabInfo: true,
...mapping,
}),
contextMentions: [],
securityProperties: new SecurityProperties(),
});
}
add_task(async function test_params_pin_the_tab_version() {
const out = await buildWith([PARAMS_V1, LEGACY_TAB, TAB_V1, TAB_V2]);
Assert.equal(
out,
"v1 tab: about:smartpage?page=my-page\nPage slug: my-page\nTitle: My page",
"The tab module and version named by the params manifest are used"
);
});
add_task(async function test_tab_without_slug_drops_the_slug_line() {
const out = await buildWith([PARAMS_V1, LEGACY_TAB, TAB_V1], {
pageSlug: "",
});
Assert.equal(
out,
"The slug line is removed when there is no slug"
);
});
add_task(async function test_no_params_falls_back_to_legacy_tab() {
const out = await buildWith([LEGACY_TAB, TAB_V1]);
Assert.equal(
out,
"legacy tab: about:smartpage?page=my-page",
"Without a params manifest the legacy tab fragment is used"
);
});
add_task(async function test_mentioned_smart_page_includes_its_slug() {
_setRemoteClientForTesting({
get: async () => [
{
kind: "module",
feature: "browser-context",
module: "mentions",
model: "generic",
version: "1.0",
prompts: "Mentioned:\n{contextUrls}",
},
],
});
const out = await buildBrowserContextPrompt("generic", {
getRealTimeMapping: async () => ({ hasTabInfo: false }),
contextMentions: [
{ type: "tab", url: "about:smartpage?page=my-page", label: "My page" },
],
securityProperties: new SecurityProperties(),
});
Assert.ok(
out.includes(
`- URL: about:smartpage?page=my-page\n` +
` Title: ${sanitizeUntrustedContent("My page")}\n` +
` Page slug: my-page\n`
),
`A mentioned smart page lists its slug: ${out}`
);
Assert.equal(
out.match(/Page slug:/g).length,
1,
"A mentioned web page has no slug line"
);
});