fixed CI commits on export

1 files changed, 7 insertions, 2 deletions

diff --git a/spec/lib/gitlab/import_export/project_tree_saver_spec.rb b/spec/lib/gitlab/import_export/project_tree_saver_spec.rb
index b9cf88d3f77..1c55d0e33c1 100644
--- a/spec/lib/gitlab/import_export/project_tree_saver_spec.rb
+++ b/spec/lib/gitlab/import_export/project_tree_saver_spec.rb
@@ -84,10 +84,15 @@ describe Gitlab::ImportExport::ProjectTreeSaver, services: true do
       it 'has author on merge requests comments' do
         expect(saved_project_json['merge_requests'].first['notes'].first['author']).not_to be_empty
       end
+
       it 'has commit statuses' do
         expect(saved_project_json['ci_commits'].first['statuses']).not_to be_empty
       end
 
+      it 'has CI builds' do
+        expect(saved_project_json['ci_commits'].first['statuses'].first['type']).to eq('Ci::Build')
+      end
+
       it 'has ci commits' do
         expect(saved_project_json['ci_commits']).not_to be_empty
       end
@@ -99,7 +104,6 @@ describe Gitlab::ImportExport::ProjectTreeSaver, services: true do
     merge_request = create(:merge_request)
     label = create(:label)
     snippet = create(:project_snippet)
-    commit_status = create(:commit_status)
     release = create(:release)
 
     project = create(:project,
@@ -111,7 +115,8 @@ describe Gitlab::ImportExport::ProjectTreeSaver, services: true do
                      releases: [release]
                     )
 
-    create(:ci_commit, project: project, sha: merge_request.last_commit.id, ref: merge_request.source_branch, statuses: [commit_status])
+    ci_commit = create(:ci_commit, project: project, sha: merge_request.last_commit.id, ref: merge_request.source_branch)
+    create(:ci_build, commit: ci_commit)
     create(:milestone, project: project)
     create(:note, noteable: issue)
     create(:note, noteable: merge_request)