discourse-ai/spec/lib/modules/ai_helper/entry_point_spec.rb

# frozen_string_literal: true

describe DiscourseAi::AiHelper::EntryPoint do
  fab!(:english_user) { Fabricate(:user) }
  fab!(:french_user) { Fabricate(:user, locale: "fr") }

  it "will correctly localize available prompts" do
    assign_fake_provider_to(:ai_helper_model)
    SiteSetting.default_locale = "en"
    SiteSetting.allow_user_locale = true
    SiteSetting.ai_helper_enabled = true
    SiteSetting.composer_ai_helper_allowed_groups = "10" # tl0
    DiscourseAi::AiHelper::Assistant.clear_prompt_cache!

    Group.refresh_automatic_groups!

    serializer = CurrentUserSerializer.new(english_user, scope: Guardian.new(english_user))
    parsed = JSON.parse(serializer.to_json)

    translate_prompt =
      parsed["current_user"]["ai_helper_prompts"].find { |prompt| prompt["name"] == "translate" }

    expect(translate_prompt["translated_name"]).to eq(
      I18n.t("discourse_ai.ai_helper.prompts.translate", language: "English (US)"),
    )

    I18n.with_locale("fr") do
      serializer = CurrentUserSerializer.new(french_user, scope: Guardian.new(french_user))
      parsed = JSON.parse(serializer.to_json)

      translate_prompt =
        parsed["current_user"]["ai_helper_prompts"].find { |prompt| prompt["name"] == "translate" }

      expect(translate_prompt["translated_name"]).to eq(
        I18n.t("discourse_ai.ai_helper.prompts.translate", locale: "fr", language: "Français"),
      )
    end
  end

  it "will include auto_image_caption field in the user_option if image caption is enabled" do
    assign_fake_provider_to(:ai_helper_model)
    assign_fake_provider_to(:ai_helper_image_caption_model)
    SiteSetting.ai_helper_enabled = true
    SiteSetting.ai_helper_enabled_features = "image_caption"
    SiteSetting.ai_auto_image_caption_allowed_groups = "10" # tl0
    serializer = CurrentUserSerializer.new(english_user, scope: Guardian.new(english_user))

    expect(serializer.user_option.auto_image_caption).to eq(false)
  end
end