84 lines
2.8 KiB
Ruby
84 lines
2.8 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
RSpec.describe Jobs::StreamTopicAiSummary do
|
|
subject(:job) { described_class.new }
|
|
|
|
describe "#execute" do
|
|
fab!(:topic) { Fabricate(:topic, highest_post_number: 2) }
|
|
fab!(:post_1) { Fabricate(:post, topic: topic, post_number: 1) }
|
|
fab!(:post_2) { Fabricate(:post, topic: topic, post_number: 2) }
|
|
fab!(:user) { Fabricate(:leader) }
|
|
|
|
before do
|
|
Group.find(Group::AUTO_GROUPS[:trust_level_3]).add(user)
|
|
assign_fake_provider_to(:ai_summarization_model)
|
|
SiteSetting.ai_summarization_enabled = true
|
|
end
|
|
|
|
def with_responses(responses)
|
|
DiscourseAi::Completions::Llm.with_prepared_responses(responses) { yield }
|
|
end
|
|
|
|
describe "validates params" do
|
|
it "does nothing if there is no topic" do
|
|
messages =
|
|
MessageBus.track_publish("/discourse-ai/summaries/topic/#{topic.id}") do
|
|
job.execute(topic_id: nil, user_id: user.id)
|
|
end
|
|
|
|
expect(messages).to be_empty
|
|
end
|
|
|
|
it "does nothing if there is no user" do
|
|
messages =
|
|
MessageBus.track_publish("/discourse-ai/summaries/topic/#{topic.id}") do
|
|
job.execute(topic_id: topic.id, user_id: nil)
|
|
end
|
|
|
|
expect(messages).to be_empty
|
|
end
|
|
|
|
it "does nothing if the user is not allowed to see the topic" do
|
|
private_topic = Fabricate(:private_message_topic)
|
|
|
|
messages =
|
|
MessageBus.track_publish("/discourse-ai/summaries/topic/#{private_topic.id}") do
|
|
job.execute(topic_id: private_topic.id, user_id: user.id)
|
|
end
|
|
|
|
expect(messages).to be_empty
|
|
end
|
|
end
|
|
|
|
it "publishes updates with a partial summary" do
|
|
with_responses(["dummy"]) do
|
|
messages =
|
|
MessageBus.track_publish("/discourse-ai/summaries/topic/#{topic.id}") do
|
|
job.execute(topic_id: topic.id, user_id: user.id)
|
|
end
|
|
|
|
partial_summary_update = messages.first.data
|
|
expect(partial_summary_update[:done]).to eq(false)
|
|
expect(partial_summary_update.dig(:ai_topic_summary, :summarized_text)).to eq("dummy")
|
|
end
|
|
end
|
|
|
|
it "publishes a final update to signal we're done and provide metadata" do
|
|
with_responses(["dummy"]) do
|
|
messages =
|
|
MessageBus.track_publish("/discourse-ai/summaries/topic/#{topic.id}") do
|
|
job.execute(topic_id: topic.id, user_id: user.id)
|
|
end
|
|
|
|
final_update = messages.last.data
|
|
expect(final_update[:done]).to eq(true)
|
|
|
|
expect(final_update.dig(:ai_topic_summary, :algorithm)).to eq("fake")
|
|
expect(final_update.dig(:ai_topic_summary, :outdated)).to eq(false)
|
|
expect(final_update.dig(:ai_topic_summary, :can_regenerate)).to eq(true)
|
|
expect(final_update.dig(:ai_topic_summary, :new_posts_since_summary)).to be_zero
|
|
end
|
|
end
|
|
end
|
|
end
|