From 8d3eafe7906f6d6cf22f7d1ff038c887a32d9a6b Mon Sep 17 00:00:00 2001 From: Robb Kidd Date: Tue, 10 Jul 2012 21:33:24 -0400 Subject: Update psql adapter to rename a default pkey sequence during rename_table. --- .../connection_adapters/postgresql_adapter.rb | 7 +++++++ activerecord/test/cases/migration_test.rb | 20 ++++++++++++++++++++ 2 files changed, 27 insertions(+) diff --git a/activerecord/lib/active_record/connection_adapters/postgresql_adapter.rb b/activerecord/lib/active_record/connection_adapters/postgresql_adapter.rb index 822f8f73de..cd565787b3 100644 --- a/activerecord/lib/active_record/connection_adapters/postgresql_adapter.rb +++ b/activerecord/lib/active_record/connection_adapters/postgresql_adapter.rb @@ -1005,12 +1005,19 @@ module ActiveRecord end # Renames a table. + # Also renames a table's primary key sequence if the sequence name matches the + # Active Record default. # # Example: # rename_table('octopuses', 'octopi') def rename_table(name, new_name) clear_cache! execute "ALTER TABLE #{quote_table_name(name)} RENAME TO #{quote_table_name(new_name)}" + pk, seq = pk_and_sequence_for(new_name) + if seq == "#{name}_#{pk}_seq" + new_seq = "#{new_name}_#{pk}_seq" + execute "ALTER TABLE #{quote_table_name(seq)} RENAME TO #{quote_table_name(new_seq)}" + end end # Adds a new column to the named table. diff --git a/activerecord/test/cases/migration_test.rb b/activerecord/test/cases/migration_test.rb index bedae76e28..fb5eb2633b 100644 --- a/activerecord/test/cases/migration_test.rb +++ b/activerecord/test/cases/migration_test.rb @@ -949,6 +949,26 @@ if ActiveRecord::Base.connection.supports_migrations? end end + if current_adapter?(:PostgreSQLAdapter) + def test_rename_table_for_postgresql_should_also_rename_default_sequence + begin + ActiveRecord::Base.connection.create_table :octopuses do |t| + t.column :url, :string + end + ActiveRecord::Base.connection.rename_table :octopuses, :octopi + + pk, seq = ActiveRecord::Base.connection.pk_and_sequence_for('octopi') + + assert_equal "octopi_#{pk}_seq", seq + + ensure + ActiveRecord::Base.connection.drop_table :octopuses rescue nil + ActiveRecord::Base.connection.drop_table :octopi rescue nil + end + end + end + + def test_change_column_nullability Person.delete_all Person.connection.add_column "people", "funny", :boolean -- cgit v1.2.3