migration_test.rb 32.1 KB
Newer Older
1
require 'active_support/testing/stream'
2
require "cases/helper"
3
require "cases/migration/helper"
4 5
require 'bigdecimal/util'

J
Jeremy Kemper 已提交
6 7
require 'models/person'
require 'models/topic'
8
require 'models/developer'
A
Arun Agrawal 已提交
9
require 'models/computer'
10 11

require MIGRATIONS_ROOT + "/valid/2_we_need_reminders"
12 13
require MIGRATIONS_ROOT + "/rename/1_we_need_things"
require MIGRATIONS_ROOT + "/rename/2_rename_things"
14
require MIGRATIONS_ROOT + "/decimal/1_give_me_big_numbers"
15

16
class BigNumber < ActiveRecord::Base
17
  unless current_adapter?(:PostgreSQLAdapter, :SQLite3Adapter)
S
Sean Griffin 已提交
18
    attribute :value_of_e, Type::Integer.new
19
  end
S
Sean Griffin 已提交
20
  attribute :my_house_population, Type::Integer.new
21
end
22

23
class Reminder < ActiveRecord::Base; end
24

25
class Thing < ActiveRecord::Base; end
26

27 28
class MigrationTest < ActiveRecord::TestCase
  self.use_transactional_fixtures = false
29

30
  fixtures :people
31

32 33
  def setup
    super
A
Akira Matsuda 已提交
34
    %w(reminders people_reminders prefix_reminders_suffix p_things_s).each do |table|
A
Aaron Patterson 已提交
35 36 37
      Reminder.connection.drop_table(table) rescue nil
    end
    Reminder.reset_column_information
38
    @verbose_was, ActiveRecord::Migration.verbose = ActiveRecord::Migration.verbose, false
A
Akira Matsuda 已提交
39
    ActiveRecord::Base.connection.schema_cache.clear!
40
  end
41

G
Guo Xiang Tan 已提交
42
  teardown do
43 44 45
    ActiveRecord::Base.table_name_prefix = ""
    ActiveRecord::Base.table_name_suffix = ""

46 47
    ActiveRecord::Base.connection.initialize_schema_migrations_table
    ActiveRecord::Base.connection.execute "DELETE FROM #{ActiveRecord::Migrator.schema_migrations_table_name}"
48

49
    %w(things awesome_things prefix_things_suffix p_awesome_things_s ).each do |table|
50
      Thing.connection.drop_table(table) rescue nil
51
    end
52
    Thing.reset_column_information
53

54 55 56
    %w(reminders people_reminders prefix_reminders_suffix).each do |table|
      Reminder.connection.drop_table(table) rescue nil
    end
57
    Reminder.reset_table_name
58
    Reminder.reset_column_information
59

60 61 62 63 64 65
    %w(last_name key bio age height wealth birthday favorite_day
       moment_of_truth male administrator funny).each do |column|
      Person.connection.remove_column('people', column) rescue nil
    end
    Person.connection.remove_column("people", "first_name") rescue nil
    Person.connection.remove_column("people", "middle_name") rescue nil
66
    Person.connection.add_column("people", "first_name", :string)
67
    Person.reset_column_information
68 69

    ActiveRecord::Migration.verbose = @verbose_was
70
  end
71

72 73
  def test_migrator_versions
    migrations_path = MIGRATIONS_ROOT + "/valid"
74
    old_path = ActiveRecord::Migrator.migrations_paths
75 76 77 78 79
    ActiveRecord::Migrator.migrations_paths = migrations_path

    ActiveRecord::Migrator.up(migrations_path)
    assert_equal 3, ActiveRecord::Migrator.current_version
    assert_equal 3, ActiveRecord::Migrator.last_version
80
    assert_equal false, ActiveRecord::Migrator.needs_migration?
81 82 83 84

    ActiveRecord::Migrator.down(MIGRATIONS_ROOT + "/valid")
    assert_equal 0, ActiveRecord::Migrator.current_version
    assert_equal 3, ActiveRecord::Migrator.last_version
85
    assert_equal true, ActiveRecord::Migrator.needs_migration?
86 87 88 89 90 91 92 93

    ActiveRecord::SchemaMigration.create!(:version => ActiveRecord::Migrator.last_version)
    assert_equal true, ActiveRecord::Migrator.needs_migration?
  ensure
    ActiveRecord::Migrator.migrations_paths = old_path
  end

  def test_migration_detection_without_schema_migration_table
94
    ActiveRecord::Base.connection.drop_table('schema_migrations') if ActiveRecord::Base.connection.table_exists?('schema_migrations')
95 96 97 98 99 100

    migrations_path = MIGRATIONS_ROOT + "/valid"
    old_path = ActiveRecord::Migrator.migrations_paths
    ActiveRecord::Migrator.migrations_paths = migrations_path

    assert_equal true, ActiveRecord::Migrator.needs_migration?
101 102
  ensure
    ActiveRecord::Migrator.migrations_paths = old_path
103 104
  end

105 106 107 108 109 110 111 112 113 114 115 116 117
  def test_any_migrations
    old_path = ActiveRecord::Migrator.migrations_paths
    ActiveRecord::Migrator.migrations_paths = MIGRATIONS_ROOT + "/valid"

    assert ActiveRecord::Migrator.any_migrations?

    ActiveRecord::Migrator.migrations_paths = MIGRATIONS_ROOT + "/empty"

    assert_not ActiveRecord::Migrator.any_migrations?
  ensure
    ActiveRecord::Migrator.migrations_paths = old_path
  end

118 119 120 121
  def test_migration_version
    ActiveRecord::Migrator.run(:up, MIGRATIONS_ROOT + "/version_check", 20131219224947)
  end

122 123 124
  def test_create_table_with_force_true_does_not_drop_nonexisting_table
    if Person.connection.table_exists?(:testings2)
      Person.connection.drop_table :testings2
125 126
    end

127 128 129
    # using a copy as we need the drop_table method to
    # continue to work for the ensure block of the test
    temp_conn = Person.connection.dup
130 131 132

    assert_not_equal temp_conn, Person.connection

133 134
    temp_conn.create_table :testings2, :force => true do |t|
      t.column :foo, :string
135
    end
136 137 138
  ensure
    Person.connection.drop_table :testings2 rescue nil
  end
139

140 141 142
  def connection
    ActiveRecord::Base.connection
  end
143

144 145 146 147
  def test_migration_instance_has_connection
    migration = Class.new(ActiveRecord::Migration).new
    assert_equal connection, migration.connection
  end
148

149 150 151 152 153 154 155 156
  def test_method_missing_delegates_to_connection
    migration = Class.new(ActiveRecord::Migration) {
      def connection
        Class.new {
          def create_table; "hi mom!"; end
        }.new
      end
    }.new
157

158
    assert_equal "hi mom!", migration.method_missing(:create_table)
159
  end
160

161 162 163 164 165 166 167 168 169 170 171 172 173 174
  def test_add_table_with_decimals
    Person.connection.drop_table :big_numbers rescue nil

    assert !BigNumber.table_exists?
    GiveMeBigNumbers.up

    assert BigNumber.create(
      :bank_balance => 1586.43,
      :big_bank_balance => BigDecimal("1000234000567.95"),
      :world_population => 6000000000,
      :my_house_population => 3,
      :value_of_e => BigDecimal("2.7182818284590452353602875")
    )

175
    b = BigNumber.first
176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218
    assert_not_nil b

    assert_not_nil b.bank_balance
    assert_not_nil b.big_bank_balance
    assert_not_nil b.world_population
    assert_not_nil b.my_house_population
    assert_not_nil b.value_of_e

    # TODO: set world_population >= 2**62 to cover 64-bit platforms and test
    # is_a?(Bignum)
    assert_kind_of Integer, b.world_population
    assert_equal 6000000000, b.world_population
    assert_kind_of Fixnum, b.my_house_population
    assert_equal 3, b.my_house_population
    assert_kind_of BigDecimal, b.bank_balance
    assert_equal BigDecimal("1586.43"), b.bank_balance
    assert_kind_of BigDecimal, b.big_bank_balance
    assert_equal BigDecimal("1000234000567.95"), b.big_bank_balance

    # This one is fun. The 'value_of_e' field is defined as 'DECIMAL' with
    # precision/scale explicitly left out.  By the SQL standard, numbers
    # assigned to this field should be truncated but that's seldom respected.
    if current_adapter?(:PostgreSQLAdapter)
      # - PostgreSQL changes the SQL spec on columns declared simply as
      # "decimal" to something more useful: instead of being given a scale
      # of 0, they take on the compile-time limit for precision and scale,
      # so the following should succeed unless you have used really wacky
      # compilation options
      # - SQLite2 has the default behavior of preserving all data sent in,
      # so this happens there too
      assert_kind_of BigDecimal, b.value_of_e
      assert_equal BigDecimal("2.7182818284590452353602875"), b.value_of_e
    elsif current_adapter?(:SQLite3Adapter)
      # - SQLite3 stores a float, in violation of SQL
      assert_kind_of BigDecimal, b.value_of_e
      assert_in_delta BigDecimal("2.71828182845905"), b.value_of_e, 0.00000000000001
    else
      # - SQL standard is an integer
      assert_kind_of Fixnum, b.value_of_e
      assert_equal 2, b.value_of_e
    end

    GiveMeBigNumbers.down
219
    assert_raise(ActiveRecord::StatementInvalid) { BigNumber.first }
220
  end
221

222
  def test_filtering_migrations
223
    assert_no_column Person, :last_name
224
    assert !Reminder.table_exists?
225

226 227
    name_filter = lambda { |migration| migration.name == "ValidPeopleHaveLastNames" }
    ActiveRecord::Migrator.up(MIGRATIONS_ROOT + "/valid", &name_filter)
228

229
    assert_column Person, :last_name
230
    assert_raise(ActiveRecord::StatementInvalid) { Reminder.first }
231

232 233
    ActiveRecord::Migrator.down(MIGRATIONS_ROOT + "/valid", &name_filter)

234
    assert_no_column Person, :last_name
235
    assert_raise(ActiveRecord::StatementInvalid) { Reminder.first }
236 237 238 239 240 241 242
  end

  class MockMigration < ActiveRecord::Migration
    attr_reader :went_up, :went_down
    def initialize
      @went_up   = false
      @went_down = false
243 244
    end

245 246 247 248
    def up
      @went_up = true
      super
    end
249

250 251 252
    def down
      @went_down = true
      super
253
    end
254
  end
255

256 257 258 259
  def test_instance_based_migration_up
    migration = MockMigration.new
    assert !migration.went_up, 'have not gone up'
    assert !migration.went_down, 'have not gone down'
260

261 262 263 264
    migration.migrate :up
    assert migration.went_up, 'have gone up'
    assert !migration.went_down, 'have not gone down'
  end
265

266 267 268 269
  def test_instance_based_migration_down
    migration = MockMigration.new
    assert !migration.went_up, 'have not gone up'
    assert !migration.went_down, 'have not gone down'
270

271 272 273 274
    migration.migrate :down
    assert !migration.went_up, 'have gone up'
    assert migration.went_down, 'have not gone down'
  end
275

276 277 278 279 280 281 282 283 284 285 286
  if ActiveRecord::Base.connection.supports_ddl_transactions?
    def test_migrator_one_up_with_exception_and_rollback
      assert_no_column Person, :last_name

      migration = Class.new(ActiveRecord::Migration) {
        def version; 100 end
        def migrate(x)
          add_column "people", "last_name", :string
          raise 'Something broke'
        end
      }.new
287

288
      migrator = ActiveRecord::Migrator.new(:up, [migration], 100)
289

290
      e = assert_raise(StandardError) { migrator.migrate }
291

292
      assert_equal "An error has occurred, this and all later migrations canceled:\n\nSomething broke", e.message
293

294 295
      assert_no_column Person, :last_name,
        "On error, the Migrator should revert schema changes but it did not."
B
bondarev 已提交
296 297
    end

298 299
    def test_migrator_one_up_with_exception_and_rollback_using_run
      assert_no_column Person, :last_name
B
bondarev 已提交
300

301 302 303 304 305 306 307
      migration = Class.new(ActiveRecord::Migration) {
        def version; 100 end
        def migrate(x)
          add_column "people", "last_name", :string
          raise 'Something broke'
        end
      }.new
B
bondarev 已提交
308

309
      migrator = ActiveRecord::Migrator.new(:up, [migration], 100)
B
bondarev 已提交
310

311
      e = assert_raise(StandardError) { migrator.run }
B
bondarev 已提交
312

313
      assert_equal "An error has occurred, this migration was canceled:\n\nSomething broke", e.message
B
bondarev 已提交
314

315 316
      assert_no_column Person, :last_name,
        "On error, the Migrator should revert schema changes but it did not."
317 318
    end

319 320
    def test_migration_without_transaction
      assert_no_column Person, :last_name
321

322 323
      migration = Class.new(ActiveRecord::Migration) {
        self.disable_ddl_transaction!
324

325 326 327 328 329 330
        def version; 101 end
        def migrate(x)
          add_column "people", "last_name", :string
          raise 'Something broke'
        end
      }.new
331

332 333 334
      migrator = ActiveRecord::Migrator.new(:up, [migration], 101)
      e = assert_raise(StandardError) { migrator.migrate }
      assert_equal "An error has occurred, all later migrations canceled:\n\nSomething broke", e.message
335

336 337 338 339 340 341 342
      assert_column Person, :last_name,
        "without ddl transactions, the Migrator should not rollback on error but it did."
    ensure
      Person.reset_column_information
      if Person.column_names.include?('last_name')
        Person.connection.remove_column('people', 'last_name')
      end
343
    end
344
  end
345

346
  def test_schema_migrations_table_name
347 348 349
    original_schema_migrations_table_name = ActiveRecord::Migrator.schema_migrations_table_name

    assert_equal "schema_migrations", ActiveRecord::Migrator.schema_migrations_table_name
350 351 352 353
    ActiveRecord::Base.table_name_prefix = "prefix_"
    ActiveRecord::Base.table_name_suffix = "_suffix"
    Reminder.reset_table_name
    assert_equal "prefix_schema_migrations_suffix", ActiveRecord::Migrator.schema_migrations_table_name
354 355 356
    ActiveRecord::Base.schema_migrations_table_name = "changed"
    Reminder.reset_table_name
    assert_equal "prefix_changed_suffix", ActiveRecord::Migrator.schema_migrations_table_name
357 358 359
    ActiveRecord::Base.table_name_prefix = ""
    ActiveRecord::Base.table_name_suffix = ""
    Reminder.reset_table_name
360 361 362 363
    assert_equal "changed", ActiveRecord::Migrator.schema_migrations_table_name
  ensure
    ActiveRecord::Base.schema_migrations_table_name = original_schema_migrations_table_name
    Reminder.reset_table_name
364 365
  end

366
  def test_proper_table_name_on_migration
367
    reminder_class = new_isolated_reminder_class
368 369 370
    migration = ActiveRecord::Migration.new
    assert_equal "table", migration.proper_table_name('table')
    assert_equal "table", migration.proper_table_name(:table)
371 372 373
    assert_equal "reminders", migration.proper_table_name(reminder_class)
    reminder_class.reset_table_name
    assert_equal reminder_class.table_name, migration.proper_table_name(reminder_class)
374 375 376 377

    # Use the model's own prefix/suffix if a model is given
    ActiveRecord::Base.table_name_prefix = "ARprefix_"
    ActiveRecord::Base.table_name_suffix = "_ARsuffix"
378 379 380 381 382 383 384
    reminder_class.table_name_prefix = 'prefix_'
    reminder_class.table_name_suffix = '_suffix'
    reminder_class.reset_table_name
    assert_equal "prefix_reminders_suffix", migration.proper_table_name(reminder_class)
    reminder_class.table_name_prefix = ''
    reminder_class.table_name_suffix = ''
    reminder_class.reset_table_name
385 386 387 388

    # Use AR::Base's prefix/suffix if string or symbol is given
    ActiveRecord::Base.table_name_prefix = "prefix_"
    ActiveRecord::Base.table_name_suffix = "_suffix"
389
    reminder_class.reset_table_name
390 391 392 393
    assert_equal "prefix_table_suffix", migration.proper_table_name('table', migration.table_name_options)
    assert_equal "prefix_table_suffix", migration.proper_table_name(:table, migration.table_name_options)
  end

394 395
  def test_rename_table_with_prefix_and_suffix
    assert !Thing.table_exists?
396 397
    ActiveRecord::Base.table_name_prefix = 'p_'
    ActiveRecord::Base.table_name_suffix = '_s'
398 399 400 401 402
    Thing.reset_table_name
    Thing.reset_sequence_name
    WeNeedThings.up

    assert Thing.create("content" => "hello world")
403
    assert_equal "hello world", Thing.first.content
404 405

    RenameThings.up
406
    Thing.table_name = "p_awesome_things_s"
407

408
    assert_equal "hello world", Thing.first.content
409 410 411 412 413 414 415 416 417 418 419 420 421
  ensure
    Thing.reset_table_name
    Thing.reset_sequence_name
  end

  def test_add_drop_table_with_prefix_and_suffix
    assert !Reminder.table_exists?
    ActiveRecord::Base.table_name_prefix = 'prefix_'
    ActiveRecord::Base.table_name_suffix = '_suffix'
    Reminder.reset_table_name
    Reminder.reset_sequence_name
    WeNeedReminders.up
    assert Reminder.create("content" => "hello world", "remind_at" => Time.now)
422
    assert_equal "hello world", Reminder.first.content
423 424

    WeNeedReminders.down
425
    assert_raise(ActiveRecord::StatementInvalid) { Reminder.first }
426 427 428
  ensure
    Reminder.reset_sequence_name
  end
429

430 431
  def test_create_table_with_binary_column
    Person.connection.drop_table :binary_testings rescue nil
432

433 434
    assert_nothing_raised {
      Person.connection.create_table :binary_testings do |t|
435
        t.column "data", :binary, :null => false
436
      end
437 438 439 440
    }

    columns = Person.connection.columns(:binary_testings)
    data_column = columns.detect { |c| c.name == "data" }
441

J
Jon Leighton 已提交
442
    assert_nil data_column.default
443

444 445 446
    Person.connection.drop_table :binary_testings rescue nil
  end

447 448 449 450
  unless mysql_enforcing_gtid_consistency?
    def test_create_table_with_query
      Person.connection.drop_table :table_from_query_testings rescue nil
      Person.connection.create_table(:person, force: true)
451

452
      Person.connection.create_table :table_from_query_testings, as: "SELECT id FROM person"
453

454 455 456
      columns = Person.connection.columns(:table_from_query_testings)
      assert_equal 1, columns.length
      assert_equal "id", columns.first.name
457

458 459
      Person.connection.drop_table :table_from_query_testings rescue nil
    end
460

461 462 463
    def test_create_table_with_query_from_relation
      Person.connection.drop_table :table_from_query_testings rescue nil
      Person.connection.create_table(:person, force: true)
464

465
      Person.connection.create_table :table_from_query_testings, as: Person.select(:id)
466

467 468 469
      columns = Person.connection.columns(:table_from_query_testings)
      assert_equal 1, columns.length
      assert_equal "id", columns.first.name
470

471 472
      Person.connection.drop_table :table_from_query_testings rescue nil
    end
473 474
  end

475 476 477 478 479 480 481 482 483 484 485
  if current_adapter? :OracleAdapter
    def test_create_table_with_custom_sequence_name
      # table name is 29 chars, the standard sequence name will
      # be 33 chars and should be shortened
      assert_nothing_raised do
        begin
          Person.connection.create_table :table_with_name_thats_just_ok do |t|
            t.column :foo, :string, :null => false
          end
        ensure
          Person.connection.drop_table :table_with_name_thats_just_ok rescue nil
486 487 488
        end
      end

489 490 491 492 493 494 495
      # should be all good w/ a custom sequence name
      assert_nothing_raised do
        begin
          Person.connection.create_table :table_with_name_thats_just_ok,
            :sequence_name => 'suitably_short_seq' do |t|
            t.column :foo, :string, :null => false
          end
496

497
          Person.connection.execute("select suitably_short_seq.nextval from dual")
498

499 500 501 502
        ensure
          Person.connection.drop_table :table_with_name_thats_just_ok,
            :sequence_name => 'suitably_short_seq' rescue nil
        end
503 504
      end

505 506 507 508
      # confirm the custom sequence got dropped
      assert_raise(ActiveRecord::StatementInvalid) do
        Person.connection.execute("select suitably_short_seq.nextval from dual")
      end
509
    end
510
  end
511

512 513 514 515 516 517 518
  if current_adapter?(:MysqlAdapter, :Mysql2Adapter, :PostgreSQLAdapter)
    def test_out_of_range_limit_should_raise
      Person.connection.drop_table :test_limits rescue nil
      assert_raise(ActiveRecord::ActiveRecordError, "integer limit didn't raise") do
        Person.connection.create_table :test_integer_limits, :force => true do |t|
          t.column :bigone, :integer, :limit => 10
        end
519 520
      end

521 522 523 524 525
      unless current_adapter?(:PostgreSQLAdapter)
        assert_raise(ActiveRecord::ActiveRecordError, "text limit didn't raise") do
          Person.connection.create_table :test_text_limits, :force => true do |t|
            t.column :bigtext, :text, :limit => 0xfffffffff
          end
526 527 528
        end
      end

529 530
      Person.connection.drop_table :test_limits rescue nil
    end
531 532
  end

533
  protected
534 535 536 537 538 539 540 541
    # This is needed to isolate class_attribute assignments like `table_name_prefix`
    # for each test case.
    def new_isolated_reminder_class
      Class.new(Reminder) {
        def self.name; "Reminder"; end
        def self.base_class; self; end
      }
    end
542
end
543

544 545 546 547 548
class ReservedWordsMigrationTest < ActiveRecord::TestCase
  def test_drop_index_from_table_named_values
    connection = Person.connection
    connection.create_table :values, :force => true do |t|
      t.integer :value
549
    end
550

551 552 553
    assert_nothing_raised do
      connection.add_index :values, :value
      connection.remove_index :values, :column => :value
554
    end
555

556 557 558 559
    connection.drop_table :values rescue nil
  end
end

560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575
class ExplicitlyNamedIndexMigrationTest < ActiveRecord::TestCase
  def test_drop_index_by_name
    connection = Person.connection
    connection.create_table :values, force: true do |t|
      t.integer :value
    end

    assert_nothing_raised ArgumentError do
      connection.add_index :values, :value, name: 'a_different_name'
      connection.remove_index :values, column: :value, name: 'a_different_name'
    end

    connection.drop_table :values rescue nil
  end
end

576 577 578 579 580
if ActiveRecord::Base.connection.supports_bulk_alter?
  class BulkAlterTableMigrationsTest < ActiveRecord::TestCase
    def setup
      @connection = Person.connection
      @connection.create_table(:delete_me, :force => true) {|t| }
A
Akira Matsuda 已提交
581 582
      Person.reset_column_information
      Person.reset_sequence_name
583
    end
584

G
Guo Xiang Tan 已提交
585
    teardown do
586 587
      Person.connection.drop_table(:delete_me) rescue nil
    end
588

589 590
    def test_adding_multiple_columns
      assert_queries(1) do
591
        with_bulk_change_table do |t|
592
          t.column :name, :string
593
          t.string :qualification, :experience
594 595
          t.integer :age, :default => 0
          t.date :birthdate
596
          t.timestamps null: true
597
        end
598
      end
599

600 601
      assert_equal 8, columns.size
      [:name, :qualification, :experience].each {|s| assert_equal :string, column(s).type }
602
      assert_equal '0', column(:age).default
603
    end
604

605 606 607
    def test_removing_columns
      with_bulk_change_table do |t|
        t.string :qualification, :experience
608 609
      end

610
      [:qualification, :experience].each {|c| assert column(c) }
611

612 613 614 615
      assert_queries(1) do
        with_bulk_change_table do |t|
          t.remove :qualification, :experience
          t.string :qualification_experience
616
        end
617
      end
618

619 620 621
      [:qualification, :experience].each {|c| assert ! column(c) }
      assert column(:qualification_experience)
    end
622

623 624 625 626 627
    def test_adding_indexes
      with_bulk_change_table do |t|
        t.string :username
        t.string :name
        t.integer :age
628 629
      end

630 631
      # Adding an index fires a query every time to check if an index already exists or not
      assert_queries(3) do
632
        with_bulk_change_table do |t|
633 634
          t.index :username, :unique => true, :name => :awesome_username_index
          t.index [:name, :age]
635
        end
636
      end
637

638
      assert_equal 2, indexes.size
639

640 641 642
      name_age_index = index(:index_delete_me_on_name_and_age)
      assert_equal ['name', 'age'].sort, name_age_index.columns.sort
      assert ! name_age_index.unique
643

644 645
      assert index(:awesome_username_index).unique
    end
646

647 648 649 650
    def test_removing_index
      with_bulk_change_table do |t|
        t.string :name
        t.index :name
651
      end
652

653 654 655
      assert index(:index_delete_me_on_name)

      assert_queries(3) do
656
        with_bulk_change_table do |t|
657 658
          t.remove_index :name
          t.index :name, :name => :new_name_index, :unique => true
659
        end
660
      end
661

662
      assert ! index(:index_delete_me_on_name)
663

664 665 666
      new_name_index = index(:new_name_index)
      assert new_name_index.unique
    end
667

668 669 670 671
    def test_changing_columns
      with_bulk_change_table do |t|
        t.string :name
        t.date :birthdate
672
      end
673

674 675
      assert ! column(:name).default
      assert_equal :date, column(:birthdate).type
676

677 678 679
      # One query for columns (delete_me table)
      # One query for primary key (delete_me table)
      # One query to do the bulk change
680
      assert_queries(3, :ignore_none => true) do
681 682 683
        with_bulk_change_table do |t|
          t.change :name, :string, :default => 'NONAME'
          t.change :birthdate, :datetime
684
        end
685 686
      end

687 688 689
      assert_equal 'NONAME', column(:name).default
      assert_equal :datetime, column(:birthdate).type
    end
690

691
    protected
692

693 694 695
    def with_bulk_change_table
      # Reset columns/indexes cache as we're changing the table
      @columns = @indexes = nil
696

697 698
      Person.connection.change_table(:delete_me, :bulk => true) do |t|
        yield t
699
      end
700
    end
701

702 703 704
    def column(name)
      columns.detect {|c| c.name == name.to_s }
    end
705

706 707
    def columns
      @columns ||= Person.connection.columns('delete_me')
708 709
    end

710 711
    def index(name)
      indexes.detect {|i| i.name == name.to_s }
712 713
    end

714 715 716 717 718 719 720 721
    def indexes
      @indexes ||= Person.connection.indexes('delete_me')
    end
  end # AlterTableMigrationsTest

end

class CopyMigrationsTest < ActiveRecord::TestCase
722 723
  include ActiveSupport::Testing::Stream

724 725 726 727 728 729 730 731 732 733 734 735 736 737 738
  def setup
  end

  def clear
    ActiveRecord::Base.timestamped_migrations = true
    to_delete = Dir[@migrations_path + "/*.rb"] - @existing_migrations
    File.delete(*to_delete)
  end

  def test_copying_migrations_without_timestamps
    ActiveRecord::Base.timestamped_migrations = false
    @migrations_path = MIGRATIONS_ROOT + "/valid"
    @existing_migrations = Dir[@migrations_path + "/*.rb"]

    copied = ActiveRecord::Migration.copy(@migrations_path, {:bukkits => MIGRATIONS_ROOT + "/to_copy"})
A
Arun Agrawal 已提交
739 740
    assert File.exist?(@migrations_path + "/4_people_have_hobbies.bukkits.rb")
    assert File.exist?(@migrations_path + "/5_people_have_descriptions.bukkits.rb")
741 742 743 744 745 746 747 748 749 750 751 752 753 754 755 756 757 758
    assert_equal [@migrations_path + "/4_people_have_hobbies.bukkits.rb", @migrations_path + "/5_people_have_descriptions.bukkits.rb"], copied.map(&:filename)

    expected = "# This migration comes from bukkits (originally 1)"
    assert_equal expected, IO.readlines(@migrations_path + "/4_people_have_hobbies.bukkits.rb")[0].chomp

    files_count = Dir[@migrations_path + "/*.rb"].length
    copied = ActiveRecord::Migration.copy(@migrations_path, {:bukkits => MIGRATIONS_ROOT + "/to_copy"})
    assert_equal files_count, Dir[@migrations_path + "/*.rb"].length
    assert copied.empty?
  ensure
    clear
  end

  def test_copying_migrations_without_timestamps_from_2_sources
    ActiveRecord::Base.timestamped_migrations = false
    @migrations_path = MIGRATIONS_ROOT + "/valid"
    @existing_migrations = Dir[@migrations_path + "/*.rb"]

759
    sources = {}
760 761 762
    sources[:bukkits] = MIGRATIONS_ROOT + "/to_copy"
    sources[:omg] = MIGRATIONS_ROOT + "/to_copy2"
    ActiveRecord::Migration.copy(@migrations_path, sources)
A
Arun Agrawal 已提交
763 764 765 766
    assert File.exist?(@migrations_path + "/4_people_have_hobbies.bukkits.rb")
    assert File.exist?(@migrations_path + "/5_people_have_descriptions.bukkits.rb")
    assert File.exist?(@migrations_path + "/6_create_articles.omg.rb")
    assert File.exist?(@migrations_path + "/7_create_comments.omg.rb")
767 768 769 770 771 772 773

    files_count = Dir[@migrations_path + "/*.rb"].length
    ActiveRecord::Migration.copy(@migrations_path, sources)
    assert_equal files_count, Dir[@migrations_path + "/*.rb"].length
  ensure
    clear
  end
774

775 776 777
  def test_copying_migrations_with_timestamps
    @migrations_path = MIGRATIONS_ROOT + "/valid_with_timestamps"
    @existing_migrations = Dir[@migrations_path + "/*.rb"]
778

779
    travel_to(Time.utc(2010, 7, 26, 10, 10, 10)) do
780
      copied = ActiveRecord::Migration.copy(@migrations_path, {:bukkits => MIGRATIONS_ROOT + "/to_copy_with_timestamps"})
A
Arun Agrawal 已提交
781 782
      assert File.exist?(@migrations_path + "/20100726101010_people_have_hobbies.bukkits.rb")
      assert File.exist?(@migrations_path + "/20100726101011_people_have_descriptions.bukkits.rb")
783 784 785
      expected = [@migrations_path + "/20100726101010_people_have_hobbies.bukkits.rb",
                  @migrations_path + "/20100726101011_people_have_descriptions.bukkits.rb"]
      assert_equal expected, copied.map(&:filename)
786

787
      files_count = Dir[@migrations_path + "/*.rb"].length
788
      copied = ActiveRecord::Migration.copy(@migrations_path, {:bukkits => MIGRATIONS_ROOT + "/to_copy_with_timestamps"})
789 790 791
      assert_equal files_count, Dir[@migrations_path + "/*.rb"].length
      assert copied.empty?
    end
792 793 794
  ensure
    clear
  end
795

796 797 798
  def test_copying_migrations_with_timestamps_from_2_sources
    @migrations_path = MIGRATIONS_ROOT + "/valid_with_timestamps"
    @existing_migrations = Dir[@migrations_path + "/*.rb"]
799

800
    sources = {}
801 802 803
    sources[:bukkits] = MIGRATIONS_ROOT + "/to_copy_with_timestamps"
    sources[:omg]     = MIGRATIONS_ROOT + "/to_copy_with_timestamps2"

804
    travel_to(Time.utc(2010, 7, 26, 10, 10, 10)) do
805
      copied = ActiveRecord::Migration.copy(@migrations_path, sources)
A
Arun Agrawal 已提交
806 807 808 809
      assert File.exist?(@migrations_path + "/20100726101010_people_have_hobbies.bukkits.rb")
      assert File.exist?(@migrations_path + "/20100726101011_people_have_descriptions.bukkits.rb")
      assert File.exist?(@migrations_path + "/20100726101012_create_articles.omg.rb")
      assert File.exist?(@migrations_path + "/20100726101013_create_comments.omg.rb")
810
      assert_equal 4, copied.length
811 812 813 814 815

      files_count = Dir[@migrations_path + "/*.rb"].length
      ActiveRecord::Migration.copy(@migrations_path, sources)
      assert_equal files_count, Dir[@migrations_path + "/*.rb"].length
    end
816 817 818
  ensure
    clear
  end
819

820 821 822
  def test_copying_migrations_with_timestamps_to_destination_with_timestamps_in_future
    @migrations_path = MIGRATIONS_ROOT + "/valid_with_timestamps"
    @existing_migrations = Dir[@migrations_path + "/*.rb"]
823

824
    travel_to(Time.utc(2010, 2, 20, 10, 10, 10)) do
825
      ActiveRecord::Migration.copy(@migrations_path, {:bukkits => MIGRATIONS_ROOT + "/to_copy_with_timestamps"})
A
Arun Agrawal 已提交
826 827
      assert File.exist?(@migrations_path + "/20100301010102_people_have_hobbies.bukkits.rb")
      assert File.exist?(@migrations_path + "/20100301010103_people_have_descriptions.bukkits.rb")
828

829 830 831 832
      files_count = Dir[@migrations_path + "/*.rb"].length
      copied = ActiveRecord::Migration.copy(@migrations_path, {:bukkits => MIGRATIONS_ROOT + "/to_copy_with_timestamps"})
      assert_equal files_count, Dir[@migrations_path + "/*.rb"].length
      assert copied.empty?
833
    end
834 835 836
  ensure
    clear
  end
837

838 839 840 841 842 843
  def test_copying_migrations_preserving_magic_comments
    ActiveRecord::Base.timestamped_migrations = false
    @migrations_path = MIGRATIONS_ROOT + "/valid"
    @existing_migrations = Dir[@migrations_path + "/*.rb"]

    copied = ActiveRecord::Migration.copy(@migrations_path, {:bukkits => MIGRATIONS_ROOT + "/magic"})
A
Arun Agrawal 已提交
844
    assert File.exist?(@migrations_path + "/4_currencies_have_symbols.bukkits.rb")
845 846 847 848 849 850 851 852 853 854 855 856 857
    assert_equal [@migrations_path + "/4_currencies_have_symbols.bukkits.rb"], copied.map(&:filename)

    expected = "# coding: ISO-8859-15\n# This migration comes from bukkits (originally 1)"
    assert_equal expected, IO.readlines(@migrations_path + "/4_currencies_have_symbols.bukkits.rb")[0..1].join.chomp

    files_count = Dir[@migrations_path + "/*.rb"].length
    copied = ActiveRecord::Migration.copy(@migrations_path, {:bukkits => MIGRATIONS_ROOT + "/magic"})
    assert_equal files_count, Dir[@migrations_path + "/*.rb"].length
    assert copied.empty?
  ensure
    clear
  end

858 859 860
  def test_skipping_migrations
    @migrations_path = MIGRATIONS_ROOT + "/valid_with_timestamps"
    @existing_migrations = Dir[@migrations_path + "/*.rb"]
861 862

    sources = {}
863 864
    sources[:bukkits] = MIGRATIONS_ROOT + "/to_copy_with_timestamps"
    sources[:omg]     = MIGRATIONS_ROOT + "/to_copy_with_name_collision"
865

866 867 868 869
    skipped = []
    on_skip = Proc.new { |name, migration| skipped << "#{name} #{migration.name}" }
    copied = ActiveRecord::Migration.copy(@migrations_path, sources, :on_skip => on_skip)
    assert_equal 2, copied.length
870

871 872 873 874 875
    assert_equal 1, skipped.length
    assert_equal ["omg PeopleHaveHobbies"], skipped
  ensure
    clear
  end
876

877 878 879
  def test_skip_is_not_called_if_migrations_are_from_the_same_plugin
    @migrations_path = MIGRATIONS_ROOT + "/valid_with_timestamps"
    @existing_migrations = Dir[@migrations_path + "/*.rb"]
880

881
    sources = {}
882
    sources[:bukkits] = MIGRATIONS_ROOT + "/to_copy_with_timestamps"
883

884 885 886 887
    skipped = []
    on_skip = Proc.new { |name, migration| skipped << "#{name} #{migration.name}" }
    copied = ActiveRecord::Migration.copy(@migrations_path, sources, :on_skip => on_skip)
    ActiveRecord::Migration.copy(@migrations_path, sources, :on_skip => on_skip)
888

889 890 891 892 893
    assert_equal 2, copied.length
    assert_equal 0, skipped.length
  ensure
    clear
  end
894

895 896 897
  def test_copying_migrations_to_non_existing_directory
    @migrations_path = MIGRATIONS_ROOT + "/non_existing"
    @existing_migrations = []
898

899
    travel_to(Time.utc(2010, 7, 26, 10, 10, 10)) do
900
      copied = ActiveRecord::Migration.copy(@migrations_path, {:bukkits => MIGRATIONS_ROOT + "/to_copy_with_timestamps"})
A
Arun Agrawal 已提交
901 902
      assert File.exist?(@migrations_path + "/20100726101010_people_have_hobbies.bukkits.rb")
      assert File.exist?(@migrations_path + "/20100726101011_people_have_descriptions.bukkits.rb")
903 904
      assert_equal 2, copied.length
    end
905 906 907 908
  ensure
    clear
    Dir.delete(@migrations_path)
  end
909

910 911 912
  def test_copying_migrations_to_empty_directory
    @migrations_path = MIGRATIONS_ROOT + "/empty"
    @existing_migrations = []
913

914
    travel_to(Time.utc(2010, 7, 26, 10, 10, 10)) do
915
      copied = ActiveRecord::Migration.copy(@migrations_path, {:bukkits => MIGRATIONS_ROOT + "/to_copy_with_timestamps"})
A
Arun Agrawal 已提交
916 917
      assert File.exist?(@migrations_path + "/20100726101010_people_have_hobbies.bukkits.rb")
      assert File.exist?(@migrations_path + "/20100726101011_people_have_descriptions.bukkits.rb")
918
      assert_equal 2, copied.length
919
    end
920 921
  ensure
    clear
922
  end
923 924 925 926 927 928 929 930 931

  def test_check_pending_with_stdlib_logger
    old, ActiveRecord::Base.logger = ActiveRecord::Base.logger, ::Logger.new($stdout)
    quietly do
      assert_nothing_raised { ActiveRecord::Migration::CheckPending.new(Proc.new {}).call({}) }
    end
  ensure
    ActiveRecord::Base.logger = old
  end
932

933
end