1 files changed, 17 insertions, 5 deletions
diff --git a/spec/requests/api/project_container_repositories_spec.rb b/spec/requests/api/project_container_repositories_spec.rb
index 506e60d19a6..52ec06d76a9 100644
--- a/spec/requests/api/project_container_repositories_spec.rb
+++ b/spec/requests/api/project_container_repositories_spec.rb
@@ -138,14 +138,26 @@ RSpec.describe API::ProjectContainerRepositories do
         context 'for maintainer' do
           let(:api_user) { maintainer }
 
-          it 'schedules removal of repository' do
-            expect(DeleteContainerRepositoryWorker).to receive(:perform_async)
-              .with(maintainer.id, root_repository.id)
-
-            subject
+          it 'marks the repository as delete_scheduled' do
+            expect(DeleteContainerRepositoryWorker).not_to receive(:perform_async)
+            expect { subject }.to change { root_repository.reload.status }.from(nil).to('delete_scheduled')
 
             expect(response).to have_gitlab_http_status(:accepted)
           end
+
+          context 'with container_registry_delete_repository_with_cron_worker disabled' do
+            before do
+              stub_feature_flags(container_registry_delete_repository_with_cron_worker: false)
+            end
+
+            it 'schedules removal of repository' do
+              expect(DeleteContainerRepositoryWorker).to receive(:perform_async)
+                .with(maintainer.id, root_repository.id)
+              expect { subject }.to change { root_repository.reload.status }.from(nil).to('delete_scheduled')
+
+              expect(response).to have_gitlab_http_status(:accepted)
+            end
+          end
         end
       end
     end