summaryrefslogtreecommitdiff
path: root/spec/lib/gitlab/background_migration/delete_orphaned_deployments_spec.rb
blob: 8f058c875a21345aa9cad3e7c4d30ffaff3456aa (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
# frozen_string_literal: true

require 'spec_helper'

RSpec.describe Gitlab::BackgroundMigration::DeleteOrphanedDeployments, :migration, schema: 20210826171758 do
  let!(:namespace) { table(:namespaces).create!(name: 'user', path: 'user') }
  let!(:project) { table(:projects).create!(namespace_id: namespace.id) }
  let!(:environment) { table(:environments).create!(name: 'production', slug: 'production', project_id: project.id) }
  let(:background_migration_jobs) { table(:background_migration_jobs) }

  before do
    create_deployment!(environment.id, project.id)
  end

  it 'deletes only orphaned deployments' do
    expect(valid_deployments.pluck(:id)).not_to be_empty

    subject.perform(table(:deployments).minimum(:id), table(:deployments).maximum(:id))

    expect(valid_deployments.pluck(:id)).not_to be_empty
  end

  it 'marks jobs as done' do
    first_job = background_migration_jobs.create!(
      class_name: 'DeleteOrphanedDeployments',
      arguments: [table(:deployments).minimum(:id), table(:deployments).minimum(:id)]
    )

    subject.perform(table(:deployments).minimum(:id), table(:deployments).minimum(:id))

    expect(first_job.reload.status).to eq(Gitlab::Database::BackgroundMigrationJob.statuses[:succeeded])
  end

  private

  def valid_deployments
    table(:deployments).where('EXISTS (SELECT 1 FROM environments WHERE deployments.environment_id = environments.id)')
  end

  def orphaned_deployments
    table(:deployments).where('NOT EXISTS (SELECT 1 FROM environments WHERE deployments.environment_id = environments.id)')
  end

  def create_deployment!(environment_id, project_id)
    table(:deployments).create!(
      environment_id: environment_id,
      project_id: project_id,
      ref: 'master',
      tag: false,
      sha: 'x',
      status: 1,
      iid: table(:deployments).count + 1)
  end
end