discourse-ai/spec/tasks/backfill_spec.rb

28 lines
1012 B
Ruby

# frozen_string_literal: true
require_relative "../support/sentiment_inference_stubs"
RSpec.describe "assets:precompile" do
before do
Rake::Task.clear
Discourse::Application.load_tasks
end
describe "ai:sentiment:backfill" do
before do
SiteSetting.ai_sentiment_model_configs =
"[{\"model_name\":\"SamLowe/roberta-base-go_emotions\",\"endpoint\":\"http://samlowe-emotion.com\",\"api_key\":\"123\"},{\"model_name\":\"j-hartmann/emotion-english-distilroberta-base\",\"endpoint\":\"http://jhartmann-emotion.com\",\"api_key\":\"123\"},{\"model_name\":\"cardiffnlp/twitter-roberta-base-sentiment-latest\",\"endpoint\":\"http://cardiffnlp-sentiment.com\",\"api_key\":\"123\"}]"
end
it "does nothing if the topic is soft-deleted" do
target = Fabricate(:post)
SentimentInferenceStubs.stub_classification(target)
target.topic.trash!
path = Rake::Task["ai:sentiment:backfill"].invoke
expect(ClassificationResult.count).to be_zero
end
end
end