debian-mirror-gitlab/spec/lib/gitlab/seeders/ci/runner/runner_fleet_pipeline_seeder_spec.rb
2023-05-08 21:46:49 +05:30

60 lines
2.2 KiB
Ruby

# frozen_string_literal: true
require 'spec_helper'
NULL_LOGGER = Gitlab::JsonLogger.new('/dev/null')
TAG_LIST = Gitlab::Seeders::Ci::Runner::RunnerFleetSeeder::TAG_LIST.to_set
RSpec.describe ::Gitlab::Seeders::Ci::Runner::RunnerFleetPipelineSeeder, feature_category: :runner_fleet do
subject(:seeder) do
described_class.new(NULL_LOGGER, projects_to_runners: projects_to_runners, job_count: job_count)
end
def runner_ids_for_project(runner_count, project)
create_list(:ci_runner, runner_count, :project, projects: [project], tag_list: TAG_LIST.to_a.sample(5)).map(&:id)
end
let_it_be(:projects) { create_list(:project, 4) }
let_it_be(:projects_to_runners) do
[
{ project_id: projects[0].id, runner_ids: runner_ids_for_project(2, projects[0]) },
{ project_id: projects[1].id, runner_ids: runner_ids_for_project(1, projects[1]) },
{ project_id: projects[2].id, runner_ids: runner_ids_for_project(2, projects[2]) },
{ project_id: projects[3].id, runner_ids: runner_ids_for_project(1, projects[3]) }
]
end
describe '#seed' do
context 'with job_count specified' do
let(:job_count) { 20 }
it 'creates expected jobs', :aggregate_failures,
quarantine: 'https://gitlab.com/gitlab-org/gitlab/-/issues/394721' do
expect { seeder.seed }.to change { Ci::Build.count }.by(job_count)
.and change { Ci::Pipeline.count }.by(4)
expect(Ci::Pipeline.where.not(started_at: nil).map(&:queued_duration)).to all(be < 5.minutes)
expect(Ci::Build.where.not(started_at: nil).map(&:queued_duration)).to all(be < 5.minutes)
projects_to_runners.first(3).each do |project|
expect(Ci::Build.where(runner_id: project[:runner_ids])).not_to be_empty
end
end
end
context 'with nil job_count' do
let(:job_count) { nil }
before do
stub_const('Gitlab::Seeders::Ci::Runner::RunnerFleetPipelineSeeder::DEFAULT_JOB_COUNT', 2)
end
it 'creates expected jobs', :aggregate_failures do
expect { seeder.seed }.to change { Ci::Build.count }.by(2)
.and change { Ci::Pipeline.count }.by(2)
expect(Ci::Build.last(2).map(&:tag_list).map(&:to_set)).to all satisfy { |r| r.subset?(TAG_LIST) }
end
end
end
end