71 lines
2.3 KiB
Ruby
71 lines
2.3 KiB
Ruby
|
# frozen_string_literal: true
|
||
|
|
|
||
|
|
RSpec.describe Jobs::UpdateHotTopicGist do
|
||
|
|
describe "#execute" do
|
||
|
|
fab!(:topic_1) { Fabricate(:topic) }
|
||
|
|
fab!(:post_1) { Fabricate(:post, topic: topic_1, post_number: 1) }
|
||
|
|
fab!(:post_2) { Fabricate(:post, topic: topic_1, post_number: 2) }
|
||
|
|
|
||
|
|
before do
|
||
|
|
assign_fake_provider_to(:ai_summarization_model)
|
||
|
|
SiteSetting.ai_summarization_enabled = true
|
||
|
|
SiteSetting.ai_summarize_max_hot_topics_gists_per_batch = 100
|
||
|
|
end
|
||
|
|
|
||
|
|
context "when the hot topic has a gist" do
|
||
|
|
before { TopicHotScore.create!(topic_id: topic_1.id, score: 0.1) }
|
||
|
|
fab!(:ai_gist) do
|
||
|
|
Fabricate(:topic_ai_gist, target: topic_1, original_content_sha: AiSummary.build_sha("12"))
|
||
|
|
end
|
||
|
|
let(:updated_gist) { "They updated me :(" }
|
||
|
|
|
||
|
|
context "when it's up to date" do
|
||
|
|
it "does nothing" do
|
||
|
|
DiscourseAi::Completions::Llm.with_prepared_responses([updated_gist]) do
|
||
|
|
subject.execute(topic_id: topic_1.id)
|
||
|
|
end
|
||
|
|
|
||
|
|
gist = AiSummary.gist.find_by(target: topic_1)
|
||
|
|
expect(AiSummary.gist.where(target: topic_1).count).to eq(1)
|
||
|
|
expect(gist.summarized_text).not_to eq(updated_gist)
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
context "when it's outdated" do
|
||
|
|
it "regenerates the gist using the latest data" do
|
||
|
|
Fabricate(:post, topic: topic_1, post_number: 3)
|
||
|
|
|
||
|
|
DiscourseAi::Completions::Llm.with_prepared_responses([updated_gist]) do
|
||
|
|
subject.execute(topic_id: topic_1.id)
|
||
|
|
end
|
||
|
|
|
||
|
|
gist = AiSummary.gist.find_by(target: topic_1)
|
||
|
|
expect(AiSummary.gist.where(target: topic_1).count).to eq(1)
|
||
|
|
expect(gist.summarized_text).to eq(updated_gist)
|
||
|
|
expect(gist.original_content_sha).to eq(AiSummary.build_sha("123"))
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
context "when the topic doesn't have a hot topic score" do
|
||
|
|
it "does nothing" do
|
||
|
|
subject.execute({})
|
||
|
|
|
||
|
|
gist = AiSummary.gist.find_by(target: topic_1)
|
||
|
|
expect(gist).to be_nil
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
context "when the topic has a hot topic score but no gist" do
|
||
|
|
before { TopicHotScore.create!(topic_id: topic_1.id, score: 0.1) }
|
||
|
|
|
||
|
|
it "does nothing" do
|
||
|
|
subject.execute({})
|
||
|
|
|
||
|
|
gist = AiSummary.gist.find_by(target: topic_1)
|
||
|
|
expect(gist).to be_nil
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|