summaryrefslogtreecommitdiff
path: root/spec/migrations/swap_snippet_user_mentions_note_id_to_bigint_for_gitlab_dot_com_spec.rb
blob: 628c0fba528500f15cd95b889688197a98e20c00 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
# frozen_string_literal: true

require 'spec_helper'
require_migration!

RSpec.describe SwapSnippetUserMentionsNoteIdToBigintForGitlabDotCom, feature_category: :database do
  describe '#up' do
    before do
      # A we call `schema_migrate_down!` before each example, and for this migration
      # `#down` is same as `#up`, we need to ensure we start from the expected state.
      connection = described_class.new.connection
      connection.execute('ALTER TABLE snippet_user_mentions ALTER COLUMN note_id TYPE integer')
      connection.execute('ALTER TABLE snippet_user_mentions ALTER COLUMN note_id_convert_to_bigint TYPE bigint')
    end

    # rubocop: disable RSpec/AnyInstanceOf
    it 'swaps the integer and bigint columns for GitLab.com, dev, or test' do
      allow_any_instance_of(described_class).to receive(:com_or_dev_or_test_but_not_jh?).and_return(true)

      user_mentions = table(:snippet_user_mentions)

      disable_migrations_output do
        reversible_migration do |migration|
          migration.before -> {
            user_mentions.reset_column_information

            expect(user_mentions.columns.find { |c| c.name == 'note_id' }.sql_type).to eq('integer')
            expect(user_mentions.columns.find { |c| c.name == 'note_id_convert_to_bigint' }.sql_type).to eq('bigint')
          }

          migration.after -> {
            user_mentions.reset_column_information

            expect(user_mentions.columns.find { |c| c.name == 'note_id' }.sql_type).to eq('bigint')
            expect(user_mentions.columns.find { |c| c.name == 'note_id_convert_to_bigint' }.sql_type).to eq('integer')
          }
        end
      end
    end

    it 'is a no-op for other instances' do
      allow_any_instance_of(described_class).to receive(:com_or_dev_or_test_but_not_jh?).and_return(false)

      user_mentions = table(:snippet_user_mentions)

      disable_migrations_output do
        reversible_migration do |migration|
          migration.before -> {
            user_mentions.reset_column_information

            expect(user_mentions.columns.find { |c| c.name == 'note_id' }.sql_type).to eq('integer')
            expect(user_mentions.columns.find { |c| c.name == 'note_id_convert_to_bigint' }.sql_type).to eq('bigint')
          }

          migration.after -> {
            user_mentions.reset_column_information

            expect(user_mentions.columns.find { |c| c.name == 'note_id' }.sql_type).to eq('integer')
            expect(user_mentions.columns.find { |c| c.name == 'note_id_convert_to_bigint' }.sql_type).to eq('bigint')
          }
        end
      end
    end
    # rubocop: enable RSpec/AnyInstanceOf
  end
end