1 files changed, 44 insertions, 4 deletions
diff --git a/spec/models/bulk_imports/file_transfer/project_config_spec.rb b/spec/models/bulk_imports/file_transfer/project_config_spec.rb
index db037528ec1..3bd79333f0c 100644
--- a/spec/models/bulk_imports/file_transfer/project_config_spec.rb
+++ b/spec/models/bulk_imports/file_transfer/project_config_spec.rb
@@ -23,10 +23,8 @@ RSpec.describe BulkImports::FileTransfer::ProjectConfig do
   end
 
   describe '#export_path' do
-    it 'returns correct export path' do
-      expect(::Gitlab::ImportExport).to receive(:storage_path).and_return('storage_path')
-
-      expect(subject.export_path).to eq("storage_path/#{exportable.disk_path}/#{hex}")
+    it 'returns tmpdir location' do
+      expect(subject.export_path).to include(File.join(Dir.tmpdir, 'bulk_imports'))
     end
   end
 
@@ -51,4 +49,46 @@ RSpec.describe BulkImports::FileTransfer::ProjectConfig do
       expect(subject.relation_excluded_keys('project')).to include('creator_id')
     end
   end
+
+  describe '#tree_relation?' do
+    context 'when it is a tree relation' do
+      it 'returns true' do
+        expect(subject.tree_relation?('labels')).to eq(true)
+      end
+    end
+
+    context 'when it is not a tree relation' do
+      it 'returns false' do
+        expect(subject.tree_relation?('example')).to eq(false)
+      end
+    end
+  end
+
+  describe '#file_relation?' do
+    context 'when it is a file relation' do
+      it 'returns true' do
+        expect(subject.file_relation?('uploads')).to eq(true)
+      end
+    end
+
+    context 'when it is not a file relation' do
+      it 'returns false' do
+        expect(subject.file_relation?('example')).to eq(false)
+      end
+    end
+  end
+
+  describe '#tree_relation_definition_for' do
+    it 'returns relation definition' do
+      expected = { service_desk_setting: { except: [:outgoing_name, :file_template_project_id], include: [] } }
+
+      expect(subject.tree_relation_definition_for('service_desk_setting')).to eq(expected)
+    end
+
+    context 'when relation is not tree relation' do
+      it 'returns' do
+        expect(subject.tree_relation_definition_for('example')).to be_nil
+      end
+    end
+  end
 end