Fix polling for transformed individual notes30299-fix-polling-for-transformed-notes

Sets `updated_at` so that the transformed note is returned together with
the new reply

4 files changed, 15 insertions, 6 deletions

diff --git a/app/models/discussion.rb b/app/models/discussion.rb
index f2678e0597d..32529ebf71d 100644
--- a/app/models/discussion.rb
+++ b/app/models/discussion.rb
@@ -17,8 +17,6 @@ class Discussion
             :for_commit?,
             :for_merge_request?,
 
-            :save,
-
             to: :first_note
 
   def project_id
diff --git a/app/models/individual_note_discussion.rb b/app/models/individual_note_discussion.rb
index aab0ff93468..b4a661ae5b4 100644
--- a/app/models/individual_note_discussion.rb
+++ b/app/models/individual_note_discussion.rb
@@ -17,8 +17,12 @@ class IndividualNoteDiscussion < Discussion
     noteable.supports_replying_to_individual_notes? && Feature.enabled?(:reply_to_individual_notes)
   end
 
-  def convert_to_discussion!
-    first_note.becomes!(Discussion.note_class).to_discussion
+  def convert_to_discussion!(save: false)
+    first_note.becomes!(Discussion.note_class).to_discussion.tap do
+      # Save needs to be called on first_note instead of the transformed note
+      # because of https://gitlab.com/gitlab-org/gitlab-ce/issues/57324
+      first_note.save if save
+    end
   end
 
   def reply_attributes
diff --git a/app/services/notes/create_service.rb b/app/services/notes/create_service.rb
index b975c3a8cb6..5a6e7338b42 100644
--- a/app/services/notes/create_service.rb
+++ b/app/services/notes/create_service.rb
@@ -35,7 +35,7 @@ module Notes
 
       if !only_commands && note.save
         if note.part_of_discussion? && note.discussion.can_convert_to_discussion?
-          note.discussion.convert_to_discussion!.save(touch: false)
+          note.discussion.convert_to_discussion!(save: true)
         end
 
         todo_service.new_note(note, current_user)
diff --git a/spec/services/notes/create_service_spec.rb b/spec/services/notes/create_service_spec.rb
index 48f1d696ff6..1645b67c329 100644
--- a/spec/services/notes/create_service_spec.rb
+++ b/spec/services/notes/create_service_spec.rb
@@ -311,7 +311,14 @@ describe Notes::CreateService do
         end
 
         it 'converts existing note to DiscussionNote' do
-          expect { subject }.to change { existing_note.reload.type }.from(nil).to('DiscussionNote')
+          expect do
+            existing_note
+
+            Timecop.freeze(Time.now + 1.minute) { subject }
+
+            existing_note.reload
+          end.to change { existing_note.type }.from(nil).to('DiscussionNote')
+             .and change { existing_note.updated_at }
         end
       end
     end