Commit 433b6d2c authored by Grzegorz Bizon's avatar Grzegorz Bizon 💡
Browse files

Add more specs for background migrations queue processing

parent 2a0ead2c

spec/lib/gitlab/background_migration_spec.rb

+18 −1
Original line number Diff line number Diff line
@@ -36,9 +36,10 @@ 
    end



    context 'when there are scheduled jobs present', :sidekiq, :redis do

      it 'steals all jobs from the schedule sets' do

      it 'steals all jobs from the scheduled sets' do

        Sidekiq::Testing.disable! do

          BackgroundMigrationWorker.perform_in(10.minutes, 'Object')



          expect(Sidekiq::ScheduledSet.new).to be_one

          expect(described_class).to receive(:perform).with('Object', any_args)
@@ -48,6 +49,22 @@ 
        end

      end

    end



    context 'when there are enqueued and scheduled jobs present', :sidekiq, :redis do

      it 'steals from the scheduled sets queue first' do

        Sidekiq::Testing.disable! do

          expect(described_class).to receive(:perform)

            .with('Object', [1]).ordered

          expect(described_class).to receive(:perform)

            .with('Object', [2]).ordered



          BackgroundMigrationWorker.perform_async('Object', [2])

          BackgroundMigrationWorker.perform_in(10.minutes, 'Object', [1])



          described_class.steal('Object')

        end

      end

    end

  end



  describe '.perform' do