1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103
|
require 'mocha/expectation_error'
module Mocha
module TestCaseAdapter
class AssertionCounter
def initialize(test_result)
@test_result = test_result
end
def increment
@test_result.add_assertion
end
end
def self.included(base)
if RUBY_VERSION < '1.8.6'
base.class_eval do
alias_method :run_before_mocha_test_case_adapter, :run
def run(result)
assertion_counter = AssertionCounter.new(result)
yield(Test::Unit::TestCase::STARTED, name)
@_result = result
begin
begin
setup
__send__(@method_name)
mocha_verify(assertion_counter)
rescue Mocha::ExpectationError => e
add_failure(e.message, e.backtrace)
rescue Test::Unit::AssertionFailedError => e
add_failure(e.message, e.backtrace)
rescue StandardError, ScriptError
add_error($!)
ensure
begin
teardown
rescue Test::Unit::AssertionFailedError => e
add_failure(e.message, e.backtrace)
rescue StandardError, ScriptError
add_error($!)
end
end
ensure
mocha_teardown
end
result.add_run
yield(Test::Unit::TestCase::FINISHED, name)
end
end
else
base.class_eval do
alias_method :run_before_mocha_test_case_adapter, :run
def run(result)
assertion_counter = AssertionCounter.new(result)
yield(Test::Unit::TestCase::STARTED, name)
@_result = result
begin
begin
setup
__send__(@method_name)
mocha_verify(assertion_counter)
rescue Mocha::ExpectationError => e
add_failure(e.message, e.backtrace)
rescue Test::Unit::AssertionFailedError => e
add_failure(e.message, e.backtrace)
rescue Exception
raise if Test::Unit::TestCase::PASSTHROUGH_EXCEPTIONS.include? $!.class
add_error($!)
ensure
begin
teardown
rescue Test::Unit::AssertionFailedError => e
add_failure(e.message, e.backtrace)
rescue Exception
raise if Test::Unit::TestCase::PASSTHROUGH_EXCEPTIONS.include? $!.class
add_error($!)
end
end
ensure
mocha_teardown
end
result.add_run
yield(Test::Unit::TestCase::FINISHED, name)
end
end
end
end
end
end
|