92 lines
3.0 KiB
Ruby
92 lines
3.0 KiB
Ruby
|
require 'active_support/callbacks'
|
||
|
|
||
|
module ActiveSupport
|
||
|
module Testing
|
||
|
module SetupAndTeardown
|
||
|
def self.included(base)
|
||
|
base.class_eval do
|
||
|
include ActiveSupport::Callbacks
|
||
|
define_callbacks :setup, :teardown
|
||
|
|
||
|
if defined?(MiniTest::Assertions) && TestCase < MiniTest::Assertions
|
||
|
include ForMiniTest
|
||
|
else
|
||
|
include ForClassicTestUnit
|
||
|
end
|
||
|
end
|
||
|
end
|
||
|
|
||
|
module ForMiniTest
|
||
|
def run(runner)
|
||
|
result = '.'
|
||
|
begin
|
||
|
run_callbacks :setup
|
||
|
result = super
|
||
|
rescue Exception => e
|
||
|
result = runner.puke(self.class, self.name, e)
|
||
|
ensure
|
||
|
begin
|
||
|
run_callbacks :teardown, :enumerator => :reverse_each
|
||
|
rescue Exception => e
|
||
|
result = runner.puke(self.class, self.name, e)
|
||
|
end
|
||
|
end
|
||
|
result
|
||
|
end
|
||
|
end
|
||
|
|
||
|
module ForClassicTestUnit
|
||
|
# For compatibility with Ruby < 1.8.6
|
||
|
PASSTHROUGH_EXCEPTIONS = Test::Unit::TestCase::PASSTHROUGH_EXCEPTIONS rescue [NoMemoryError, SignalException, Interrupt, SystemExit]
|
||
|
|
||
|
# This redefinition is unfortunate but test/unit shows us no alternative.
|
||
|
# Doubly unfortunate: hax to support Mocha's hax.
|
||
|
def run(result)
|
||
|
return if @method_name.to_s == "default_test"
|
||
|
|
||
|
if using_mocha = respond_to?(:mocha_verify)
|
||
|
assertion_counter_klass = if defined?(Mocha::TestCaseAdapter::AssertionCounter)
|
||
|
Mocha::TestCaseAdapter::AssertionCounter
|
||
|
else
|
||
|
Mocha::Integration::TestUnit::AssertionCounter
|
||
|
end
|
||
|
assertion_counter = assertion_counter_klass.new(result)
|
||
|
end
|
||
|
|
||
|
yield(Test::Unit::TestCase::STARTED, name)
|
||
|
@_result = result
|
||
|
begin
|
||
|
begin
|
||
|
run_callbacks :setup
|
||
|
setup
|
||
|
__send__(@method_name)
|
||
|
mocha_verify(assertion_counter) if using_mocha
|
||
|
rescue Mocha::ExpectationError => e
|
||
|
add_failure(e.message, e.backtrace)
|
||
|
rescue Test::Unit::AssertionFailedError => e
|
||
|
add_failure(e.message, e.backtrace)
|
||
|
rescue Exception => e
|
||
|
raise if PASSTHROUGH_EXCEPTIONS.include?(e.class)
|
||
|
add_error(e)
|
||
|
ensure
|
||
|
begin
|
||
|
teardown
|
||
|
run_callbacks :teardown, :enumerator => :reverse_each
|
||
|
rescue Test::Unit::AssertionFailedError => e
|
||
|
add_failure(e.message, e.backtrace)
|
||
|
rescue Exception => e
|
||
|
raise if PASSTHROUGH_EXCEPTIONS.include?(e.class)
|
||
|
add_error(e)
|
||
|
end
|
||
|
end
|
||
|
ensure
|
||
|
mocha_teardown if using_mocha
|
||
|
end
|
||
|
result.add_run
|
||
|
yield(Test::Unit::TestCase::FINISHED, name)
|
||
|
end
|
||
|
end
|
||
|
end
|
||
|
end
|
||
|
end
|