1 files changed, 24 insertions, 14 deletions
diff --git a/spec/services/pages/delete_services_spec.rb b/spec/services/pages/delete_services_spec.rb
index f6d4694b4dd..440549020a2 100644
--- a/spec/services/pages/delete_services_spec.rb
+++ b/spec/services/pages/delete_services_spec.rb
@@ -3,25 +3,35 @@
 require 'spec_helper'
 
 RSpec.describe Pages::DeleteService do
-  let_it_be(:project) { create(:project, path: "my.project")}
-  let_it_be(:admin) { create(:admin) }
-  let_it_be(:domain) { create(:pages_domain, project: project) }
-  let_it_be(:service) { described_class.new(project, admin)}
+  shared_examples 'remove pages' do
+    let_it_be(:project) { create(:project, path: "my.project")}
+    let_it_be(:admin) { create(:admin) }
+    let_it_be(:domain) { create(:pages_domain, project: project) }
+    let_it_be(:service) { described_class.new(project, admin)}
 
-  it 'deletes published pages' do
-    expect_any_instance_of(Gitlab::PagesTransfer).to receive(:rename_project).and_return true
-    expect(PagesWorker).to receive(:perform_in).with(5.minutes, :remove, project.namespace.full_path, anything)
+    it 'deletes published pages' do
+      expect_any_instance_of(Gitlab::PagesTransfer).to receive(:rename_project).and_return true
+      expect(PagesWorker).to receive(:perform_in).with(5.minutes, :remove, project.namespace.full_path, anything)
 
-    service.execute
+      Sidekiq::Testing.inline! { service.execute }
 
-    expect(project.reload.pages_metadatum.deployed?).to be(false)
-  end
+      expect(project.reload.pages_metadatum.deployed?).to be(false)
+    end
+
+    it 'deletes all domains' do
+      expect(project.pages_domains.count).to be 1
 
-  it 'deletes all domains' do
-    expect(project.pages_domains.count).to be 1
+      Sidekiq::Testing.inline! { service.execute }
+
+      expect(project.reload.pages_domains.count).to be 0
+    end
+  end
 
-    service.execute
+  context 'with feature flag enabled' do
+    before do
+      expect(PagesRemoveWorker).to receive(:perform_async).and_call_original
+    end
 
-    expect(project.reload.pages_domains.count).to be 0
+    it_behaves_like 'remove pages'
   end
 end