Add latest changes from gitlab-org/gitlab@master

1 files changed, 70 insertions, 0 deletions

diff --git a/spec/services/ci/abort_pipelines_service_spec.rb b/spec/services/ci/abort_pipelines_service_spec.rb
new file mode 100644
index 00000000000..21c95606b51
--- /dev/null
+++ b/spec/services/ci/abort_pipelines_service_spec.rb
@@ -0,0 +1,70 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+RSpec.describe Ci::AbortPipelinesService do
+  let_it_be(:user) { create(:user) }
+  let_it_be(:project) { create(:project, namespace: user.namespace) }
+
+  let_it_be(:cancelable_pipeline, reload: true) { create(:ci_pipeline, :running, project: project, user: user) }
+  let_it_be(:manual_pipeline, reload: true) { create(:ci_pipeline, status: :manual, project: project, user: user) } # not cancelable
+  let_it_be(:other_users_pipeline, reload: true) { create(:ci_pipeline, :running, project: project, user: create(:user)) } # not this user's pipeline
+  let_it_be(:cancelable_build, reload: true) { create(:ci_build, :running, pipeline: cancelable_pipeline) }
+  let_it_be(:non_cancelable_build, reload: true) { create(:ci_build, :success, pipeline: cancelable_pipeline) }
+  let_it_be(:cancelable_stage, reload: true) { create(:ci_stage_entity, name: 'stageA', status: :running, pipeline: cancelable_pipeline, project: project) }
+  let_it_be(:non_cancelable_stage, reload: true) { create(:ci_stage_entity, name: 'stageB', status: :success, pipeline: cancelable_pipeline, project: project) }
+
+  describe '#execute' do
+    def expect_correct_cancellations
+      expect(cancelable_pipeline.finished_at).not_to be_nil
+      expect(cancelable_pipeline).to be_canceled
+      expect(cancelable_pipeline.stages - [non_cancelable_stage]).to all(be_canceled)
+      expect(cancelable_build).to be_canceled
+
+      expect(manual_pipeline).not_to be_canceled
+      expect(non_cancelable_stage).not_to be_canceled
+      expect(non_cancelable_build).not_to be_canceled
+    end
+
+    context 'with project pipelines' do
+      it 'cancels all running pipelines and related jobs' do
+        expect(described_class.new.execute(project.all_pipelines)).to be_success
+
+        expect_correct_cancellations
+
+        expect(other_users_pipeline).to be_canceled
+        expect(other_users_pipeline.stages).to all(be_canceled)
+      end
+
+      it 'avoids N+1 queries' do
+        project_pipelines = project.all_pipelines
+        control_count = ActiveRecord::QueryRecorder.new { described_class.new.execute(project_pipelines) }.count
+
+        pipelines = create_list(:ci_pipeline, 5, :running, project: project)
+        create_list(:ci_build, 5, :running, pipeline: pipelines.first)
+
+        expect { described_class.new.execute(project_pipelines) }.not_to exceed_query_limit(control_count)
+      end
+    end
+
+    context 'with user pipelines' do
+      it 'cancels all running pipelines and related jobs' do
+        expect(described_class.new.execute(user.pipelines)).to be_success
+
+        expect_correct_cancellations
+
+        expect(other_users_pipeline).not_to be_canceled
+      end
+
+      it 'avoids N+1 queries' do
+        user_pipelines = user.pipelines
+        control_count = ActiveRecord::QueryRecorder.new { described_class.new.execute(user_pipelines) }.count
+
+        pipelines = create_list(:ci_pipeline, 5, :running, project: project, user: user)
+        create_list(:ci_build, 5, :running, pipeline: pipelines.first)
+
+        expect { described_class.new.execute(user_pipelines) }.not_to exceed_query_limit(control_count)
+      end
+    end
+  end
+end