122 lines
4.4 KiB
Ruby
122 lines
4.4 KiB
Ruby
# 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
|