gitlab-foss/spec/initializers/active_job_shard_support_spec.rb

# frozen_string_literal: true

require 'spec_helper'

RSpec.describe 'ActionMailer::MailDeliveryJob', :sidekiq_mailers, feature_category: :scalability do
  let(:mailer_class) do
    Class.new(ApplicationMailer) do
      def self.name
        'Notify'
      end

      def test_mail; end
    end
  end

  before do
    # The client middleware is invoked in a Sidekiq::Client.push, where
    # ActiveJob::QueueAdapters::SidekiqAdapter::JobWrapper is passed into the Router.
    allow(Gitlab::SidekiqSharding::Router).to receive(:get_shard_instance).with(nil).and_call_original
  end

  context 'when routing is not enabled' do
    before do
      allow(Gitlab::SidekiqSharding::Router).to receive(:enabled?).and_return(false)
    end

    it 'does not check for shard instance' do
      expect(ActionMailer::MailDeliveryJob).not_to receive(:sidekiq_options)

      mailer_class.test_mail.deliver_later
    end
  end

  context 'when routing is enabled' do
    before do
      allow(Gitlab::SidekiqSharding::Router).to receive(:enabled?).and_return(true)
    end

    it 'checks for shard instance and sets Sidekiq redis pool' do
      expect(Gitlab::SidekiqSharding::Router)
        .to receive(:route).with(ActionMailer::MailDeliveryJob).ordered.and_call_original
      expect(Sidekiq::Client).to receive(:via).ordered.and_call_original

      mailer_class.test_mail.deliver_later
    end
  end
end