Permalink
Browse files

Crazy hacks to get the Isolation testing module to work on non forkin…

…g environments
  • Loading branch information...
1 parent 45e6f19 commit eea7b5db1db5d7e6020c5bcb6b4d85afcbc2e696 Yehuda Katz + Carl Lerche committed Jul 1, 2009
@@ -16,24 +16,83 @@ def method_missing(name, *args)
end
module Isolation
+ def self.forking_env?
+ !ENV["NO_FORK"] && RUBY_PLATFORM !~ /mswin|mingw|java/
+ end
+
def run(result)
yield(Test::Unit::TestCase::STARTED, name)
- read, write = IO.pipe
+ @_result = result
- pid = fork do
- # child
- read.close
- proxy = ProxyTestResult.new
+ proxy = run_in_isolation do |proxy|
super(proxy) { }
- write.puts [Marshal.dump(proxy)].pack("m")
- exit!
end
- write.close
- Marshal.load(read.read.unpack("m")[0]).__replay__(result)
- Process.wait2(pid)
+ proxy.__replay__(@_result)
+
yield(Test::Unit::TestCase::FINISHED, name)
end
+
+ module Forking
+ def run_in_isolation(&blk)
+ read, write = IO.pipe
+
+ pid = fork do
+ read.close
+ proxy = ProxyTestResult.new
+ yield proxy
+ write.puts [Marshal.dump(proxy)].pack("m")
+ exit!
+ end
+
+ write.close
+ result = read.read
+ Process.wait2(pid)
+ Marshal.load(result.unpack("m")[0])
+ end
+ end
+
+ module Subprocess
+ # Crazy H4X to get this working in windows / jruby with
+ # no forking.
+ def run_in_isolation(&blk)
+ require "tempfile"
+
+ if ENV["ISOLATION_TEST"]
+ proxy = ProxyTestResult.new
+ yield proxy
+ File.open(ENV["ISOLATION_OUTPUT"], "w") do |file|
+ file.puts [Marshal.dump(proxy)].pack("m")
+ end
+ exit!
+ else
+ Tempfile.open("isolation") do |tmpfile|
+ ENV["ISOLATION_TEST"] = @method_name
+ ENV["ISOLATION_OUTPUT"] = tmpfile.path
+
+ load_paths = $-I.map {|p| "-I\"#{File.expand_path(p)}\"" }.join(" ")
+ `#{Gem.ruby} #{load_paths} #{$0} #{ORIG_ARGV.join(" ")} -t\"#{self.class}\"`
+
+ ENV.delete("ISOLATION_TEST")
+ ENV.delete("ISOLATION_OUTPUT")
+
+ return Marshal.load(tmpfile.read.unpack("m")[0])
+ end
+ end
+ end
+ end
+
+ include forking_env? ? Forking : Subprocess
+ end
+end
+
+# Only in subprocess for windows / jruby.
+if ENV['ISOLATION_TEST']
+ require "test/unit/collector/objectspace"
+ class Test::Unit::Collector::ObjectSpace
+ def include?(test)
+ super && test.method_name == ENV['ISOLATION_TEST']
+ end
end
end
@@ -1,3 +1,5 @@
+ORIG_ARGV = ARGV.dup
+
require 'rubygems'
require 'test/unit'
@@ -1,3 +1,5 @@
+ORIG_ARGV = ARGV.dup
+
$:.unshift File.dirname(__FILE__) + "/../../activesupport/lib"
$:.unshift File.dirname(__FILE__) + "/../../activerecord/lib"
$:.unshift File.dirname(__FILE__) + "/../../actionpack/lib"

0 comments on commit eea7b5d

Please sign in to comment.