aboutsummaryrefslogtreecommitdiffstats
path: root/activerecord/test/cases/migration/helper.rb
diff options
context:
space:
mode:
Diffstat (limited to 'activerecord/test/cases/migration/helper.rb')
-rw-r--r--activerecord/test/cases/migration/helper.rb9
1 files changed, 9 insertions, 0 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