1 files changed, 2 insertions, 19 deletions
diff --git a/spec/support/helpers/migrations_helpers/work_item_types_helper.rb b/spec/support/helpers/migrations_helpers/work_item_types_helper.rb
index b05caf265ee..40f84486537 100644
--- a/spec/support/helpers/migrations_helpers/work_item_types_helper.rb
+++ b/spec/support/helpers/migrations_helpers/work_item_types_helper.rb
@@ -2,26 +2,9 @@
 
 module MigrationHelpers
   module WorkItemTypesHelper
-    DEFAULT_WORK_ITEM_TYPES = {
-      issue: { name: 'Issue', icon_name: 'issue-type-issue', enum_value: 0 },
-      incident: { name: 'Incident', icon_name: 'issue-type-incident', enum_value: 1 },
-      test_case: { name: 'Test Case', icon_name: 'issue-type-test-case', enum_value: 2 },
-      requirement: { name: 'Requirement', icon_name: 'issue-type-requirements', enum_value: 3 },
-      task: { name: 'Task', icon_name: 'issue-type-task', enum_value: 4 }
-    }.freeze
-
     def reset_work_item_types
-      work_item_types_table.delete_all
-
-      DEFAULT_WORK_ITEM_TYPES.each do |type, attributes|
-        work_item_types_table.create!(base_type: attributes[:enum_value], **attributes.slice(:name, :icon_name))
-      end
-    end
-
-    private
-
-    def work_item_types_table
-      table(:work_item_types)
+      Gitlab::DatabaseImporters::WorkItems::BaseTypeImporter.upsert_types
+      Gitlab::DatabaseImporters::WorkItems::HierarchyRestrictionsImporter.upsert_restrictions
     end
   end
 end