1 files changed, 28 insertions, 0 deletions
diff --git a/app/models/experiment_subject.rb b/app/models/experiment_subject.rb
new file mode 100644
index 00000000000..51ffc0b304e
--- /dev/null
+++ b/app/models/experiment_subject.rb
@@ -0,0 +1,28 @@
+# 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