Added support for minitest under Ruby 1.9.

JEG2 · Aug 16, 2010 · f2ac27f · f2ac27f
1 parent c4549eb
commit f2ac27f
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/src/edgecase.rb b/src/edgecase.rb
@@ -46,7 +46,13 @@ module EdgeCase
   class Sensei
     attr_reader :failure, :failed_test
 
-    AssertionError = Test::Unit::AssertionFailedError
+    in_ruby_version("1.8") do
+      AssertionError = Test::Unit::AssertionFailedError
+    end
+
+    in_ruby_version("1.9") do
+      AssertionError = MiniTest::Assertion
+    end
 
     def initialize
       @pass_count = 0
@@ -169,12 +175,12 @@ def run_test(method, accumulator)
         test.setup
         begin
           test.send(method)
-        rescue StandardError => ex
+        rescue StandardError, EdgeCase::Sensei::AssertionError => ex
           test.failed(ex)
         ensure
           begin
             test.teardown
-          rescue StandardError => ex
+          rescue StandardError, EdgeCase::Sensei::AssertionError => ex
             test.failed(ex) if test.passed?
           end
         end