discourse-ai/spec/system/ai_helper/ai_post_helper_spec.rb

# frozen_string_literal: true

RSpec.describe "AI Composer helper", type: :system, js: true do
  fab!(:user) { Fabricate(:admin) }
  fab!(:non_member_group) { Fabricate(:group) }
  fab!(:topic) { Fabricate(:topic) }
  fab!(:post) do
    Fabricate(
      :post,
      topic: topic,
      raw:
        "I like to eat pie. It is a very good dessert. Some people are wasteful by throwing pie at others but I do not do that. I always eat the pie.",
    )
  end
  fab!(:post_2) do
    Fabricate(:post, topic: topic, raw: "La lluvia en España se queda principalmente en el avión.")
  end
  let(:topic_page) { PageObjects::Pages::Topic.new }
  let(:post_ai_helper) { PageObjects::Components::AIHelperPostOptions.new }

  let(:explain_response) { <<~STRING }
    In this context, \"pie\" refers to a baked dessert typically consisting of a pastry crust and filling. 
    The person states they enjoy eating pie, considering it a good dessert. They note that some people wastefully 
    throw pie at others, but the person themselves chooses to eat the pie rather than throwing it. Overall, \"pie\" 
    is being used to refer the the baked dessert food item.
  STRING

  before do
    Group.find_by(id: Group::AUTO_GROUPS[:admins]).add(user)
    SiteSetting.composer_ai_helper_enabled = true
    sign_in(user)
  end

  def select_post_text(selected_post)
    topic_page.visit_topic(topic)
    page.execute_script(
      "var element = document.querySelector('#{topic_page.post_by_number_selector(selected_post.post_number)} .cooked p'); " +
        "var range = document.createRange(); " + "range.selectNodeContents(element); " +
        "var selection = window.getSelection(); " + "selection.removeAllRanges(); " +
        "selection.addRange(range);",
    )
  end

  context "when triggering AI helper in post" do
    it "shows the Ask AI button in the post selection toolbar" do
      select_post_text(post)
      expect(post_ai_helper).to have_post_selection_toolbar
      expect(post_ai_helper).to have_post_ai_helper
    end

    it "shows AI helper options after clicking the AI button" do
      select_post_text(post)
      post_ai_helper.click_ai_button
      expect(post_ai_helper).to have_no_post_selection_primary_buttons
      expect(post_ai_helper).to have_post_ai_helper_options
    end

    context "when using explain mode" do
      skip "TODO: Fix explain mode option not appearing in spec" do
        let(:mode) { CompletionPrompt::EXPLAIN }

        it "shows an explanation of the selected text" do
          select_post_text(post)
          post_ai_helper.click_ai_button

          DiscourseAi::Completions::LLM.with_prepared_responses([explain_response]) do
            post_ai_helper.select_helper_model(mode)

            wait_for { post_ai_helper.suggestion_value == explain_response }

            expect(post_ai_helper.suggestion_value).to eq(explain_response)
          end
        end
      end
    end

    context "when using translate mode" do
      skip "TODO: Fix WebMock request for translate mode not working" do
        let(:mode) { CompletionPrompt::TRANSLATE }

        let(:translated_input) { "The rain in Spain, stays mainly in the Plane." }

        it "shows a translation of the selected text" do
          select_post_text(post_2)
          post_ai_helper.click_ai_button

          DiscourseAi::Completions::LLM.with_prepared_responses([translated_input]) do
            post_ai_helper.select_helper_model(mode)

            wait_for { post_ai_helper.suggestion_value == translated_input }

            expect(post_ai_helper.suggestion_value).to eq(translated_input)
          end
        end
      end
    end
  end

  context "when AI helper is disabled" do
    before { SiteSetting.composer_ai_helper_enabled = false }

    it "does not show the Ask AI button in the post selection toolbar" do
      select_post_text(post)
      expect(post_ai_helper).to have_post_selection_toolbar
      expect(post_ai_helper).to have_no_post_ai_helper
    end
  end

  context "when user is not a member of the post AI helper allowed group" do
    before do
      SiteSetting.composer_ai_helper_enabled = true
      SiteSetting.post_ai_helper_allowed_groups = non_member_group.id.to_s
    end

    it "does not show the Ask AI button in the post selection toolbar" do
      select_post_text(post)
      expect(post_ai_helper).to have_post_selection_toolbar
      expect(post_ai_helper).to have_no_post_ai_helper
    end
  end
end