vulnerability_feedback.rb

class VulnerabilityFeedback < ActiveRecord::Base
  belongs_to :project
  belongs_to :author, class_name: "User"
  belongs_to :issue
  belongs_to :pipeline, class_name: 'Ci::Pipeline', foreign_key: :pipeline_id

  attr_accessor :vulnerability_data

  enum feedback_type: { dismissal: 0, issue: 1 }
  enum category: { sast: 0, dependency_scanning: 1, container_scanning: 2, dast: 3 }

  validates :project, presence: true
  validates :author, presence: true
  validates :issue, presence: true, if: :issue?
  validates :vulnerability_data, presence: true, if: :issue?
  validates :feedback_type, presence: true
  validates :category, presence: true
  validates :project_fingerprint, presence: true, uniqueness: { scope: [:project_id, :category, :feedback_type] }

  scope :with_associations, -> { includes(:pipeline, :issue, :author) }
end