migration_test.rb 77.7 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
  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)
33 34 35
      # Use shorter prefix and suffix as in Oracle database identifier cannot be larger than 30 characters
      ActiveRecord::Base.table_name_prefix = 'p_'
      ActiveRecord::Base.table_name_suffix = '_s'
36 37 38 39
      conn.drop_table(ActiveRecord::Migrator.schema_migrations_table_name) if conn.table_exists?(ActiveRecord::Migrator.schema_migrations_table_name)

      conn.initialize_schema_migrations_table

40
      assert_equal "p_unique_schema_migrations_s", conn.indexes(ActiveRecord::Migrator.schema_migrations_table_name)[0][:name]
41 42 43 44 45 46
    ensure
      ActiveRecord::Base.table_name_prefix = ""
      ActiveRecord::Base.table_name_suffix = ""
    end
  end

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

50
    fixtures :people
J
Jamis Buck 已提交
51

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

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

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

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

76
    def test_add_index
77 78 79
      # 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
80
      Person.connection.add_column "people", "administrator", :boolean
81

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

85
      # Orcl nds shrt indx nms.  Sybs 2.
86
      # OpenBase does not have named indexes.  You must specify a single column name
87
      unless current_adapter?(:SybaseAdapter, :OpenBaseAdapter)
88 89
        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"]) }
90 91 92 93 94 95 96 97
        # Oracle adapter cannot have specified index name larger than 30 characters
        # Oracle adapter is shortening index name when just column list is given
        unless current_adapter?(:OracleAdapter)
          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") }
        end
98 99
        assert_nothing_raised { Person.connection.add_index("people", ["last_name", "first_name"]) }
        assert_nothing_raised { Person.connection.remove_index("people", ["last_name", "first_name"]) }
100 101 102 103 104 105 106 107
        assert_nothing_raised { Person.connection.add_index("people", ["last_name"], :length => 10) }
        assert_nothing_raised { Person.connection.remove_index("people", "last_name") }
        assert_nothing_raised { Person.connection.add_index("people", ["last_name"], :length => {:last_name => 10}) }
        assert_nothing_raised { Person.connection.remove_index("people", ["last_name"]) }
        assert_nothing_raised { Person.connection.add_index("people", ["last_name", "first_name"], :length => 10) }
        assert_nothing_raised { Person.connection.remove_index("people", ["last_name", "first_name"]) }
        assert_nothing_raised { Person.connection.add_index("people", ["last_name", "first_name"], :length => {:last_name => 10, :first_name => 20}) }
        assert_nothing_raised { Person.connection.remove_index("people", ["last_name", "first_name"]) }
108
      end
109

110
      # quoting
111
      # Note: changed index name from "key" to "key_idx" since "key" is a Firebird reserved word
112 113
      # OpenBase does not have named indexes.  You must specify a single column name
      unless current_adapter?(:OpenBaseAdapter)
114
        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)
115 116 117
        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 已提交
118

119
      # Sybase adapter does not support indexes on :boolean columns
120 121
      # OpenBase does not have named indexes.  You must specify a single column
      unless current_adapter?(:SybaseAdapter, :OpenBaseAdapter)
122 123 124
        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
125
    end
126

127 128 129
    def test_add_index_length_limit
      good_index_name = 'x' * Person.connection.index_name_length
      too_long_index_name = good_index_name + 'x'
130
      assert_raise(ArgumentError)  { Person.connection.add_index("people", "first_name", :name => too_long_index_name) }
131
      assert !Person.connection.index_name_exists?("people", too_long_index_name, false)
132
      assert_nothing_raised { Person.connection.add_index("people", "first_name", :name => good_index_name) }
133
      assert Person.connection.index_name_exists?("people", good_index_name, false)
134
      Person.connection.remove_index("people", :name => good_index_name)
135 136 137 138 139
    end

    def test_remove_nonexistent_index
      # we do this by name, so OpenBase is a wash as noted above
      unless current_adapter?(:OpenBaseAdapter)
140
        assert_raise(ArgumentError) { Person.connection.remove_index("people", "no_such_index") }
141 142 143 144 145 146 147 148 149
      end
    end

    def test_rename_index
      unless current_adapter?(:OpenBaseAdapter)
        # keep the names short to make Oracle and similar behave
        Person.connection.add_index('people', [:first_name], :name => 'old_idx')
        assert_nothing_raised { Person.connection.rename_index('people', 'old_idx', 'new_idx') }
        # if the adapter doesn't support the indexes call, pick defaults that let the test pass
150 151
        assert !Person.connection.index_name_exists?('people', 'old_idx', false)
        assert Person.connection.index_name_exists?('people', 'new_idx', true)
152 153 154 155 156 157
      end
    end

    def test_double_add_index
      unless current_adapter?(:OpenBaseAdapter)
        Person.connection.add_index('people', [:first_name], :name => 'some_idx')
158
        assert_raise(ArgumentError) { Person.connection.add_index('people', [:first_name], :name => 'some_idx') }
159 160 161
      end
    end

162 163 164 165 166 167 168 169 170 171 172 173 174 175 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
    def test_index_exists
      Person.connection.create_table :testings do |t|
        t.column :foo, :string, :limit => 100
        t.column :bar, :string, :limit => 100
      end
      Person.connection.add_index :testings, :foo

      assert Person.connection.index_exists?(:testings, :foo)
      assert !Person.connection.index_exists?(:testings, :bar)
    ensure
      Person.connection.drop_table :testings rescue nil
    end

    def test_index_exists_on_multiple_columns
      Person.connection.create_table :testings do |t|
        t.column :foo, :string, :limit => 100
        t.column :bar, :string, :limit => 100
      end
      Person.connection.add_index :testings, [:foo, :bar]

      assert Person.connection.index_exists?(:testings, [:foo, :bar])
    ensure
      Person.connection.drop_table :testings rescue nil
    end

    def test_unique_index_exists
      Person.connection.create_table :testings do |t|
        t.column :foo, :string, :limit => 100
      end
      Person.connection.add_index :testings, :foo, :unique => true

      assert Person.connection.index_exists?(:testings, :foo, :unique => true)
    ensure
      Person.connection.drop_table :testings rescue nil
    end

    def test_named_index_exists
      Person.connection.create_table :testings do |t|
        t.column :foo, :string, :limit => 100
      end
      Person.connection.add_index :testings, :foo, :name => "custom_index_name"

      assert Person.connection.index_exists?(:testings, :foo, :name => "custom_index_name")
    ensure
      Person.connection.drop_table :testings rescue nil
    end

209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232
    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

233 234 235 236 237 238 239 240 241 242
    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
243 244

    def test_create_table_with_not_null_column
245 246 247 248
      assert_nothing_raised do
        Person.connection.create_table :testings do |t|
          t.column :foo, :string, :null => false
        end
249 250
      end

251
      assert_raise(ActiveRecord::StatementInvalid) do
252 253 254 255 256
        Person.connection.execute "insert into testings (foo) values (NULL)"
      end
    ensure
      Person.connection.drop_table :testings rescue nil
    end
257 258

    def test_create_table_with_defaults
259
      # MySQL doesn't allow defaults on TEXT or BLOB columns.
B
Brian Lopez 已提交
260
      mysql = current_adapter?(:MysqlAdapter) || current_adapter?(:Mysql2Adapter)
261

262 263 264 265 266
      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
267
        t.column :five, :text, :default => "hello" unless mysql
268 269 270 271 272 273 274
      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" }
275
      five = columns.detect { |c| c.name == "five" } unless mysql
276 277

      assert_equal "hello", one.default
278 279
      assert_equal true, two.default
      assert_equal false, three.default
280
      assert_equal 1, four.default
281
      assert_equal "hello", five.default unless mysql
282 283 284 285

    ensure
      Person.connection.drop_table :testings rescue nil
    end
286 287

    def test_create_table_with_limits
288 289 290
      assert_nothing_raised do
        Person.connection.create_table :testings do |t|
          t.column :foo, :string, :limit => 255
291

292
          t.column :default_int, :integer
293

294 295 296 297
          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
298
        end
299 300 301 302 303 304 305 306 307 308
      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"   }
309
      eleven  = columns.detect { |c| c.name == "eleven_int"   }
310 311 312 313 314 315

      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
316
        assert_equal 'integer', eleven.sql_type
B
Brian Lopez 已提交
317
      elsif current_adapter?(:MysqlAdapter) or current_adapter?(:Mysql2Adapter)
318 319 320 321 322
        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
323 324 325 326 327
      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
328 329 330 331
      end
    ensure
      Person.connection.drop_table :testings rescue nil
    end
332

333 334 335 336 337 338 339
    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

340
      assert_equal %w(foo testing_id), Person.connection.columns(:testings).map { |c| c.name }.sort
341 342 343 344 345 346 347 348 349 350 351 352
    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

353
      assert_equal %w(foo testingid), Person.connection.columns(:testings).map { |c| c.name }.sort
354 355 356 357 358
    ensure
      Person.connection.drop_table :testings rescue nil
      ActiveRecord::Base.primary_key_prefix_type = nil
    end

359 360 361 362
    def test_create_table_with_force_true_does_not_drop_nonexisting_table
      if Person.connection.table_exists?(:testings2)
        Person.connection.drop_table :testings2
      end
363

364 365 366 367 368 369
      # 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
370
      end
371 372
    ensure
      Person.connection.drop_table :testings2 rescue nil
373 374
    end

375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407
    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
408

409 410 411 412 413 414 415
    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

416
    # Sybase, and SQLite3 will not allow you to add a NOT NULL
417
    # column to a table without a default value.
418
    unless current_adapter?(:SybaseAdapter, :SQLiteAdapter)
419 420 421 422 423
      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
424

425
        assert_raise(ActiveRecord::StatementInvalid) do
426 427 428 429 430 431
          Person.connection.execute "insert into testings (foo, bar) values ('hello', NULL)"
        end
      ensure
        Person.connection.drop_table :testings rescue nil
      end
    end
432

433
    def test_add_column_not_null_with_default
434
      Person.connection.create_table :testings do |t|
435 436
        t.column :foo, :string
      end
J
Jeremy Kemper 已提交
437 438

      con = Person.connection
439
      Person.connection.enable_identity_insert("testings", true) if current_adapter?(:SybaseAdapter)
440
      Person.connection.execute "insert into testings (#{con.quote_column_name('id')}, #{con.quote_column_name('foo')}) values (1, 'hello')"
441
      Person.connection.enable_identity_insert("testings", false) if current_adapter?(:SybaseAdapter)
442
      assert_nothing_raised {Person.connection.add_column :testings, :bar, :string, :null => false, :default => "default" }
443

444
      assert_raise(ActiveRecord::StatementInvalid) do
445 446 447 448 449 450
        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
451 452 453 454
      end
    ensure
      Person.connection.drop_table :testings rescue nil
    end
455

456 457 458 459
    # 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
460
      correct_value = '0012345678901234567890.0123456789'.to_d
461 462 463 464 465 466

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

      # Do a manual insertion
467 468
      if current_adapter?(:OracleAdapter)
        Person.connection.execute "insert into people (id, wealth) values (people_seq.nextval, 12345678901234567890.0123456789)"
469
      elsif current_adapter?(:OpenBaseAdapter) || (current_adapter?(:MysqlAdapter) && Mysql.client_version < 50003) #before mysql 5.0.3 decimals stored as strings
470
        Person.connection.execute "insert into people (wealth) values ('12345678901234567890.0123456789')"
471 472 473
      else
        Person.connection.execute "insert into people (wealth) values (12345678901234567890.0123456789)"
      end
474 475 476 477 478 479

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

      # If this assert fails, that means the SELECT is broken!
480 481 482
      unless current_adapter?(:SQLite3Adapter)
        assert_equal correct_value, row.wealth
      end
483 484 485 486 487 488 489 490 491 492 493 494

      # 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!
495 496 497
      unless current_adapter?(:SQLite3Adapter)
        assert_equal correct_value, row.wealth
      end
498 499 500 501 502 503

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

504 505 506 507 508 509 510 511
    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
512

513 514 515 516 517 518
    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
519
      Person.connection.add_column "people", "wealth", :decimal, :precision => '30', :scale => '10'
520 521
      Person.connection.add_column "people", "birthday", :datetime
      Person.connection.add_column "people", "favorite_day", :date
522
      Person.connection.add_column "people", "moment_of_truth", :datetime
523
      Person.connection.add_column "people", "male", :boolean
524
      Person.reset_column_information
525

526 527 528 529 530
      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,
531
          :moment_of_truth => "1782-10-10 21:40:18", :male => true
532 533 534
      end

      bob = Person.find(:first)
535 536 537 538 539 540 541
      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 已提交
542

543 544 545
      unless current_adapter?(:SQLite3Adapter)
        assert_equal BigDecimal.new("0012345678901234567890.0123456789"), bob.wealth
      end
J
Jeremy Kemper 已提交
546

547
      assert_equal true, bob.male?
548

549 550 551 552 553
      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
554

555
      if current_adapter?(:OracleAdapter, :SybaseAdapter)
556
        # Sybase, and Oracle don't differentiate between date/time
557 558 559 560 561
        assert_equal Time, bob.favorite_day.class
      else
        assert_equal Date, bob.favorite_day.class
      end

562 563 564 565 566 567 568 569
      # 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
570
            bob.reload
571 572 573 574 575
            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
576
            assert_match(/\A-05:?00\Z/, bob.moment_of_truth.zone) #ruby 1.8.6 uses HH:MM, prior versions use HHMM
577 578
            assert_equal DateTime::ITALY, bob.moment_of_truth.start
          end
579
        end
580
      end
581

582
      assert_instance_of TrueClass, bob.male?
583
      assert_kind_of BigDecimal, bob.wealth
584 585
    end

B
Brian Lopez 已提交
586
    if current_adapter?(:MysqlAdapter) or current_adapter?(:Mysql2Adapter)
587 588 589 590
      def test_unabstracted_database_dependent_types
        Person.delete_all

        ActiveRecord::Migration.add_column :people, :intelligence_quotient, :tinyint
591
        Person.reset_column_information
592
        assert_match(/tinyint/, Person.columns_hash['intelligence_quotient'].sql_type)
593
      ensure
594
        ActiveRecord::Migration.remove_column :people, :intelligence_quotient rescue nil
595 596 597
      end
    end

598
    def test_add_remove_single_field_using_string_arguments
599 600
      assert !Person.column_methods_hash.include?(:last_name)

601
      ActiveRecord::Migration.add_column 'people', 'last_name', :string
602 603 604

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

606 607 608 609 610 611 612 613 614 615 616 617 618 619 620
      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
621 622 623 624

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

B
Brian Lopez 已提交
626
    if current_adapter?(:MysqlAdapter) or current_adapter?(:Mysql2Adapter)
627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655 656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672
      def testing_table_for_positioning
        Person.connection.create_table :testings, :id => false do |t|
          t.column :first, :integer
          t.column :second, :integer
          t.column :third, :integer
        end

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

      def test_column_positioning
        testing_table_for_positioning do |conn|
          assert_equal %w(first second third), conn.columns(:testings).map {|c| c.name }
        end
      end

      def test_add_column_with_positioning
        testing_table_for_positioning do |conn|
          conn.add_column :testings, :new_col, :integer
          assert_equal %w(first second third new_col), conn.columns(:testings).map {|c| c.name }
        end
        testing_table_for_positioning do |conn|
          conn.add_column :testings, :new_col, :integer, :first => true
          assert_equal %w(new_col first second third), conn.columns(:testings).map {|c| c.name }
        end
        testing_table_for_positioning do |conn|
          conn.add_column :testings, :new_col, :integer, :after => :first
          assert_equal %w(first new_col second third), conn.columns(:testings).map {|c| c.name }
        end
      end

      def test_change_column_with_positioning
        testing_table_for_positioning do |conn|
          conn.change_column :testings, :second, :integer, :first => true
          assert_equal %w(second first third), conn.columns(:testings).map {|c| c.name }
        end
        testing_table_for_positioning do |conn|
          conn.change_column :testings, :second, :integer, :after => :third
          assert_equal %w(first third second), conn.columns(:testings).map {|c| c.name }
        end
      end
    end

673 674
    def test_add_rename
      Person.delete_all
675

676
      begin
677
        Person.connection.add_column "people", "girlfriend", :string
678
        Person.reset_column_information
679 680
        Person.create :girlfriend => 'bobette'

681
        Person.connection.rename_column "people", "girlfriend", "exgirlfriend"
682 683

        Person.reset_column_information
684
        bob = Person.find(:first)
685

686 687 688 689 690
        assert_equal "bobette", bob.exgirlfriend
      ensure
        Person.connection.remove_column("people", "girlfriend") rescue nil
        Person.connection.remove_column("people", "exgirlfriend") rescue nil
      end
691

692
    end
693

694 695
    def test_rename_column_using_symbol_arguments
      begin
696
        names_before = Person.find(:all).map(&:first_name)
697 698 699
        Person.connection.rename_column :people, :first_name, :nick_name
        Person.reset_column_information
        assert Person.column_names.include?("nick_name")
700
        assert_equal names_before, Person.find(:all).map(&:nick_name)
701 702 703 704 705
      ensure
        Person.connection.remove_column("people","nick_name")
        Person.connection.add_column("people","first_name", :string)
      end
    end
706

707 708
    def test_rename_column
      begin
709
        names_before = Person.find(:all).map(&:first_name)
710 711 712
        Person.connection.rename_column "people", "first_name", "nick_name"
        Person.reset_column_information
        assert Person.column_names.include?("nick_name")
713
        assert_equal names_before, Person.find(:all).map(&:nick_name)
714 715 716 717 718
      ensure
        Person.connection.remove_column("people","nick_name")
        Person.connection.add_column("people","first_name", :string)
      end
    end
719 720 721 722 723 724 725 726 727 728 729 730 731 732 733 734 735 736 737 738

    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
739
      exception = if current_adapter?(:PostgreSQLAdapter, :OracleAdapter)
740 741 742 743
        ActiveRecord::StatementInvalid
      else
        ActiveRecord::ActiveRecordError
      end
744
      assert_raise(exception) do
745 746 747 748 749
        Person.connection.rename_column "hats", "nonexistent", "should_fail"
      end
    ensure
      ActiveRecord::Base.connection.drop_table(:hats)
    end
750

751 752 753 754 755 756 757 758 759 760
    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 已提交
761

762 763 764 765 766
    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
767
      Person.connection.add_index :hats, :hat_name
768 769 770 771 772 773 774 775 776 777 778 779 780 781 782 783 784 785 786 787 788 789 790 791 792
      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
793
      ActiveRecord::Base.connection.add_index "hats", ["hat_style", "hat_size"], :unique => true
794 795 796 797 798 799

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

800 801 802 803
    def test_remove_column_no_second_parameter_raises_exception
      assert_raise(ArgumentError) { Person.connection.remove_column("funny") }
    end

804 805 806 807
    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 已提交
808

809 810
        Topic.connection.change_column "topics", "written_on", :datetime, :null => false
        Topic.reset_column_information
811 812 813

        Topic.connection.change_column "topics", "written_on", :datetime, :null => true
        Topic.reset_column_information
814 815
      end
    end
816

817 818 819 820 821 822 823 824 825 826 827 828 829 830 831 832 833 834 835 836 837 838 839 840 841 842
    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

843 844 845 846
    def test_rename_table
      begin
        ActiveRecord::Base.connection.create_table :octopuses do |t|
          t.column :url, :string
847
        end
848
        ActiveRecord::Base.connection.rename_table :octopuses, :octopi
849

850
        # Using explicit id in insert for compatibility across all databases
J
Jeremy Kemper 已提交
851
        con = ActiveRecord::Base.connection
852
        con.enable_identity_insert("octopi", true) if current_adapter?(:SybaseAdapter)
853
        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')" }
854
        con.enable_identity_insert("octopi", false) if current_adapter?(:SybaseAdapter)
855 856

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

858 859 860 861 862
      ensure
        ActiveRecord::Base.connection.drop_table :octopuses rescue nil
        ActiveRecord::Base.connection.drop_table :octopi rescue nil
      end
    end
J
Jeremy Kemper 已提交
863

864
    def test_change_column_nullability
J
Jeremy Kemper 已提交
865
      Person.delete_all
866 867 868 869 870 871 872 873 874 875
      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
876

877 878 879 880 881 882
    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 已提交
883

884 885 886
        ActiveRecord::Base.connection.rename_table :octopuses, :octopi

        # Using explicit id in insert for compatibility across all databases
J
Jeremy Kemper 已提交
887
        con = ActiveRecord::Base.connection
888
        con.enable_identity_insert("octopi", true) if current_adapter?(:SybaseAdapter)
889
        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')" }
890
        con.enable_identity_insert("octopi", false) if current_adapter?(:SybaseAdapter)
891 892 893 894 895 896 897 898 899

        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

900
    def test_change_column
901
      Person.connection.add_column 'people', 'age', :integer
902 903
      label = "test_change_column Columns"
      old_columns = Person.connection.columns(Person.table_name, label)
904 905 906
      assert old_columns.find { |c| c.name == 'age' and c.type == :integer }

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

908
      new_columns = Person.connection.columns(Person.table_name, label)
909 910
      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 }
911

912
      old_columns = Topic.connection.columns(Topic.table_name, label)
913 914
      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 }
915
      new_columns = Topic.connection.columns(Topic.table_name, label)
916 917
      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 }
918
      assert_nothing_raised { Topic.connection.change_column :topics, :approved, :boolean, :default => true }
919
    end
J
Jeremy Kemper 已提交
920

921 922 923 924
    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 已提交
925

926 927 928 929
      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
930 931
    ensure
      Person.connection.remove_column("people", "contributor") rescue nil
932
    end
933 934

    def test_change_column_with_new_default
935
      Person.connection.add_column "people", "administrator", :boolean, :default => true
936
      Person.reset_column_information
937
      assert Person.new.administrator?
938

939
      assert_nothing_raised { Person.connection.change_column "people", "administrator", :boolean, :default => false }
940
      Person.reset_column_information
941
      assert !Person.new.administrator?
942 943
    ensure
      Person.connection.remove_column("people", "administrator") rescue nil
944
    end
J
Jeremy Kemper 已提交
945

946 947 948 949 950
    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
951 952 953 954 955 956 957 958

    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 }

959 960 961 962 963 964
      # 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
965 966 967 968
    ensure
      Person.connection.drop_table :testings rescue nil
    end

969 970 971 972 973 974 975 976 977 978 979
    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
980 981 982 983 984 985
      # 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
986 987 988 989 990 991 992 993 994 995 996 997 998 999 1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010 1011 1012 1013 1014 1015 1016 1017 1018 1019 1020 1021 1022

      # 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

1023 1024 1025 1026 1027
    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
1028

1029 1030 1031 1032 1033 1034 1035 1036 1037 1038 1039 1040 1041 1042 1043 1044 1045 1046 1047 1048 1049 1050 1051 1052 1053 1054 1055 1056 1057 1058 1059 1060 1061 1062 1063 1064 1065 1066 1067
    def test_column_exists
      Person.connection.create_table :testings do |t|
        t.column :foo, :string
      end

      assert Person.connection.column_exists?(:testings, :foo)
      assert !Person.connection.column_exists?(:testings, :bar)
    ensure
      Person.connection.drop_table :testings rescue nil
    end

    def test_column_exists_with_type
      Person.connection.create_table :testings do |t|
        t.column :foo, :string
        t.column :bar, :decimal, :precision => 8, :scale => 2
      end

      assert Person.connection.column_exists?(:testings, :foo, :string)
      assert !Person.connection.column_exists?(:testings, :foo, :integer)
      assert Person.connection.column_exists?(:testings, :bar, :decimal)
      assert !Person.connection.column_exists?(:testings, :bar, :integer)
    ensure
      Person.connection.drop_table :testings rescue nil
    end

    def test_column_exists_with_definition
      Person.connection.create_table :testings do |t|
        t.column :foo, :string, :limit => 100
        t.column :bar, :decimal, :precision => 8, :scale => 2
      end

      assert Person.connection.column_exists?(:testings, :foo, :string, :limit => 100)
      assert !Person.connection.column_exists?(:testings, :foo, :string, :limit => 50)
      assert Person.connection.column_exists?(:testings, :bar, :decimal, :precision => 8, :scale => 2)
      assert !Person.connection.column_exists?(:testings, :bar, :decimal, :precision => 10, :scale => 2)
    ensure
      Person.connection.drop_table :testings rescue nil
    end

1068
    def test_add_table
1069
      assert !Reminder.table_exists?
1070

1071
      WeNeedReminders.up
J
Jeremy Kemper 已提交
1072 1073

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

1076
      WeNeedReminders.down
1077
      assert_raise(ActiveRecord::StatementInvalid) { Reminder.find(:first) }
1078 1079
    end

1080 1081 1082 1083 1084 1085 1086 1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098 1099 1100 1101 1102 1103 1104 1105 1106 1107 1108 1109 1110 1111 1112 1113 1114
    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
1115
      # precision/scale explicitly left out.  By the SQL standard, numbers
1116
      # assigned to this field should be truncated but that's seldom respected.
1117
      if current_adapter?(:PostgreSQLAdapter)
1118 1119 1120 1121 1122 1123 1124 1125 1126 1127 1128 1129
        # - 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
1130
        assert_in_delta BigDecimal("2.71828182845905"), b.value_of_e, 0.00000000000001
1131 1132 1133 1134 1135 1136 1137
      else
        # - SQL standard is an integer
        assert_kind_of Fixnum, b.value_of_e
        assert_equal 2, b.value_of_e
      end

      GiveMeBigNumbers.down
1138
      assert_raise(ActiveRecord::StatementInvalid) { BigNumber.find(:first) }
1139 1140
    end

1141 1142
    def test_migrator
      assert !Person.column_methods_hash.include?(:last_name)
1143
      assert !Reminder.table_exists?
1144

1145
      ActiveRecord::Migrator.up(MIGRATIONS_ROOT + "/valid")
1146

1147
      assert_equal 3, ActiveRecord::Migrator.current_version
1148 1149 1150 1151 1152
      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

1153
      ActiveRecord::Migrator.down(MIGRATIONS_ROOT + "/valid")
1154 1155 1156 1157

      assert_equal 0, ActiveRecord::Migrator.current_version
      Person.reset_column_information
      assert !Person.column_methods_hash.include?(:last_name)
1158
      assert_raise(ActiveRecord::StatementInvalid) { Reminder.find(:first) }
1159 1160 1161 1162
    end

    def test_migrator_one_up
      assert !Person.column_methods_hash.include?(:last_name)
1163
      assert !Reminder.table_exists?
1164

1165
      ActiveRecord::Migrator.up(MIGRATIONS_ROOT + "/valid", 1)
1166 1167 1168

      Person.reset_column_information
      assert Person.column_methods_hash.include?(:last_name)
1169
      assert !Reminder.table_exists?
1170

1171
      ActiveRecord::Migrator.up(MIGRATIONS_ROOT + "/valid", 2)
1172 1173 1174 1175

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

1177
    def test_migrator_one_down
1178
      ActiveRecord::Migrator.up(MIGRATIONS_ROOT + "/valid")
1179

1180
      ActiveRecord::Migrator.down(MIGRATIONS_ROOT + "/valid", 1)
1181

1182 1183
      Person.reset_column_information
      assert Person.column_methods_hash.include?(:last_name)
1184
      assert !Reminder.table_exists?
1185
    end
1186

1187
    def test_migrator_one_up_one_down
1188 1189
      ActiveRecord::Migrator.up(MIGRATIONS_ROOT + "/valid", 1)
      ActiveRecord::Migrator.down(MIGRATIONS_ROOT + "/valid", 0)
1190 1191

      assert !Person.column_methods_hash.include?(:last_name)
1192
      assert !Reminder.table_exists?
1193
    end
1194

1195 1196 1197 1198 1199 1200 1201 1202 1203 1204 1205 1206 1207 1208 1209
    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

1210
    if ActiveRecord::Base.connection.supports_ddl_transactions?
1211 1212 1213
      def test_migrator_one_up_with_exception_and_rollback
        assert !Person.column_methods_hash.include?(:last_name)

1214
        e = assert_raise(StandardError) do
1215 1216 1217 1218 1219 1220 1221 1222 1223 1224
          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

1225 1226
    def test_finds_migrations
      migrations = ActiveRecord::Migrator.new(:up, MIGRATIONS_ROOT + "/valid").migrations
1227 1228 1229 1230

      [[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
1231 1232 1233 1234 1235 1236
      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
1237

1238
      assert_equal 1, migrations.size
1239 1240
      assert_equal migrations[0].version, 3
      assert_equal migrations[0].name, 'InnocentJointable'
1241 1242
    end

1243 1244 1245 1246 1247 1248 1249 1250 1251 1252 1253 1254 1255
    def test_relative_migrations
      $".delete_if do |fname|
        fname == (MIGRATIONS_ROOT + "/valid/1_people_have_last_names.rb")
      end
      Object.send(:remove_const, :PeopleHaveLastNames)

      Dir.chdir(MIGRATIONS_ROOT) do
        ActiveRecord::Migrator.up("valid/", 1)
      end

      assert defined?(PeopleHaveLastNames)
    end

1256 1257 1258 1259 1260
    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
1261
      Object.send(:remove_const, :PeopleHaveLastNames)
1262 1263 1264 1265 1266 1267 1268 1269 1270 1271 1272 1273 1274

      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

1275 1276 1277 1278 1279 1280 1281 1282 1283 1284 1285 1286 1287 1288 1289 1290 1291 1292 1293
    def test_target_version_zero_should_run_only_once
      # migrate up to 1
      ActiveRecord::Migrator.migrate(MIGRATIONS_ROOT + "/valid", 1)

      # migrate down to 0
      ActiveRecord::Migrator.migrate(MIGRATIONS_ROOT + "/valid", 0)

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

      # migrate down to 0 again
      ActiveRecord::Migrator.migrate(MIGRATIONS_ROOT + "/valid", 0)

      assert !defined? PeopleHaveLastNames
    ensure
      load(MIGRATIONS_ROOT + "/valid/1_people_have_last_names.rb")
    end

1294
    def test_migrator_db_has_no_schema_migrations_table
1295 1296 1297 1298 1299 1300
      # 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
1301 1302 1303 1304 1305
      assert_nothing_raised do
        ActiveRecord::Migrator.migrate(MIGRATIONS_ROOT + "/valid", 1)
      end
    end

J
Jamis Buck 已提交
1306
    def test_migrator_verbosity
1307
      ActiveRecord::Migrator.up(MIGRATIONS_ROOT + "/valid", 1)
J
Jamis Buck 已提交
1308 1309 1310
      assert PeopleHaveLastNames.message_count > 0
      PeopleHaveLastNames.message_count = 0

1311
      ActiveRecord::Migrator.down(MIGRATIONS_ROOT + "/valid", 0)
J
Jamis Buck 已提交
1312 1313 1314
      assert PeopleHaveLastNames.message_count > 0
      PeopleHaveLastNames.message_count = 0
    end
1315

J
Jamis Buck 已提交
1316 1317
    def test_migrator_verbosity_off
      PeopleHaveLastNames.verbose = false
1318
      ActiveRecord::Migrator.up(MIGRATIONS_ROOT + "/valid", 1)
J
Jamis Buck 已提交
1319
      assert PeopleHaveLastNames.message_count.zero?
1320
      ActiveRecord::Migrator.down(MIGRATIONS_ROOT + "/valid", 0)
J
Jamis Buck 已提交
1321 1322
      assert PeopleHaveLastNames.message_count.zero?
    end
1323

1324
    def test_migrator_going_down_due_to_version_target
1325 1326
      ActiveRecord::Migrator.up(MIGRATIONS_ROOT + "/valid", 1)
      ActiveRecord::Migrator.migrate(MIGRATIONS_ROOT + "/valid", 0)
1327 1328

      assert !Person.column_methods_hash.include?(:last_name)
1329
      assert !Reminder.table_exists?
1330

1331
      ActiveRecord::Migrator.migrate(MIGRATIONS_ROOT + "/valid")
1332 1333 1334 1335 1336 1337

      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
1338

1339 1340 1341
    def test_migrator_rollback
      ActiveRecord::Migrator.migrate(MIGRATIONS_ROOT + "/valid")
      assert_equal(3, ActiveRecord::Migrator.current_version)
1342

1343
      ActiveRecord::Migrator.rollback(MIGRATIONS_ROOT + "/valid")
1344
      assert_equal(2, ActiveRecord::Migrator.current_version)
1345

1346
      ActiveRecord::Migrator.rollback(MIGRATIONS_ROOT + "/valid")
1347
      assert_equal(1, ActiveRecord::Migrator.current_version)
1348

1349
      ActiveRecord::Migrator.rollback(MIGRATIONS_ROOT + "/valid")
1350
      assert_equal(0, ActiveRecord::Migrator.current_version)
1351

1352
      ActiveRecord::Migrator.rollback(MIGRATIONS_ROOT + "/valid")
1353
      assert_equal(0, ActiveRecord::Migrator.current_version)
1354
    end
1355

1356 1357 1358 1359 1360 1361 1362 1363 1364 1365
    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
1366

1367 1368
    def test_get_all_versions
      ActiveRecord::Migrator.migrate(MIGRATIONS_ROOT + "/valid")
1369
      assert_equal([1,2,3], ActiveRecord::Migrator.get_all_versions)
1370 1371 1372 1373 1374 1375 1376 1377 1378 1379

      ActiveRecord::Migrator.rollback(MIGRATIONS_ROOT + "/valid")
      assert_equal([1,2], ActiveRecord::Migrator.get_all_versions)

      ActiveRecord::Migrator.rollback(MIGRATIONS_ROOT + "/valid")
      assert_equal([1], ActiveRecord::Migrator.get_all_versions)

      ActiveRecord::Migrator.rollback(MIGRATIONS_ROOT + "/valid")
      assert_equal([], ActiveRecord::Migrator.get_all_versions)
    end
1380

1381
    def test_schema_migrations_table_name
1382 1383
      ActiveRecord::Base.table_name_prefix = "prefix_"
      ActiveRecord::Base.table_name_suffix = "_suffix"
1384
      Reminder.reset_table_name
1385
      assert_equal "prefix_schema_migrations_suffix", ActiveRecord::Migrator.schema_migrations_table_name
1386 1387
      ActiveRecord::Base.table_name_prefix = ""
      ActiveRecord::Base.table_name_suffix = ""
1388
      Reminder.reset_table_name
1389
      assert_equal "schema_migrations", ActiveRecord::Migrator.schema_migrations_table_name
1390 1391 1392
    ensure
      ActiveRecord::Base.table_name_prefix = ""
      ActiveRecord::Base.table_name_suffix = ""
1393
    end
1394

1395 1396 1397 1398
    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)
1399
      Reminder.reset_table_name
1400
      assert_equal Reminder.table_name, ActiveRecord::Migrator.proper_table_name(Reminder)
1401

1402 1403 1404 1405 1406
      # 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'
1407
      Reminder.reset_table_name
1408 1409 1410
      assert_equal "prefix_reminders_suffix", ActiveRecord::Migrator.proper_table_name(Reminder)
      Reminder.table_name_prefix = ''
      Reminder.table_name_suffix = ''
1411
      Reminder.reset_table_name
1412 1413

      # Use AR::Base's prefix/suffix if string or symbol is given
1414 1415
      ActiveRecord::Base.table_name_prefix = "prefix_"
      ActiveRecord::Base.table_name_suffix = "_suffix"
1416
      Reminder.reset_table_name
1417 1418 1419 1420
      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 = ""
1421
      Reminder.reset_table_name
1422
    end
1423 1424

    def test_add_drop_table_with_prefix_and_suffix
1425
      assert !Reminder.table_exists?
1426 1427
      ActiveRecord::Base.table_name_prefix = 'prefix_'
      ActiveRecord::Base.table_name_suffix = '_suffix'
1428
      Reminder.reset_table_name
1429
      Reminder.reset_sequence_name
1430 1431 1432 1433 1434
      WeNeedReminders.up
      assert Reminder.create("content" => "hello world", "remind_at" => Time.now)
      assert_equal "hello world", Reminder.find(:first).content

      WeNeedReminders.down
1435
      assert_raise(ActiveRecord::StatementInvalid) { Reminder.find(:first) }
1436
    ensure
1437 1438
      ActiveRecord::Base.table_name_prefix = ''
      ActiveRecord::Base.table_name_suffix = ''
1439
      Reminder.reset_table_name
1440
      Reminder.reset_sequence_name
1441
    end
1442

1443 1444
    def test_create_table_with_binary_column
      Person.connection.drop_table :binary_testings rescue nil
1445

1446 1447 1448 1449 1450
      assert_nothing_raised {
        Person.connection.create_table :binary_testings do |t|
          t.column "data", :binary, :null => false
        end
      }
1451

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

B
Brian Lopez 已提交
1455
      if current_adapter?(:MysqlAdapter) or current_adapter?(:Mysql2Adapter)
1456 1457 1458 1459
        assert_equal '', data_column.default
      else
        assert_nil data_column.default
      end
1460

1461
      Person.connection.drop_table :binary_testings rescue nil
1462
    end
1463

1464
    def test_migrator_with_duplicates
1465
      assert_raise(ActiveRecord::DuplicateMigrationVersionError) do
1466
        ActiveRecord::Migrator.migrate(MIGRATIONS_ROOT + "/duplicate", nil)
1467 1468
      end
    end
1469

1470
    def test_migrator_with_duplicate_names
1471
      assert_raise(ActiveRecord::DuplicateMigrationNameError, "Multiple migrations have the name Chunky") do
1472 1473 1474 1475
        ActiveRecord::Migrator.migrate(MIGRATIONS_ROOT + "/duplicate_names", nil)
      end
    end

1476
    def test_migrator_with_missing_version_numbers
1477 1478 1479
      assert_raise(ActiveRecord::UnknownMigrationVersionError) do
        ActiveRecord::Migrator.migrate(MIGRATIONS_ROOT + "/missing", 500)
      end
1480
    end
1481

1482 1483 1484 1485
    def test_create_table_with_custom_sequence_name
      return unless current_adapter? :OracleAdapter

      # table name is 29 chars, the standard sequence name will
1486 1487
      # be 33 chars and should be shortened
      assert_nothing_raised do
1488 1489 1490 1491 1492 1493 1494 1495 1496 1497 1498 1499 1500 1501 1502 1503 1504 1505 1506 1507 1508 1509 1510 1511 1512 1513
        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
1514
      assert_raise(ActiveRecord::StatementInvalid) do
1515 1516 1517
        Person.connection.execute("select suitably_short_seq.nextval from dual")
      end
    end
1518 1519 1520 1521 1522 1523 1524 1525 1526

    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

1527
  end
1528

1529 1530 1531 1532 1533
  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
1534
      end
1535
    end
J
Jeremy Kemper 已提交
1536

1537 1538 1539 1540 1541
    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
1542
      end
1543
    end
J
Jeremy Kemper 已提交
1544

1545 1546 1547 1548 1549
    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
1550
      end
1551
    end
J
Jeremy Kemper 已提交
1552

1553 1554 1555 1556
    def test_belongs_to_works_like_references
      with_new_table do |t|
        t.expects(:column).with('customer_id', :integer, {})
        t.belongs_to :customer
1557
      end
1558
    end
J
Jeremy Kemper 已提交
1559

1560 1561 1562 1563 1564
    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
1565
      end
1566
    end
J
Jeremy Kemper 已提交
1567

1568 1569 1570 1571 1572
    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
1573
      end
1574
    end
J
Jeremy Kemper 已提交
1575

1576 1577 1578 1579 1580
    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
1581
      end
1582
    end
J
Jeremy Kemper 已提交
1583

1584 1585 1586 1587 1588 1589 1590 1591 1592
    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

1593 1594 1595 1596
    protected
    def with_new_table
      Person.connection.create_table :delete_me, :force => true do |t|
        yield t
1597
      end
1598 1599 1600
    ensure
      Person.connection.drop_table :delete_me rescue nil
    end
J
Jeremy Kemper 已提交
1601

1602
  end # SexyMigrationsTest
1603

1604 1605 1606 1607 1608 1609 1610 1611 1612 1613 1614 1615 1616 1617 1618 1619 1620 1621 1622 1623 1624 1625 1626 1627 1628 1629 1630 1631 1632 1633 1634 1635 1636 1637 1638 1639 1640 1641 1642
  class MigrationLoggerTest < ActiveRecord::TestCase
    def setup
      Object.send(:remove_const, :InnocentJointable)
    end

    def test_migration_should_be_run_without_logger
      previous_logger = ActiveRecord::Base.logger
      ActiveRecord::Base.logger = nil
      assert_nothing_raised do
        ActiveRecord::Migrator.migrate(MIGRATIONS_ROOT + "/valid")
      end
    ensure
      ActiveRecord::Base.logger = previous_logger
    end
  end

  class InterleavedMigrationsTest < ActiveRecord::TestCase
    def setup
      Object.send(:remove_const, :PeopleHaveLastNames)
    end

    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)

      Object.send(:remove_const, :PeopleHaveLastNames)
      Object.send(:remove_const, :InnocentJointable)
      assert_nothing_raised do
        ActiveRecord::Migrator.down(MIGRATIONS_ROOT + "/interleaved/pass_3")
      end
    end
  end

1643 1644 1645 1646 1647 1648 1649
  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
      end

1650
      assert_nothing_raised do
1651
        connection.add_index :values, :value
1652 1653
        connection.remove_index :values, :column => :value
      end
1654 1655 1656 1657 1658 1659

      connection.drop_table :values rescue nil
    end
  end


1660 1661 1662 1663
  class ChangeTableMigrationsTest < ActiveRecord::TestCase
    def setup
      @connection = Person.connection
      @connection.create_table :delete_me, :force => true do |t|
1664
      end
1665
    end
1666

1667 1668 1669
    def teardown
      Person.connection.drop_table :delete_me rescue nil
    end
1670

1671 1672 1673 1674
    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
1675
      end
1676
    end
1677

1678 1679 1680 1681
    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
1682
      end
1683
    end
1684

1685 1686 1687 1688
    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
1689
      end
1690
    end
1691

1692 1693 1694 1695
    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
1696
      end
1697
    end
1698

1699 1700 1701 1702 1703
    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
1704
      end
1705
    end
1706

1707 1708 1709 1710 1711
    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
1712
      end
1713
    end
1714

1715 1716 1717 1718 1719
    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
1720
      end
1721
    end
1722

1723 1724 1725 1726 1727
    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
1728
      end
1729
    end
1730

1731 1732 1733 1734
    def test_timestamps_creates_updated_at_and_created_at
      with_change_table do |t|
        @connection.expects(:add_timestamps).with(:delete_me)
        t.timestamps
1735
      end
1736
    end
1737

1738 1739 1740 1741
    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
1742
      end
1743
    end
1744

1745 1746 1747
    def string_column
      if current_adapter?(:PostgreSQLAdapter)
        "character varying(255)"
1748 1749
      elsif current_adapter?(:OracleAdapter)
        'VARCHAR2(255)'
1750 1751
      else
        'varchar(255)'
1752
      end
1753
    end
1754

1755
    def integer_column
B
Brian Lopez 已提交
1756
      if current_adapter?(:MysqlAdapter) or current_adapter?(:Mysql2Adapter)
1757
        'int(11)'
1758 1759
      elsif current_adapter?(:OracleAdapter)
        'NUMBER(38)'
1760 1761
      else
        'integer'
1762
      end
1763
    end
1764

1765 1766 1767 1768 1769
    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
1770
      end
1771
    end
1772

1773 1774 1775 1776 1777
    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
1778
      end
1779
    end
1780

1781 1782 1783 1784
    def test_column_creates_column
      with_change_table do |t|
        @connection.expects(:add_column).with(:delete_me, :bar, :integer, {})
        t.column :bar, :integer
1785
      end
1786
    end
1787

1788 1789 1790 1791
    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
1792
      end
1793
    end
1794

1795 1796 1797 1798
    def test_index_creates_index
      with_change_table do |t|
        @connection.expects(:add_index).with(:delete_me, :bar, {})
        t.index :bar
1799
      end
1800
    end
1801

1802 1803 1804 1805
    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
1806
      end
1807
    end
1808

1809 1810 1811 1812 1813 1814 1815 1816 1817 1818 1819 1820 1821 1822
    def test_index_exists
      with_change_table do |t|
        @connection.expects(:index_exists?).with(:delete_me, :bar, {})
        t.index_exists?(:bar)
      end
    end

    def test_index_exists_with_options
      with_change_table do |t|
        @connection.expects(:index_exists?).with(:delete_me, :bar, {:unique => true})
        t.index_exists?(:bar, :unique => true)
      end
    end

1823 1824 1825 1826
    def test_change_changes_column
      with_change_table do |t|
        @connection.expects(:change_column).with(:delete_me, :bar, :string, {})
        t.change :bar, :string
1827
      end
1828
    end
1829

1830 1831 1832 1833
    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
1834
      end
1835
    end
1836

1837 1838 1839 1840
    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
1841
      end
1842
    end
1843

1844 1845 1846 1847
    def test_remove_drops_single_column
      with_change_table do |t|
        @connection.expects(:remove_column).with(:delete_me, [:bar])
        t.remove :bar
1848
      end
1849
    end
1850

1851 1852 1853 1854
    def test_remove_drops_multiple_columns
      with_change_table do |t|
        @connection.expects(:remove_column).with(:delete_me, [:bar, :baz])
        t.remove :bar, :baz
1855
      end
1856
    end
1857

1858 1859 1860 1861
    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
1862
      end
1863
    end
1864

1865 1866 1867 1868
    def test_rename_renames_column
      with_change_table do |t|
        @connection.expects(:rename_column).with(:delete_me, :bar, :baz)
        t.rename :bar, :baz
1869
      end
1870
    end
1871

1872 1873 1874 1875
    protected
    def with_change_table
      Person.connection.change_table :delete_me do |t|
        yield t
1876
      end
1877 1878
    end
  end
1879 1880 1881 1882 1883 1884 1885 1886 1887 1888 1889 1890 1891 1892 1893 1894 1895 1896 1897 1898 1899 1900 1901 1902 1903 1904 1905 1906 1907 1908 1909 1910 1911 1912 1913 1914 1915 1916 1917 1918 1919 1920 1921 1922 1923 1924 1925 1926 1927 1928 1929 1930 1931 1932 1933 1934 1935 1936 1937 1938 1939 1940 1941 1942 1943 1944 1945 1946 1947 1948 1949 1950 1951 1952 1953 1954 1955 1956 1957 1958 1959 1960 1961 1962 1963 1964 1965 1966 1967 1968 1969 1970 1971 1972 1973 1974 1975 1976 1977 1978 1979 1980 1981 1982 1983 1984 1985 1986 1987 1988

  class CopyMigrationsTest < ActiveRecord::TestCase
    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"})
      assert File.exists?(@migrations_path + "/4_people_have_hobbies.bukkits.rb")
      assert File.exists?(@migrations_path + "/5_people_have_descriptions.bukkits.rb")
      assert_equal [@migrations_path + "/4_people_have_hobbies.bukkits.rb", @migrations_path + "/5_people_have_descriptions.bukkits.rb"], copied

      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"]

      sources = ActiveSupport::OrderedHash.new
      sources[:bukkits] = sources[:omg] = MIGRATIONS_ROOT + "/to_copy"
      ActiveRecord::Migration.copy(@migrations_path, sources)
      assert File.exists?(@migrations_path + "/4_people_have_hobbies.omg.rb")
      assert File.exists?(@migrations_path + "/5_people_have_descriptions.omg.rb")
      assert File.exists?(@migrations_path + "/6_people_have_hobbies.bukkits.rb")
      assert File.exists?(@migrations_path + "/7_people_have_descriptions.bukkits.rb")

      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

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

      Time.travel_to(created_at = Time.utc(2010, 7, 26, 10, 10, 10)) do
        copied = ActiveRecord::Migration.copy(@migrations_path, {:bukkits => MIGRATIONS_ROOT + "/to_copy_with_timestamps"})
        assert File.exists?(@migrations_path + "/20100726101010_people_have_hobbies.bukkits.rb")
        assert File.exists?(@migrations_path + "/20100726101011_people_have_descriptions.bukkits.rb")
        expected = [@migrations_path + "/20100726101010_people_have_hobbies.bukkits.rb",
                    @migrations_path + "/20100726101011_people_have_descriptions.bukkits.rb"]
        assert_equal expected, copied

        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?
      end
    ensure
      clear
    end

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

      sources = ActiveSupport::OrderedHash.new
      sources[:bukkits] = sources[:omg] = MIGRATIONS_ROOT + "/to_copy_with_timestamps"

      Time.travel_to(created_at = Time.utc(2010, 7, 26, 10, 10, 10)) do
        copied = ActiveRecord::Migration.copy(@migrations_path, sources)
        assert File.exists?(@migrations_path + "/20100726101010_people_have_hobbies.omg.rb")
        assert File.exists?(@migrations_path + "/20100726101011_people_have_descriptions.omg.rb")
        assert File.exists?(@migrations_path + "/20100726101012_people_have_hobbies.bukkits.rb")
        assert File.exists?(@migrations_path + "/20100726101013_people_have_descriptions.bukkits.rb")
        assert_equal 4, copied.length

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

    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"]

      Time.travel_to(created_at = Time.utc(2010, 2, 20, 10, 10, 10)) do
        ActiveRecord::Migration.copy(@migrations_path, {:bukkits => MIGRATIONS_ROOT + "/to_copy_with_timestamps"})
        assert File.exists?(@migrations_path + "/20100301010102_people_have_hobbies.bukkits.rb")
        assert File.exists?(@migrations_path + "/20100301010103_people_have_descriptions.bukkits.rb")

        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?
      end
    ensure
      clear
    end
1989 1990 1991 1992 1993 1994 1995 1996 1997 1998 1999 2000 2001 2002

    def test_copying_migrations_to_empty_directory
      @migrations_path = MIGRATIONS_ROOT + "/empty"
      @existing_migrations = []

      Time.travel_to(created_at = Time.utc(2010, 7, 26, 10, 10, 10)) do
        copied = ActiveRecord::Migration.copy(@migrations_path, {:bukkits => MIGRATIONS_ROOT + "/to_copy_with_timestamps"})
        assert File.exists?(@migrations_path + "/20100726101010_people_have_hobbies.bukkits.rb")
        assert File.exists?(@migrations_path + "/20100726101011_people_have_descriptions.bukkits.rb")
        assert_equal 2, copied.length
      end
    ensure
      clear
    end
2003
  end
2004
end
2005