Welcome to mirror list, hosted at ThFree Co, Russian Federation.

migrate_from_legacy_storage_service_spec.rb « pages « services « spec - gitlab.com/gitlab-org/gitlab-foss.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
blob: 2a275bab4cc5f7f6ac065bf56f929d6202e8efe7 (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
# frozen_string_literal: true

require 'spec_helper'

RSpec.describe Pages::MigrateFromLegacyStorageService do
  let(:batch_size) { 10 }
  let(:mark_projects_as_not_deployed) { false }
  let(:service) { described_class.new(Rails.logger, migration_threads: 3, batch_size: batch_size, ignore_invalid_entries: false, mark_projects_as_not_deployed: mark_projects_as_not_deployed) }

  it 'does not try to migrate pages if pages are not deployed' do
    expect(::Pages::MigrateLegacyStorageToDeploymentService).not_to receive(:new)

    expect(service.execute).to eq(migrated: 0, errored: 0)
  end

  context 'when there is work for multiple threads' do
    let(:batch_size) { 2 } # override to force usage of multiple threads

    it 'uses multiple threads' do
      projects = create_list(:project, 20)
      projects.each do |project|
        project.mark_pages_as_deployed

        FileUtils.mkdir_p File.join(project.pages_path, "public")
        File.open(File.join(project.pages_path, "public/index.html"), "w") do |f|
          f.write("Hello!")
        end
      end

      threads = Concurrent::Set.new

      expect(service).to receive(:migrate_project).exactly(20).times.and_wrap_original do |m, *args|
        threads.add(Thread.current)

        # sleep to be 100% certain that once thread can't consume all the queue
        # it works without it, but I want to avoid making this test flaky
        sleep(0.01)

        m.call(*args)
      end

      expect(service.execute).to eq(migrated: 20, errored: 0)
      expect(threads.length).to eq(3)
    end
  end

  context 'when pages are marked as deployed' do
    let(:project) { create(:project) }

    before do
      project.mark_pages_as_deployed
    end

    context 'when pages directory does not exist' do
      context 'when mark_projects_as_not_deployed is set' do
        let(:mark_projects_as_not_deployed) { true }

        it 'counts project as migrated' do
          expect_next_instance_of(::Pages::MigrateLegacyStorageToDeploymentService, project, ignore_invalid_entries: false, mark_projects_as_not_deployed: true) do |service|
            expect(service).to receive(:execute).and_call_original
          end

          expect(service.execute).to eq(migrated: 1, errored: 0)
        end
      end

      it 'counts project as errored' do
        expect_next_instance_of(::Pages::MigrateLegacyStorageToDeploymentService, project, ignore_invalid_entries: false, mark_projects_as_not_deployed: false) do |service|
          expect(service).to receive(:execute).and_call_original
        end

        expect(service.execute).to eq(migrated: 0, errored: 1)
      end
    end

    context 'when pages directory exists on disk' do
      before do
        FileUtils.mkdir_p File.join(project.pages_path, "public")
        File.open(File.join(project.pages_path, "public/index.html"), "w") do |f|
          f.write("Hello!")
        end
      end

      it 'migrates pages projects without deployments' do
        expect_next_instance_of(::Pages::MigrateLegacyStorageToDeploymentService, project, ignore_invalid_entries: false, mark_projects_as_not_deployed: false) do |service|
          expect(service).to receive(:execute).and_call_original
        end

        expect do
          expect(service.execute).to eq(migrated: 1, errored: 0)
        end.to change { project.pages_metadatum.reload.pages_deployment }.from(nil)
      end

      context 'when deployed already exists for the project' do
        before do
          deployment = create(:pages_deployment, project: project)
          project.set_first_pages_deployment!(deployment)
        end

        it 'does not try to migrate project' do
          expect(::Pages::MigrateLegacyStorageToDeploymentService).not_to receive(:new)

          expect(service.execute).to eq(migrated: 0, errored: 0)
        end
      end
    end
  end
end