etest-unit 0.7.0 → 0.7.1

data/lib/etest-unit/version.rb

@@ -1,3 +1,3 @@
 module Etest
-  VERSION = "0.7.0"
+  VERSION = "0.7.1"
 end

data/lib/etest-unit.rb

@@ -6,6 +6,26 @@ require File.dirname(__FILE__) + "/module_ext"
 require "test/unit/ui/console/testrunner"
 
 module Etest
+  class TestRunner < Test::Unit::UI::Console::TestRunner
+    def setup_mediator
+      super
+      @mediator = Etest::TestRunnerMediator.new(@suite)
+    end
+  end
+
+  class TestRunnerMediator < Test::Unit::UI::TestRunnerMediator
+    def run
+      if defined?(ActiveRecord)
+        ActiveRecord::Base.transaction do
+          super
+          raise ActiveRecord::Rollback, "Rollback test transaction"
+        end
+      else
+        super
+      end
+    end
+  end
+  
   class TestCase < Test::Unit::TestCase
     def self.etest=(etest)
       include @etest = etest
@@ -21,7 +41,7 @@ module Etest
   def self.run(etest)
     test_case_klass = Class.new(TestCase)
     test_case_klass.etest = etest
-    Test::Unit::UI::Console::TestRunner.new(test_case_klass).start
+    TestRunner.new(test_case_klass).start
   end
 end
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: etest-unit
 version: !ruby/object:Gem::Version
-  version: 0.7.0
+  version: 0.7.1
   prerelease: 
 platform: ruby
 authors:
@@ -53,7 +53,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: 1268382848584802831
+      hash: 2430133208310339074
 required_rubygems_version: !ruby/object:Gem::Requirement
   none: false
   requirements: