1 files changed, 19 insertions, 1 deletions
diff --git a/spec/lib/gitlab/database/async_indexes_spec.rb b/spec/lib/gitlab/database/async_indexes_spec.rb
index 74e30ea2c4e..8a5509f892f 100644
--- a/spec/lib/gitlab/database/async_indexes_spec.rb
+++ b/spec/lib/gitlab/database/async_indexes_spec.rb
@@ -11,7 +11,7 @@ RSpec.describe Gitlab::Database::AsyncIndexes do
     end
 
     it 'takes 2 pending indexes and creates those' do
-      Gitlab::Database::AsyncIndexes::PostgresAsyncIndex.order(:id).limit(2).each do |index|
+      Gitlab::Database::AsyncIndexes::PostgresAsyncIndex.to_create.order(:id).limit(2).each do |index|
         creator = double('index creator')
         expect(Gitlab::Database::AsyncIndexes::IndexCreator).to receive(:new).with(index).and_return(creator)
         expect(creator).to receive(:perform)
@@ -20,4 +20,22 @@ RSpec.describe Gitlab::Database::AsyncIndexes do
       subject
     end
   end
+
+  describe '.drop_pending_indexes!' do
+    subject { described_class.drop_pending_indexes! }
+
+    before do
+      create_list(:postgres_async_index, 4, :with_drop)
+    end
+
+    it 'takes 2 pending indexes and destroys those' do
+      Gitlab::Database::AsyncIndexes::PostgresAsyncIndex.to_drop.order(:id).limit(2).each do |index|
+        destructor = double('index destructor')
+        expect(Gitlab::Database::AsyncIndexes::IndexDestructor).to receive(:new).with(index).and_return(destructor)
+        expect(destructor).to receive(:perform)
+      end
+
+      subject
+    end
+  end
 end