diff options
Diffstat (limited to 'spec/tasks/gitlab/storage_rake_spec.rb')
-rw-r--r-- | spec/tasks/gitlab/storage_rake_spec.rb | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/spec/tasks/gitlab/storage_rake_spec.rb b/spec/tasks/gitlab/storage_rake_spec.rb index fefcd05af3b..570f67c8bb7 100644 --- a/spec/tasks/gitlab/storage_rake_spec.rb +++ b/spec/tasks/gitlab/storage_rake_spec.rb @@ -90,7 +90,7 @@ RSpec.describe 'rake gitlab:storage:*', :silence_stdout do shared_examples 'wait until database is ready' do it 'checks if the database is ready once' do - expect(Gitlab::Database).to receive(:exists?).once + expect(Gitlab::Database.main).to receive(:exists?).once run_rake_task(task) end @@ -102,7 +102,7 @@ RSpec.describe 'rake gitlab:storage:*', :silence_stdout do end it 'tries for 3 times, polling every 0.1 seconds' do - expect(Gitlab::Database).to receive(:exists?).exactly(3).times.and_return(false) + expect(Gitlab::Database.main).to receive(:exists?).exactly(3).times.and_return(false) run_rake_task(task) end |