discourse-ai/spec/tasks/backfill_spec.rb

# 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