Allow ActiveSupport's isolation tests to run with MiniTest on 1.9

activesupport/lib/active_support/testing/isolation.rb

@@ -36,6 +36,15 @@ module ActiveSupport
         !ENV["NO_FORK"] && RUBY_PLATFORM !~ /mswin|mingw|java/
       end
 
+      def self.included(base)
+        if defined?(::MiniTest) && base < ::MiniTest::Unit::TestCase
+          base.send :include, MiniTest
+        elsif defined?(Test::Unit)
+          base.send :include, TestUnit
+        end
+      end
+
+      module TestUnit
         def run(result)
           unless defined?(@@ran_class_setup)
             self.class.setup if self.class.respond_to?(:setup)
@@ -54,10 +63,29 @@ module ActiveSupport
             end
           end
 
-        proxy = Marshal.load(serialized)
+          retval, proxy = Marshal.load(serialized)
           proxy.__replay__(@_result)
 
           yield(Test::Unit::TestCase::FINISHED, name)
+          retval
+        end
+      end
+
+      module MiniTest
+        def run(runner)
+          unless defined?(@@ran_class_setup)
+            self.class.setup if self.class.respond_to?(:setup)
+            @@ran_class_setup = true
+          end
+
+          serialized = run_in_isolation do |runner|
+            super(runner)
+          end
+
+          retval, proxy = Marshal.load(serialized)
+          proxy.__replay__(runner)
+          retval
+        end
       end
 
       module Forking
@@ -67,8 +95,8 @@ module ActiveSupport
           pid = fork do
             read.close
             proxy = ProxyTestResult.new
-            yield proxy
-            write.puts [Marshal.dump(proxy)].pack("m")
+            retval = yield proxy
+            write.puts [Marshal.dump([retval, proxy])].pack("m")
             exit!
           end
 
@@ -87,9 +115,9 @@ module ActiveSupport
 
           if ENV["ISOLATION_TEST"]
             proxy = ProxyTestResult.new
-            yield proxy
+            retval = yield proxy
             File.open(ENV["ISOLATION_OUTPUT"], "w") do |file|
-              file.puts [Marshal.dump(proxy)].pack("m")
+              file.puts [Marshal.dump([retval, proxy])].pack("m")
             end
             exit!
           else

activesupport/test/isolation_test.rb

@@ -1,11 +1,9 @@
 require 'abstract_unit'
 require 'rbconfig'
 
-if defined?(MiniTest) || defined?(Test::Unit::TestResultFailureSupport)
-  $stderr.puts "Isolation tests can test test-unit 1 only"
+# if defined?(MiniTest) || defined?(Test::Unit::TestResultFailureSupport)
+#   $stderr.puts "Isolation tests can test test-unit 1 only"
 
-else
-  # Does awesome
 if ENV['CHILD']
   class ChildIsolationTest < ActiveSupport::TestCase
     include ActiveSupport::Testing::Isolation
@@ -82,7 +80,10 @@ else
     ENV.delete("CHILD")
 
     def setup
-        # Extract the results
+      defined?(::MiniTest) ? parse_minitest : parse_testunit
+    end
+
+    def parse_testunit
       @results = {}
       OUTPUT[/Started\n\s*(.*)\s*\nFinished/mi, 1].to_s.split(/\s*\n\s*/).each do |result|
         result =~ %r'^(\w+)\(\w+\):\s*(\.|E|F)$'
@@ -98,16 +99,36 @@ else
       end
     end
 
+    def parse_minitest
+      @results = {}
+      OUTPUT[/Started\n\s*(.*)\s*\nFinished/mi, 1].to_s.split(/\s*\n\s*/).each do |result|
+        result =~ %r'^\w+#(\w+):.*:\s*(.*Assertion.*|.*RuntimeError.*|\.\s*)$'
+        val = :success
+        val = :error if $2.include?('RuntimeError')
+        val = :failure if $2.include?('Assertion')
+
+        @results[$1] = val
+      end
+
+      # Extract the backtraces
+      @backtraces = {}
+      OUTPUT.scan(/^\s*\d+\).*?\n\n/m).each do |backtrace|
+        # \n  1) Error:\ntest_captures_errors(ChildIsolationTest):
+        backtrace =~ %r'\s*\d+\)\s*(Error|Failure):\n(\w+)'i
+        @backtraces[$2] = { :type => $1, :output => backtrace }
+      end
+    end
+
     def assert_failing(name)
-        assert_equal :failure, @results[name.to_s], "Test #{name} did not fail"
+      assert_equal :failure, @results[name.to_s], "Test #{name} failed"
     end
 
     def assert_passing(name)
-        assert_equal :success, @results[name.to_s], "Test #{name} did not pass"
+      assert_equal :success, @results[name.to_s], "Test #{name} passed"
     end
 
     def assert_erroring(name)
-        assert_equal :error, @results[name.to_s], "Test #{name} did not error"
+      assert_equal :error, @results[name.to_s], "Test #{name} errored"
     end
 
     test "has all tests" do
@@ -144,12 +165,12 @@ else
 
     test "backtrace is printed for errors" do
       assert_equal 'Error', @backtraces["test_captures_errors"][:type]
-        assert_match %r{isolation_test.rb:\d+:in `test_captures_errors'}, @backtraces["test_captures_errors"][:output]
+      assert_match %r{isolation_test.rb:\d+}, @backtraces["test_captures_errors"][:output]
     end
 
     test "backtrace is printed for failures" do
       assert_equal 'Failure', @backtraces["test_captures_failures"][:type]
-        assert_match %r{isolation_test.rb:\d+:in `test_captures_failures'}, @backtraces["test_captures_failures"][:output]
+      assert_match %r{isolation_test.rb:\d+}, @backtraces["test_captures_failures"][:output]
     end
 
     test "self.setup is run only once" do
@@ -159,4 +180,3 @@ else
 
   end
 end
-end