1
0
Fork 0
mirror of https://github.com/rails/rails.git synced 2022-11-09 12:12:34 -05:00

Crazy hacks to get the Isolation testing module to work on non forking environments

This commit is contained in:
Yehuda Katz + Carl Lerche 2009-06-30 17:26:46 -07:00
parent 45e6f19925
commit eea7b5db1d
3 changed files with 73 additions and 10 deletions

View file

@ -16,24 +16,83 @@ module ActiveSupport::Testing
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

View file

@ -1,3 +1,5 @@
ORIG_ARGV = ARGV.dup
require 'rubygems'
require 'test/unit'

View file

@ -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"