mocha 0.12.10 → 0.13.0

data/lib/mocha/integration/mini_test/version_230_to_2101.rb

@@ -1,59 +1,65 @@
-require 'mocha/integration/mini_test/assertion_counter'
-require 'mocha/expectation_error'
+require 'mocha/integration/assertion_counter'
+require 'mocha/integration/monkey_patcher'
+require 'mocha/integration/mini_test/exception_translation'
 
 module Mocha
-
   module Integration
-
     module MiniTest
-
       module Version230To2101
+        def self.applicable_to?(mini_test_version)
+          Gem::Requirement.new('>= 2.3.0', '<= 2.10.1').satisfied_by?(mini_test_version)
+        end
+
+        def self.description
+          "monkey patch for MiniTest gem >= v2.3.0 <= v2.10.1"
+        end
+
         def self.included(mod)
-          $stderr.puts "Monkey patching MiniTest >= v2.3.0 <= v2.10.1" if $mocha_options['debug']
+          MonkeyPatcher.apply(mod, RunMethodPatch)
         end
-        def run runner
-          trap 'INFO' do
-            time = runner.start_time ? Time.now - runner.start_time : 0
-            warn "%s#%s %.2fs" % [self.class, self.__name__, time]
-            runner.status $stderr
-          end if ::MiniTest::Unit::TestCase::SUPPORTS_INFO_SIGNAL
 
-          assertion_counter = AssertionCounter.new(self)
-          result = ""
-          begin
+        module RunMethodPatch
+          def run runner
+            trap 'INFO' do
+              time = runner.start_time ? Time.now - runner.start_time : 0
+              warn "%s#%s %.2fs" % [self.class, self.__name__, time]
+              runner.status $stderr
+            end if ::MiniTest::Unit::TestCase::SUPPORTS_INFO_SIGNAL
+
+            assertion_counter = AssertionCounter.new(self)
+            result = ""
             begin
-              @passed = nil
-              self.setup
-              self.run_setup_hooks
-              self.__send__ self.__name__
-              mocha_verify(assertion_counter)
-              result = "." unless io?
-              @passed = true
-            rescue *::MiniTest::Unit::TestCase::PASSTHROUGH_EXCEPTIONS
-              raise
-            rescue Exception => e
-              @passed = false
-              result = runner.puke self.class, self.__name__, Mocha::Integration::MiniTest.translate(e)
-            ensure
               begin
-                self.run_teardown_hooks
-                self.teardown
+                @passed = nil
+                self.setup
+                self.run_setup_hooks
+                self.__send__ self.__name__
+                mocha_verify(assertion_counter)
+                result = "." unless io?
+                @passed = true
               rescue *::MiniTest::Unit::TestCase::PASSTHROUGH_EXCEPTIONS
                 raise
               rescue Exception => e
+                @passed = false
                 result = runner.puke self.class, self.__name__, Mocha::Integration::MiniTest.translate(e)
+              ensure
+                begin
+                  self.run_teardown_hooks
+                  self.teardown
+                rescue *::MiniTest::Unit::TestCase::PASSTHROUGH_EXCEPTIONS
+                  raise
+                rescue Exception => e
+                  result = runner.puke self.class, self.__name__, Mocha::Integration::MiniTest.translate(e)
+                end
+                trap 'INFO', 'DEFAULT' if ::MiniTest::Unit::TestCase::SUPPORTS_INFO_SIGNAL
               end
-              trap 'INFO', 'DEFAULT' if ::MiniTest::Unit::TestCase::SUPPORTS_INFO_SIGNAL
+            ensure
+              mocha_teardown
             end
-          ensure
-            mocha_teardown
+            result
           end
-          result
         end
       end
-
     end
-
   end
-
 end

data/lib/mocha/integration/monkey_patcher.rb

@@ -0,0 +1,18 @@
+require 'mocha/api'
+
+module Mocha
+  module Integration
+    module MonkeyPatcher
+      def self.apply(mod, run_method_patch)
+        unless mod < Mocha::API
+          mod.send(:include, Mocha::API)
+        end
+        unless mod.method_defined?(:run_before_mocha)
+          mod.send(:alias_method, :run_before_mocha, :run)
+          mod.send(:remove_method, :run)
+          mod.send(:include, run_method_patch)
+        end
+      end
+    end
+  end
+end

data/lib/mocha/integration/test_unit.rb

@@ -1,56 +1,49 @@
-require 'mocha/api'
-require 'mocha/options'
-
-if !Test::Unit::TestCase.ancestors.include?(Mocha::API)
-
-  require 'mocha/integration/test_unit/gem_version_200'
-  require 'mocha/integration/test_unit/gem_version_201_to_202'
-  require 'mocha/integration/test_unit/gem_version_203_to_220'
-  require 'mocha/integration/test_unit/gem_version_230_to_254'
-  require 'mocha/integration/test_unit/ruby_version_185_and_below'
-  require 'mocha/integration/test_unit/ruby_version_186_and_above'
-
-  module Test
-    module Unit
-      class TestCase
-
-        include Mocha::API
-
-        alias_method :run_before_mocha, :run
-        remove_method :run
+require 'mocha/debug'
+
+require 'mocha/integration/test_unit/nothing'
+require 'mocha/integration/test_unit/ruby_version_185_and_below'
+require 'mocha/integration/test_unit/ruby_version_186_and_above'
+require 'mocha/integration/test_unit/gem_version_200'
+require 'mocha/integration/test_unit/gem_version_201_to_202'
+require 'mocha/integration/test_unit/gem_version_203_to_220'
+require 'mocha/integration/test_unit/gem_version_230_to_250'
+require 'mocha/integration/test_unit/adapter'
+
+module Mocha
+  module Integration
+    module TestUnit
+      def self.activate
+        return false unless defined?(::Test::Unit::TestCase) && !(defined?(::MiniTest::Unit::TestCase) && (::Test::Unit::TestCase < ::MiniTest::Unit::TestCase))
 
         test_unit_version = begin
           load 'test/unit/version.rb'
-          Gem::Version.new(Test::Unit::VERSION)
+          Gem::Version.new(::Test::Unit::VERSION)
         rescue LoadError
-          Gem::Version.new('1.x')
+          Gem::Version.new('1.0.0')
         end
 
-        if $mocha_options['debug']
-          $stderr.puts "Detected Ruby version: #{RUBY_VERSION}"
-          $stderr.puts "Detected Test::Unit version: #{test_unit_version}"
+        ruby_version = Gem::Version.new(RUBY_VERSION.dup)
+
+        Debug.puts "Detected Ruby version: #{ruby_version}"
+        Debug.puts "Detected Test::Unit version: #{test_unit_version}"
+
+        integration_module = [
+          TestUnit::Adapter,
+          TestUnit::GemVersion230To250,
+          TestUnit::GemVersion203To220,
+          TestUnit::GemVersion201To202,
+          TestUnit::GemVersion200,
+          TestUnit::RubyVersion186AndAbove,
+          TestUnit::RubyVersion185AndBelow,
+          TestUnit::Nothing
+        ].detect { |m| m.applicable_to?(test_unit_version, ruby_version) }
+
+        unless ::Test::Unit::TestCase < integration_module
+          Debug.puts "Applying #{integration_module.description}"
+          ::Test::Unit::TestCase.send(:include, integration_module)
         end
-
-        if (test_unit_version == Gem::Version.new('1.x')) || (test_unit_version == Gem::Version.new('1.2.3'))
-          if RUBY_VERSION < '1.8.6'
-            include Mocha::Integration::TestUnit::RubyVersion185AndBelow
-          else
-            include Mocha::Integration::TestUnit::RubyVersion186AndAbove
-          end
-        elsif Gem::Requirement.new('2.0.0').satisfied_by?(test_unit_version)
-          include Mocha::Integration::TestUnit::GemVersion200
-        elsif Gem::Requirement.new('>= 2.0.1', '<= 2.0.2').satisfied_by?(test_unit_version)
-          include Mocha::Integration::TestUnit::GemVersion201To202
-        elsif Gem::Requirement.new('>= 2.0.3', '<= 2.2.0').satisfied_by?(test_unit_version)
-          include Mocha::Integration::TestUnit::GemVersion203To220
-        elsif Gem::Requirement.new('>= 2.3.0', '<= 2.5.4').satisfied_by?(test_unit_version)
-          include Mocha::Integration::TestUnit::GemVersion230To254
-        else
-          raise "No Mocha monkey-patch for Test::Unit version"
-        end
-
+        true
       end
     end
   end
-
 end

data/lib/mocha/integration/test_unit/adapter.rb

@@ -0,0 +1,51 @@
+require 'mocha/api'
+require 'mocha/integration/assertion_counter'
+require 'mocha/expectation_error'
+
+module Mocha
+  module Integration
+    module TestUnit
+
+      # Integrates Mocha into recent versions of Test::Unit.
+      #
+      # See the source code for an example of how to integrate Mocha into a test library.
+      module Adapter
+        include Mocha::API
+
+        # @private
+        def self.applicable_to?(test_unit_version, ruby_version = nil)
+          Gem::Requirement.new('>= 2.5.1').satisfied_by?(test_unit_version)
+        end
+
+        # @private
+        def self.description
+          "adapter for Test::Unit gem >= v2.5.1"
+        end
+
+        # @private
+        def self.included(mod)
+          mod.setup :mocha_setup, :before => :prepend
+
+          mod.exception_handler(:handle_mocha_expectation_error)
+
+          mod.cleanup :after => :append do
+            assertion_counter = Integration::AssertionCounter.new(self)
+            mocha_verify(assertion_counter)
+          end
+
+          mod.teardown :mocha_teardown, :after => :append
+        end
+
+        private
+
+        # @private
+        def handle_mocha_expectation_error(e)
+          return false unless e.is_a?(Mocha::ExpectationError)
+          problem_occurred
+          add_failure(e.message, e.backtrace)
+          true
+        end
+      end
+    end
+  end
+end

data/lib/mocha/integration/test_unit/gem_version_200.rb

@@ -1,52 +1,59 @@
-require 'test/unit/testcase'
-require 'mocha/integration/test_unit/assertion_counter'
+require 'mocha/integration/assertion_counter'
+require 'mocha/integration/monkey_patcher'
 require 'mocha/expectation_error'
 
 module Mocha
-
   module Integration
-
     module TestUnit
-
       module GemVersion200
+        def self.applicable_to?(test_unit_version, ruby_version = nil)
+          Gem::Requirement.new('2.0.0').satisfied_by?(test_unit_version)
+        end
+
+        def self.description
+          "monkey patch for Test::Unit gem v2.0.0"
+        end
+
         def self.included(mod)
-          $stderr.puts "Monkey patching Test::Unit gem v2.0.0" if $mocha_options['debug']
+          MonkeyPatcher.apply(mod, RunMethodPatch)
         end
-        def run(result)
-          assertion_counter = AssertionCounter.new(result)
-          begin
-            @_result = result
-            yield(Test::Unit::TestCase::STARTED, name)
+
+        module RunMethodPatch
+          def run(result)
+            assertion_counter = AssertionCounter.new(self)
             begin
+              @_result = result
+              yield(Test::Unit::TestCase::STARTED, name)
               begin
-                run_setup
-                __send__(@method_name)
-                mocha_verify(assertion_counter)
-              rescue Mocha::ExpectationError => e
-                add_failure(e.message, e.backtrace)
-              rescue Exception
-                @interrupted = true
-                raise unless handle_exception($!)
-              ensure
                 begin
-                  run_teardown
+                  run_setup
+                  __send__(@method_name)
+                  mocha_verify(assertion_counter)
+                rescue Mocha::ExpectationError => e
+                  add_failure(e.message, e.backtrace)
                 rescue Exception
+                  @interrupted = true
                   raise unless handle_exception($!)
+                ensure
+                  begin
+                    run_teardown
+                  rescue Mocha::ExpectationError => e
+                    add_failure(e.message, e.backtrace)
+                  rescue Exception
+                    raise unless handle_exception($!)
+                  end
                 end
+              ensure
+                mocha_teardown
               end
+              result.add_run
+              yield(Test::Unit::TestCase::FINISHED, name)
             ensure
-              mocha_teardown
+              @_result = nil
             end
-            result.add_run
-            yield(Test::Unit::TestCase::FINISHED, name)
-          ensure
-            @_result = nil
           end
         end
       end
-
     end
-
   end
-
 end

data/lib/mocha/integration/test_unit/gem_version_201_to_202.rb

@@ -1,52 +1,59 @@
-require 'test/unit/testcase'
-require 'mocha/integration/test_unit/assertion_counter'
+require 'mocha/integration/assertion_counter'
+require 'mocha/integration/monkey_patcher'
 require 'mocha/expectation_error'
 
 module Mocha
-
   module Integration
-
     module TestUnit
-
       module GemVersion201To202
+        def self.applicable_to?(test_unit_version, ruby_version = nil)
+          Gem::Requirement.new('>= 2.0.1', '<= 2.0.2').satisfied_by?(test_unit_version)
+        end
+
+        def self.description
+          "monkey patch for Test::Unit gem >= v2.0.1 and <= v2.0.2"
+        end
+
         def self.included(mod)
-          $stderr.puts "Monkey patching Test::Unit gem >= v2.0.1 and <= v2.0.2" if $mocha_options['debug']
+          MonkeyPatcher.apply(mod, RunMethodPatch)
         end
-        def run(result)
-          assertion_counter = AssertionCounter.new(result)
-          begin
-            @_result = result
-            yield(Test::Unit::TestCase::STARTED, name)
+
+        module RunMethodPatch
+          def run(result)
+            assertion_counter = AssertionCounter.new(self)
             begin
+              @_result = result
+              yield(Test::Unit::TestCase::STARTED, name)
               begin
-                run_setup
-                run_test
-                mocha_verify(assertion_counter)
-              rescue Mocha::ExpectationError => e
-                add_failure(e.message, e.backtrace)
-              rescue Exception
-                @interrupted = true
-                raise unless handle_exception($!)
-              ensure
                 begin
-                  run_teardown
+                  run_setup
+                  run_test
+                  mocha_verify(assertion_counter)
+                rescue Mocha::ExpectationError => e
+                  add_failure(e.message, e.backtrace)
                 rescue Exception
+                  @interrupted = true
                   raise unless handle_exception($!)
+                ensure
+                  begin
+                    run_teardown
+                  rescue Mocha::ExpectationError => e
+                    add_failure(e.message, e.backtrace)
+                  rescue Exception
+                    raise unless handle_exception($!)
+                  end
                 end
+              ensure
+                mocha_teardown
               end
+              result.add_run
+              yield(Test::Unit::TestCase::FINISHED, name)
             ensure
-              mocha_teardown
+              @_result = nil
             end
-            result.add_run
-            yield(Test::Unit::TestCase::FINISHED, name)
-          ensure
-            @_result = nil
           end
         end
       end
-
     end
-
   end
-
 end

data/lib/mocha/integration/test_unit/gem_version_203_to_220.rb

@@ -1,52 +1,59 @@
-require 'test/unit/testcase'
-require 'mocha/integration/test_unit/assertion_counter'
+require 'mocha/integration/assertion_counter'
+require 'mocha/integration/monkey_patcher'
 require 'mocha/expectation_error'
 
 module Mocha
-
   module Integration
-
     module TestUnit
-
       module GemVersion203To220
+        def self.applicable_to?(test_unit_version, ruby_version = nil)
+          Gem::Requirement.new('>= 2.0.3', '<= 2.2.0').satisfied_by?(test_unit_version)
+        end
+
+        def self.description
+          "monkey patch for Test::Unit gem >= v2.0.3 and <= v2.2.0"
+        end
+
         def self.included(mod)
-          $stderr.puts "Monkey patching Test::Unit gem >= v2.0.3 and <= v2.2.0" if $mocha_options['debug']
+          MonkeyPatcher.apply(mod, RunMethodPatch)
         end
-        def run(result)
-          assertion_counter = AssertionCounter.new(result)
-          begin
-            @_result = result
-            yield(Test::Unit::TestCase::STARTED, name)
+
+        module RunMethodPatch
+          def run(result)
+            assertion_counter = AssertionCounter.new(self)
             begin
+              @_result = result
+              yield(Test::Unit::TestCase::STARTED, name)
               begin
-                run_setup
-                run_test
-                mocha_verify(assertion_counter)
-              rescue Mocha::ExpectationError => e
-                add_failure(e.message, e.backtrace)
-              rescue Exception
-                @interrupted = true
-                raise unless handle_exception($!)
-              ensure
                 begin
-                  run_teardown
+                  run_setup
+                  run_test
+                  mocha_verify(assertion_counter)
+                rescue Mocha::ExpectationError => e
+                  add_failure(e.message, e.backtrace)
                 rescue Exception
+                  @interrupted = true
                   raise unless handle_exception($!)
+                ensure
+                  begin
+                    run_teardown
+                  rescue Mocha::ExpectationError => e
+                    add_failure(e.message, e.backtrace)
+                  rescue Exception
+                    raise unless handle_exception($!)
+                  end
                 end
+              ensure
+                mocha_teardown
               end
+              result.add_run
+              yield(Test::Unit::TestCase::FINISHED, name)
             ensure
-              mocha_teardown
+              # @_result = nil # For test-spec's after_all :<
             end
-            result.add_run
-            yield(Test::Unit::TestCase::FINISHED, name)
-          ensure
-            # @_result = nil # For test-spec's after_all :<
           end
         end
       end
-
     end
-
   end
-
 end