1 files changed, 27 insertions, 0 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
new file mode 100644
index 00000000000..59b1f1b1305
--- /dev/null
+++ b/spec/support/helpers/migrations_helpers/work_item_types_helper.rb
@@ -0,0 +1,27 @@
+# frozen_string_literal: true
+
+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)
+    end
+  end
+end