Add latest changes from gitlab-org/gitlab@master

1 files changed, 24 insertions, 0 deletions

diff --git a/spec/experiments/force_company_trial_experiment_spec.rb b/spec/experiments/force_company_trial_experiment_spec.rb
new file mode 100644
index 00000000000..42a3245771a
--- /dev/null
+++ b/spec/experiments/force_company_trial_experiment_spec.rb
@@ -0,0 +1,24 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+RSpec.describe ForceCompanyTrialExperiment, :experiment do
+  subject { described_class.new(current_user: user) }
+
+  let(:user) { create(:user, setup_for_company: setup_for_company) }
+  let(:setup_for_company) { true }
+
+  context 'when a user is setup_for_company' do
+    it 'is not excluded' do
+      expect(subject).not_to exclude(user: user)
+    end
+  end
+
+  context 'when a user is not setup_for_company' do
+    let(:setup_for_company) { nil }
+
+    it 'is excluded' do
+      expect(subject).to exclude(user: user)
+    end
+  end
+end