1 files changed, 32 insertions, 0 deletions
diff --git a/spec/models/projects/import_export/relation_export_spec.rb b/spec/models/projects/import_export/relation_export_spec.rb
index c74ca82e161..8643fbc7b46 100644
--- a/spec/models/projects/import_export/relation_export_spec.rb
+++ b/spec/models/projects/import_export/relation_export_spec.rb
@@ -20,4 +20,36 @@ RSpec.describe Projects::ImportExport::RelationExport, type: :model do
     it { is_expected.to validate_length_of(:jid).is_at_most(255) }
     it { is_expected.to validate_length_of(:export_error).is_at_most(300) }
   end
+
+  describe '.by_relation' do
+    it 'returns export relations filtered by relation name' do
+      project_relation_export_1 = create(:project_relation_export, relation: 'labels')
+      project_relation_export_2 = create(:project_relation_export, relation: 'labels')
+      create(:project_relation_export, relation: 'uploads')
+
+      relations = described_class.by_relation('labels').to_a
+
+      expect(relations).to match_array([project_relation_export_1, project_relation_export_2])
+    end
+  end
+
+  describe '.relation_names_list' do
+    it 'includes extra relations list' do
+      expect(described_class.relation_names_list).to include(
+        'design_repository', 'lfs_objects', 'repository', 'snippets_repository', 'uploads', 'wiki_repository'
+      )
+    end
+
+    it 'includes root tree relation name project' do
+      expect(described_class.relation_names_list).to include('project')
+    end
+
+    it 'includes project tree top level relation nodes' do
+      expect(described_class.relation_names_list).to include('milestones', 'issues', 'snippets', 'releases')
+    end
+
+    it 'includes project tree nested relation nodes' do
+      expect(described_class.relation_names_list).not_to include('events', 'notes')
+    end
+  end
 end