etest-unit 0.8.0 → 0.8.1

data/lib/etest-unit/version.rb

@@ -1,3 +1,3 @@
 module EtestUnit
-  VERSION = "0.8.0"
+  VERSION = "0.8.1"
 end

data/lib/etest-unit.rb

@@ -21,6 +21,38 @@ module EtestUnit
   class TestSuiteCreator < Test::Unit::TestSuiteCreator
     attr :tests, true
     
+    module ActiveRecordAdditions
+      def set_log_level(level)
+        return unless ActiveRecord::Base.logger
+        
+        r = ActiveRecord::Base.logger.level
+        ActiveRecord::Base.logger.level = level
+        r
+      end
+      
+      def run_test(test, result)
+        old_log_level = set_log_level(Logger::ERROR)
+        ActiveRecord::Base.transaction do
+          set_log_level(old_log_level)
+
+          super(test, result)
+
+          set_log_level(Logger::ERROR)
+          raise ActiveRecord::Rollback, "Rollback test transaction"
+        end
+      ensure
+        set_log_level(old_log_level)
+      end
+    end
+    
+    def create
+      suite = super
+      if defined?(::ActiveRecord)
+        suite.extend ActiveRecordAdditions
+      end
+      suite
+    end
+    
     private
 
     def collect_test_names
@@ -53,17 +85,6 @@ module EtestUnit
       suite.name = @etest.name
       suite
     end
-    
-    def run_test
-      if defined?(ActiveRecord)
-        ActiveRecord::Base.transaction do
-          super
-          raise ActiveRecord::Rollback, "Rollback test transaction"
-        end
-      else
-        super
-      end
-    end
   end
   
   def self.run(etest, *tests)

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: etest-unit
 version: !ruby/object:Gem::Version
-  version: 0.8.0
+  version: 0.8.1
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-07-17 00:00:00.000000000 Z
+date: 2013-07-19 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: test-unit
@@ -53,7 +53,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: 3935101071244175937
+      hash: -2665298691011624838
 required_rubygems_version: !ruby/object:Gem::Requirement
   none: false
   requirements: