1 files changed, 16 insertions, 0 deletions
diff --git a/spec/lib/gitlab/import_export/project/tree_restorer_spec.rb b/spec/lib/gitlab/import_export/project/tree_restorer_spec.rb
index 47d7555c8f4..299e107c881 100644
--- a/spec/lib/gitlab/import_export/project/tree_restorer_spec.rb
+++ b/spec/lib/gitlab/import_export/project/tree_restorer_spec.rb
@@ -192,10 +192,26 @@ RSpec.describe Gitlab::ImportExport::Project::TreeRestorer do
           expect(Issue.find_by(title: 'Voluptatem').resource_label_events).not_to be_empty
         end
 
+        it 'restores issue resource milestone events' do
+          expect(Issue.find_by(title: 'Voluptatem').resource_milestone_events).not_to be_empty
+        end
+
+        it 'restores issue resource state events' do
+          expect(Issue.find_by(title: 'Voluptatem').resource_state_events).not_to be_empty
+        end
+
         it 'restores merge requests resource label events' do
           expect(MergeRequest.find_by(title: 'MR1').resource_label_events).not_to be_empty
         end
 
+        it 'restores merge request resource milestone events' do
+          expect(MergeRequest.find_by(title: 'MR1').resource_milestone_events).not_to be_empty
+        end
+
+        it 'restores merge request resource state events' do
+          expect(MergeRequest.find_by(title: 'MR1').resource_state_events).not_to be_empty
+        end
+
         it 'restores suggestion' do
           note = Note.find_by("note LIKE 'Saepe asperiores exercitationem non dignissimos laborum reiciendis et ipsum%'")