fixed CI commits on export

lib/gitlab/import_export/attributes_finder.rb

@@ -2,9 +2,10 @@ module Gitlab
   module ImportExport
     class AttributesFinder
     
-      def initialize(included_attributes:, excluded_attributes:)
+      def initialize(included_attributes:, excluded_attributes:, methods:)
         @included_attributes = included_attributes || {}
         @excluded_attributes = excluded_attributes || {}
+        @methods = methods || {}
       end
 
       def find(model_object)
@@ -27,10 +28,15 @@ module Gitlab
         @excluded_attributes[key].nil? ? {} : { except: @excluded_attributes[key] }
       end
 
+      def find_method(value)
+        key = key_from_hash(value)
+        @methods[key].nil? ? {} : { methods: @methods[key] }
+      end
+
       private
 
       def find_attributes_only(value)
-        find_included(value).merge(find_excluded(value))
+        find_included(value).merge(find_excluded(value)).merge(find_method(value))
       end
 
       def key_from_hash(value)

lib/gitlab/import_export/import_export.yml

@@ -46,3 +46,7 @@ included_attributes:
 excluded_attributes:
   snippets:
     - :expired_at
+
+methods:
+  statuses:
+    - :type
\ No newline at end of file

lib/gitlab/import_export/import_export_reader.rb

@@ -7,7 +7,8 @@ module Gitlab
         config_hash = YAML.load_file(config).deep_symbolize_keys
         @tree = config_hash[:project_tree]
         @attributes_parser = Gitlab::ImportExport::AttributesFinder.new(included_attributes: config_hash[:included_attributes],
-                                                                        excluded_attributes: config_hash[:excluded_attributes])
+                                                                        excluded_attributes: config_hash[:excluded_attributes],
+                                                                        methods: config_hash[:methods])
       end
 
       def project_tree

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)