Fix template labels

1 files changed, 36 insertions, 0 deletions

diff --git a/spec/migrations/fix_null_type_labels_spec.rb b/spec/migrations/fix_null_type_labels_spec.rb
new file mode 100644
index 00000000000..462ae9b913f
--- /dev/null
+++ b/spec/migrations/fix_null_type_labels_spec.rb
@@ -0,0 +1,36 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+require Rails.root.join('db', 'post_migrate', '20190131122559_fix_null_type_labels')
+
+describe FixNullTypeLabels, :migration do
+  let(:migration) { described_class.new }
+  let(:projects) { table(:projects) }
+  let(:namespaces) { table(:namespaces) }
+  let(:labels) { table(:labels) }
+
+  before do
+    group = namespaces.create(name: 'labels-test-project', path: 'labels-test-project', type: 'Group')
+    project = projects.create!(namespace_id: group.id, name: 'labels-test-group', path: 'labels-test-group')
+
+    @template_label = labels.create(title: 'template', template: true)
+    @project_label = labels.create(title: 'project label', project_id: project.id, type: 'ProjectLabel')
+    @group_label = labels.create(title: 'group_label', group_id: group.id, type: 'GroupLabel')
+    @broken_label_1 = labels.create(title: 'broken 1', project_id: project.id)
+    @broken_label_2 = labels.create(title: 'broken 2', project_id: project.id)
+  end
+
+  describe '#up' do
+    it 'fix labels with type missing' do
+      migration.up
+
+      # Labels that requires type change
+      expect(@broken_label_1.reload.type).to eq('ProjectLabel')
+      expect(@broken_label_2.reload.type).to eq('ProjectLabel')
+      # Labels out of scope
+      expect(@template_label.reload.type).to be_nil
+      expect(@project_label.reload.type).to eq('ProjectLabel')
+      expect(@group_label.reload.type).to eq('GroupLabel')
+    end
+  end
+end