1 files changed, 37 insertions, 0 deletions
diff --git a/spec/services/releases/update_service_spec.rb b/spec/services/releases/update_service_spec.rb
index 14e6a5f13c8..944f3d8c9ad 100644
--- a/spec/services/releases/update_service_spec.rb
+++ b/spec/services/releases/update_service_spec.rb
@@ -48,5 +48,42 @@ describe Releases::UpdateService do
 
       it_behaves_like 'a failed update'
     end
+
+    context 'when a milestone is passed in' do
+      let(:old_title) { 'v1.0' }
+      let(:new_title) { 'v2.0' }
+      let(:milestone) { create(:milestone, project: project, title: old_title) }
+      let(:new_milestone) { create(:milestone, project: project, title: new_title) }
+      let(:params_with_milestone) { params.merge!({ milestone: new_title }) }
+
+      before do
+        release.milestone = milestone
+        release.save!
+
+        described_class.new(new_milestone.project, user, params_with_milestone).execute
+        release.reload
+      end
+
+      it 'updates the related milestone accordingly' do
+        expect(release.milestone.title).to eq(new_title)
+      end
+    end
+
+    context "when an 'empty' milestone is passed in" do
+      let(:milestone) { create(:milestone, project: project, title: 'v1.0') }
+      let(:params_with_empty_milestone) { params.merge!({ milestone: '' }) }
+
+      before do
+        release.milestone = milestone
+        release.save!
+
+        described_class.new(milestone.project, user, params_with_empty_milestone).execute
+        release.reload
+      end
+
+      it 'removes the old milestone and does not associate any new milestone' do
+        expect(release.milestone).to be_nil
+      end
+    end
   end
 end