summaryrefslogtreecommitdiff
path: root/spec/services/pages/migrate_from_legacy_storage_service_spec.rb
blob: b3328d540e79a2c8e35c7f3b11e5e61aa6aa66c9 (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
# frozen_string_literal: true

require 'spec_helper'

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

  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

  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

    service = described_class.new(Rails.logger, migration_threads: 3, batch_size: 2, ignore_invalid_entries: false)

    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

  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
      it 'counts project as migrated' do
        expect_next_instance_of(::Pages::MigrateLegacyStorageToDeploymentService, project, ignore_invalid_entries: false) do |service|
          expect(service).to receive(:execute).and_call_original
        end

        expect(service.execute).to eq(migrated: 1, errored: 0)
      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) 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