# frozen_string_literal: true

require "rails_helper"
require_relative "../../../support/toxicity_inference_stubs"

describe DiscourseAI::Toxicity::ToxicityClassification do
  describe "#request" do
    fab!(:target) { Fabricate(:post) }

    it "returns the classification and the model used for it" do
      ToxicityInferenceStubs.stub_post_classification(target, toxic: false)

      result = subject.request(target)

      expect(result[SiteSetting.ai_toxicity_inference_service_api_model]).to eq(
        ToxicityInferenceStubs.civilized_response,
      )
    end
  end

  describe "#should_flag_based_on?" do
    before { SiteSetting.ai_toxicity_flag_automatically = true }

    let(:toxic_response) do
      {
        SiteSetting.ai_toxicity_inference_service_api_model =>
          ToxicityInferenceStubs.toxic_response,
      }
    end

    it "returns false when toxicity flaggin is disabled" do
      SiteSetting.ai_toxicity_flag_automatically = false

      should_flag = subject.should_flag_based_on?(toxic_response)

      expect(should_flag).to eq(false)
    end

    it "returns true if the response is toxic based on our thresholds" do
      should_flag = subject.should_flag_based_on?(toxic_response)

      expect(should_flag).to eq(true)
    end

    it "returns false if the response is civilized based on our thresholds" do
      civilized_response = {
        SiteSetting.ai_toxicity_inference_service_api_model =>
          ToxicityInferenceStubs.civilized_response,
      }

      should_flag = subject.should_flag_based_on?(civilized_response)

      expect(should_flag).to eq(false)
    end
  end
end