discourse-ai/spec/system/summarization/topic_summarization_spec.rb

74 lines
2.0 KiB
Ruby

# frozen_string_literal: true
RSpec.describe "Summarize a topic ", type: :system do
fab!(:current_user) { Fabricate(:user) }
fab!(:group)
fab!(: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
let(:summarization_result) { "This is a summary" }
let(:topic_page) { PageObjects::Pages::Topic.new }
let(:summary_box) { PageObjects::Components::AiSummaryBox.new }
before do
group.add(current_user)
assign_fake_provider_to(:ai_summarization_model)
SiteSetting.ai_summarization_enabled = true
SiteSetting.ai_custom_summarization_allowed_groups = group.id.to_s
sign_in(current_user)
end
context "when a summary is cached" do
before do
AiSummary.create!(
target: topic,
summarized_text: summarization_result,
algorithm: "test",
original_content_sha: "test",
summary_type: AiSummary.summary_types[:complete],
)
end
it "displays it" do
topic_page.visit_topic(topic)
summary_box.click_summarize
expect(summary_box).to have_summary(summarization_result)
end
end
context "when a summary is outdated" do
before do
AiSummary.create!(
target: topic,
summarized_text: summarization_result,
algorithm: "test",
original_content_sha: "test",
summary_type: AiSummary.summary_types[:complete],
)
end
fab!(:new_post) do
Fabricate(
:post,
topic: topic,
raw:
"Idk, I think pie is overrated. I prefer cake. Cake is the best dessert. I always eat cake. I never throw it at people.",
)
end
it "displays the new summary instead of a cached one" do
topic_page.visit_topic(topic)
summary_box.click_summarize
summary_box.click_regenerate_summary
expect(summary_box).to have_generating_summary_indicator
end
end
end