activerecord/test/cases/adapters/postgresql/legacy_migration_test.rb


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

require "cases/helper"

class PostgresqlLegacyMigrationTest < ActiveRecord::PostgreSQLTestCase
  class GenerateTableWithoutBigserial < ActiveRecord::Migration[5.0]
    def change
      create_table :legacy_integer_pk do |table|
        table.string :foo
      end

      create_table :override_pk, id: :bigint do |table|
        table.string :bar
      end
    end
  end

  def setup
    super

    @migration_verbose_old = ActiveRecord::Migration.verbose
    ActiveRecord::Migration.verbose = false

    migrations = [GenerateTableWithoutBigserial.new(nil, 1)]
    ActiveRecord::Migrator.new(:up, migrations).migrate
  end

  def teardown
    ActiveRecord::Migration.verbose = @migration_verbose_old

    super
  end

  def test_create_table_uses_serial_as_pkey_by_default
    col = column(:legacy_integer_pk, :id)
    assert_equal "integer", sql_type_for(col)
    assert col.serial?
  end

  def test_create_tables_respects_pk_column_type_override
    col = column(:override_pk, :id)
    assert_equal "bigint", sql_type_for(col)
  end

  private

    def column(table_name, column_name)
      ActiveRecord::Base.connection.
        columns(table_name.to_s).
        detect { |c| c.name == column_name.to_s }
    end

    def sql_type_for(col)
      col && col.sql_type
    end
end