Fix mocks in background migrations specs

1 files changed, 5 insertions, 1 deletions

diff --git a/spec/lib/gitlab/background_migration_spec.rb b/spec/lib/gitlab/background_migration_spec.rb
index 55bf7040ef3..2f3e7aba1f4 100644
--- a/spec/lib/gitlab/background_migration_spec.rb
+++ b/spec/lib/gitlab/background_migration_spec.rb
@@ -10,12 +10,16 @@ describe Gitlab::BackgroundMigration do
 
   describe '.steal' do
     context 'when there are enqueued jobs present' do
-      let(:queue) { [double(:job, args: ['Foo', [10, 20]])] }
+      let(:job) { double(:job, args: ['Foo', [10, 20]]) }
+      let(:queue) { [job] }
 
       before do
         allow(Sidekiq::Queue).to receive(:new)
           .with(described_class.queue)
           .and_return(queue)
+
+        allow(job).to receive(:queue)
+          .and_return(described_class.queue)
       end
 
       it 'steals jobs from a queue' do