spec/migrations/reseed_repository_storages_weighted_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
35
36
37
38
39
40
41
42
43

# frozen_string_literal: true

require 'spec_helper'
require Rails.root.join('db', 'migrate', '20200509203901_reseed_repository_storages_weighted.rb')

RSpec.describe ReseedRepositoryStoragesWeighted do
  let(:storages) { { "foo" => {}, "baz" => {} } }
  let(:application_settings) do
    table(:application_settings).tap do |klass|
      klass.class_eval do
        serialize :repository_storages
      end
    end
  end

  before do
    allow(Gitlab.config.repositories).to receive(:storages).and_return(storages)
  end

  let(:repository_storages) { ["foo"] }
  let!(:application_setting) { application_settings.create!(repository_storages: repository_storages) }

  context 'with empty repository_storages_weighted column' do
    it 'populates repository_storages_weighted properly' do
      migrate!

      expect(application_settings.find(application_setting.id).repository_storages_weighted).to eq({ "foo" => 100, "baz" => 0 })
    end
  end

  context 'with already-populated repository_storages_weighted column' do
    let(:existing_weights) { { "foo" => 100, "baz" => 50 } }

    it 'does not change repository_storages_weighted properly' do
      application_setting.repository_storages_weighted = existing_weights
      application_setting.save!

      migrate!

      expect(application_settings.find(application_setting.id).repository_storages_weighted).to eq(existing_weights)
    end
  end
end