3 files changed, 11 insertions, 22 deletions
diff --git a/activerecord/test/cases/migration/helper.rb b/activerecord/test/cases/migration/helper.rb
index 1a5ac1333e..fe53510ba2 100644
--- a/activerecord/test/cases/migration/helper.rb
+++ b/activerecord/test/cases/migration/helper.rb
@@ -2,6 +2,15 @@ require "cases/helper"
 
 module ActiveRecord
   class Migration
+    class << self
+      attr_accessor :message_count
+    end
+
+    def puts(text="")
+      ActiveRecord::Migration.message_count ||= 0
+      ActiveRecord::Migration.message_count += 1
+    end
+
     module TestHelper
       attr_reader :connection, :table_name
 
diff --git a/activerecord/test/cases/migration_test.rb b/activerecord/test/cases/migration_test.rb
index efb47b1844..9b80ea92a2 100644
--- a/activerecord/test/cases/migration_test.rb
+++ b/activerecord/test/cases/migration_test.rb
@@ -1,4 +1,5 @@
 require "cases/helper"
+require "cases/migration/helper"
 require 'bigdecimal/util'
 
 require 'models/person'
@@ -16,28 +17,6 @@ class Reminder < ActiveRecord::Base; end
 
 class Thing < ActiveRecord::Base; end
 
-class ActiveRecord::Migration
-  class << self
-    attr_accessor :message_count
-  end
-
-  def puts(text="")
-    ActiveRecord::Migration.message_count ||= 0
-    ActiveRecord::Migration.message_count += 1
-  end
-end
-
-module ActiveRecord
-  class MigrationTest < ActiveRecord::TestCase
-    attr_reader :connection
-
-    def setup
-      super
-      @connection = Base.connection
-    end
-  end
-end
-
 class MigrationTest < ActiveRecord::TestCase
   self.use_transactional_fixtures = false
 
diff --git a/activerecord/test/cases/migrator_test.rb b/activerecord/test/cases/migrator_test.rb
index c14c485842..5553798cc7 100644
--- a/activerecord/test/cases/migrator_test.rb
+++ b/activerecord/test/cases/migrator_test.rb
@@ -1,4 +1,5 @@
 require "cases/helper"
+require "cases/migration/helper"
 
 module ActiveRecord
   class MigratorTest < ActiveRecord::TestCase