1 files changed, 59 insertions, 0 deletions
diff --git a/spec/support/shared_examples/work_item_hierarchy_restrictions_importer.rb b/spec/support/shared_examples/work_item_hierarchy_restrictions_importer.rb
new file mode 100644
index 00000000000..b75aa27b2b7
--- /dev/null
+++ b/spec/support/shared_examples/work_item_hierarchy_restrictions_importer.rb
@@ -0,0 +1,59 @@
+# frozen_string_literal: true
+
+RSpec.shared_examples 'work item hierarchy restrictions importer' do
+  shared_examples_for 'adds restrictions' do
+    it "adds all restrictions if they don't exist" do
+      expect { subject }.to change { WorkItems::HierarchyRestriction.count }.from(0).to(4)
+    end
+  end
+
+  context 'when restrictions are missing' do
+    before do
+      WorkItems::HierarchyRestriction.delete_all
+    end
+
+    it_behaves_like 'adds restrictions'
+  end
+
+  context 'when base types are missing' do
+    before do
+      WorkItems::Type.delete_all
+    end
+
+    it_behaves_like 'adds restrictions'
+  end
+
+  context 'when restrictions already exist' do
+    before do
+      Gitlab::DatabaseImporters::WorkItems::HierarchyRestrictionsImporter.upsert_restrictions
+    end
+
+    it 'upserts restrictions' do
+      restriction = WorkItems::HierarchyRestriction.first
+      depth = restriction.maximum_depth
+
+      restriction.update!(maximum_depth: depth + 1)
+
+      expect do
+        subject
+        restriction.reload
+      end.to not_change { WorkItems::HierarchyRestriction.count }.and(
+        change { restriction.maximum_depth }.from(depth + 1).to(depth)
+      )
+    end
+  end
+
+  context 'when some restrictions are missing' do
+    before do
+      Gitlab::DatabaseImporters::WorkItems::HierarchyRestrictionsImporter.upsert_restrictions
+      WorkItems::HierarchyRestriction.limit(1).delete_all
+    end
+
+    it 'inserts missing restrictions and does nothing if some already existed' do
+      expect { subject }.to make_queries_matching(/INSERT/, 1).and(
+        change { WorkItems::HierarchyRestriction.count }.by(1)
+      )
+      expect(WorkItems::HierarchyRestriction.count).to eq(4)
+    end
+  end
+end