diff --git a/lib/flexmock/test_unit_integration.rb b/lib/flexmock/test_unit_integration.rb index f918324..d5b146a 100755 --- a/lib/flexmock/test_unit_integration.rb +++ b/lib/flexmock/test_unit_integration.rb @@ -45,6 +45,12 @@ def teardown class TestUnitFrameworkAdapter include Test::Unit::Assertions + attr_accessor :assertions + + def initialize + self.assertions = 0 + end + def make_assertion(msg, &block) unless yield msg = msg.call if msg.is_a?(Proc)