From c7b425a3e50a3b348c17e8df2d44e8019d6afe3c Mon Sep 17 00:00:00 2001
From: Jarka Kadlecova <jarka@gitlab.com>
Date: Thu, 9 Feb 2017 11:34:02 +0100
Subject: [PATCH] Mark as WIP based on MR commits only

---
 .../merge_requests/refresh_service.rb         |  6 ++-
 .../merge_requests/refresh_service_spec.rb    | 41 +++++++++++++++----
 2 files changed, 37 insertions(+), 10 deletions(-)

diff --git a/app/services/merge_requests/refresh_service.rb b/app/services/merge_requests/refresh_service.rb
index b4bfb0e5e8c..581d18032e6 100644
--- a/app/services/merge_requests/refresh_service.rb
+++ b/app/services/merge_requests/refresh_service.rb
@@ -144,7 +144,11 @@ module MergeRequests
       return unless @commits.present?
 
       merge_requests_for_source_branch.each do |merge_request|
-        wip_commit = @commits.detect(&:work_in_progress?)
+        commit_shas = merge_request.commits_sha
+
+        wip_commit = @commits.detect do |commit|
+          commit.work_in_progress? && commit_shas.include?(commit.sha)
+        end
 
         if wip_commit && !merge_request.work_in_progress?
           merge_request.update(title: merge_request.wip_title)
diff --git a/spec/services/merge_requests/refresh_service_spec.rb b/spec/services/merge_requests/refresh_service_spec.rb
index 2cc21acab7b..983dac6efdb 100644
--- a/spec/services/merge_requests/refresh_service_spec.rb
+++ b/spec/services/merge_requests/refresh_service_spec.rb
@@ -287,41 +287,64 @@ describe MergeRequests::RefreshService, services: true do
 
       it 'references the commit that caused the Work in Progress status' do
         refresh_service.execute(@oldrev, @newrev, 'refs/heads/master')
-
         allow(refresh_service).to receive(:find_new_commits)
         refresh_service.instance_variable_set("@commits", [
-          instance_double(
-            Commit,
+          double(
             id: 'aaaaaaa',
+            sha: '38008cb17ce1466d8fec2dfa6f6ab8dcfe5cf49e',
             short_id: 'aaaaaaa',
             title: 'Fix issue',
             work_in_progress?: false
           ),
-          instance_double(
-            Commit,
+          double(
             id: 'bbbbbbb',
+            sha: '498214de67004b1da3d820901307bed2a68a8ef6',
             short_id: 'bbbbbbb',
             title: 'fixup! Fix issue',
             work_in_progress?: true,
             to_reference: 'bbbbbbb'
           ),
-          instance_double(
-            Commit,
+          double(
             id: 'ccccccc',
+            sha: '1b12f15a11fc6e62177bef08f47bc7b5ce50b141',
             short_id: 'ccccccc',
             title: 'fixup! Fix issue',
             work_in_progress?: true,
             to_reference: 'ccccccc'
           ),
         ])
-
         refresh_service.execute(@oldrev, @newrev, 'refs/heads/wip')
         reload_mrs
-
         expect(@merge_request.notes.last.note).to eq(
           "marked as a **Work In Progress** from bbbbbbb"
         )
       end
+
+      it 'does not mark as WIP based on commits that do not belong to an MR' do
+        allow(refresh_service).to receive(:find_new_commits)
+        refresh_service.instance_variable_set("@commits", [
+          double(
+            id: 'aaaaaaa',
+            sha: 'aaaaaaa',
+            short_id: 'aaaaaaa',
+            title: 'Fix issue',
+            work_in_progress?: false
+          ),
+          double(
+            id: 'bbbbbbb',
+            sha: 'bbbbbbbb',
+            short_id: 'bbbbbbb',
+            title: 'fixup! Fix issue',
+            work_in_progress?: true,
+            to_reference: 'bbbbbbb'
+          )
+        ])
+
+        refresh_service.execute(@oldrev, @newrev, 'refs/heads/master')
+        reload_mrs
+
+        expect(@merge_request.work_in_progress?).to be_falsey
+      end
     end
 
     def reload_mrs