floehopper-mocha 0.9.3.20081220175348

data/lib/mocha/any_instance_method.rb

@@ -0,0 +1,54 @@
+require 'mocha/class_method'
+
+module Mocha
+
+  class AnyInstanceMethod < ClassMethod
+  
+    def unstub
+      remove_new_method
+      restore_original_method
+      stubbee.any_instance.reset_mocha
+    end
+    
+    def mock
+      stubbee.any_instance.mocha
+    end
+   
+    def hide_original_method
+      if method_exists?(method)
+        begin
+          stubbee.class_eval("alias_method :#{hidden_method}, :#{method}", __FILE__, __LINE__)
+        rescue NameError
+          # deal with nasties like ActiveRecord::Associations::AssociationProxy
+        end
+      end
+    end
+
+    def define_new_method
+      stubbee.class_eval("def #{method}(*args, &block); self.class.any_instance.mocha.method_missing(:#{method}, *args, &block); end", __FILE__, __LINE__)
+    end
+
+    def remove_new_method
+      stubbee.class_eval("remove_method :#{method}", __FILE__, __LINE__)
+    end
+
+    def restore_original_method
+      if method_exists?(hidden_method)
+        begin
+          stubbee.class_eval("alias_method :#{method}, :#{hidden_method}; remove_method :#{hidden_method}", __FILE__, __LINE__)
+        rescue NameError
+          # deal with nasties like ActiveRecord::Associations::AssociationProxy
+        end
+      end
+    end
+
+    def method_exists?(method)
+      return true if stubbee.public_instance_methods(false).include?(method)
+      return true if stubbee.protected_instance_methods(false).include?(method)
+      return true if stubbee.private_instance_methods(false).include?(method)
+      return false
+    end
+    
+  end
+  
+end

data/lib/mocha/argument_iterator.rb

@@ -0,0 +1,21 @@
+module Mocha
+  
+  class ArgumentIterator
+  
+    def initialize(argument)
+      @argument = argument
+    end
+  
+    def each(&block)
+      if @argument.is_a?(Hash) then
+        @argument.each do |method_name, return_value|
+          block.call(method_name, return_value)
+        end
+      else
+        block.call(@argument)
+      end
+    end
+  
+  end
+  
+end

data/lib/mocha/backtrace_filter.rb

@@ -0,0 +1,17 @@
+module Mocha
+
+  class BacktraceFilter
+  
+    LIB_DIRECTORY = File.expand_path(File.join(File.dirname(__FILE__), "..")) + File::SEPARATOR
+
+    def initialize(lib_directory = LIB_DIRECTORY)
+      @lib_directory = lib_directory
+    end
+
+    def filtered(backtrace)
+      backtrace.reject { |location| Regexp.new(@lib_directory).match(File.expand_path(location)) }
+    end
+
+  end
+
+end

data/lib/mocha/cardinality.rb

@@ -0,0 +1,95 @@
+module Mocha
+
+  class Cardinality
+    
+    INFINITY = 1 / 0.0
+    
+    class << self
+      
+      def exactly(count)
+        new(count, count)
+      end
+      
+      def at_least(count)
+        new(count, INFINITY)
+      end
+      
+      def at_most(count)
+        new(0, count)
+      end
+      
+      def times(range_or_count)
+        case range_or_count
+          when Range then new(range_or_count.first, range_or_count.last)
+          else new(range_or_count, range_or_count)
+        end
+      end
+      
+    end
+    
+    def initialize(required, maximum)
+      @required, @maximum = required, maximum
+    end
+    
+    def invocations_allowed?(invocation_count)
+      invocation_count < maximum
+    end
+    
+    def satisfied?(invocations_so_far)
+      invocations_so_far >= required
+    end
+    
+    def needs_verifying?
+      !allowed_any_number_of_times?
+    end
+    
+    def verified?(invocation_count)
+      (invocation_count >= required) && (invocation_count <= maximum)
+    end
+    
+    def allowed_any_number_of_times?
+      required == 0 && infinite?(maximum)
+    end
+    
+    def used?(invocation_count)
+      (invocation_count > 0) || (maximum == 0)
+    end
+    
+    def mocha_inspect
+      if allowed_any_number_of_times?
+        "allowed any number of times"
+      else
+        if required == 0 && maximum == 0
+          "expected never"
+        elsif required == maximum
+          "expected exactly #{times(required)}"
+        elsif infinite?(maximum)
+          "expected at least #{times(required)}"
+        elsif required == 0
+          "expected at most #{times(maximum)}"
+        else
+          "expected between #{required} and #{times(maximum)}"
+        end
+      end
+    end
+    
+    protected
+    
+    attr_reader :required, :maximum
+    
+    def times(number)
+      case number
+        when 0 then "no times"
+        when 1 then "once"
+        when 2 then "twice"
+        else "#{number} times"
+      end
+    end
+    
+    def infinite?(number)
+      number.respond_to?(:infinite?) && number.infinite?
+    end
+    
+  end
+  
+end

data/lib/mocha/central.rb

@@ -0,0 +1,27 @@
+module Mocha
+  
+  class Central
+  
+    attr_accessor :stubba_methods
+  
+    def initialize
+      self.stubba_methods = []
+    end
+   
+    def stub(method)
+      unless stubba_methods.include?(method)
+        method.stub 
+        stubba_methods.push(method)
+      end
+    end
+    
+    def unstub_all
+      while stubba_methods.length > 0
+        method = stubba_methods.pop
+        method.unstub
+      end
+    end
+
+  end
+
+end

data/lib/mocha/change_state_side_effect.rb

@@ -0,0 +1,19 @@
+module Mocha
+
+  class ChangeStateSideEffect
+  
+    def initialize(state)
+      @state = state
+    end
+  
+    def perform
+      @state.activate
+    end
+  
+    def mocha_inspect
+      "then #{@state.mocha_inspect}"
+    end
+  
+  end
+
+end

data/lib/mocha/class_method.rb

@@ -0,0 +1,86 @@
+require 'mocha/metaclass'
+
+module Mocha
+
+  class ClassMethod
+  
+    attr_reader :stubbee, :method
+   
+    def initialize(stubbee, method)
+      @stubbee = stubbee
+      @method = RUBY_VERSION < '1.9' ? method.to_s : method.to_sym
+    end
+  
+    def stub
+      hide_original_method
+      define_new_method
+    end
+  
+    def unstub
+      remove_new_method
+      restore_original_method
+      stubbee.reset_mocha
+    end
+    
+    def mock
+      stubbee.mocha
+    end
+  
+    def hide_original_method
+      if method_exists?(method)
+        begin
+          stubbee.__metaclass__.class_eval("alias_method :#{hidden_method}, :#{method}", __FILE__, __LINE__)
+        rescue NameError
+          # deal with nasties like ActiveRecord::Associations::AssociationProxy
+        end
+      end
+    end
+  
+    def define_new_method
+      stubbee.__metaclass__.class_eval("def #{method}(*args, &block); mocha.method_missing(:#{method}, *args, &block); end", __FILE__, __LINE__)
+    end
+  
+    def remove_new_method
+      stubbee.__metaclass__.class_eval("remove_method :#{method}", __FILE__, __LINE__)
+    end
+  
+    def restore_original_method
+      if method_exists?(hidden_method)
+        begin
+          stubbee.__metaclass__.class_eval("alias_method :#{method}, :#{hidden_method}; remove_method :#{hidden_method}", __FILE__, __LINE__)
+        rescue NameError
+          # deal with nasties like ActiveRecord::Associations::AssociationProxy
+        end
+      end
+    end
+  
+    def hidden_method
+      if RUBY_VERSION < '1.9'
+        method_name = method.to_s.gsub(/\W/) { |s| "_substituted_character_#{s[0]}_" }
+      else
+        method_name = method.to_s.gsub(/\W/) { |s| "_substituted_character_#{s.ord}_" }
+      end
+      hidden_method = "__stubba__#{method_name}__stubba__"
+      RUBY_VERSION < '1.9' ? hidden_method.to_s : hidden_method.to_sym
+    end  
+  
+    def eql?(other)
+      return false unless (other.class == self.class)
+      (stubbee.object_id == other.stubbee.object_id) and (method == other.method)
+    end
+  
+    alias_method :==, :eql?
+  
+    def to_s
+      "#{stubbee}.#{method}"
+    end
+    
+    def method_exists?(method)
+      symbol = method.to_sym
+      metaclass = stubbee.__metaclass__
+      metaclass.public_method_defined?(symbol) || metaclass.protected_method_defined?(symbol) || metaclass.private_method_defined?(symbol)
+    end
+
+  end
+  
+end

data/lib/mocha/configuration.rb

@@ -0,0 +1,60 @@
+module Mocha # :nodoc:
+  
+  # Configuration settings
+  class Configuration
+    
+    DEFAULTS = { :stubbing_method_unnecessarily => :allow, :stubbing_method_on_non_mock_object => :allow, :stubbing_non_existent_method => :allow, :stubbing_non_public_method => :allow }
+    
+    class << self
+    
+      # :call-seq: allow(action)
+      #
+      # Allow the specified <tt>action</tt> (as a symbol).
+      # The <tt>actions</tt> currently available are <tt>:stubbing_method_unnecessarily, :stubbing_method_on_non_mock_object, :stubbing_non_existent_method, :stubbing_non_public_method</tt>.
+      def allow(action)
+        configuration[action] = :allow
+      end
+    
+      def allow?(action) # :nodoc:
+        configuration[action] == :allow
+      end
+    
+      # :call-seq: warn_when(action)
+      #
+      # Warn if the specified <tt>action</tt> (as a symbol) is attempted.
+      # The <tt>actions</tt> currently available are <tt>:stubbing_method_unnecessarily, :stubbing_method_on_non_mock_object, :stubbing_non_existent_method, :stubbing_non_public_method</tt>.
+      def warn_when(action)
+        configuration[action] = :warn
+      end
+    
+      def warn_when?(action) # :nodoc:
+        configuration[action] == :warn
+      end
+    
+      # :call-seq: prevent(action)
+      #
+      # Raise a StubbingError if the specified <tt>action</tt> (as a symbol) is attempted.
+      # The <tt>actions</tt> currently available are <tt>:stubbing_method_unnecessarily, :stubbing_method_on_non_mock_object, :stubbing_non_existent_method, :stubbing_non_public_method</tt>.
+      def prevent(action)
+        configuration[action] = :prevent
+      end
+    
+      def prevent?(action) # :nodoc:
+        configuration[action] == :prevent
+      end
+    
+      def reset_configuration # :nodoc:
+        @configuration = nil
+      end
+    
+      private
+    
+      def configuration # :nodoc:
+        @configuration ||= DEFAULTS.dup
+      end
+    
+    end
+    
+  end
+  
+end

data/lib/mocha/deprecation.rb

@@ -0,0 +1,22 @@
+module Mocha
+  
+  class Deprecation
+    
+    class << self
+      
+      attr_accessor :mode, :messages
+      
+      def warning(message)
+        @messages << message
+        $stderr.puts "Mocha deprecation warning: #{message}" unless mode == :disabled
+        $stderr.puts caller.join("\n  ") if mode == :debug
+      end
+
+    end
+  
+    self.mode = :enabled
+    self.messages = []
+    
+  end
+   
+end

data/lib/mocha/exception_raiser.rb

@@ -0,0 +1,17 @@
+module Mocha # :nodoc:
+  
+  class ExceptionRaiser # :nodoc:
+    
+    def initialize(exception, message)
+      @exception, @message = exception, message
+    end
+    
+    def evaluate
+      raise @exception, @exception.to_s if @exception.is_a?(Module) && @exception.ancestors.include?(Interrupt)
+      raise @exception, @message if @message
+      raise @exception
+    end
+    
+  end
+  
+end