spec/migrations/add_default_and_free_plans_spec.rb


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

# frozen_string_literal: true

require 'spec_helper'
require_migration!('add_default_and_free_plans')

RSpec.describe AddDefaultAndFreePlans do
  describe 'migrate' do
    let(:plans) { table(:plans) }

    context 'when on Gitlab.com' do
      before do
        expect(Gitlab).to receive(:com?) { true }
      end

      it 'creates free and default plans' do
        expect { migrate! }.to change { plans.count }.by 2

        expect(plans.last(2).pluck(:name)).to eq %w[free default]
      end
    end

    context 'when on self-hosted' do
      before do
        expect(Gitlab).to receive(:com?) { false }
      end

      it 'creates only a default plan' do
        expect { migrate! }.to change { plans.count }.by 1

        expect(plans.last.name).to eq 'default'
      end
    end
  end
end