mirror of
https://github.com/discourse/discourse-ai.git
synced 2025-07-01 12:02:16 +00:00
6827147362
The AiApiAuditLog per translation event doesn't trace back easily to a post or topic. This commit adds support to that, and also switches the translators to named arguments rather than positional arguments.
100 lines
3.2 KiB
Ruby
100 lines
3.2 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
describe DiscourseAi::Translation::CategoryLocalizer do
|
|
subject(:localizer) { described_class }
|
|
|
|
before do
|
|
Fabricate(:fake_model).tap do |fake_llm|
|
|
SiteSetting.public_send("ai_translation_model=", "custom:#{fake_llm.id}")
|
|
end
|
|
end
|
|
|
|
def post_raw_translator_stub(opts)
|
|
mock = instance_double(DiscourseAi::Translation::PostRawTranslator)
|
|
allow(DiscourseAi::Translation::PostRawTranslator).to receive(:new).with(
|
|
text: opts[:text],
|
|
target_locale: opts[:target_locale],
|
|
).and_return(mock)
|
|
allow(mock).to receive(:translate).and_return(opts[:translated])
|
|
end
|
|
|
|
def short_text_translator_stub(opts)
|
|
mock = instance_double(DiscourseAi::Translation::ShortTextTranslator)
|
|
allow(DiscourseAi::Translation::ShortTextTranslator).to receive(:new).with(
|
|
text: opts[:text],
|
|
target_locale: opts[:target_locale],
|
|
).and_return(mock)
|
|
allow(mock).to receive(:translate).and_return(opts[:translated])
|
|
end
|
|
|
|
fab!(:category) do
|
|
Fabricate(:category, name: "Test Category", description: "This is a test category")
|
|
end
|
|
|
|
describe ".localize" do
|
|
let(:target_locale) { "fr" }
|
|
|
|
it "translates the category name and description" do
|
|
translated_cat_desc = "C'est une catégorie de test"
|
|
translated_cat_name = "Catégorie de Test"
|
|
short_text_translator_stub(
|
|
{ text: category.name, target_locale: target_locale, translated: translated_cat_name },
|
|
)
|
|
post_raw_translator_stub(
|
|
{
|
|
text: category.description,
|
|
target_locale: target_locale,
|
|
translated: translated_cat_desc,
|
|
},
|
|
)
|
|
|
|
res = localizer.localize(category, target_locale)
|
|
|
|
expect(res.name).to eq(translated_cat_name)
|
|
expect(res.description).to eq(translated_cat_desc)
|
|
end
|
|
|
|
it "handles locale format standardization" do
|
|
translated_cat_desc = "C'est une catégorie de test"
|
|
translated_cat_name = "Catégorie de Test"
|
|
short_text_translator_stub(
|
|
{ text: category.name, target_locale:, translated: translated_cat_name },
|
|
)
|
|
post_raw_translator_stub(
|
|
{ text: category.description, target_locale:, translated: translated_cat_desc },
|
|
)
|
|
|
|
res = localizer.localize(category, target_locale)
|
|
|
|
expect(res.name).to eq(translated_cat_name)
|
|
expect(res.description).to eq(translated_cat_desc)
|
|
end
|
|
|
|
it "returns nil if category is blank" do
|
|
expect(localizer.localize(nil)).to be_nil
|
|
end
|
|
|
|
it "returns nil if target locale is blank" do
|
|
expect(localizer.localize(category, nil)).to be_nil
|
|
end
|
|
|
|
it "uses I18n.locale as default when no target locale is provided" do
|
|
I18n.locale = :es
|
|
translated_cat_desc = "C'est une catégorie de test"
|
|
translated_cat_name = "Esta es una categoría de prueba"
|
|
short_text_translator_stub(
|
|
{ text: category.name, target_locale: "es", translated: translated_cat_name },
|
|
)
|
|
post_raw_translator_stub(
|
|
{ text: category.description, target_locale: "es", translated: translated_cat_desc },
|
|
)
|
|
|
|
res = localizer.localize(category)
|
|
|
|
expect(res.name).to eq(translated_cat_name)
|
|
expect(res.description).to eq(translated_cat_desc)
|
|
expect(res.locale).to eq("es")
|
|
end
|
|
end
|
|
end
|