ibm_db 3.0.4 → 3.0.5

data/test/cases/migration/foreign_key_test.rb

@@ -1,360 +1,328 @@
-require 'cases/helper'
-require 'support/ddl_helper'
-require 'support/schema_dumping_helper'
-
-if ActiveRecord::Base.connection.supports_foreign_keys?
-module ActiveRecord
-  class Migration
-    class ForeignKeyTest < ActiveRecord::TestCase
-     include DdlHelper
-     include SchemaDumpingHelper
-
-      class Rocket < ActiveRecord::Base
-      end
-
-      class Astronaut < ActiveRecord::Base
-      end
-
-      setup do	    
-        @connection = ActiveRecord::Base.connection
-        @connection.create_table "rockets", force: true do |t|
-          t.string :name
-        end
-
-        @connection.create_table "astronauts", force: true do |t|
-          t.string :name
-          t.references :rocket
-        end			
-      end
-	  	  
-
-      teardown do	    
-        if defined?(@connection)
-          @connection.drop_table "astronauts" if @connection.table_exists? 'astronauts' 
-          @connection.drop_table "rockets" if @connection.table_exists? 'rockets'
-        end
-      end
-
-
-      def test_foreign_keys
-        foreign_keys = @connection.foreign_keys("fk_test_has_fk")
-        assert_equal 1, foreign_keys.size
-
-        fk = foreign_keys.first
-		if current_adapter?(:IBM_DBAdapter)
-			assert_equal "fk_test_has_fk".upcase, fk.from_table.upcase
-			assert_equal "fk_test_has_pk".upcase, fk.to_table.upcase
-			assert_equal "fk_id".upcase, fk.column.upcase
-			assert_equal "pk_id".upcase, fk.primary_key.upcase
-			assert_equal "fk_name".upcase, fk.name.upcase
-		else
-			assert_equal "fk_test_has_fk", fk.from_table
-			assert_equal "fk_test_has_pk", fk.to_table
-			assert_equal "fk_id", fk.column
-			assert_equal "pk_id", fk.primary_key
-			assert_equal "fk_name", fk.name
-		end        
-      end
-
-
-      def test_add_foreign_key_inferes_column	  		        
-        @connection.add_foreign_key :astronauts, :rockets		
-
-        foreign_keys = @connection.foreign_keys("astronauts")
-        assert_equal 1, foreign_keys.size
-
-        fk = foreign_keys.first
-						
-		if current_adapter?(:IBM_DBAdapter)
-			assert_equal "astronauts".upcase, fk.from_table.upcase
-		else
-			assert_equal "astronauts", fk.from_table
-		end
-        
-		if current_adapter?(:IBM_DBAdapter)
-			assert_equal "rockets".upcase, fk.to_table.upcase
-			assert_equal "rocket_id".upcase, fk.column.upcase
-			assert_equal "id".upcase, fk.primary_key.upcase
-			assert_equal("fk_rails_78146ddd2e".upcase, fk.name.upcase)
-		else
-			assert_equal "rockets", fk.to_table
-			assert_equal "rocket_id", fk.column
-			assert_equal "id", fk.primary_key
-			assert_equal("fk_rails_78146ddd2e", fk.name)
-		end
-        
-      end
-
-	  
-
-      def test_add_foreign_key_with_column
-        @connection.add_foreign_key :astronauts, :rockets, column: "rocket_id"
-
-        foreign_keys = @connection.foreign_keys("astronauts")
-        assert_equal 1, foreign_keys.size
-
-        fk = foreign_keys.first
-		if current_adapter?(:IBM_DBAdapter)
-			assert_equal "astronauts".upcase, fk.from_table.upcase
-			assert_equal "rockets".upcase, fk.to_table.upcase
-			assert_equal "rocket_id".upcase, fk.column.upcase
-			assert_equal "id".upcase, fk.primary_key.upcase
-			assert_equal("fk_rails_78146ddd2e".upcase, fk.name.upcase)
-		else
-			assert_equal "astronauts", fk.from_table
-			assert_equal "rockets", fk.to_table
-			assert_equal "rocket_id", fk.column
-			assert_equal "id", fk.primary_key
-			assert_equal("fk_rails_78146ddd2e", fk.name)
-		end
-                
-      end
-
-      def test_add_foreign_key_with_non_standard_primary_key	  
-		if current_adapter?(:IBM_DBAdapter)
-			@connection.create_table :space_shuttles, force: true, id: false do |t|
-			t.primary_key :pk
-			end			
-			@connection.add_foreign_key(:astronauts, :space_shuttles,
-								  column: "rocket_id", primary_key: "pk", name: "custom_pk")
-			foreign_keys = @connection.foreign_keys("astronauts")
-			assert_equal 1, foreign_keys.size
-			fk = foreign_keys.first			  			 
-			assert_equal "astronauts".upcase, fk.from_table.upcase
-			assert_equal "space_shuttles".upcase, fk.to_table.upcase
-			assert_equal "pk".upcase, fk.primary_key.upcase			  
-			@connection.remove_foreign_key :astronauts, name: "custom_pk"			
-		else			
-			with_example_table @connection, "space_shuttles", "pk integer PRIMARY KEY" do
-			@connection.add_foreign_key(:astronauts, :space_shuttles,
-									column: "rocket_id", primary_key: "pk", name: "custom_pk")
-			foreign_keys = @connection.foreign_keys("astronauts")
-			assert_equal 1, foreign_keys.size
-			fk = foreign_keys.first
-			assert_equal "astronauts", fk.from_table
-			assert_equal "space_shuttles", fk.to_table
-			assert_equal "pk", fk.primary_key		  
-			@connection.remove_foreign_key :astronauts, name: "custom_pk"
-			end		
-		end		
-      end
-
-      def test_add_on_delete_restrict_foreign_key
-        @connection.add_foreign_key :astronauts, :rockets, column: "rocket_id", on_delete: :restrict
-
-        foreign_keys = @connection.foreign_keys("astronauts")
-        assert_equal 1, foreign_keys.size
-
-        fk = foreign_keys.first
-        if current_adapter?(:MysqlAdapter, :Mysql2Adapter)
-          # ON DELETE RESTRICT is the default on MySQL
-          assert_equal "noaction", fk.on_delete
-        elsif current_adapter?(:IBM_DBAdapter)
-			assert_equal :noaction, fk.on_delete		
-		else 
-          assert_equal :restrict, fk.on_delete
-        end
-      end
-
-      def test_add_on_delete_cascade_foreign_key
-        @connection.add_foreign_key :astronauts, :rockets, column: "rocket_id", on_delete: :cascade
-
-        foreign_keys = @connection.foreign_keys("astronauts")
-        assert_equal 1, foreign_keys.size
-
-        fk = foreign_keys.first		
-		if current_adapter?(:IBM_DBAdapter)
-			assert_equal :noaction, fk.on_delete		
-		else 
-			assert_equal :cascade, fk.on_delete
-		end
-        
-      end
-
-      def test_add_on_delete_nullify_foreign_key
-        @connection.add_foreign_key :astronauts, :rockets, column: "rocket_id", on_delete: :nullify
-
-        foreign_keys = @connection.foreign_keys("astronauts")
-        assert_equal 1, foreign_keys.size
-
-        fk = foreign_keys.first
-		
-		if current_adapter?(:IBM_DBAdapter)
-			assert_equal :noaction, fk.on_delete		
-		else 
-			assert_equal :nullify, fk.on_delete
-		end
-        
-      end
-
-      def test_on_update_and_on_delete_raises_with_invalid_values
-        assert_raises ArgumentError do
-          @connection.add_foreign_key :astronauts, :rockets, column: "rocket_id", on_delete: :invalid
-        end
-
-        assert_raises ArgumentError do
-          @connection.add_foreign_key :astronauts, :rockets, column: "rocket_id", on_update: :invalid
-        end
-      end
-
-      def test_add_foreign_key_with_on_update
-        #@connection.add_foreign_key :astronauts, :rockets, column: "rocket_id" , on_update: :nullify
-		@connection.add_foreign_key :astronauts, :rockets, column: "rocket_id"
-
-        foreign_keys = @connection.foreign_keys("astronauts")
-        assert_equal 1, foreign_keys.size
-
-        fk = foreign_keys.first
-		if current_adapter?(:IBM_DBAdapter)
-			assert_equal :noaction, fk.on_update		
-		else 
-			assert_equal :nullify, fk.on_update		
-		end
-      end
-
-      def test_remove_foreign_key_inferes_column
-        @connection.add_foreign_key :astronauts, :rockets
-
-        assert_equal 1, @connection.foreign_keys("astronauts").size
-        @connection.remove_foreign_key :astronauts, :rockets
-        assert_equal [], @connection.foreign_keys("astronauts")
-      end
-
-      def test_remove_foreign_key_by_column
-        @connection.add_foreign_key :astronauts, :rockets, column: "rocket_id"
-
-        assert_equal 1, @connection.foreign_keys("astronauts").size
-        @connection.remove_foreign_key :astronauts, column: "rocket_id"
-        assert_equal [], @connection.foreign_keys("astronauts")
-      end
-
-      def test_remove_foreign_key_by_symbol_column
-        @connection.add_foreign_key :astronauts, :rockets, column: :rocket_id
-		
-		if current_adapter?(:IBM_DBAdapter)			
-			foreign_keys = @connection.foreign_keys("astronauts")
-			assert_equal 1, foreign_keys.size											
-			@connection.remove_foreign_key "astronauts".upcase, column: "rocket_id".upcase
-		else
-			assert_equal 1, @connection.foreign_keys("astronauts").size
-			@connection.remove_foreign_key :astronauts, column: :rocket_id
-		end
-        
-		assert_equal [], @connection.foreign_keys("astronauts")
-      end
-
-      def test_remove_foreign_key_by_name
-        @connection.add_foreign_key :astronauts, :rockets, column: "rocket_id", name: "fancy_named_fk"
-
-        assert_equal 1, @connection.foreign_keys("astronauts").size
-        @connection.remove_foreign_key :astronauts, name: "fancy_named_fk"
-        assert_equal [], @connection.foreign_keys("astronauts")
-      end
-
-      def test_remove_foreign_non_existing_foreign_key_raises
-        assert_raises ArgumentError do
-          @connection.remove_foreign_key :astronauts, :rockets
-        end
-      end
-
-      def test_schema_dumping
-        @connection.add_foreign_key :astronauts, :rockets
-        output = dump_table_schema "astronauts"
-        assert_match %r{\s+add_foreign_key "astronauts", "rockets"$}, output
-      end
-
-      def test_schema_dumping_with_options
-        output = dump_table_schema "fk_test_has_fk"		
-		assert_match %r{\s+add_foreign_key "fk_test_has_fk", "fk_test_has_pk", column: "fk_id", primary_key: "pk_id", name: "fk_name"$}, output		
-      end
-
-      def test_schema_dumping_on_delete_and_on_update_options
-	    if current_adapter?(:IBM_DBAdapter)
-			@connection.add_foreign_key :astronauts, :rockets, column: "rocket_id", on_delete: :nullify, on_update: :restrict
-		else
-			@connection.add_foreign_key :astronauts, :rockets, column: "rocket_id", on_delete: :nullify, on_update: :cascade
-		end
-		
-		output = dump_table_schema "astronauts"
-		assert_match %r{\s+add_foreign_key "astronauts",.+on_update: :restrict,.+on_delete: :nullify$}, output		
-      end
-
-      class CreateCitiesAndHousesMigration < ActiveRecord::Migration
-        def change
-          create_table("cities") { |t| }
-
-          create_table("houses") do |t|
-            t.column :city_id, :integer
-          end
-          add_foreign_key :houses, :cities, column: "city_id"
-        end
-      end
-
-      def test_add_foreign_key_is_reversible
-        migration = CreateCitiesAndHousesMigration.new
-        silence_stream($stdout) { migration.migrate(:up) }
-        assert_equal 1, @connection.foreign_keys("houses").size
-      ensure
-        silence_stream($stdout) { migration.migrate(:down) }
-      end
-
-      class CreateSchoolsAndClassesMigration < ActiveRecord::Migration
-        def change
-          create_table(:schools)
-
-          create_table(:classes) do |t|
-            t.column :school_id, :integer
-          end
-          add_foreign_key :classes, :schools
-        end
-      end
-
-      def test_add_foreign_key_with_prefix
-        ActiveRecord::Base.table_name_prefix = 'p_'
-        migration = CreateSchoolsAndClassesMigration.new
-        silence_stream($stdout) { migration.migrate(:up) }
-        assert_equal 1, @connection.foreign_keys("p_classes").size
-      ensure
-        silence_stream($stdout) { migration.migrate(:down) }
-        ActiveRecord::Base.table_name_prefix = nil
-      end
-
-      def test_add_foreign_key_with_suffix
-        ActiveRecord::Base.table_name_suffix = '_s'
-        migration = CreateSchoolsAndClassesMigration.new
-        silence_stream($stdout) { migration.migrate(:up) }
-        assert_equal 1, @connection.foreign_keys("classes_s").size
-      ensure
-        silence_stream($stdout) { migration.migrate(:down) }
-        ActiveRecord::Base.table_name_suffix = nil
-      end
-	  
-
-    end
-  end
-end
-else
-module ActiveRecord
-  class Migration
-    class NoForeignKeySupportTest < ActiveRecord::TestCase
-      setup do
-        @connection = ActiveRecord::Base.connection
-      end
-
-      def test_add_foreign_key_should_be_noop
-        @connection.add_foreign_key :clubs, :categories
-      end
-
-      def test_remove_foreign_key_should_be_noop
-        @connection.remove_foreign_key :clubs, :categories
-      end
-
-      def test_foreign_keys_should_raise_not_implemented
-        assert_raises NotImplementedError do
-          @connection.foreign_keys("clubs")
-        end
-      end
-    end
-  end
-end
-end
+require 'cases/helper'
+require 'support/ddl_helper'
+require 'support/schema_dumping_helper'
+
+if ActiveRecord::Base.connection.supports_foreign_keys?
+module ActiveRecord
+  class Migration
+    class ForeignKeyTest < ActiveRecord::TestCase
+      include DdlHelper
+      include SchemaDumpingHelper
+
+      class Rocket < ActiveRecord::Base
+      end
+
+      class Astronaut < ActiveRecord::Base
+      end
+
+      setup do
+        @connection = ActiveRecord::Base.connection
+        @connection.create_table "rockets", force: true do |t|
+          t.string :name
+        end
+
+        @connection.create_table "astronauts", force: true do |t|
+          t.string :name
+          t.references :rocket
+        end
+      end
+
+      teardown do
+        if defined?(@connection)
+          @connection.drop_table "astronauts" if @connection.table_exists? 'astronauts' 
+          @connection.drop_table "rockets" if @connection.table_exists? 'rockets'
+        end
+      end
+
+      def test_foreign_keys
+        foreign_keys = @connection.foreign_keys("fk_test_has_fk")
+        assert_equal 1, foreign_keys.size
+
+        fk = foreign_keys.first
+		if current_adapter?(:IBM_DBAdapter)
+			assert_equal "fk_test_has_fk".upcase, fk.from_table.upcase
+			assert_equal "fk_test_has_pk".upcase, fk.to_table.upcase
+			assert_equal "fk_id".upcase, fk.column.upcase
+			assert_equal "pk_id".upcase, fk.primary_key.upcase
+			assert_equal "fk_name".upcase, fk.name.upcase
+		else
+			assert_equal "fk_test_has_fk", fk.from_table
+			assert_equal "fk_test_has_pk", fk.to_table
+			assert_equal "fk_id", fk.column
+			assert_equal "pk_id", fk.primary_key
+			assert_equal "fk_name", fk.name
+		end        
+      end
+
+      def test_add_foreign_key_inferes_column
+        @connection.add_foreign_key :astronauts, :rockets
+
+        foreign_keys = @connection.foreign_keys("astronauts")
+        assert_equal 1, foreign_keys.size
+
+        fk = foreign_keys.first
+						
+		if current_adapter?(:IBM_DBAdapter)
+			assert_equal "astronauts".upcase, fk.from_table.upcase
+		else
+			assert_equal "astronauts", fk.from_table
+		end
+        
+		if current_adapter?(:IBM_DBAdapter)
+			assert_equal "rockets".upcase, fk.to_table.upcase
+			assert_equal "rocket_id".upcase, fk.column.upcase
+			assert_equal "id".upcase, fk.primary_key.upcase
+			assert_equal("fk_rails_78146ddd2e".upcase, fk.name.upcase)
+		else
+			assert_equal "rockets", fk.to_table
+			assert_equal "rocket_id", fk.column
+			assert_equal "id", fk.primary_key
+			assert_equal("fk_rails_78146ddd2e", fk.name)
+		end
+        
+      end
+
+      def test_add_foreign_key_with_column
+        @connection.add_foreign_key :astronauts, :rockets, column: "rocket_id"
+
+        foreign_keys = @connection.foreign_keys("astronauts")
+        assert_equal 1, foreign_keys.size
+
+        fk = foreign_keys.first
+		if current_adapter?(:IBM_DBAdapter)
+			assert_equal "astronauts".upcase, fk.from_table.upcase
+			assert_equal "rockets".upcase, fk.to_table.upcase
+			assert_equal "rocket_id".upcase, fk.column.upcase
+			assert_equal "id".upcase, fk.primary_key.upcase
+			assert_equal("fk_rails_78146ddd2e".upcase, fk.name.upcase)
+		else
+			assert_equal "astronauts", fk.from_table
+			assert_equal "rockets", fk.to_table
+			assert_equal "rocket_id", fk.column
+			assert_equal "id", fk.primary_key
+			assert_equal("fk_rails_78146ddd2e", fk.name)
+		end
+                
+      end
+
+      def test_add_foreign_key_with_non_standard_primary_key
+		if current_adapter?(:IBM_DBAdapter)
+			@connection.create_table :space_shuttles, force: true, id: false do |t|
+			t.primary_key :pk
+			end			
+			@connection.add_foreign_key(:astronauts, :space_shuttles,
+								  column: "rocket_id", primary_key: "pk", name: "custom_pk")
+			foreign_keys = @connection.foreign_keys("astronauts")
+			assert_equal 1, foreign_keys.size
+			fk = foreign_keys.first			  			 
+			assert_equal "astronauts".upcase, fk.from_table.upcase
+			assert_equal "space_shuttles".upcase, fk.to_table.upcase
+			assert_equal "pk".upcase, fk.primary_key.upcase			  
+			@connection.remove_foreign_key :astronauts, name: "custom_pk"			
+		else			
+			with_example_table @connection, "space_shuttles", "pk integer PRIMARY KEY" do
+			@connection.add_foreign_key(:astronauts, :space_shuttles,
+									column: "rocket_id", primary_key: "pk", name: "custom_pk")
+			foreign_keys = @connection.foreign_keys("astronauts")
+			assert_equal 1, foreign_keys.size
+			fk = foreign_keys.first
+			assert_equal "astronauts", fk.from_table
+			assert_equal "space_shuttles", fk.to_table
+			assert_equal "pk", fk.primary_key		  
+			@connection.remove_foreign_key :astronauts, name: "custom_pk"
+			end		
+        end
+      end
+
+      def test_add_on_delete_restrict_foreign_key
+        @connection.add_foreign_key :astronauts, :rockets, column: "rocket_id", on_delete: :restrict
+
+        foreign_keys = @connection.foreign_keys("astronauts")
+        assert_equal 1, foreign_keys.size
+
+        fk = foreign_keys.first
+        if current_adapter?(:MysqlAdapter, :Mysql2Adapter)
+          # ON DELETE RESTRICT is the default on MySQL
+          assert_equal nil, fk.on_delete
+		else 
+          assert_equal :restrict, fk.on_delete
+        end
+      end
+
+      def test_add_on_delete_cascade_foreign_key
+        @connection.add_foreign_key :astronauts, :rockets, column: "rocket_id", on_delete: :cascade
+
+        foreign_keys = @connection.foreign_keys("astronauts")
+        assert_equal 1, foreign_keys.size
+
+        fk = foreign_keys.first		
+        assert_equal :cascade, fk.on_delete		
+      end
+
+      def test_add_on_delete_nullify_foreign_key
+        @connection.add_foreign_key :astronauts, :rockets, column: "rocket_id", on_delete: :nullify
+
+        foreign_keys = @connection.foreign_keys("astronauts")
+        assert_equal 1, foreign_keys.size
+
+        fk = foreign_keys.first
+        assert_equal :nullify, fk.on_delete
+      end
+
+      def test_on_update_and_on_delete_raises_with_invalid_values
+        assert_raises ArgumentError do
+          @connection.add_foreign_key :astronauts, :rockets, column: "rocket_id", on_delete: :invalid
+        end
+
+        assert_raises ArgumentError do
+          @connection.add_foreign_key :astronauts, :rockets, column: "rocket_id", on_update: :invalid
+        end
+      end
+
+      def test_add_foreign_key_with_on_update
+        #@connection.add_foreign_key :astronauts, :rockets, column: "rocket_id" , on_update: :nullify
+		@connection.add_foreign_key :astronauts, :rockets, column: "rocket_id"
+
+        foreign_keys = @connection.foreign_keys("astronauts")
+        assert_equal 1, foreign_keys.size
+
+        fk = foreign_keys.first
+        assert_equal :noaction, fk.on_update
+		#assert_equal nil, fk.on_update
+      end
+
+      def test_remove_foreign_key_inferes_column
+        @connection.add_foreign_key :astronauts, :rockets
+
+        assert_equal 1, @connection.foreign_keys("astronauts").size
+        @connection.remove_foreign_key :astronauts, :rockets
+        assert_equal [], @connection.foreign_keys("astronauts")
+      end
+
+      def test_remove_foreign_key_by_column
+        @connection.add_foreign_key :astronauts, :rockets, column: "rocket_id"
+
+        assert_equal 1, @connection.foreign_keys("astronauts").size
+        @connection.remove_foreign_key :astronauts, column: "rocket_id"
+        assert_equal [], @connection.foreign_keys("astronauts")
+      end
+
+      def test_remove_foreign_key_by_symbol_column
+        @connection.add_foreign_key :astronauts, :rockets, column: :rocket_id
+
+        assert_equal 1, @connection.foreign_keys("astronauts").size
+        @connection.remove_foreign_key :astronauts, column: :rocket_id
+        assert_equal [], @connection.foreign_keys("astronauts")
+      end
+
+      def test_remove_foreign_key_by_name
+        @connection.add_foreign_key :astronauts, :rockets, column: "rocket_id", name: "fancy_named_fk"
+
+        assert_equal 1, @connection.foreign_keys("astronauts").size
+        @connection.remove_foreign_key :astronauts, name: "fancy_named_fk"
+        assert_equal [], @connection.foreign_keys("astronauts")
+      end
+
+      def test_remove_foreign_non_existing_foreign_key_raises
+        assert_raises ArgumentError do
+          @connection.remove_foreign_key :astronauts, :rockets
+        end
+      end
+
+      def test_schema_dumping
+        @connection.add_foreign_key :astronauts, :rockets
+        output = dump_table_schema "astronauts"
+        assert_match %r{\s+add_foreign_key "ASTRONAUTS", "ROCKETS"$}, output
+      end
+
+      def test_schema_dumping_with_options
+        output = dump_table_schema "fk_test_has_fk"
+        assert_match %r{\s+add_foreign_key "FK_TEST_HAS_FK", "FK_TEST_HAS_PK", column: "FK_ID", primary_key: "PK_ID", name: "FK_NAME"$}, output
+      end
+
+      def test_schema_dumping_on_delete_and_on_update_options
+        #@connection.add_foreign_key :astronauts, :rockets, column: "rocket_id", on_delete: :nullify, on_update: :cascade
+		@connection.add_foreign_key :astronauts, :rockets, column: "rocket_id", on_delete: :cascade, on_update: :restrict
+
+        output = dump_table_schema "astronauts"
+        #assert_match %r{\s+add_foreign_key "astronauts",.+on_update: :cascade,.+on_delete: :nullify$}, output
+		assert_match %r{\s+add_foreign_key "ASTRONAUTS",.+on_update: :restrict,.+on_delete: :cascade$}, output
+      end
+
+      class CreateCitiesAndHousesMigration < ActiveRecord::Migration
+        def change
+          create_table("cities") { |t| }
+
+          create_table("houses") do |t|
+            t.column :city_id, :integer
+          end
+          add_foreign_key :houses, :cities, column: "city_id"
+        end
+      end
+
+      def test_add_foreign_key_is_reversible
+        migration = CreateCitiesAndHousesMigration.new
+        silence_stream($stdout) { migration.migrate(:up) }
+        assert_equal 1, @connection.foreign_keys("houses").size
+      ensure
+        silence_stream($stdout) { migration.migrate(:down) }
+      end
+
+      class CreateSchoolsAndClassesMigration < ActiveRecord::Migration
+        def change
+          create_table(:schools)
+
+          create_table(:classes) do |t|
+            t.column :school_id, :integer
+          end
+          add_foreign_key :classes, :schools
+        end
+      end
+
+      def test_add_foreign_key_with_prefix
+        ActiveRecord::Base.table_name_prefix = 'p_'
+        migration = CreateSchoolsAndClassesMigration.new
+        silence_stream($stdout) { migration.migrate(:up) }
+        assert_equal 1, @connection.foreign_keys("p_classes").size
+      ensure
+        silence_stream($stdout) { migration.migrate(:down) }
+        ActiveRecord::Base.table_name_prefix = nil
+      end
+
+      def test_add_foreign_key_with_suffix
+        ActiveRecord::Base.table_name_suffix = '_s'
+        migration = CreateSchoolsAndClassesMigration.new
+        silence_stream($stdout) { migration.migrate(:up) }
+        assert_equal 1, @connection.foreign_keys("classes_s").size
+      ensure
+        silence_stream($stdout) { migration.migrate(:down) }
+        ActiveRecord::Base.table_name_suffix = nil
+      end
+    end
+  end
+end
+else
+module ActiveRecord
+  class Migration
+    class NoForeignKeySupportTest < ActiveRecord::TestCase
+      setup do
+        @connection = ActiveRecord::Base.connection
+      end
+
+      def test_add_foreign_key_should_be_noop
+        @connection.add_foreign_key :clubs, :categories
+      end
+
+      def test_remove_foreign_key_should_be_noop
+        @connection.remove_foreign_key :clubs, :categories
+      end
+
+      def test_foreign_keys_should_raise_not_implemented
+        assert_raises NotImplementedError do
+          @connection.foreign_keys("clubs")
+        end
+      end
+    end
+  end
+end
+end