1 files changed, 43 insertions, 0 deletions
diff --git a/activerecord/test/cases/migration/helper.rb b/activerecord/test/cases/migration/helper.rb
new file mode 100644
index 0000000000..e28feedcf9
--- /dev/null
+++ b/activerecord/test/cases/migration/helper.rb
@@ -0,0 +1,43 @@
+require "cases/helper"
+
+module ActiveRecord
+  class Migration
+    class << self; attr_accessor :message_count; end
+    self.message_count = 0
+
+    def puts(text="")
+      ActiveRecord::Migration.message_count += 1
+    end
+
+    module TestHelper
+      attr_reader :connection, :table_name
+
+      CONNECTION_METHODS = %w[add_column remove_column rename_column add_index change_column rename_table column_exists? index_exists? add_reference add_belongs_to remove_reference remove_references remove_belongs_to]
+
+      class TestModel < ActiveRecord::Base
+        self.table_name = :test_models
+      end
+
+      def setup
+        super
+        @connection = ActiveRecord::Base.connection
+        connection.create_table :test_models do |t|
+          t.timestamps
+        end
+
+        TestModel.reset_column_information
+      end
+
+      def teardown
+        super
+        TestModel.reset_table_name
+        TestModel.reset_sequence_name
+        connection.drop_table :test_models rescue nil
+      end
+
+      private
+
+      delegate(*CONNECTION_METHODS, to: :connection)
+    end
+  end
+end