remove_repetition

app/models/procedure_revision.rb

@@ -157,9 +157,13 @@ class ProcedureRevision < ApplicationRecord
   def children_of(tdc)
     parent_revision_type_de_champ = revision_types_de_champ.find_by(type_de_champ: tdc)
 
+    if parent_revision_type_de_champ.present?
       types_de_champ
         .where(procedure_revision_types_de_champ: { parent_id: parent_revision_type_de_champ.id })
         .order("procedure_revision_types_de_champ.position")
+    else
+      TypeDeChamp.none
+    end
   end
 
   private

app/models/type_de_champ.rb

@@ -433,8 +433,11 @@ class TypeDeChamp < ApplicationRecord
   end
 
   def remove_repetition
-    if !repetition?
+    if !repetition? && procedure.present?
-      types_de_champ.destroy_all
+      procedure
+        .draft_revision # action occurs only on draft
+        .children_of(self)
+        .destroy_all
     end
   end
 end

spec/models/procedure_revision_spec.rb

@@ -409,7 +409,7 @@ describe ProcedureRevision do
       let(:procedure) { create(:procedure, :with_repetition) }
 
       before do
-        child = new_draft.types_de_champ_public.last.types_de_champ.first
+        child = new_draft.children_of(new_draft.types_de_champ_public.last).first
         new_draft.find_or_clone_type_de_champ(child.stable_id).update(type_champ: :drop_down_list, drop_down_options: ['one', 'two'])
       end
 

spec/models/type_de_champ_shared_example.rb

@@ -101,7 +101,7 @@ shared_examples 'type_de_champ_spec' do
         let(:target_type_champ) { TypeDeChamp.type_champs.fetch(:text) }
 
         it 'removes the children types de champ' do
-          expect(tdc.types_de_champ).to be_empty
+          expect(procedure.draft_revision.children_of(tdc)).to be_empty
         end
       end
     end