Create a pending task when an MR is assigned to someone

3 files changed, 109 insertions, 8 deletions

diff --git a/spec/services/merge_requests/create_service_spec.rb b/spec/services/merge_requests/create_service_spec.rb
index be8f1676eeb..2e3c9f88369 100644
--- a/spec/services/merge_requests/create_service_spec.rb
+++ b/spec/services/merge_requests/create_service_spec.rb
@@ -3,6 +3,7 @@ require 'spec_helper'
 describe MergeRequests::CreateService, services: true do
   let(:project) { create(:project) }
   let(:user) { create(:user) }
+  let(:assignee) { create(:user) }
 
   describe :execute do
     context 'valid params' do
@@ -14,10 +15,12 @@ describe MergeRequests::CreateService, services: true do
           target_branch: 'master'
         }
       end
+
       let(:service) { MergeRequests::CreateService.new(project, user, opts) }
 
       before do
         project.team << [user, :master]
+        project.team << [assignee, :developer]
         allow(service).to receive(:execute_hooks)
 
         @merge_request = service.execute
@@ -25,10 +28,47 @@ describe MergeRequests::CreateService, services: true do
 
       it { expect(@merge_request).to be_valid }
       it { expect(@merge_request.title).to eq('Awesome merge_request') }
+      it { expect(@merge_request.assignee).to be_nil }
 
       it 'should execute hooks with default action' do
         expect(service).to have_received(:execute_hooks).with(@merge_request)
       end
+
+      it 'does not creates a pending task' do
+        attributes = {
+          project: project,
+          target: @merge_request
+        }
+
+        expect(Task.where(attributes).count).to be_zero
+      end
+
+      context 'when merge request is assigned to someone' do
+        let(:opts) do
+          {
+            title: 'Awesome merge_request',
+            description: 'please fix',
+            source_branch: 'feature',
+            target_branch: 'master',
+            assignee: assignee
+          }
+        end
+
+        it { expect(@merge_request.assignee).to eq assignee }
+
+        it 'creates a pending task for new assignee' do
+          attributes = {
+            project: project,
+            author: user,
+            user: assignee,
+            target: @merge_request,
+            action: Task::ASSIGNED,
+            state: :pending
+          }
+
+          expect(Task.where(attributes).count).to eq 1
+        end
+      end
     end
   end
 end
diff --git a/spec/services/merge_requests/update_service_spec.rb b/spec/services/merge_requests/update_service_spec.rb
index 2e9e6e0870d..88853483784 100644
--- a/spec/services/merge_requests/update_service_spec.rb
+++ b/spec/services/merge_requests/update_service_spec.rb
@@ -98,6 +98,31 @@ describe MergeRequests::UpdateService, services: true do
       end
     end
 
+    context 'task queue' do
+      let!(:pending_task) do
+        create(:pending_assigned_task, user: user, project: project, target: merge_request, author: user2)
+      end
+
+      context 'when is reassigned' do
+        before do
+          update_merge_request({ assignee: user2 })
+        end
+
+        it 'creates a pending task for new assignee' do
+          attributes = {
+            project: project,
+            author: user,
+            user: user2,
+            target: merge_request,
+            action: Task::ASSIGNED,
+            state: :pending
+          }
+
+          expect(Task.where(attributes).count).to eq 1
+        end
+      end
+    end
+
     context 'when MergeRequest has tasks' do
       before { update_merge_request({ description: "- [ ] Task 1\n- [ ] Task 2" }) }
 
diff --git a/spec/services/task_service_spec.rb b/spec/services/task_service_spec.rb
index 2fd75d10b6c..0a072dda94b 100644
--- a/spec/services/task_service_spec.rb
+++ b/spec/services/task_service_spec.rb
@@ -1,20 +1,20 @@
 require 'spec_helper'
 
 describe TaskService, services: true do
+  let(:author) { create(:user) }
+  let(:john_doe) { create(:user) }
+  let(:project) { create(:project) }
   let(:service) { described_class.new }
 
+  before do
+    project.team << [author, :developer]
+    project.team << [john_doe, :developer]
+  end
+
   describe 'Issues' do
-    let(:author) { create(:user) }
-    let(:john_doe) { create(:user) }
-    let(:project) { create(:empty_project, :public) }
     let(:assigned_issue) { create(:issue, project: project, assignee: john_doe) }
     let(:unassigned_issue) { create(:issue, project: project, assignee: nil) }
 
-    before do
-      project.team << [author, :developer]
-      project.team << [john_doe, :developer]
-    end
-
     describe '#new_issue' do
       it 'creates a pending task if assigned' do
         service.new_issue(assigned_issue, author)
@@ -116,6 +116,42 @@ describe TaskService, services: true do
     end
   end
 
+  describe 'Merge Requests' do
+    let(:mr_assigned) { create(:merge_request, source_project: project, assignee: john_doe) }
+    let(:mr_unassigned) { create(:merge_request, source_project: project, assignee: nil) }
+
+    describe '#new_merge_request' do
+      it 'creates a pending task if assigned' do
+        service.new_merge_request(mr_assigned, author)
+
+        is_expected_to_create_pending_task(user: john_doe, target: mr_assigned, action: Task::ASSIGNED)
+      end
+
+      it 'does not create a task if unassigned' do
+        is_expected_to_not_create_task { service.new_merge_request(mr_unassigned, author) }
+      end
+
+      it 'does not create a task if assignee is the current user' do
+        is_expected_to_not_create_task { service.new_merge_request(mr_unassigned, john_doe) }
+      end
+    end
+
+    describe '#reassigned_merge_request' do
+      it 'creates a pending task for new assignee' do
+        mr_unassigned.update_attribute(:assignee, john_doe)
+        service.reassigned_merge_request(mr_unassigned, author)
+
+        is_expected_to_create_pending_task(user: john_doe, target: mr_unassigned, action: Task::ASSIGNED)
+      end
+
+      it 'does not create a task if unassigned' do
+        mr_assigned.update_attribute(:assignee, nil)
+
+        is_expected_to_not_create_task { service.reassigned_merge_request(mr_assigned, author) }
+      end
+    end
+  end
+
   def is_expected_to_create_pending_task(attributes = {})
     attributes.reverse_merge!(
       project: project,