Add labels notes

2 files changed, 20 insertions, 2 deletions

diff --git a/spec/services/issues/update_service_spec.rb b/spec/services/issues/update_service_spec.rb
index 36030577835..964b3a707e4 100644
--- a/spec/services/issues/update_service_spec.rb
+++ b/spec/services/issues/update_service_spec.rb
@@ -5,6 +5,7 @@ describe Issues::UpdateService do
   let(:user) { create(:user) }
   let(:user2) { create(:user) }
   let(:issue) { create(:issue) }
+  let(:label) { create(:label) }
 
   before do
     project.team << [user, :master]
@@ -18,7 +19,8 @@ describe Issues::UpdateService do
           title: 'New title',
           description: 'Also please fix',
           assignee_id: user2.id,
-          state_event: 'close'
+          state_event: 'close',
+          label_ids: [label.id]
         }
 
         @issue = Issues::UpdateService.new(project, user, opts).execute(issue)
@@ -29,6 +31,8 @@ describe Issues::UpdateService do
       it { @issue.title.should == 'New title' }
       it { @issue.assignee.should == user2 }
       it { @issue.should be_closed }
+      it { @issue.labels.count.should == 1 }
+      it { @issue.labels.first.title.should == 'Bug' }
 
       it 'should send email to user2 about assign of new issue' do
         email = ActionMailer::Base.deliveries.last
@@ -40,6 +44,11 @@ describe Issues::UpdateService do
         note = @issue.notes.last
         note.note.should include "Reassigned to \@#{user2.username}"
       end
+
+      it 'should create system note about issue label edit' do
+        note = @issue.notes[1]
+        note.note.should include "Added ~#{label.id} label"
+      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 0e60baae2c4..b27acb47711 100644
--- a/spec/services/merge_requests/update_service_spec.rb
+++ b/spec/services/merge_requests/update_service_spec.rb
@@ -5,6 +5,7 @@ describe MergeRequests::UpdateService do
   let(:user2) { create(:user) }
   let(:merge_request) { create(:merge_request, :simple) }
   let(:project) { merge_request.project }
+  let(:label) { create(:label) }
 
   before do
     project.team << [user, :master]
@@ -18,7 +19,8 @@ describe MergeRequests::UpdateService do
           title: 'New title',
           description: 'Also please fix',
           assignee_id: user2.id,
-          state_event: 'close'
+          state_event: 'close',
+          label_ids: [label.id]
         }
       end
 
@@ -35,6 +37,8 @@ describe MergeRequests::UpdateService do
       it { @merge_request.title.should == 'New title' }
       it { @merge_request.assignee.should == user2 }
       it { @merge_request.should be_closed }
+      it { @merge_request.labels.count.should == 1 }
+      it { @merge_request.labels.first.title.should == 'Bug' }
 
       it 'should execute hooks with update action' do
         expect(service).to have_received(:execute_hooks).
@@ -51,6 +55,11 @@ describe MergeRequests::UpdateService do
         note = @merge_request.notes.last
         note.note.should include "Reassigned to \@#{user2.username}"
       end
+
+      it 'should create system note about merge_request label edit' do
+        note = @merge_request.notes[1]
+        note.note.should include "Added ~#{label.id} label"
+      end
     end
   end
 end