summaryrefslogtreecommitdiff
path: root/spec/models/experiment_spec.rb
blob: 64cd2da4621762a62bab1a92f73d45f0faf8b8b4 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
# frozen_string_literal: true

require 'spec_helper'

RSpec.describe Experiment do
  subject { build(:experiment) }

  describe 'associations' do
    it { is_expected.to have_many(:experiment_users) }
    it { is_expected.to have_many(:users) }
    it { is_expected.to have_many(:control_group_users) }
    it { is_expected.to have_many(:experimental_group_users) }

    describe 'control_group_users and experimental_group_users' do
      let(:experiment) { create(:experiment) }
      let(:control_group_user) { build(:user) }
      let(:experimental_group_user) { build(:user) }

      before do
        experiment.control_group_users << control_group_user
        experiment.experimental_group_users << experimental_group_user
      end

      describe 'control_group_users' do
        subject { experiment.control_group_users }

        it { is_expected.to contain_exactly(control_group_user) }
      end

      describe 'experimental_group_users' do
        subject { experiment.experimental_group_users }

        it { is_expected.to contain_exactly(experimental_group_user) }
      end
    end
  end

  describe 'validations' do
    it { is_expected.to validate_presence_of(:name) }
    it { is_expected.to validate_uniqueness_of(:name) }
    it { is_expected.to validate_length_of(:name).is_at_most(255) }
  end

  describe '.add_user' do
    let(:name) { :experiment_key }
    let(:user) { build(:user) }

    let!(:experiment) { create(:experiment, name: name) }

    subject { described_class.add_user(name, :control, user) }

    describe 'creating a new experiment record' do
      context 'an experiment with the provided name already exists' do
        it 'does not create a new experiment record' do
          expect { subject }.not_to change(Experiment, :count)
        end
      end

      context 'an experiment with the provided name does not exist yet' do
        let(:experiment) { nil }

        it 'creates a new experiment record' do
          expect { subject }.to change(Experiment, :count).by(1)
        end
      end
    end

    describe 'creating a new experiment_user record' do
      context 'an experiment_user record for this experiment already exists' do
        before do
          subject
        end

        it 'does not create a new experiment_user record' do
          expect { subject }.not_to change(ExperimentUser, :count)
        end
      end

      context 'an experiment_user record for this experiment does not exist yet' do
        it 'creates a new experiment_user record' do
          expect { subject }.to change(ExperimentUser, :count).by(1)
        end

        it 'assigns the correct group_type to the experiment_user' do
          expect { subject }.to change { experiment.control_group_users.count }.by(1)
        end
      end
    end
  end

  describe '#add_control_user' do
    let(:experiment) { create(:experiment) }
    let(:user) { build(:user) }

    subject { experiment.add_control_user(user) }

    it 'creates a new experiment_user record and assigns the correct group_type' do
      expect { subject }.to change { experiment.control_group_users.count }.by(1)
    end
  end

  describe '#add_experimental_user' do
    let(:experiment) { create(:experiment) }
    let(:user) { build(:user) }

    subject { experiment.add_experimental_user(user) }

    it 'creates a new experiment_user record and assigns the correct group_type' do
      expect { subject }.to change { experiment.experimental_group_users.count }.by(1)
    end
  end
end