Contents

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 verify_all(&block)
      unique_mocks.each { |mock| mock.verify(&block) }
    end
    
    def unique_mocks
      stubba_methods.inject({}) { |mocks, method| mocks[method.mock.__id__] = method.mock; mocks }.values
    end
  
    def unstub_all
      while stubba_methods.length > 0
        method = stubba_methods.pop
        method.unstub
      end
    end

  end

end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
mocha-0.5.6	lib/mocha/central.rb