spec/tasks/gitlab/pages_rake_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

# frozen_string_literal: true

require 'rake_helper'

RSpec.describe 'gitlab:pages:migrate_legacy_storagerake task' do
  before(:context) do
    Rake.application.rake_require 'tasks/gitlab/pages'
  end

  subject { run_rake_task('gitlab:pages:migrate_legacy_storage') }

  it 'calls migration service' do
    expect_next_instance_of(::Pages::MigrateFromLegacyStorageService, anything, 3, 10) do |service|
      expect(service).to receive(:execute).and_call_original
    end

    subject
  end

  it 'uses PAGES_MIGRATION_THREADS environment variable' do
    stub_env('PAGES_MIGRATION_THREADS', '5')

    expect_next_instance_of(::Pages::MigrateFromLegacyStorageService, anything, 5, 10) do |service|
      expect(service).to receive(:execute).and_call_original
    end

    subject
  end

  it 'uses PAGES_MIGRATION_BATCH_SIZE environment variable' do
    stub_env('PAGES_MIGRATION_BATCH_SIZE', '100')

    expect_next_instance_of(::Pages::MigrateFromLegacyStorageService, anything, 3, 100) do |service|
      expect(service).to receive(:execute).and_call_original
    end

    subject
  end
end