migration_test.rb 60.9 KB
Newer Older
1
require "cases/helper"
2 3
require 'bigdecimal/util'

J
Jeremy Kemper 已提交
4 5
require 'models/person'
require 'models/topic'
6
require 'models/developer'
7 8 9 10

require MIGRATIONS_ROOT + "/valid/1_people_have_last_names"
require MIGRATIONS_ROOT + "/valid/2_we_need_reminders"
require MIGRATIONS_ROOT + "/decimal/1_give_me_big_numbers"
11
require MIGRATIONS_ROOT + "/interleaved/pass_3/2_i_raise_on_down"
12

13
if ActiveRecord::Base.connection.supports_migrations?
14 15
  class BigNumber < ActiveRecord::Base; end

16 17
  class Reminder < ActiveRecord::Base; end

J
Jamis Buck 已提交
18 19 20 21 22 23 24 25 26 27
  class ActiveRecord::Migration
    class <<self
      attr_accessor :message_count
      def puts(text="")
        self.message_count ||= 0
        self.message_count += 1
      end
    end
  end

28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45
  class MigrationTableAndIndexTest < ActiveRecord::TestCase
    def test_add_schema_info_respects_prefix_and_suffix
      conn = ActiveRecord::Base.connection

      conn.drop_table(ActiveRecord::Migrator.schema_migrations_table_name) if conn.table_exists?(ActiveRecord::Migrator.schema_migrations_table_name)
      ActiveRecord::Base.table_name_prefix = 'foo_'
      ActiveRecord::Base.table_name_suffix = '_bar'
      conn.drop_table(ActiveRecord::Migrator.schema_migrations_table_name) if conn.table_exists?(ActiveRecord::Migrator.schema_migrations_table_name)

      conn.initialize_schema_migrations_table

      assert_equal "foo_unique_schema_migrations_bar", conn.indexes(ActiveRecord::Migrator.schema_migrations_table_name)[0][:name]
    ensure
      ActiveRecord::Base.table_name_prefix = ""
      ActiveRecord::Base.table_name_suffix = ""
    end
  end

46
  class MigrationTest < ActiveRecord::TestCase
J
Jamis Buck 已提交
47
    self.use_transactional_fixtures = false
J
Jeremy Kemper 已提交
48

49
    fixtures :people
J
Jamis Buck 已提交
50

51
    def setup
J
Jamis Buck 已提交
52 53
      ActiveRecord::Migration.verbose = true
      PeopleHaveLastNames.message_count = 0
54 55 56
    end

    def teardown
57 58
      ActiveRecord::Base.connection.initialize_schema_migrations_table
      ActiveRecord::Base.connection.execute "DELETE FROM #{ActiveRecord::Migrator.schema_migrations_table_name}"
59

60 61 62
      %w(reminders people_reminders prefix_reminders_suffix).each do |table|
        Reminder.connection.drop_table(table) rescue nil
      end
63 64
      Reminder.reset_column_information

65
      %w(last_name key bio age height wealth birthday favorite_day
66
         moment_of_truth male administrator funny).each do |column|
67 68
        Person.connection.remove_column('people', column) rescue nil
      end
69
      Person.connection.remove_column("people", "first_name") rescue nil
70
      Person.connection.remove_column("people", "middle_name") rescue nil
71
      Person.connection.add_column("people", "first_name", :string, :limit => 40)
72 73
      Person.reset_column_information
    end
74

75
    def test_add_index
76 77 78
      # Limit size of last_name and key columns to support Firebird index limitations
      Person.connection.add_column "people", "last_name", :string, :limit => 100
      Person.connection.add_column "people", "key", :string, :limit => 100
79
      Person.connection.add_column "people", "administrator", :boolean
80

81 82 83
      assert_nothing_raised { Person.connection.add_index("people", "last_name") }
      assert_nothing_raised { Person.connection.remove_index("people", "last_name") }

84
      # Orcl nds shrt indx nms.  Sybs 2.
85 86
      # OpenBase does not have named indexes.  You must specify a single column name
      unless current_adapter?(:OracleAdapter, :SybaseAdapter, :OpenBaseAdapter)
87 88 89 90 91 92 93 94 95
        assert_nothing_raised { Person.connection.add_index("people", ["last_name", "first_name"]) }
        assert_nothing_raised { Person.connection.remove_index("people", :column => ["last_name", "first_name"]) }
        assert_nothing_raised { Person.connection.add_index("people", ["last_name", "first_name"]) }
        assert_nothing_raised { Person.connection.remove_index("people", :name => "index_people_on_last_name_and_first_name") }
        assert_nothing_raised { Person.connection.add_index("people", ["last_name", "first_name"]) }
        assert_nothing_raised { Person.connection.remove_index("people", "last_name_and_first_name") }
        assert_nothing_raised { Person.connection.add_index("people", ["last_name", "first_name"]) }
        assert_nothing_raised { Person.connection.remove_index("people", ["last_name", "first_name"]) }
      end
96

97
      # quoting
98
      # Note: changed index name from "key" to "key_idx" since "key" is a Firebird reserved word
99 100
      # OpenBase does not have named indexes.  You must specify a single column name
      unless current_adapter?(:OpenBaseAdapter)
101
        Person.update_all "#{Person.connection.quote_column_name 'key'}=#{Person.connection.quote_column_name 'id'}" #some databases (including sqlite2 won't add a unique index if existing data non unique)
102 103 104
        assert_nothing_raised { Person.connection.add_index("people", ["key"], :name => "key_idx", :unique => true) }
        assert_nothing_raised { Person.connection.remove_index("people", :name => "key_idx", :unique => true) }
      end
J
Jeremy Kemper 已提交
105

106
      # Sybase adapter does not support indexes on :boolean columns
107 108
      # OpenBase does not have named indexes.  You must specify a single column
      unless current_adapter?(:SybaseAdapter, :OpenBaseAdapter)
109 110 111
        assert_nothing_raised { Person.connection.add_index("people", %w(last_name first_name administrator), :name => "named_admin") }
        assert_nothing_raised { Person.connection.remove_index("people", :name => "named_admin") }
      end
112
    end
113

114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137
    def testing_table_with_only_foo_attribute
      Person.connection.create_table :testings, :id => false do |t|
        t.column :foo, :string
      end

      yield Person.connection
    ensure
      Person.connection.drop_table :testings rescue nil
    end
    protected :testing_table_with_only_foo_attribute

    def test_create_table_without_id
      testing_table_with_only_foo_attribute do |connection|
        assert_equal connection.columns(:testings).size, 1
      end
    end

    def test_add_column_with_primary_key_attribute
      testing_table_with_only_foo_attribute do |connection|
        assert_nothing_raised { connection.add_column :testings, :id, :primary_key }
        assert_equal connection.columns(:testings).size, 2
      end
    end

138 139 140 141 142 143 144 145 146 147
    def test_create_table_adds_id
      Person.connection.create_table :testings do |t|
        t.column :foo, :string
      end

      assert_equal %w(foo id),
        Person.connection.columns(:testings).map { |c| c.name }.sort
    ensure
      Person.connection.drop_table :testings rescue nil
    end
148 149

    def test_create_table_with_not_null_column
150 151 152 153
      assert_nothing_raised do
        Person.connection.create_table :testings do |t|
          t.column :foo, :string, :null => false
        end
154 155
      end

156
      assert_raise(ActiveRecord::StatementInvalid) do
157 158 159 160 161
        Person.connection.execute "insert into testings (foo) values (NULL)"
      end
    ensure
      Person.connection.drop_table :testings rescue nil
    end
162 163

    def test_create_table_with_defaults
164 165 166
      # MySQL doesn't allow defaults on TEXT or BLOB columns.
      mysql = current_adapter?(:MysqlAdapter)

167 168 169 170 171
      Person.connection.create_table :testings do |t|
        t.column :one, :string, :default => "hello"
        t.column :two, :boolean, :default => true
        t.column :three, :boolean, :default => false
        t.column :four, :integer, :default => 1
172
        t.column :five, :text, :default => "hello" unless mysql
173 174 175 176 177 178 179
      end

      columns = Person.connection.columns(:testings)
      one = columns.detect { |c| c.name == "one" }
      two = columns.detect { |c| c.name == "two" }
      three = columns.detect { |c| c.name == "three" }
      four = columns.detect { |c| c.name == "four" }
180
      five = columns.detect { |c| c.name == "five" } unless mysql
181 182

      assert_equal "hello", one.default
183 184
      assert_equal true, two.default
      assert_equal false, three.default
185
      assert_equal 1, four.default
186
      assert_equal "hello", five.default unless mysql
187 188 189 190

    ensure
      Person.connection.drop_table :testings rescue nil
    end
191 192

    def test_create_table_with_limits
193 194 195
      assert_nothing_raised do
        Person.connection.create_table :testings do |t|
          t.column :foo, :string, :limit => 255
196

197
          t.column :default_int, :integer
198

199 200 201 202
          t.column :one_int,    :integer, :limit => 1
          t.column :four_int,   :integer, :limit => 4
          t.column :eight_int,  :integer, :limit => 8
          t.column :eleven_int, :integer, :limit => 11
203
        end
204 205 206 207 208 209 210 211 212 213
      end

      columns = Person.connection.columns(:testings)
      foo = columns.detect { |c| c.name == "foo" }
      assert_equal 255, foo.limit

      default = columns.detect { |c| c.name == "default_int" }
      one     = columns.detect { |c| c.name == "one_int"     }
      four    = columns.detect { |c| c.name == "four_int"    }
      eight   = columns.detect { |c| c.name == "eight_int"   }
214
      eleven  = columns.detect { |c| c.name == "eleven_int"   }
215 216 217 218 219 220

      if current_adapter?(:PostgreSQLAdapter)
        assert_equal 'integer', default.sql_type
        assert_equal 'smallint', one.sql_type
        assert_equal 'integer', four.sql_type
        assert_equal 'bigint', eight.sql_type
221
        assert_equal 'integer', eleven.sql_type
222
      elsif current_adapter?(:MysqlAdapter)
223 224 225 226 227
        assert_match 'int(11)', default.sql_type
        assert_match 'tinyint', one.sql_type
        assert_match 'int', four.sql_type
        assert_match 'bigint', eight.sql_type
        assert_match 'int(11)', eleven.sql_type
228 229 230 231 232
      elsif current_adapter?(:OracleAdapter)
        assert_equal 'NUMBER(38)', default.sql_type
        assert_equal 'NUMBER(1)', one.sql_type
        assert_equal 'NUMBER(4)', four.sql_type
        assert_equal 'NUMBER(8)', eight.sql_type
233 234 235 236
      end
    ensure
      Person.connection.drop_table :testings rescue nil
    end
237

238 239 240 241 242 243 244
    def test_create_table_with_primary_key_prefix_as_table_name_with_underscore
      ActiveRecord::Base.primary_key_prefix_type = :table_name_with_underscore

      Person.connection.create_table :testings do |t|
          t.column :foo, :string
      end

245
      assert_equal %w(foo testing_id), Person.connection.columns(:testings).map { |c| c.name }.sort
246 247 248 249 250 251 252 253 254 255 256 257
    ensure
      Person.connection.drop_table :testings rescue nil
      ActiveRecord::Base.primary_key_prefix_type = nil
    end

    def test_create_table_with_primary_key_prefix_as_table_name
      ActiveRecord::Base.primary_key_prefix_type = :table_name

      Person.connection.create_table :testings do |t|
          t.column :foo, :string
      end

258
      assert_equal %w(foo testingid), Person.connection.columns(:testings).map { |c| c.name }.sort
259 260 261 262 263
    ensure
      Person.connection.drop_table :testings rescue nil
      ActiveRecord::Base.primary_key_prefix_type = nil
    end

264 265 266 267
    def test_create_table_with_force_true_does_not_drop_nonexisting_table
      if Person.connection.table_exists?(:testings2)
        Person.connection.drop_table :testings2
      end
268

269 270 271 272 273 274
      # 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
      temp_conn.expects(:drop_table).never
      temp_conn.create_table :testings2, :force => true do |t|
        t.column :foo, :string
275
      end
276 277
    ensure
      Person.connection.drop_table :testings2 rescue nil
278 279
    end

280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312
    def test_create_table_with_timestamps_should_create_datetime_columns
      table_name = :testings

      Person.connection.create_table table_name do |t|
        t.timestamps
      end
      created_columns = Person.connection.columns(table_name)

      created_at_column = created_columns.detect {|c| c.name == 'created_at' }
      updated_at_column = created_columns.detect {|c| c.name == 'updated_at' }

      assert created_at_column.null
      assert updated_at_column.null
    ensure
      Person.connection.drop_table table_name rescue nil
    end

    def test_create_table_with_timestamps_should_create_datetime_columns_with_options
      table_name = :testings

      Person.connection.create_table table_name do |t|
        t.timestamps :null => false
      end
      created_columns = Person.connection.columns(table_name)

      created_at_column = created_columns.detect {|c| c.name == 'created_at' }
      updated_at_column = created_columns.detect {|c| c.name == 'updated_at' }

      assert !created_at_column.null
      assert !updated_at_column.null
    ensure
      Person.connection.drop_table table_name rescue nil
    end
313

314 315 316 317 318 319 320
    def test_create_table_without_a_block
      table_name = :testings
      Person.connection.create_table table_name
    ensure
      Person.connection.drop_table table_name rescue nil
    end

321
    # Sybase, and SQLite3 will not allow you to add a NOT NULL
322
    # column to a table without a default value.
323
    unless current_adapter?(:SybaseAdapter, :SQLiteAdapter)
324 325 326 327 328
      def test_add_column_not_null_without_default
        Person.connection.create_table :testings do |t|
          t.column :foo, :string
        end
        Person.connection.add_column :testings, :bar, :string, :null => false
329

330
        assert_raise(ActiveRecord::StatementInvalid) do
331 332 333 334 335 336
          Person.connection.execute "insert into testings (foo, bar) values ('hello', NULL)"
        end
      ensure
        Person.connection.drop_table :testings rescue nil
      end
    end
337

338
    def test_add_column_not_null_with_default
339
      Person.connection.create_table :testings do |t|
340 341
        t.column :foo, :string
      end
J
Jeremy Kemper 已提交
342 343

      con = Person.connection
344
      Person.connection.enable_identity_insert("testings", true) if current_adapter?(:SybaseAdapter)
345
      Person.connection.execute "insert into testings (#{con.quote_column_name('id')}, #{con.quote_column_name('foo')}) values (1, 'hello')"
346
      Person.connection.enable_identity_insert("testings", false) if current_adapter?(:SybaseAdapter)
347
      assert_nothing_raised {Person.connection.add_column :testings, :bar, :string, :null => false, :default => "default" }
348

349
      assert_raise(ActiveRecord::StatementInvalid) do
350 351 352 353 354 355
        unless current_adapter?(:OpenBaseAdapter)
          Person.connection.execute "insert into testings (#{con.quote_column_name('id')}, #{con.quote_column_name('foo')}, #{con.quote_column_name('bar')}) values (2, 'hello', NULL)"
        else
          Person.connection.insert("INSERT INTO testings (#{con.quote_column_name('id')}, #{con.quote_column_name('foo')}, #{con.quote_column_name('bar')}) VALUES (2, 'hello', NULL)",
            "Testing Insert","id",2)
        end
356 357 358 359
      end
    ensure
      Person.connection.drop_table :testings rescue nil
    end
360

361 362 363 364
    # We specifically do a manual INSERT here, and then test only the SELECT
    # functionality. This allows us to more easily catch INSERT being broken,
    # but SELECT actually working fine.
    def test_native_decimal_insert_manual_vs_automatic
365
      correct_value = '0012345678901234567890.0123456789'.to_d
366 367 368 369 370 371

      Person.delete_all
      Person.connection.add_column "people", "wealth", :decimal, :precision => '30', :scale => '10'
      Person.reset_column_information

      # Do a manual insertion
372 373
      if current_adapter?(:OracleAdapter)
        Person.connection.execute "insert into people (id, wealth) values (people_seq.nextval, 12345678901234567890.0123456789)"
374
      elsif current_adapter?(:OpenBaseAdapter) || (current_adapter?(:MysqlAdapter) && Mysql.client_version < 50003) #before mysql 5.0.3 decimals stored as strings
375
        Person.connection.execute "insert into people (wealth) values ('12345678901234567890.0123456789')"
376 377 378
      else
        Person.connection.execute "insert into people (wealth) values (12345678901234567890.0123456789)"
      end
379 380 381 382 383 384

      # SELECT
      row = Person.find(:first)
      assert_kind_of BigDecimal, row.wealth

      # If this assert fails, that means the SELECT is broken!
385 386 387
      unless current_adapter?(:SQLite3Adapter)
        assert_equal correct_value, row.wealth
      end
388 389 390 391 392 393 394 395 396 397 398 399

      # Reset to old state
      Person.delete_all

      # Now use the Rails insertion
      assert_nothing_raised { Person.create :wealth => BigDecimal.new("12345678901234567890.0123456789") }

      # SELECT
      row = Person.find(:first)
      assert_kind_of BigDecimal, row.wealth

      # If these asserts fail, that means the INSERT (create function, or cast to SQL) is broken!
400 401 402
      unless current_adapter?(:SQLite3Adapter)
        assert_equal correct_value, row.wealth
      end
403 404 405 406 407 408

      # Reset to old state
      Person.connection.del_column "people", "wealth" rescue nil
      Person.reset_column_information
    end

409 410 411 412 413 414 415 416
    def test_add_column_with_precision_and_scale
      Person.connection.add_column 'people', 'wealth', :decimal, :precision => 9, :scale => 7
      Person.reset_column_information

      wealth_column = Person.columns_hash['wealth']
      assert_equal 9, wealth_column.precision
      assert_equal 7, wealth_column.scale
    end
417

418 419 420 421 422 423
    def test_native_types
      Person.delete_all
      Person.connection.add_column "people", "last_name", :string
      Person.connection.add_column "people", "bio", :text
      Person.connection.add_column "people", "age", :integer
      Person.connection.add_column "people", "height", :float
424
      Person.connection.add_column "people", "wealth", :decimal, :precision => '30', :scale => '10'
425 426
      Person.connection.add_column "people", "birthday", :datetime
      Person.connection.add_column "people", "favorite_day", :date
427
      Person.connection.add_column "people", "moment_of_truth", :datetime
428
      Person.connection.add_column "people", "male", :boolean
429
      Person.reset_column_information
430

431 432 433 434 435
      assert_nothing_raised do
        Person.create :first_name => 'bob', :last_name => 'bobsen',
          :bio => "I was born ....", :age => 18, :height => 1.78,
          :wealth => BigDecimal.new("12345678901234567890.0123456789"),
          :birthday => 18.years.ago, :favorite_day => 10.days.ago,
436
          :moment_of_truth => "1782-10-10 21:40:18", :male => true
437 438 439
      end

      bob = Person.find(:first)
440 441 442 443 444 445 446
      assert_equal 'bob', bob.first_name
      assert_equal 'bobsen', bob.last_name
      assert_equal "I was born ....", bob.bio
      assert_equal 18, bob.age

      # Test for 30 significent digits (beyond the 16 of float), 10 of them
      # after the decimal place.
J
Jeremy Kemper 已提交
447

448 449 450
      unless current_adapter?(:SQLite3Adapter)
        assert_equal BigDecimal.new("0012345678901234567890.0123456789"), bob.wealth
      end
J
Jeremy Kemper 已提交
451

452
      assert_equal true, bob.male?
453

454 455 456 457 458
      assert_equal String, bob.first_name.class
      assert_equal String, bob.last_name.class
      assert_equal String, bob.bio.class
      assert_equal Fixnum, bob.age.class
      assert_equal Time, bob.birthday.class
459

460
      if current_adapter?(:OracleAdapter, :SybaseAdapter)
461
        # Sybase, and Oracle don't differentiate between date/time
462 463 464 465 466
        assert_equal Time, bob.favorite_day.class
      else
        assert_equal Date, bob.favorite_day.class
      end

467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482
      # Oracle adapter stores Time or DateTime with timezone value already in _before_type_cast column
      # therefore no timezone change is done afterwards when default timezone is changed
      unless current_adapter?(:OracleAdapter)
        # Test DateTime column and defaults, including timezone.
        # FIXME: moment of truth may be Time on 64-bit platforms.
        if bob.moment_of_truth.is_a?(DateTime)

          with_env_tz 'US/Eastern' do
            assert_equal DateTime.local_offset, bob.moment_of_truth.offset
            assert_not_equal 0, bob.moment_of_truth.offset
            assert_not_equal "Z", bob.moment_of_truth.zone
            # US/Eastern is -5 hours from GMT
            assert_equal Rational(-5, 24), bob.moment_of_truth.offset
            assert_match /\A-05:?00\Z/, bob.moment_of_truth.zone #ruby 1.8.6 uses HH:MM, prior versions use HHMM
            assert_equal DateTime::ITALY, bob.moment_of_truth.start
          end
483
        end
484
      end
485

486
      assert_equal TrueClass, bob.male?.class
487
      assert_kind_of BigDecimal, bob.wealth
488 489
    end

490 491 492 493 494
    if current_adapter?(:MysqlAdapter)
      def test_unabstracted_database_dependent_types
        Person.delete_all

        ActiveRecord::Migration.add_column :people, :intelligence_quotient, :tinyint
495
        Person.reset_column_information
496
        assert_match /tinyint/, Person.columns_hash['intelligence_quotient'].sql_type
497
      ensure
498
        ActiveRecord::Migration.remove_column :people, :intelligence_quotient rescue nil
499 500 501
      end
    end

502
    def test_add_remove_single_field_using_string_arguments
503 504
      assert !Person.column_methods_hash.include?(:last_name)

505
      ActiveRecord::Migration.add_column 'people', 'last_name', :string
506 507 508

      Person.reset_column_information
      assert Person.column_methods_hash.include?(:last_name)
509

510 511 512 513 514 515 516 517 518 519 520 521 522 523 524
      ActiveRecord::Migration.remove_column 'people', 'last_name'

      Person.reset_column_information
      assert !Person.column_methods_hash.include?(:last_name)
    end

    def test_add_remove_single_field_using_symbol_arguments
      assert !Person.column_methods_hash.include?(:last_name)

      ActiveRecord::Migration.add_column :people, :last_name, :string

      Person.reset_column_information
      assert Person.column_methods_hash.include?(:last_name)

      ActiveRecord::Migration.remove_column :people, :last_name
525 526 527 528

      Person.reset_column_information
      assert !Person.column_methods_hash.include?(:last_name)
    end
529

530 531
    def test_add_rename
      Person.delete_all
532

533
      begin
534
        Person.connection.add_column "people", "girlfriend", :string
535
        Person.reset_column_information
536 537
        Person.create :girlfriend => 'bobette'

538
        Person.connection.rename_column "people", "girlfriend", "exgirlfriend"
539 540

        Person.reset_column_information
541
        bob = Person.find(:first)
542

543 544 545 546 547
        assert_equal "bobette", bob.exgirlfriend
      ensure
        Person.connection.remove_column("people", "girlfriend") rescue nil
        Person.connection.remove_column("people", "exgirlfriend") rescue nil
      end
548

549
    end
550

551 552
    def test_rename_column_using_symbol_arguments
      begin
553
        names_before = Person.find(:all).map(&:first_name)
554 555 556
        Person.connection.rename_column :people, :first_name, :nick_name
        Person.reset_column_information
        assert Person.column_names.include?("nick_name")
557
        assert_equal names_before, Person.find(:all).map(&:nick_name)
558 559 560 561 562
      ensure
        Person.connection.remove_column("people","nick_name")
        Person.connection.add_column("people","first_name", :string)
      end
    end
563

564 565
    def test_rename_column
      begin
566
        names_before = Person.find(:all).map(&:first_name)
567 568 569
        Person.connection.rename_column "people", "first_name", "nick_name"
        Person.reset_column_information
        assert Person.column_names.include?("nick_name")
570
        assert_equal names_before, Person.find(:all).map(&:nick_name)
571 572 573 574 575
      ensure
        Person.connection.remove_column("people","nick_name")
        Person.connection.add_column("people","first_name", :string)
      end
    end
576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593 594 595

    def test_rename_column_preserves_default_value_not_null
      begin
        default_before = Developer.connection.columns("developers").find { |c| c.name == "salary" }.default
        assert_equal 70000, default_before
        Developer.connection.rename_column "developers", "salary", "anual_salary"
        Developer.reset_column_information
        assert Developer.column_names.include?("anual_salary")
        default_after = Developer.connection.columns("developers").find { |c| c.name == "anual_salary" }.default
        assert_equal 70000, default_after
      ensure
        Developer.connection.rename_column "developers", "anual_salary", "salary"
        Developer.reset_column_information
      end
    end

    def test_rename_nonexistent_column
      ActiveRecord::Base.connection.create_table(:hats) do |table|
        table.column :hat_name, :string, :default => nil
      end
596
      exception = if current_adapter?(:PostgreSQLAdapter, :OracleAdapter)
597 598 599 600
        ActiveRecord::StatementInvalid
      else
        ActiveRecord::ActiveRecordError
      end
601
      assert_raise(exception) do
602 603 604 605 606
        Person.connection.rename_column "hats", "nonexistent", "should_fail"
      end
    ensure
      ActiveRecord::Base.connection.drop_table(:hats)
    end
607

608 609 610 611 612 613 614 615 616 617
    def test_rename_column_with_sql_reserved_word
      begin
        assert_nothing_raised { Person.connection.rename_column "people", "first_name", "group" }
        Person.reset_column_information
        assert Person.column_names.include?("group")
      ensure
        Person.connection.remove_column("people", "group") rescue nil
        Person.connection.add_column("people", "first_name", :string) rescue nil
      end
    end
J
Jeremy Kemper 已提交
618

619 620 621 622 623
    def test_rename_column_with_an_index
      ActiveRecord::Base.connection.create_table(:hats) do |table|
        table.column :hat_name, :string, :limit => 100
        table.column :hat_size, :integer
      end
624
      Person.connection.add_index :hats, :hat_name
625 626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649
      assert_nothing_raised do
        Person.connection.rename_column "hats", "hat_name", "name"
      end
    ensure
      ActiveRecord::Base.connection.drop_table(:hats)
    end

    def test_remove_column_with_index
      ActiveRecord::Base.connection.create_table(:hats) do |table|
        table.column :hat_name, :string, :limit => 100
        table.column :hat_size, :integer
      end
      ActiveRecord::Base.connection.add_index "hats", "hat_size"

      assert_nothing_raised { Person.connection.remove_column("hats", "hat_size") }
    ensure
      ActiveRecord::Base.connection.drop_table(:hats)
    end

    def test_remove_column_with_multi_column_index
      ActiveRecord::Base.connection.create_table(:hats) do |table|
        table.column :hat_name, :string, :limit => 100
        table.column :hat_size, :integer
        table.column :hat_style, :string, :limit => 100
      end
650 651 652 653 654 655 656
      # Oracle index names should be 30 or less characters
      if current_adapter?(:OracleAdapter)
        ActiveRecord::Base.connection.add_index "hats", ["hat_style", "hat_size"], :unique => true,
          :name => 'index_hats_on_hat_style_size'
      else
        ActiveRecord::Base.connection.add_index "hats", ["hat_style", "hat_size"], :unique => true
      end
657 658 659 660 661 662

      assert_nothing_raised { Person.connection.remove_column("hats", "hat_size") }
    ensure
      ActiveRecord::Base.connection.drop_table(:hats)
    end

663 664 665 666
    def test_change_type_of_not_null_column
      assert_nothing_raised do
        Topic.connection.change_column "topics", "written_on", :datetime, :null => false
        Topic.reset_column_information
J
Jeremy Kemper 已提交
667

668 669 670 671
        Topic.connection.change_column "topics", "written_on", :datetime, :null => false
        Topic.reset_column_information
      end
    end
672

673 674 675 676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691 692 693 694 695 696 697 698
    if current_adapter?(:SQLiteAdapter)
      def test_rename_table_for_sqlite_should_work_with_reserved_words
        begin
          assert_nothing_raised do
            ActiveRecord::Base.connection.rename_table :references, :old_references
            ActiveRecord::Base.connection.create_table :octopuses do |t|
              t.column :url, :string
            end
          end

          assert_nothing_raised { ActiveRecord::Base.connection.rename_table :octopuses, :references }

          # Using explicit id in insert for compatibility across all databases
          con = ActiveRecord::Base.connection
          assert_nothing_raised do
            con.execute "INSERT INTO 'references' (#{con.quote_column_name('id')}, #{con.quote_column_name('url')}) VALUES (1, 'http://rubyonrails.com')"
          end
          assert_equal 'http://rubyonrails.com', ActiveRecord::Base.connection.select_value("SELECT url FROM 'references' WHERE id=1")

        ensure
          ActiveRecord::Base.connection.drop_table :references
          ActiveRecord::Base.connection.rename_table :old_references, :references
        end
      end
    end

699 700 701 702
    def test_rename_table
      begin
        ActiveRecord::Base.connection.create_table :octopuses do |t|
          t.column :url, :string
703
        end
704
        ActiveRecord::Base.connection.rename_table :octopuses, :octopi
705

706
        # Using explicit id in insert for compatibility across all databases
J
Jeremy Kemper 已提交
707
        con = ActiveRecord::Base.connection
708
        con.enable_identity_insert("octopi", true) if current_adapter?(:SybaseAdapter)
709
        assert_nothing_raised { con.execute "INSERT INTO octopi (#{con.quote_column_name('id')}, #{con.quote_column_name('url')}) VALUES (1, 'http://www.foreverflying.com/octopus-black7.jpg')" }
710
        con.enable_identity_insert("octopi", false) if current_adapter?(:SybaseAdapter)
711 712

        assert_equal 'http://www.foreverflying.com/octopus-black7.jpg', ActiveRecord::Base.connection.select_value("SELECT url FROM octopi WHERE id=1")
713

714 715 716 717 718
      ensure
        ActiveRecord::Base.connection.drop_table :octopuses rescue nil
        ActiveRecord::Base.connection.drop_table :octopi rescue nil
      end
    end
J
Jeremy Kemper 已提交
719

720
    def test_change_column_nullability
J
Jeremy Kemper 已提交
721
      Person.delete_all
722 723 724 725 726 727 728 729 730 731
      Person.connection.add_column "people", "funny", :boolean
      Person.reset_column_information
      assert Person.columns_hash["funny"].null, "Column 'funny' must initially allow nulls"
      Person.connection.change_column "people", "funny", :boolean, :null => false, :default => true
      Person.reset_column_information
      assert !Person.columns_hash["funny"].null, "Column 'funny' must *not* allow nulls at this point"
      Person.connection.change_column "people", "funny", :boolean, :null => true
      Person.reset_column_information
      assert Person.columns_hash["funny"].null, "Column 'funny' must allow nulls again at this point"
    end
732

733 734 735 736 737 738
    def test_rename_table_with_an_index
      begin
        ActiveRecord::Base.connection.create_table :octopuses do |t|
          t.column :url, :string
        end
        ActiveRecord::Base.connection.add_index :octopuses, :url
J
Jeremy Kemper 已提交
739

740 741 742
        ActiveRecord::Base.connection.rename_table :octopuses, :octopi

        # Using explicit id in insert for compatibility across all databases
J
Jeremy Kemper 已提交
743
        con = ActiveRecord::Base.connection
744
        con.enable_identity_insert("octopi", true) if current_adapter?(:SybaseAdapter)
745
        assert_nothing_raised { con.execute "INSERT INTO octopi (#{con.quote_column_name('id')}, #{con.quote_column_name('url')}) VALUES (1, 'http://www.foreverflying.com/octopus-black7.jpg')" }
746
        con.enable_identity_insert("octopi", false) if current_adapter?(:SybaseAdapter)
747 748 749 750 751 752 753 754 755

        assert_equal 'http://www.foreverflying.com/octopus-black7.jpg', ActiveRecord::Base.connection.select_value("SELECT url FROM octopi WHERE id=1")
        assert ActiveRecord::Base.connection.indexes(:octopi).first.columns.include?("url")
      ensure
        ActiveRecord::Base.connection.drop_table :octopuses rescue nil
        ActiveRecord::Base.connection.drop_table :octopi rescue nil
      end
    end

756
    def test_change_column
757
      Person.connection.add_column 'people', 'age', :integer
758 759
      label = "test_change_column Columns"
      old_columns = Person.connection.columns(Person.table_name, label)
760 761 762
      assert old_columns.find { |c| c.name == 'age' and c.type == :integer }

      assert_nothing_raised { Person.connection.change_column "people", "age", :string }
763

764
      new_columns = Person.connection.columns(Person.table_name, label)
765 766
      assert_nil new_columns.find { |c| c.name == 'age' and c.type == :integer }
      assert new_columns.find { |c| c.name == 'age' and c.type == :string }
767

768
      old_columns = Topic.connection.columns(Topic.table_name, label)
769 770
      assert old_columns.find { |c| c.name == 'approved' and c.type == :boolean and c.default == true }
      assert_nothing_raised { Topic.connection.change_column :topics, :approved, :boolean, :default => false }
771
      new_columns = Topic.connection.columns(Topic.table_name, label)
772 773
      assert_nil new_columns.find { |c| c.name == 'approved' and c.type == :boolean and c.default == true }
      assert new_columns.find { |c| c.name == 'approved' and c.type == :boolean and c.default == false }
774
      assert_nothing_raised { Topic.connection.change_column :topics, :approved, :boolean, :default => true }
775
    end
J
Jeremy Kemper 已提交
776

777 778 779 780
    def test_change_column_with_nil_default
      Person.connection.add_column "people", "contributor", :boolean, :default => true
      Person.reset_column_information
      assert Person.new.contributor?
J
Jeremy Kemper 已提交
781

782 783 784 785
      assert_nothing_raised { Person.connection.change_column "people", "contributor", :boolean, :default => nil }
      Person.reset_column_information
      assert !Person.new.contributor?
      assert_nil Person.new.contributor
786 787
    ensure
      Person.connection.remove_column("people", "contributor") rescue nil
788
    end
789 790

    def test_change_column_with_new_default
791
      Person.connection.add_column "people", "administrator", :boolean, :default => true
792
      Person.reset_column_information
793
      assert Person.new.administrator?
794

795
      assert_nothing_raised { Person.connection.change_column "people", "administrator", :boolean, :default => false }
796
      Person.reset_column_information
797
      assert !Person.new.administrator?
798 799
    ensure
      Person.connection.remove_column("people", "administrator") rescue nil
800
    end
J
Jeremy Kemper 已提交
801

802 803 804 805 806
    def test_change_column_default
      Person.connection.change_column_default "people", "first_name", "Tester"
      Person.reset_column_information
      assert_equal "Tester", Person.new.first_name
    end
807 808 809 810 811 812 813 814

    def test_change_column_quotes_column_names
      Person.connection.create_table :testings do |t|
        t.column :select, :string
      end

      assert_nothing_raised { Person.connection.change_column :testings, :select, :string, :limit => 10 }

815 816 817 818 819 820
      # Oracle needs primary key value from sequence
      if current_adapter?(:OracleAdapter)
        assert_nothing_raised { Person.connection.execute "insert into testings (id, #{Person.connection.quote_column_name('select')}) values (testings_seq.nextval, '7 chars')" }
      else
        assert_nothing_raised { Person.connection.execute "insert into testings (#{Person.connection.quote_column_name('select')}) values ('7 chars')" }
      end
821 822 823 824
    ensure
      Person.connection.drop_table :testings rescue nil
    end

825 826 827 828 829 830 831 832 833 834 835
    def test_keeping_default_and_notnull_constaint_on_change
      Person.connection.create_table :testings do |t|
        t.column :title, :string
      end
      person_klass = Class.new(Person)
      person_klass.set_table_name 'testings'

      person_klass.connection.add_column "testings", "wealth", :integer, :null => false, :default => 99
      person_klass.reset_column_information
      assert_equal 99, person_klass.columns_hash["wealth"].default
      assert_equal false, person_klass.columns_hash["wealth"].null
836 837 838 839 840 841
      # Oracle needs primary key value from sequence
      if current_adapter?(:OracleAdapter)
        assert_nothing_raised {person_klass.connection.execute("insert into testings (id, title) values (testings_seq.nextval, 'tester')")}
      else
        assert_nothing_raised {person_klass.connection.execute("insert into testings (title) values ('tester')")}
      end
842 843 844 845 846 847 848 849 850 851 852 853 854 855 856 857 858 859 860 861 862 863 864 865 866 867 868 869 870 871 872 873 874 875 876 877 878

      # change column default to see that column doesn't lose its not null definition
      person_klass.connection.change_column_default "testings", "wealth", 100
      person_klass.reset_column_information
      assert_equal 100, person_klass.columns_hash["wealth"].default
      assert_equal false, person_klass.columns_hash["wealth"].null

      # rename column to see that column doesn't lose its not null and/or default definition
      person_klass.connection.rename_column "testings", "wealth", "money"
      person_klass.reset_column_information
      assert_nil person_klass.columns_hash["wealth"]
      assert_equal 100, person_klass.columns_hash["money"].default
      assert_equal false, person_klass.columns_hash["money"].null

      # change column
      person_klass.connection.change_column "testings", "money", :integer, :null => false, :default => 1000
      person_klass.reset_column_information
      assert_equal 1000, person_klass.columns_hash["money"].default
      assert_equal false, person_klass.columns_hash["money"].null

      # change column, make it nullable and clear default
      person_klass.connection.change_column "testings", "money", :integer, :null => true, :default => nil
      person_klass.reset_column_information
      assert_nil person_klass.columns_hash["money"].default
      assert_equal true, person_klass.columns_hash["money"].null

      # change_column_null, make it not nullable and set null values to a default value
      person_klass.connection.execute('UPDATE testings SET money = NULL')
      person_klass.connection.change_column_null "testings", "money", false, 2000
      person_klass.reset_column_information
      assert_nil person_klass.columns_hash["money"].default
      assert_equal false, person_klass.columns_hash["money"].null
      assert_equal [2000], Person.connection.select_values("SELECT money FROM testings").map { |s| s.to_i }.sort
    ensure
      Person.connection.drop_table :testings rescue nil
    end

879 880 881 882 883
    def test_change_column_default_to_null
      Person.connection.change_column_default "people", "first_name", nil
      Person.reset_column_information
      assert_nil Person.new.first_name
    end
884 885

    def test_add_table
886
      assert !Reminder.table_exists?
887

888
      WeNeedReminders.up
J
Jeremy Kemper 已提交
889 890

      assert Reminder.create("content" => "hello world", "remind_at" => Time.now)
891
      assert_equal "hello world", Reminder.find(:first).content
892

893
      WeNeedReminders.down
894
      assert_raise(ActiveRecord::StatementInvalid) { Reminder.find(:first) }
895 896
    end

897 898 899 900 901 902 903 904 905 906 907 908 909 910 911 912 913 914 915 916 917 918 919 920 921 922 923 924 925 926 927 928 929 930 931
    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")
      )

      b = BigNumber.find(:first)
      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
932
      # precision/scale explicitly left out.  By the SQL standard, numbers
933 934 935 936 937 938 939 940 941 942 943 944 945 946 947 948 949 950 951 952 953 954
      # assigned to this field should be truncated but that's seldom respected.
      if current_adapter?(:PostgreSQLAdapter, :SQLite2Adapter)
        # - 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?(:SQLiteAdapter)
        # - SQLite3 stores a float, in violation of SQL
        assert_kind_of BigDecimal, b.value_of_e
        assert_equal BigDecimal("2.71828182845905"), b.value_of_e
      else
        # - SQL standard is an integer
        assert_kind_of Fixnum, b.value_of_e
        assert_equal 2, b.value_of_e
      end

      GiveMeBigNumbers.down
955
      assert_raise(ActiveRecord::StatementInvalid) { BigNumber.find(:first) }
956 957
    end

958 959
    def test_migrator
      assert !Person.column_methods_hash.include?(:last_name)
960
      assert !Reminder.table_exists?
961

962
      ActiveRecord::Migrator.up(MIGRATIONS_ROOT + "/valid")
963

964
      assert_equal 3, ActiveRecord::Migrator.current_version
965 966 967 968 969
      Person.reset_column_information
      assert Person.column_methods_hash.include?(:last_name)
      assert Reminder.create("content" => "hello world", "remind_at" => Time.now)
      assert_equal "hello world", Reminder.find(:first).content

970
      ActiveRecord::Migrator.down(MIGRATIONS_ROOT + "/valid")
971 972 973 974

      assert_equal 0, ActiveRecord::Migrator.current_version
      Person.reset_column_information
      assert !Person.column_methods_hash.include?(:last_name)
975
      assert_raise(ActiveRecord::StatementInvalid) { Reminder.find(:first) }
976 977 978 979
    end

    def test_migrator_one_up
      assert !Person.column_methods_hash.include?(:last_name)
980
      assert !Reminder.table_exists?
981

982
      ActiveRecord::Migrator.up(MIGRATIONS_ROOT + "/valid", 1)
983 984 985

      Person.reset_column_information
      assert Person.column_methods_hash.include?(:last_name)
986
      assert !Reminder.table_exists?
987

988
      ActiveRecord::Migrator.up(MIGRATIONS_ROOT + "/valid", 2)
989 990 991 992

      assert Reminder.create("content" => "hello world", "remind_at" => Time.now)
      assert_equal "hello world", Reminder.find(:first).content
    end
993

994
    def test_migrator_one_down
995
      ActiveRecord::Migrator.up(MIGRATIONS_ROOT + "/valid")
996

997
      ActiveRecord::Migrator.down(MIGRATIONS_ROOT + "/valid", 1)
998

999 1000
      Person.reset_column_information
      assert Person.column_methods_hash.include?(:last_name)
1001
      assert !Reminder.table_exists?
1002
    end
1003

1004
    def test_migrator_one_up_one_down
1005 1006
      ActiveRecord::Migrator.up(MIGRATIONS_ROOT + "/valid", 1)
      ActiveRecord::Migrator.down(MIGRATIONS_ROOT + "/valid", 0)
1007 1008

      assert !Person.column_methods_hash.include?(:last_name)
1009
      assert !Reminder.table_exists?
1010
    end
1011

1012 1013 1014 1015 1016 1017 1018 1019 1020 1021 1022 1023 1024 1025 1026
    def test_migrator_double_up
      assert_equal(0, ActiveRecord::Migrator.current_version)
      ActiveRecord::Migrator.run(:up, MIGRATIONS_ROOT + "/valid", 1)
      assert_nothing_raised { ActiveRecord::Migrator.run(:up, MIGRATIONS_ROOT + "/valid", 1) }
      assert_equal(1, ActiveRecord::Migrator.current_version)
    end

    def test_migrator_double_down
      assert_equal(0, ActiveRecord::Migrator.current_version)
      ActiveRecord::Migrator.run(:up, MIGRATIONS_ROOT + "/valid", 1)
      ActiveRecord::Migrator.run(:down, MIGRATIONS_ROOT + "/valid", 1)
      assert_nothing_raised { ActiveRecord::Migrator.run(:down, MIGRATIONS_ROOT + "/valid", 1) }
      assert_equal(0, ActiveRecord::Migrator.current_version)
    end

1027
    if ActiveRecord::Base.connection.supports_ddl_transactions?
1028 1029 1030
      def test_migrator_one_up_with_exception_and_rollback
        assert !Person.column_methods_hash.include?(:last_name)

1031
        e = assert_raise(StandardError) do
1032 1033 1034 1035 1036 1037 1038 1039 1040 1041
          ActiveRecord::Migrator.up(MIGRATIONS_ROOT + "/broken", 100)
        end

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

        Person.reset_column_information
        assert !Person.column_methods_hash.include?(:last_name)
      end
    end

1042 1043
    def test_finds_migrations
      migrations = ActiveRecord::Migrator.new(:up, MIGRATIONS_ROOT + "/valid").migrations
1044 1045 1046 1047

      [[1, 'PeopleHaveLastNames'], [2, 'WeNeedReminders'], [3, 'InnocentJointable']].each_with_index do |pair, i|
        assert_equal migrations[i].version, pair.first
        assert_equal migrations[i].name, pair.last
1048 1049 1050 1051 1052 1053
      end
    end

    def test_finds_pending_migrations
      ActiveRecord::Migrator.up(MIGRATIONS_ROOT + "/interleaved/pass_2", 1)
      migrations = ActiveRecord::Migrator.new(:up, MIGRATIONS_ROOT + "/interleaved/pass_2").pending_migrations
1054

1055
      assert_equal 1, migrations.size
1056 1057
      assert_equal migrations[0].version, 3
      assert_equal migrations[0].name, 'InnocentJointable'
1058 1059
    end

1060 1061 1062 1063 1064 1065 1066 1067 1068 1069 1070 1071 1072 1073 1074 1075 1076 1077 1078 1079
    def test_only_loads_pending_migrations
      # migrate up to 1
      ActiveRecord::Migrator.up(MIGRATIONS_ROOT + "/valid", 1)

      # now unload the migrations that have been defined
      PeopleHaveLastNames.unloadable
      ActiveSupport::Dependencies.remove_unloadable_constants!

      ActiveRecord::Migrator.migrate(MIGRATIONS_ROOT + "/valid", nil)

      assert !defined? PeopleHaveLastNames

      %w(WeNeedReminders, InnocentJointable).each do |migration|
        assert defined? migration
      end

    ensure
      load(MIGRATIONS_ROOT + "/valid/1_people_have_last_names.rb")
    end

1080 1081 1082 1083 1084 1085 1086 1087 1088 1089 1090 1091 1092 1093 1094
    def test_migrator_interleaved_migrations
      ActiveRecord::Migrator.up(MIGRATIONS_ROOT + "/interleaved/pass_1")

      assert_nothing_raised do
        ActiveRecord::Migrator.up(MIGRATIONS_ROOT + "/interleaved/pass_2")
      end

      Person.reset_column_information
      assert Person.column_methods_hash.include?(:last_name)

      assert_nothing_raised do
        ActiveRecord::Migrator.down(MIGRATIONS_ROOT + "/interleaved/pass_3")
      end
    end

1095
    def test_migrator_db_has_no_schema_migrations_table
1096 1097 1098 1099 1100 1101
      # Oracle adapter raises error if semicolon is present as last character
      if current_adapter?(:OracleAdapter)
        ActiveRecord::Base.connection.execute("DROP TABLE schema_migrations")
      else
        ActiveRecord::Base.connection.execute("DROP TABLE schema_migrations;")
      end
1102 1103 1104 1105 1106
      assert_nothing_raised do
        ActiveRecord::Migrator.migrate(MIGRATIONS_ROOT + "/valid", 1)
      end
    end

J
Jamis Buck 已提交
1107
    def test_migrator_verbosity
1108
      ActiveRecord::Migrator.up(MIGRATIONS_ROOT + "/valid", 1)
J
Jamis Buck 已提交
1109 1110 1111
      assert PeopleHaveLastNames.message_count > 0
      PeopleHaveLastNames.message_count = 0

1112
      ActiveRecord::Migrator.down(MIGRATIONS_ROOT + "/valid", 0)
J
Jamis Buck 已提交
1113 1114 1115
      assert PeopleHaveLastNames.message_count > 0
      PeopleHaveLastNames.message_count = 0
    end
1116

J
Jamis Buck 已提交
1117 1118
    def test_migrator_verbosity_off
      PeopleHaveLastNames.verbose = false
1119
      ActiveRecord::Migrator.up(MIGRATIONS_ROOT + "/valid", 1)
J
Jamis Buck 已提交
1120
      assert PeopleHaveLastNames.message_count.zero?
1121
      ActiveRecord::Migrator.down(MIGRATIONS_ROOT + "/valid", 0)
J
Jamis Buck 已提交
1122 1123
      assert PeopleHaveLastNames.message_count.zero?
    end
1124

1125
    def test_migrator_going_down_due_to_version_target
1126 1127
      ActiveRecord::Migrator.up(MIGRATIONS_ROOT + "/valid", 1)
      ActiveRecord::Migrator.migrate(MIGRATIONS_ROOT + "/valid", 0)
1128 1129

      assert !Person.column_methods_hash.include?(:last_name)
1130
      assert !Reminder.table_exists?
1131

1132
      ActiveRecord::Migrator.migrate(MIGRATIONS_ROOT + "/valid")
1133 1134 1135 1136 1137 1138

      Person.reset_column_information
      assert Person.column_methods_hash.include?(:last_name)
      assert Reminder.create("content" => "hello world", "remind_at" => Time.now)
      assert_equal "hello world", Reminder.find(:first).content
    end
1139

1140 1141 1142
    def test_migrator_rollback
      ActiveRecord::Migrator.migrate(MIGRATIONS_ROOT + "/valid")
      assert_equal(3, ActiveRecord::Migrator.current_version)
1143

1144
      ActiveRecord::Migrator.rollback(MIGRATIONS_ROOT + "/valid")
1145
      assert_equal(2, ActiveRecord::Migrator.current_version)
1146

1147
      ActiveRecord::Migrator.rollback(MIGRATIONS_ROOT + "/valid")
1148
      assert_equal(1, ActiveRecord::Migrator.current_version)
1149

1150
      ActiveRecord::Migrator.rollback(MIGRATIONS_ROOT + "/valid")
1151
      assert_equal(0, ActiveRecord::Migrator.current_version)
1152

1153
      ActiveRecord::Migrator.rollback(MIGRATIONS_ROOT + "/valid")
1154
      assert_equal(0, ActiveRecord::Migrator.current_version)
1155
    end
1156

1157 1158 1159 1160 1161 1162 1163 1164 1165 1166 1167
    def test_migrator_forward
      ActiveRecord::Migrator.migrate(MIGRATIONS_ROOT + "/valid", 1)
      assert_equal(1, ActiveRecord::Migrator.current_version)

      ActiveRecord::Migrator.forward(MIGRATIONS_ROOT + "/valid", 2)
      assert_equal(3, ActiveRecord::Migrator.current_version)

      ActiveRecord::Migrator.forward(MIGRATIONS_ROOT + "/valid")
      assert_equal(3, ActiveRecord::Migrator.current_version)
    end

1168
    def test_schema_migrations_table_name
1169 1170
      ActiveRecord::Base.table_name_prefix = "prefix_"
      ActiveRecord::Base.table_name_suffix = "_suffix"
1171
      Reminder.reset_table_name
1172
      assert_equal "prefix_schema_migrations_suffix", ActiveRecord::Migrator.schema_migrations_table_name
1173 1174
      ActiveRecord::Base.table_name_prefix = ""
      ActiveRecord::Base.table_name_suffix = ""
1175
      Reminder.reset_table_name
1176
      assert_equal "schema_migrations", ActiveRecord::Migrator.schema_migrations_table_name
1177 1178 1179
    ensure
      ActiveRecord::Base.table_name_prefix = ""
      ActiveRecord::Base.table_name_suffix = ""
1180
    end
1181

1182 1183 1184 1185
    def test_proper_table_name
      assert_equal "table", ActiveRecord::Migrator.proper_table_name('table')
      assert_equal "table", ActiveRecord::Migrator.proper_table_name(:table)
      assert_equal "reminders", ActiveRecord::Migrator.proper_table_name(Reminder)
1186
      Reminder.reset_table_name
1187
      assert_equal Reminder.table_name, ActiveRecord::Migrator.proper_table_name(Reminder)
1188

1189 1190 1191 1192 1193
      # Use the model's own prefix/suffix if a model is given
      ActiveRecord::Base.table_name_prefix = "ARprefix_"
      ActiveRecord::Base.table_name_suffix = "_ARsuffix"
      Reminder.table_name_prefix = 'prefix_'
      Reminder.table_name_suffix = '_suffix'
1194
      Reminder.reset_table_name
1195 1196 1197
      assert_equal "prefix_reminders_suffix", ActiveRecord::Migrator.proper_table_name(Reminder)
      Reminder.table_name_prefix = ''
      Reminder.table_name_suffix = ''
1198
      Reminder.reset_table_name
1199 1200

      # Use AR::Base's prefix/suffix if string or symbol is given
1201 1202
      ActiveRecord::Base.table_name_prefix = "prefix_"
      ActiveRecord::Base.table_name_suffix = "_suffix"
1203
      Reminder.reset_table_name
1204 1205 1206 1207
      assert_equal "prefix_table_suffix", ActiveRecord::Migrator.proper_table_name('table')
      assert_equal "prefix_table_suffix", ActiveRecord::Migrator.proper_table_name(:table)
      ActiveRecord::Base.table_name_prefix = ""
      ActiveRecord::Base.table_name_suffix = ""
1208
      Reminder.reset_table_name
1209
    end
1210 1211

    def test_add_drop_table_with_prefix_and_suffix
1212
      assert !Reminder.table_exists?
1213 1214
      ActiveRecord::Base.table_name_prefix = 'prefix_'
      ActiveRecord::Base.table_name_suffix = '_suffix'
1215
      Reminder.reset_table_name
1216
      Reminder.reset_sequence_name
1217 1218 1219 1220 1221
      WeNeedReminders.up
      assert Reminder.create("content" => "hello world", "remind_at" => Time.now)
      assert_equal "hello world", Reminder.find(:first).content

      WeNeedReminders.down
1222
      assert_raise(ActiveRecord::StatementInvalid) { Reminder.find(:first) }
1223
    ensure
1224 1225
      ActiveRecord::Base.table_name_prefix = ''
      ActiveRecord::Base.table_name_suffix = ''
1226
      Reminder.reset_table_name
1227
      Reminder.reset_sequence_name
1228
    end
1229

1230 1231
    def test_create_table_with_binary_column
      Person.connection.drop_table :binary_testings rescue nil
1232

1233 1234 1235 1236 1237
      assert_nothing_raised {
        Person.connection.create_table :binary_testings do |t|
          t.column "data", :binary, :null => false
        end
      }
1238

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

1242 1243 1244 1245 1246
      if current_adapter?(:MysqlAdapter)
        assert_equal '', data_column.default
      else
        assert_nil data_column.default
      end
1247

1248
      Person.connection.drop_table :binary_testings rescue nil
1249
    end
1250

1251
    def test_migrator_with_duplicates
1252
      assert_raise(ActiveRecord::DuplicateMigrationVersionError) do
1253
        ActiveRecord::Migrator.migrate(MIGRATIONS_ROOT + "/duplicate", nil)
1254 1255
      end
    end
1256

1257
    def test_migrator_with_duplicate_names
1258
      assert_raise(ActiveRecord::DuplicateMigrationNameError, "Multiple migrations have the name Chunky") do
1259 1260 1261 1262
        ActiveRecord::Migrator.migrate(MIGRATIONS_ROOT + "/duplicate_names", nil)
      end
    end

1263
    def test_migrator_with_missing_version_numbers
1264 1265 1266
      assert_raise(ActiveRecord::UnknownMigrationVersionError) do
        ActiveRecord::Migrator.migrate(MIGRATIONS_ROOT + "/missing", 500)
      end
1267
    end
1268

1269 1270 1271 1272 1273
    def test_create_table_with_custom_sequence_name
      return unless current_adapter? :OracleAdapter

      # table name is 29 chars, the standard sequence name will
      # be 33 chars and fail
1274
      assert_raise(ActiveRecord::StatementInvalid) do
1275 1276 1277 1278 1279 1280 1281 1282 1283 1284 1285 1286 1287 1288 1289 1290 1291 1292 1293 1294 1295 1296 1297 1298 1299 1300
        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
        end
      end

      # 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

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

        ensure
          Person.connection.drop_table :table_with_name_thats_just_ok,
                                       :sequence_name => 'suitably_short_seq' rescue nil
        end
      end

      # confirm the custom sequence got dropped
1301
      assert_raise(ActiveRecord::StatementInvalid) do
1302 1303 1304
        Person.connection.execute("select suitably_short_seq.nextval from dual")
      end
    end
1305 1306 1307 1308 1309 1310 1311 1312 1313

    protected
      def with_env_tz(new_tz = 'US/Eastern')
        old_tz, ENV['TZ'] = ENV['TZ'], new_tz
        yield
      ensure
        old_tz ? ENV['TZ'] = old_tz : ENV.delete('TZ')
      end

1314
  end
1315

1316 1317 1318 1319 1320
  class SexyMigrationsTest < ActiveRecord::TestCase
    def test_references_column_type_adds_id
      with_new_table do |t|
        t.expects(:column).with('customer_id', :integer, {})
        t.references :customer
1321
      end
1322
    end
J
Jeremy Kemper 已提交
1323

1324 1325 1326 1327 1328
    def test_references_column_type_with_polymorphic_adds_type
      with_new_table do |t|
        t.expects(:column).with('taggable_type', :string, {})
        t.expects(:column).with('taggable_id', :integer, {})
        t.references :taggable, :polymorphic => true
1329
      end
1330
    end
J
Jeremy Kemper 已提交
1331

1332 1333 1334 1335 1336
    def test_references_column_type_with_polymorphic_and_options_null_is_false_adds_table_flag
      with_new_table do |t|
        t.expects(:column).with('taggable_type', :string, {:null => false})
        t.expects(:column).with('taggable_id', :integer, {:null => false})
        t.references :taggable, :polymorphic => true, :null => false
1337
      end
1338
    end
J
Jeremy Kemper 已提交
1339

1340 1341 1342 1343
    def test_belongs_to_works_like_references
      with_new_table do |t|
        t.expects(:column).with('customer_id', :integer, {})
        t.belongs_to :customer
1344
      end
1345
    end
J
Jeremy Kemper 已提交
1346

1347 1348 1349 1350 1351
    def test_timestamps_creates_updated_at_and_created_at
      with_new_table do |t|
        t.expects(:column).with(:created_at, :datetime, kind_of(Hash))
        t.expects(:column).with(:updated_at, :datetime, kind_of(Hash))
        t.timestamps
1352
      end
1353
    end
J
Jeremy Kemper 已提交
1354

1355 1356 1357 1358 1359
    def test_integer_creates_integer_column
      with_new_table do |t|
        t.expects(:column).with(:foo, 'integer', {})
        t.expects(:column).with(:bar, 'integer', {})
        t.integer :foo, :bar
1360
      end
1361
    end
J
Jeremy Kemper 已提交
1362

1363 1364 1365 1366 1367
    def test_string_creates_string_column
      with_new_table do |t|
        t.expects(:column).with(:foo, 'string', {})
        t.expects(:column).with(:bar, 'string', {})
        t.string :foo, :bar
1368
      end
1369
    end
J
Jeremy Kemper 已提交
1370

1371 1372 1373 1374 1375 1376 1377 1378 1379
    if current_adapter?(:PostgreSQLAdapter)
      def test_xml_creates_xml_column
        with_new_table do |t|
          t.expects(:column).with(:data, 'xml', {})
          t.xml :data
        end
      end
    end

1380 1381 1382 1383
    protected
    def with_new_table
      Person.connection.create_table :delete_me, :force => true do |t|
        yield t
1384
      end
1385 1386 1387
    ensure
      Person.connection.drop_table :delete_me rescue nil
    end
J
Jeremy Kemper 已提交
1388

1389
  end # SexyMigrationsTest
1390

1391 1392 1393 1394
  class ChangeTableMigrationsTest < ActiveRecord::TestCase
    def setup
      @connection = Person.connection
      @connection.create_table :delete_me, :force => true do |t|
1395
      end
1396
    end
1397

1398 1399 1400
    def teardown
      Person.connection.drop_table :delete_me rescue nil
    end
1401

1402 1403 1404 1405
    def test_references_column_type_adds_id
      with_change_table do |t|
        @connection.expects(:add_column).with(:delete_me, 'customer_id', :integer, {})
        t.references :customer
1406
      end
1407
    end
1408

1409 1410 1411 1412
    def test_remove_references_column_type_removes_id
      with_change_table do |t|
        @connection.expects(:remove_column).with(:delete_me, 'customer_id')
        t.remove_references :customer
1413
      end
1414
    end
1415

1416 1417 1418 1419
    def test_add_belongs_to_works_like_add_references
      with_change_table do |t|
        @connection.expects(:add_column).with(:delete_me, 'customer_id', :integer, {})
        t.belongs_to :customer
1420
      end
1421
    end
1422

1423 1424 1425 1426
    def test_remove_belongs_to_works_like_remove_references
      with_change_table do |t|
        @connection.expects(:remove_column).with(:delete_me, 'customer_id')
        t.remove_belongs_to :customer
1427
      end
1428
    end
1429

1430 1431 1432 1433 1434
    def test_references_column_type_with_polymorphic_adds_type
      with_change_table do |t|
        @connection.expects(:add_column).with(:delete_me, 'taggable_type', :string, {})
        @connection.expects(:add_column).with(:delete_me, 'taggable_id', :integer, {})
        t.references :taggable, :polymorphic => true
1435
      end
1436
    end
1437

1438 1439 1440 1441 1442
    def test_remove_references_column_type_with_polymorphic_removes_type
      with_change_table do |t|
        @connection.expects(:remove_column).with(:delete_me, 'taggable_type')
        @connection.expects(:remove_column).with(:delete_me, 'taggable_id')
        t.remove_references :taggable, :polymorphic => true
1443
      end
1444
    end
1445

1446 1447 1448 1449 1450
    def test_references_column_type_with_polymorphic_and_options_null_is_false_adds_table_flag
      with_change_table do |t|
        @connection.expects(:add_column).with(:delete_me, 'taggable_type', :string, {:null => false})
        @connection.expects(:add_column).with(:delete_me, 'taggable_id', :integer, {:null => false})
        t.references :taggable, :polymorphic => true, :null => false
1451
      end
1452
    end
1453

1454 1455 1456 1457 1458
    def test_remove_references_column_type_with_polymorphic_and_options_null_is_false_removes_table_flag
      with_change_table do |t|
        @connection.expects(:remove_column).with(:delete_me, 'taggable_type')
        @connection.expects(:remove_column).with(:delete_me, 'taggable_id')
        t.remove_references :taggable, :polymorphic => true, :null => false
1459
      end
1460
    end
1461

1462 1463 1464 1465
    def test_timestamps_creates_updated_at_and_created_at
      with_change_table do |t|
        @connection.expects(:add_timestamps).with(:delete_me)
        t.timestamps
1466
      end
1467
    end
1468

1469 1470 1471 1472
    def test_remove_timestamps_creates_updated_at_and_created_at
      with_change_table do |t|
        @connection.expects(:remove_timestamps).with(:delete_me)
        t.remove_timestamps
1473
      end
1474
    end
1475

1476 1477 1478
    def string_column
      if current_adapter?(:PostgreSQLAdapter)
        "character varying(255)"
1479 1480
      elsif current_adapter?(:OracleAdapter)
        'VARCHAR2(255)'
1481 1482
      else
        'varchar(255)'
1483
      end
1484
    end
1485

1486 1487 1488
    def integer_column
      if current_adapter?(:MysqlAdapter)
        'int(11)'
1489 1490
      elsif current_adapter?(:OracleAdapter)
        'NUMBER(38)'
1491 1492
      else
        'integer'
1493
      end
1494
    end
1495

1496 1497 1498 1499 1500
    def test_integer_creates_integer_column
      with_change_table do |t|
        @connection.expects(:add_column).with(:delete_me, :foo, integer_column, {})
        @connection.expects(:add_column).with(:delete_me, :bar, integer_column, {})
        t.integer :foo, :bar
1501
      end
1502
    end
1503

1504 1505 1506 1507 1508
    def test_string_creates_string_column
      with_change_table do |t|
        @connection.expects(:add_column).with(:delete_me, :foo, string_column, {})
        @connection.expects(:add_column).with(:delete_me, :bar, string_column, {})
        t.string :foo, :bar
1509
      end
1510
    end
1511

1512 1513 1514 1515
    def test_column_creates_column
      with_change_table do |t|
        @connection.expects(:add_column).with(:delete_me, :bar, :integer, {})
        t.column :bar, :integer
1516
      end
1517
    end
1518

1519 1520 1521 1522
    def test_column_creates_column_with_options
      with_change_table do |t|
        @connection.expects(:add_column).with(:delete_me, :bar, :integer, {:null => false})
        t.column :bar, :integer, :null => false
1523
      end
1524
    end
1525

1526 1527 1528 1529
    def test_index_creates_index
      with_change_table do |t|
        @connection.expects(:add_index).with(:delete_me, :bar, {})
        t.index :bar
1530
      end
1531
    end
1532

1533 1534 1535 1536
    def test_index_creates_index_with_options
      with_change_table do |t|
        @connection.expects(:add_index).with(:delete_me, :bar, {:unique => true})
        t.index :bar, :unique => true
1537
      end
1538
    end
1539

1540 1541 1542 1543
    def test_change_changes_column
      with_change_table do |t|
        @connection.expects(:change_column).with(:delete_me, :bar, :string, {})
        t.change :bar, :string
1544
      end
1545
    end
1546

1547 1548 1549 1550
    def test_change_changes_column_with_options
      with_change_table do |t|
        @connection.expects(:change_column).with(:delete_me, :bar, :string, {:null => true})
        t.change :bar, :string, :null => true
1551
      end
1552
    end
1553

1554 1555 1556 1557
    def test_change_default_changes_column
      with_change_table do |t|
        @connection.expects(:change_column_default).with(:delete_me, :bar, :string)
        t.change_default :bar, :string
1558
      end
1559
    end
1560

1561 1562 1563 1564
    def test_remove_drops_single_column
      with_change_table do |t|
        @connection.expects(:remove_column).with(:delete_me, [:bar])
        t.remove :bar
1565
      end
1566
    end
1567

1568 1569 1570 1571
    def test_remove_drops_multiple_columns
      with_change_table do |t|
        @connection.expects(:remove_column).with(:delete_me, [:bar, :baz])
        t.remove :bar, :baz
1572
      end
1573
    end
1574

1575 1576 1577 1578
    def test_remove_index_removes_index_with_options
      with_change_table do |t|
        @connection.expects(:remove_index).with(:delete_me, {:unique => true})
        t.remove_index :unique => true
1579
      end
1580
    end
1581

1582 1583 1584 1585
    def test_rename_renames_column
      with_change_table do |t|
        @connection.expects(:rename_column).with(:delete_me, :bar, :baz)
        t.rename :bar, :baz
1586
      end
1587
    end
1588

1589 1590 1591 1592
    protected
    def with_change_table
      Person.connection.change_table :delete_me do |t|
        yield t
1593
      end
1594 1595
    end
  end
1596
end
1597