gitlab-org--gitlab-foss/spec/initializers/forbid_sidekiq_in_transactions_spec.rb

59 lines
1.5 KiB
Ruby

# frozen_string_literal: true
require 'spec_helper'
RSpec.describe 'Sidekiq::Worker' do
shared_examples_for 'a forbiddable operation within a transaction' do
it 'allows the operation outside of a transaction' do
expect { operation }.not_to raise_error
end
it 'forbids the operation within a transaction' do
ApplicationRecord.transaction do
expect { operation }.to raise_error(Sidekiq::Worker::EnqueueFromTransactionError)
end
end
it 'allows the oepration within a transaction if skipped' do
Sidekiq::Worker.skipping_transaction_check do
ApplicationRecord.transaction do
expect { operation }.not_to raise_error
end
end
end
it 'forbids the operation if it is within a Ci::ApplicationRecord transaction' do
Ci::Pipeline.transaction do
expect { operation }.to raise_error(Sidekiq::Worker::EnqueueFromTransactionError)
end
end
end
context 'for sidekiq workers' do
let(:worker_class) do
Class.new do
include Sidekiq::Worker
def perform
end
end
end
let(:operation) { worker_class.perform_async }
it_behaves_like 'a forbiddable operation within a transaction'
end
context 'for mailers' do
let(:mailer_class) do
Class.new(ApplicationMailer) do
def test_mail
end
end
end
let(:operation) { mailer_class.test_mail.deliver_later }
it_behaves_like 'a forbiddable operation within a transaction'
end
end