spec/migrations/add_default_project_approval_rules_vuln_allowed_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

# frozen_string_literal: true

require 'spec_helper'
require_migration!

RSpec.describe AddDefaultProjectApprovalRulesVulnAllowed do
  let(:namespaces) { table(:namespaces) }
  let(:projects) { table(:projects) }
  let(:namespace) { namespaces.create!(name: 'namespace', path: 'namespace') }
  let(:project) { projects.create!(name: 'project', path: 'project', namespace_id: namespace.id) }
  let(:approval_project_rules) { table(:approval_project_rules) }

  it 'updates records when vulnerabilities_allowed is nil' do
    records_to_migrate = 10

    records_to_migrate.times do |i|
      approval_project_rules.create!(name: "rule #{i}", project_id: project.id)
    end

    expect { migrate! }
      .to change { approval_project_rules.where(vulnerabilities_allowed: nil).count }
        .from(records_to_migrate)
        .to(0)
  end

  it 'defaults vulnerabilities_allowed to 0' do
    approval_project_rule = approval_project_rules.create!(name: "new rule", project_id: project.id)

    expect(approval_project_rule.vulnerabilities_allowed).to be_nil

    migrate!

    expect(approval_project_rule.reload.vulnerabilities_allowed).to eq(0)
  end
end