90 lines
2.5 KiB
Ruby
90 lines
2.5 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require 'spec_helper'
|
|
|
|
RSpec.describe Ci::Pipelines::AddJobService do
|
|
include ExclusiveLeaseHelpers
|
|
|
|
let_it_be_with_reload(:pipeline) { create(:ci_pipeline) }
|
|
|
|
let(:job) { build(:ci_build) }
|
|
|
|
subject(:service) { described_class.new(pipeline) }
|
|
|
|
context 'when the pipeline is not persisted' do
|
|
let(:pipeline) { build(:ci_pipeline) }
|
|
|
|
it 'raises error' do
|
|
expect { service }.to raise_error('Pipeline must be persisted for this service to be used')
|
|
end
|
|
end
|
|
|
|
describe '#execute!' do
|
|
subject(:execute) do
|
|
service.execute!(job) do |job|
|
|
job.save!
|
|
end
|
|
end
|
|
|
|
it 'assigns pipeline attributes to the job' do
|
|
expect do
|
|
execute
|
|
end.to change { job.slice(:pipeline, :project, :ref) }.to(
|
|
pipeline: pipeline, project: pipeline.project, ref: pipeline.ref
|
|
).and change { job.metadata.project }.to(pipeline.project)
|
|
end
|
|
|
|
it 'assigns partition_id to job and metadata' do
|
|
pipeline.partition_id = ci_testing_partition_id
|
|
|
|
expect { execute }
|
|
.to change(job, :partition_id).to(pipeline.partition_id)
|
|
.and change { job.metadata.partition_id }.to(pipeline.partition_id)
|
|
end
|
|
|
|
it 'returns a service response with the job as payload' do
|
|
expect(execute).to be_success
|
|
expect(execute.payload[:job]).to eq(job)
|
|
end
|
|
|
|
it 'calls update_older_statuses_retried!' do
|
|
expect(job).to receive(:update_older_statuses_retried!)
|
|
|
|
execute
|
|
end
|
|
|
|
context 'when the block raises an error' do
|
|
subject(:execute) do
|
|
service.execute!(job) do |job|
|
|
raise "this is an error"
|
|
end
|
|
end
|
|
|
|
it 'returns a service response with the error and the job as payload' do
|
|
expect(execute).to be_error
|
|
expect(execute.payload[:job]).to eq(job)
|
|
expect(execute.message).to eq('this is an error')
|
|
end
|
|
end
|
|
|
|
context 'exclusive lock' do
|
|
let(:lock_uuid) { 'test' }
|
|
let(:lock_key) { "ci:pipelines:#{pipeline.id}:add-job" }
|
|
let(:lock_timeout) { 1.minute }
|
|
|
|
before do
|
|
# "Please stub a default value first if message might be received with other args as well."
|
|
allow(Gitlab::ExclusiveLease).to receive(:new).and_call_original
|
|
end
|
|
|
|
it 'uses exclusive lock' do
|
|
lease = stub_exclusive_lease(lock_key, lock_uuid, timeout: lock_timeout)
|
|
expect(lease).to receive(:try_obtain)
|
|
expect(lease).to receive(:cancel)
|
|
|
|
expect(execute).to be_success
|
|
expect(execute.payload[:job]).to eq(job)
|
|
end
|
|
end
|
|
end
|
|
end
|