Source code
Revision control
Copy as Markdown
Other Tools
Test Info:
/* Any copyright is dedicated to the Public Domain.
const { GenAI } = ChromeUtils.importESModule(
"resource:///modules/GenAI.sys.mjs"
);
registerCleanupFunction(() => {
Services.prefs.clearUserPref("browser.ml.chat.provider");
Services.prefs.clearUserPref("browser.ml.chat.providers");
});
/**
* Check various provider ids are converted
*/
add_task(async function test_provider_id() {
Assert.equal(GenAI.getProviderId(""), "none", "Empty is none");
Assert.equal(
"custom",
"Unknown is custom"
);
Assert.equal(
"localhost",
"Known gets an id"
);
Assert.equal(GenAI.getProviderId(), "none", "Default to empty");
Services.prefs.setStringPref(
"browser.ml.chat.provider",
);
Assert.equal(GenAI.getProviderId(), "custom", "Used custom pref");
});
/**
* Check that providers can be hidden
*/
add_task(async function test_hide_providers() {
GenAI.init();
Assert.ok(!chatgpt.hidden, "ChatGPT shown by default");
Services.prefs.setStringPref("browser.ml.chat.providers", "");
Assert.ok(chatgpt.hidden, "ChatGPT hidden");
});
/**
* Check that providers can be ordered
*/
add_task(async function test_providers_order() {
Services.prefs.setStringPref(
"browser.ml.chat.providers",
"huggingchat,chatgpt"
);
const shown = [];
GenAI.chatProviders.forEach(val => {
if (!val.hidden) {
shown.push(val.id);
}
});
Assert.equal(shown, "huggingchat,chatgpt", "Providers reordered");
});