Uses M::Unit::after_tests instead of M::Unit#status to trigger finish…

… method. Fixes #3
bhenderson · Apr 30, 2012 · e4d9850 · e4d9850
1 parent f38f833
commit e4d9850
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/lib/minitest/ci.rb b/lib/minitest/ci.rb
@@ -97,14 +97,13 @@ def type e
 
   class CiUnit < Unit
 
-    def record suite, method, assertions, time, error
+    after_tests do
-      MiniTest::Ci.push suite, method, assertions, time, error
+      MiniTest::Ci.finish self.output
-      super
     end
 
-    def status io = self.output
+    def record suite, method, assertions, time, error
+      MiniTest::Ci.push suite, method, assertions, time, error
       super
-      MiniTest::Ci.finish io
     end
 
   end