28 lines
726 B
Ruby
28 lines
726 B
Ruby
# frozen_string_literal: true
|
|
|
|
class ExperimentSubject < ApplicationRecord
|
|
include ::Gitlab::Experimentation::GroupTypes
|
|
|
|
belongs_to :experiment, inverse_of: :experiment_subjects
|
|
belongs_to :user
|
|
belongs_to :group
|
|
belongs_to :project
|
|
|
|
validates :experiment, presence: true
|
|
validates :variant, presence: true
|
|
validate :must_have_one_subject_present
|
|
|
|
enum variant: { GROUP_CONTROL => 0, GROUP_EXPERIMENTAL => 1 }
|
|
|
|
private
|
|
|
|
def must_have_one_subject_present
|
|
if non_nil_subjects.length != 1
|
|
errors.add(:base, s_("ExperimentSubject|Must have exactly one of User, Group, or Project."))
|
|
end
|
|
end
|
|
|
|
def non_nil_subjects
|
|
@non_nil_subjects ||= [user, group, project].reject(&:blank?)
|
|
end
|
|
end
|