require 'abstract_unit' require 'fixtures/person' require File.dirname(__FILE__) + '/fixtures/migrations/1_people_have_last_names' require File.dirname(__FILE__) + '/fixtures/migrations/2_we_need_reminders' if ActiveRecord::Base.connection.supports_migrations? class Reminder < ActiveRecord::Base; end class MigrationTest < Test::Unit::TestCase self.use_transactional_fixtures = false def setup end def teardown ActiveRecord::Base.connection.initialize_schema_information ActiveRecord::Base.connection.update "UPDATE #{ActiveRecord::Migrator.schema_info_table_name} SET version = 0" Reminder.connection.drop_table("reminders") rescue nil Reminder.connection.drop_table("people_reminders") rescue nil Reminder.connection.drop_table("prefix_reminders_suffix") rescue nil Reminder.reset_column_information Person.connection.remove_column("people", "last_name") rescue nil Person.connection.remove_column("people", "bio") rescue nil Person.connection.remove_column("people", "age") rescue nil Person.connection.remove_column("people", "height") rescue nil Person.connection.remove_column("people", "birthday") rescue nil Person.connection.remove_column("people", "favorite_day") rescue nil Person.connection.remove_column("people", "male") rescue nil Person.connection.remove_column("people", "administrator") rescue nil Person.reset_column_information end def test_add_index Person.connection.add_column "people", "last_name", :string Person.connection.add_column "people", "administrator", :boolean assert_nothing_raised { Person.connection.add_index("people", "last_name") } assert_nothing_raised { Person.connection.remove_index("people", "last_name") } assert_nothing_raised { Person.connection.add_index("people", ["last_name", "first_name"]) } assert_nothing_raised { Person.connection.remove_index("people", "last_name") } 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 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 def test_create_table_with_not_null_column Person.connection.create_table :testings do |t| t.column :foo, :string, :null => false end assert_raises(ActiveRecord::StatementInvalid) do Person.connection.execute "insert into testings (foo) values (NULL)" end ensure Person.connection.drop_table :testings rescue nil end def test_create_table_with_defaults 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 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" } assert_equal "hello", one.default if current_adapter?(:OracleAdapter) # Oracle doesn't support native booleans assert_equal true, two.default == 1 assert_equal false, three.default != 0 else assert_equal true, two.default assert_equal false, three.default end assert_equal 1, four.default ensure Person.connection.drop_table :testings rescue nil end # SQL Server will not allow you to add a NOT NULL column # to a table without specifying a default value, so the # following test must be skipped unless current_adapter?(:SQLServerAdapter) 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 assert_raises(ActiveRecord::StatementInvalid) do Person.connection.execute "insert into testings (foo, bar) values ('hello', NULL)" end ensure Person.connection.drop_table :testings rescue nil end end def test_add_column_not_null_with_default Person.connection.create_table :testings do |t| t.column :foo, :string end Person.connection.add_column :testings, :bar, :string, :null => false, :default => "default" assert_raises(ActiveRecord::StatementInvalid) do Person.connection.execute "insert into testings (foo, bar) values ('hello', NULL)" end ensure Person.connection.drop_table :testings rescue nil end 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 Person.connection.add_column "people", "birthday", :datetime Person.connection.add_column "people", "favorite_day", :date Person.connection.add_column "people", "male", :boolean assert_nothing_raised { Person.create :first_name => 'bob', :last_name => 'bobsen', :bio => "I was born ....", :age => 18, :height => 1.78, :birthday => 18.years.ago, :favorite_day => 10.days.ago, :male => true } bob = Person.find(:first) assert_equal bob.first_name, 'bob' assert_equal bob.last_name, 'bobsen' assert_equal bob.bio, "I was born ...." assert_equal bob.age, 18 assert_equal bob.male?, true 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 if current_adapter?(:SQLServerAdapter) or current_adapter?(:OracleAdapter) # SQL Server and Oracle don't differentiate between date/time assert_equal Time, bob.favorite_day.class else assert_equal Date, bob.favorite_day.class end assert_equal TrueClass, bob.male?.class end def test_add_remove_single_field_using_string_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' 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 Person.reset_column_information assert !Person.column_methods_hash.include?(:last_name) end def test_add_rename Person.delete_all begin Person.connection.add_column "people", "girlfriend", :string Person.create :girlfriend => 'bobette' Person.connection.rename_column "people", "girlfriend", "exgirlfriend" Person.reset_column_information bob = Person.find(:first) assert_equal "bobette", bob.exgirlfriend ensure Person.connection.remove_column("people", "girlfriend") rescue nil Person.connection.remove_column("people", "exgirlfriend") rescue nil end end def test_rename_column_using_symbol_arguments begin Person.connection.rename_column :people, :first_name, :nick_name Person.reset_column_information assert Person.column_names.include?("nick_name") ensure Person.connection.remove_column("people","nick_name") Person.connection.add_column("people","first_name", :string) end end def test_rename_column begin Person.connection.rename_column "people", "first_name", "nick_name" Person.reset_column_information assert Person.column_names.include?("nick_name") ensure Person.connection.remove_column("people","nick_name") Person.connection.add_column("people","first_name", :string) end end def test_rename_table begin ActiveRecord::Base.connection.create_table :octopuses do |t| t.column :url, :string end ActiveRecord::Base.connection.rename_table :octopuses, :octopi assert_nothing_raised do if current_adapter?(:OracleAdapter) # Oracle requires the explicit sequence for the pk ActiveRecord::Base.connection.execute "INSERT INTO octopi (id, url) VALUES (octopi_seq.nextval, 'http://www.foreverflying.com/octopus-black7.jpg')" else ActiveRecord::Base.connection.execute "INSERT INTO octopi (url) VALUES ('http://www.foreverflying.com/octopus-black7.jpg')" end end assert_equal 'http://www.foreverflying.com/octopus-black7.jpg', ActiveRecord::Base.connection.select_value("SELECT url FROM octopi WHERE id=1") ensure ActiveRecord::Base.connection.drop_table :octopuses rescue nil ActiveRecord::Base.connection.drop_table :octopi rescue nil end end def test_change_column Person.connection.add_column 'people', 'age', :integer old_columns = Person.connection.columns(Person.table_name, "#{name} Columns") assert old_columns.find { |c| c.name == 'age' and c.type == :integer } assert_nothing_raised { Person.connection.change_column "people", "age", :string } new_columns = Person.connection.columns(Person.table_name, "#{name} Columns") 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 } end def test_change_column_with_new_default Person.connection.add_column "people", "administrator", :boolean, :default => 1 Person.reset_column_information assert Person.new.administrator? assert_nothing_raised { Person.connection.change_column "people", "administrator", :boolean, :default => 0 } Person.reset_column_information assert !Person.new.administrator? end def test_add_table assert !Reminder.table_exists? WeNeedReminders.up assert Reminder.create("content" => "hello world", "remind_at" => Time.now) assert_equal "hello world", Reminder.find(:first).content WeNeedReminders.down assert_raises(ActiveRecord::StatementInvalid) { Reminder.find(:first) } end def test_migrator assert !Person.column_methods_hash.include?(:last_name) assert !Reminder.table_exists? ActiveRecord::Migrator.up(File.dirname(__FILE__) + '/fixtures/migrations/') assert_equal 3, ActiveRecord::Migrator.current_version 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 ActiveRecord::Migrator.down(File.dirname(__FILE__) + '/fixtures/migrations/') assert_equal 0, ActiveRecord::Migrator.current_version Person.reset_column_information assert !Person.column_methods_hash.include?(:last_name) assert_raises(ActiveRecord::StatementInvalid) { Reminder.find(:first) } end def test_migrator_one_up assert !Person.column_methods_hash.include?(:last_name) assert !Reminder.table_exists? ActiveRecord::Migrator.up(File.dirname(__FILE__) + '/fixtures/migrations/', 1) Person.reset_column_information assert Person.column_methods_hash.include?(:last_name) assert !Reminder.table_exists? ActiveRecord::Migrator.up(File.dirname(__FILE__) + '/fixtures/migrations/', 2) assert Reminder.create("content" => "hello world", "remind_at" => Time.now) assert_equal "hello world", Reminder.find(:first).content end def test_migrator_one_down ActiveRecord::Migrator.up(File.dirname(__FILE__) + '/fixtures/migrations/') ActiveRecord::Migrator.down(File.dirname(__FILE__) + '/fixtures/migrations/', 1) Person.reset_column_information assert Person.column_methods_hash.include?(:last_name) assert !Reminder.table_exists? end def test_migrator_one_up_one_down ActiveRecord::Migrator.up(File.dirname(__FILE__) + '/fixtures/migrations/', 1) ActiveRecord::Migrator.down(File.dirname(__FILE__) + '/fixtures/migrations/', 0) assert !Person.column_methods_hash.include?(:last_name) assert !Reminder.table_exists? end def test_migrator_going_down_due_to_version_target ActiveRecord::Migrator.up(File.dirname(__FILE__) + '/fixtures/migrations/', 1) ActiveRecord::Migrator.migrate(File.dirname(__FILE__) + '/fixtures/migrations/', 0) assert !Person.column_methods_hash.include?(:last_name) assert !Reminder.table_exists? ActiveRecord::Migrator.migrate(File.dirname(__FILE__) + '/fixtures/migrations/') 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 def test_schema_info_table_name ActiveRecord::Base.table_name_prefix = "prefix_" ActiveRecord::Base.table_name_suffix = "_suffix" Reminder.reset_table_name assert_equal "prefix_schema_info_suffix", ActiveRecord::Migrator.schema_info_table_name ActiveRecord::Base.table_name_prefix = "" ActiveRecord::Base.table_name_suffix = "" Reminder.reset_table_name assert_equal "schema_info", ActiveRecord::Migrator.schema_info_table_name ensure ActiveRecord::Base.table_name_prefix = "" ActiveRecord::Base.table_name_suffix = "" end 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) Reminder.reset_table_name assert_equal Reminder.table_name, ActiveRecord::Migrator.proper_table_name(Reminder) # 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' Reminder.reset_table_name assert_equal "prefix_reminders_suffix", ActiveRecord::Migrator.proper_table_name(Reminder) Reminder.table_name_prefix = '' Reminder.table_name_suffix = '' Reminder.reset_table_name # Use AR::Base's prefix/suffix if string or symbol is given ActiveRecord::Base.table_name_prefix = "prefix_" ActiveRecord::Base.table_name_suffix = "_suffix" Reminder.reset_table_name 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 = "" Reminder.reset_table_name end def test_add_drop_table_with_prefix_and_suffix assert !Reminder.table_exists? ActiveRecord::Base.table_name_prefix = 'prefix_' ActiveRecord::Base.table_name_suffix = '_suffix' Reminder.reset_table_name Reminder.reset_sequence_name WeNeedReminders.up assert Reminder.create("content" => "hello world", "remind_at" => Time.now) assert_equal "hello world", Reminder.find(:first).content WeNeedReminders.down assert_raises(ActiveRecord::StatementInvalid) { Reminder.find(:first) } ensure ActiveRecord::Base.table_name_prefix = '' ActiveRecord::Base.table_name_suffix = '' Reminder.reset_table_name Reminder.reset_sequence_name end def test_migrator_with_duplicates assert_raises(ActiveRecord::DuplicateMigrationVersionError) do ActiveRecord::Migrator.migrate(File.dirname(__FILE__) + '/fixtures/migrations_with_duplicate/', nil) end end end end