1 files changed, 62 insertions, 0 deletions
diff --git a/spec/lib/gitlab/background_migration/populate_external_pipeline_source_spec.rb b/spec/lib/gitlab/background_migration/populate_external_pipeline_source_spec.rb
new file mode 100644
index 00000000000..c7b272cd6ca
--- /dev/null
+++ b/spec/lib/gitlab/background_migration/populate_external_pipeline_source_spec.rb
@@ -0,0 +1,62 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+describe Gitlab::BackgroundMigration::PopulateExternalPipelineSource, :migration, schema: 20180916011959 do
+  let(:migration) { described_class.new }
+
+  let!(:internal_pipeline) { create(:ci_pipeline, source: :web) }
+  let(:pipelines) { [internal_pipeline, unknown_pipeline].map(&:id) }
+
+  let!(:unknown_pipeline) do
+    build(:ci_pipeline, source: :unknown)
+      .tap { |pipeline| pipeline.save(validate: false) }
+  end
+
+  subject { migration.perform(pipelines.min, pipelines.max) }
+
+  shared_examples 'no changes' do
+    it 'does not change the pipeline source' do
+      expect { subject }.not_to change { unknown_pipeline.reload.source }
+    end
+  end
+
+  context 'when unknown pipeline is external' do
+    before do
+      create(:generic_commit_status, pipeline: unknown_pipeline)
+    end
+
+    it 'populates the pipeline source' do
+      subject
+
+      expect(unknown_pipeline.reload.source).to eq('external')
+    end
+
+    it 'can be repeated without effect' do
+      subject
+
+      expect { subject }.not_to change { unknown_pipeline.reload.source }
+    end
+  end
+
+  context 'when unknown pipeline has just a build' do
+    before do
+      create(:ci_build, pipeline: unknown_pipeline)
+    end
+
+    it_behaves_like 'no changes'
+  end
+
+  context 'when unknown pipeline has no statuses' do
+    it_behaves_like 'no changes'
+  end
+
+  context 'when unknown pipeline has a build and a status' do
+    before do
+      create(:generic_commit_status, pipeline: unknown_pipeline)
+      create(:ci_build, pipeline: unknown_pipeline)
+    end
+
+    it_behaves_like 'no changes'
+  end
+end