jferris-mocha 0.9.5.0.1240002286

data/test/method_definer.rb

@@ -0,0 +1,24 @@
+require 'mocha/metaclass'
+
+module Mocha
+  
+  module ObjectMethods
+    def define_instance_method(method_symbol, &block)
+      __metaclass__.send(:define_method, method_symbol, block)
+    end
+
+    def replace_instance_method(method_symbol, &block)
+      raise "Cannot replace #{method_symbol} as #{self} does not respond to it." unless self.respond_to?(method_symbol)
+      define_instance_method(method_symbol, &block)
+    end
+
+    def define_instance_accessor(*symbols)
+      symbols.each { |symbol| __metaclass__.send(:attr_accessor, symbol) }
+    end
+  end
+  
+end
+
+class Object
+  include Mocha::ObjectMethods
+end

data/test/simple_counter.rb

@@ -0,0 +1,13 @@
+class SimpleCounter
+  
+  attr_reader :count
+  
+  def initialize
+    @count = 0
+  end
+  
+  def increment
+    @count += 1
+  end
+  
+end

data/test/test_helper.rb

@@ -0,0 +1,16 @@
+unless defined?(STANDARD_OBJECT_PUBLIC_INSTANCE_METHODS)
+  STANDARD_OBJECT_PUBLIC_INSTANCE_METHODS = Object.public_instance_methods
+end
+
+$:.unshift File.expand_path(File.join(File.dirname(__FILE__), "..", "lib"))
+$:.unshift File.expand_path(File.join(File.dirname(__FILE__)))
+$:.unshift File.expand_path(File.join(File.dirname(__FILE__), 'unit'))
+$:.unshift File.expand_path(File.join(File.dirname(__FILE__), 'unit', 'parameter_matchers'))
+$:.unshift File.expand_path(File.join(File.dirname(__FILE__), 'acceptance'))
+
+if ENV['MOCHA_OPTIONS'] == 'use_test_unit_gem'
+  gem 'test-unit'
+end
+
+require 'test/unit'
+

data/test/test_runner.rb

@@ -0,0 +1,33 @@
+require 'test/unit/testresult'
+require 'test/unit/testcase'
+
+module TestRunner
+  
+  def run_as_test(test_result = Test::Unit::TestResult.new, &block)
+    test_class = Class.new(Test::Unit::TestCase) do
+      define_method(:test_me, &block)
+    end
+    test = test_class.new(:test_me)
+    test.run(test_result) {}
+    class << test_result
+      attr_reader :failures, :errors
+      def failure_messages
+        failures.map { |failure| failure.message }
+      end
+      def error_messages
+        errors.map { |error| error.message }
+      end
+    end
+    test_result
+  end
+  
+  def assert_passed(test_result)
+    flunk "Test failed unexpectedly with message: #{test_result.failures}" if test_result.failure_count > 0
+    flunk "Test failed unexpectedly with message: #{test_result.errors}" if test_result.error_count > 0
+  end
+
+  def assert_failed(test_result)
+    flunk "Test passed unexpectedly" if test_result.passed?
+  end
+
+end

data/test/unit/any_instance_method_test.rb

@@ -0,0 +1,126 @@
+require File.join(File.dirname(__FILE__), "..", "test_helper")
+require 'method_definer'
+require 'mocha/mock'
+require 'mocha/any_instance_method'
+
+class AnyInstanceMethodTest < Test::Unit::TestCase
+  
+  include Mocha
+  
+  def test_should_hide_original_method
+    klass = Class.new { def method_x; end }
+    method = AnyInstanceMethod.new(klass, :method_x)
+    hidden_method_x = method.hidden_method.to_sym
+    
+    method.hide_original_method
+    
+    assert klass.method_defined?(hidden_method_x)
+  end
+  
+  def test_should_not_hide_original_method_if_it_is_not_defined
+    klass = Class.new
+    method = AnyInstanceMethod.new(klass, :method_x)
+    hidden_method_x = method.hidden_method.to_sym
+    
+    method.hide_original_method
+    
+    assert_equal false, klass.method_defined?(hidden_method_x)
+  end
+  
+  def test_should_define_a_new_method
+    klass = Class.new { def method_x; end } 
+    method = AnyInstanceMethod.new(klass, :method_x)
+    mocha = Mock.new
+    mocha.expects(:method_x).with(:param1, :param2).returns(:result)
+    any_instance = Object.new
+    any_instance.define_instance_method(:mocha) { mocha }
+    klass.define_instance_method(:any_instance) { any_instance }
+    
+    method.hide_original_method
+    method.define_new_method
+
+    instance = klass.new
+    result = instance.method_x(:param1, :param2)
+        
+    assert_equal :result, result
+    assert mocha.__verified__?
+  end
+
+  def test_should_restore_original_method
+    klass = Class.new { def method_x; end }
+    method = AnyInstanceMethod.new(klass, :method_x)
+    hidden_method_x = method.hidden_method.to_sym
+    klass.send(:define_method, hidden_method_x, Proc.new { :original_result }) 
+    
+    method.remove_new_method
+    method.restore_original_method
+    
+    instance = klass.new
+    assert_equal :original_result, instance.method_x 
+    assert !klass.method_defined?(hidden_method_x)
+  end
+
+  def test_should_not_restore_original_method_if_hidden_method_not_defined
+    klass = Class.new { def method_x; :new_result; end }
+    method = AnyInstanceMethod.new(klass, :method_x)
+    
+    method.restore_original_method
+    
+    instance = klass.new
+    assert_equal :new_result, instance.method_x 
+  end
+
+  def test_should_call_remove_new_method
+    klass = Class.new { def method_x; end }
+    any_instance = Mock.new
+    any_instance.stubs(:reset_mocha)
+    klass.define_instance_method(:any_instance) { any_instance }
+    method = AnyInstanceMethod.new(klass, :method_x)
+    method.replace_instance_method(:restore_original_method) { }
+    method.define_instance_accessor(:remove_called)
+    method.replace_instance_method(:remove_new_method) { self.remove_called = true }
+    
+    method.unstub
+    
+    assert method.remove_called
+  end
+
+  def test_should_call_restore_original_method
+    klass = Class.new { def method_x; end }
+    any_instance = Mock.new
+    any_instance.stubs(:reset_mocha)
+    klass.define_instance_method(:any_instance) { any_instance }
+    method = AnyInstanceMethod.new(klass, :method_x)
+    method.replace_instance_method(:remove_new_method) { }
+    method.define_instance_accessor(:restore_called)
+    method.replace_instance_method(:restore_original_method) { self.restore_called = true }
+    
+    method.unstub
+    
+    assert method.restore_called
+  end
+
+  def test_should_call_reset_mocha
+    klass = Class.new { def method_x; end }
+    any_instance = Class.new { attr_accessor :mocha_was_reset; def reset_mocha; self.mocha_was_reset = true; end }.new
+    klass.define_instance_method(:any_instance) { any_instance }
+    method = AnyInstanceMethod.new(klass, :method_x)
+    method.replace_instance_method(:remove_new_method) { }
+    method.replace_instance_method(:restore_original_method) { }
+    
+    method.unstub
+
+    assert any_instance.mocha_was_reset
+  end
+
+  def test_should_return_any_instance_mocha_for_stubbee
+    mocha = Object.new
+    any_instance = Object.new
+    any_instance.define_instance_method(:mocha) { mocha }
+    stubbee = Class.new
+    stubbee.define_instance_method(:any_instance) { any_instance }
+    method = AnyInstanceMethod.new(stubbee, :method_name)
+    assert_equal stubbee.any_instance.mocha, method.mock
+  end
+
+end

data/test/unit/array_inspect_test.rb

@@ -0,0 +1,16 @@
+require File.join(File.dirname(__FILE__), "..", "test_helper")
+require 'mocha/inspect'
+
+class ArrayInspectTest < Test::Unit::TestCase
+  
+  def test_should_use_inspect
+    array = [1, 2]
+    assert_equal array.inspect, array.mocha_inspect
+  end
+  
+  def test_should_use_mocha_inspect_on_each_item
+    array = [1, 2, "chris"]
+    assert_equal "[1, 2, 'chris']", array.mocha_inspect
+  end
+  
+end

data/test/unit/assert_received_test.rb

@@ -0,0 +1,136 @@
+require File.join(File.dirname(__FILE__), "..", "test_helper")
+require 'test_runner'
+require 'mocha/api'
+require 'mocha/mockery'
+require 'mocha/object'
+
+class AssertReceivedTest < Test::Unit::TestCase
+
+  include Mocha
+  include TestRunner
+  include Mocha::API
+
+  def teardown
+    Mockery.reset_instance
+  end
+
+  class FakeMock
+    def initialize(name)
+      @name = name
+    end
+
+    def inspect
+      @name
+    end
+
+    def mocha
+      self
+    end
+  end
+
+  def test_passes_if_invocation_exists
+    method = :a_method
+    mock   = FakeMock.new('a mock')
+    Mockery.instance.invocation(mock, method, [])
+    assert_passes do
+      assert_received(mock, method)
+    end
+  end
+
+  def test_fails_if_invocation_doesnt_exist
+    method = :a_method
+    mock   = FakeMock.new('a mock')
+    assert_fails do
+      assert_received(mock, method)
+    end
+  end
+
+  def test_fails_if_invocation_exists_with_different_arguments
+    method = :a_method
+    mock   = FakeMock.new('a mock')
+    Mockery.instance.invocation(mock, method, [2, 1])
+    assert_fails do
+      assert_received(mock, method) {|expect| expect.with(1, 2) }
+    end
+  end
+
+  def test_passes_if_invocation_exists_with_wildcard_arguments
+    method = :a_method
+    mock   = FakeMock.new('a mock')
+    Mockery.instance.invocation(mock, method, ['hello'])
+    assert_passes do
+      assert_received(mock, method) {|expect| expect.with(is_a(String)) }
+    end
+  end
+
+  def test_passes_if_invocation_exists_with_exact_arguments
+    method = :a_method
+    mock   = FakeMock.new('a mock')
+    Mockery.instance.invocation(mock, method, ['hello'])
+    assert_passes do
+      assert_received(mock, method) {|expect| expect.with('hello') }
+    end
+  end
+
+  def test_fails_if_invocation_exists_only_on_other_mock
+    method = :a_method
+    mock   = FakeMock.new('a mock')
+    other  = 'another mock'
+    Mockery.instance.invocation(other, method, ['hello'])
+    assert_fails do
+      assert_received(mock, method)
+    end
+  end
+
+  def test_passes_if_invocation_exists_for_impersonating_mock
+    method = :a_method
+    object = Object.new
+    mock   = FakeMock.new('a mock')
+
+    class << object
+      attr_accessor :mocha
+    end
+    object.mocha = mock
+
+    Mockery.instance.invocation(mock, method, ['hello'])
+    assert_passes do
+      assert_received(object, method) {|expect| expect.with('hello') }
+    end
+  end
+
+  def test_passes_if_invocation_count_correct
+    method = :a_method
+    mock   = FakeMock.new('a mock')
+    2.times { Mockery.instance.invocation(mock, method, []) }
+    assert_passes do
+      assert_received(mock, method) {|expect| expect.twice }
+    end
+  end
+
+  def test_fails_if_invocation_count_incorrect
+    method = :a_method
+    mock   = FakeMock.new('a mock')
+    Mockery.instance.invocation(mock, method, [])
+    assert_fails do
+      assert_received(mock, method) {|expect| expect.twice }
+    end
+  end
+
+  def assert_passes(&block)
+    assert ! fails?(&block)
+  end
+
+  def assert_fails(&block)
+    assert fails?(&block)
+  end
+
+  def fails?
+    begin
+      yield
+      false
+    rescue Test::Unit::AssertionFailedError
+      true
+    end
+  end
+
+end

data/test/unit/backtrace_filter_test.rb

@@ -0,0 +1,19 @@
+require File.join(File.dirname(__FILE__), "..", "test_helper")
+require 'mocha/backtrace_filter'
+
+class BacktraceFilterTest < Test::Unit::TestCase
+  
+  include Mocha
+  
+  def test_should_exclude_mocha_locations_from_backtrace
+    mocha_lib = "/username/workspace/mocha_wibble/lib/"
+    backtrace = [ mocha_lib + 'exclude/me/1', mocha_lib + 'exclude/me/2', '/keep/me', mocha_lib + 'exclude/me/3']
+    filter = BacktraceFilter.new(mocha_lib)
+    assert_equal ['/keep/me'], filter.filtered(backtrace)
+  end
+
+  def test_should_determine_path_for_mocha_lib_directory
+    assert_match Regexp.new("/lib/$"), BacktraceFilter::LIB_DIRECTORY
+  end
+  
+end

data/test/unit/cardinality_test.rb

@@ -0,0 +1,56 @@
+require File.join(File.dirname(__FILE__), "..", "test_helper")
+require 'mocha/cardinality'
+
+class CardinalityTest < Test::Unit::TestCase
+  
+  include Mocha
+  
+  def test_should_allow_invocations_if_invocation_count_has_not_yet_reached_maximum
+    cardinality = Cardinality.new(2, 3)
+    assert cardinality.invocations_allowed?(0)
+    assert cardinality.invocations_allowed?(1)
+    assert cardinality.invocations_allowed?(2)
+    assert !cardinality.invocations_allowed?(3)
+  end
+  
+  def test_should_be_satisfied_if_invocations_so_far_have_reached_required_threshold
+    cardinality = Cardinality.new(2, 3)
+    assert !cardinality.satisfied?(0)
+    assert !cardinality.satisfied?(1)
+    assert cardinality.satisfied?(2)
+    assert cardinality.satisfied?(3)
+  end
+  
+  def test_should_describe_cardinality
+    assert_equal 'allowed any number of times', Cardinality.at_least(0).mocha_inspect
+    
+    assert_equal 'expected at most once', Cardinality.at_most(1).mocha_inspect
+    assert_equal 'expected at most twice', Cardinality.at_most(2).mocha_inspect
+    assert_equal 'expected at most 3 times', Cardinality.at_most(3).mocha_inspect
+    
+    assert_equal 'expected at least once', Cardinality.at_least(1).mocha_inspect
+    assert_equal 'expected at least twice', Cardinality.at_least(2).mocha_inspect
+    assert_equal 'expected at least 3 times', Cardinality.at_least(3).mocha_inspect
+    
+    assert_equal 'expected never', Cardinality.exactly(0).mocha_inspect
+    assert_equal 'expected exactly once', Cardinality.exactly(1).mocha_inspect
+    assert_equal 'expected exactly twice', Cardinality.exactly(2).mocha_inspect
+    assert_equal 'expected exactly 3 times', Cardinality.times(3).mocha_inspect
+    
+    assert_equal 'expected between 2 and 4 times', Cardinality.times(2..4).mocha_inspect
+    assert_equal 'expected between 1 and 3 times', Cardinality.times(1..3).mocha_inspect
+  end
+  
+  def test_should_need_verifying
+    assert Cardinality.exactly(2).needs_verifying?
+    assert Cardinality.at_least(3).needs_verifying?
+    assert Cardinality.at_most(2).needs_verifying?
+    assert Cardinality.times(4).needs_verifying?
+    assert Cardinality.times(2..4).needs_verifying?
+  end
+  
+  def test_should_not_need_verifying
+    assert_equal false, Cardinality.at_least(0).needs_verifying?
+  end
+  
+end

data/test/unit/central_test.rb

@@ -0,0 +1,65 @@
+require File.join(File.dirname(__FILE__), "..", "test_helper")
+
+require 'mocha/central'
+require 'mocha/mock'
+require 'method_definer'
+
+class CentralTest < Test::Unit::TestCase
+  
+  include Mocha
+  
+  def test_should_start_with_empty_stubba_methods
+    stubba = Central.new
+    
+    assert_equal [], stubba.stubba_methods
+  end
+  
+  def test_should_stub_method_if_not_already_stubbed
+    method = Mock.new
+    method.expects(:stub)
+    stubba = Central.new
+    
+    stubba.stub(method)
+    
+    assert method.__verified__?
+  end
+  
+  def test_should_not_stub_method_if_already_stubbed
+    method = Mock.new
+    method.expects(:stub).times(0)
+    stubba = Central.new
+    stubba_methods = Mock.new
+    stubba_methods.stubs(:include?).with(method).returns(true)
+    stubba.stubba_methods = stubba_methods
+    
+    stubba.stub(method)
+    
+    assert method.__verified__?
+  end
+  
+  def test_should_record_method
+    method = Mock.new
+    method.expects(:stub)
+    stubba = Central.new
+    
+    stubba.stub(method)
+    
+    assert_equal [method], stubba.stubba_methods
+  end
+  
+  def test_should_unstub_all_methods
+    stubba = Central.new
+    method_1 = Mock.new
+    method_1.expects(:unstub)
+    method_2 = Mock.new
+    method_2.expects(:unstub)
+    stubba.stubba_methods = [method_1, method_2]
+
+    stubba.unstub_all
+    
+    assert_equal [], stubba.stubba_methods
+    assert method_1.__verified__?
+    assert method_2.__verified__?
+  end
+  
+end

data/test/unit/change_state_side_effect_test.rb

@@ -0,0 +1,41 @@
+require File.join(File.dirname(__FILE__), "..", "test_helper")
+
+require 'mocha/change_state_side_effect'
+
+class ChangeStateSideEffectTest < Test::Unit::TestCase
+  
+  include Mocha
+  
+  class FakeState
+    
+    attr_reader :active
+    attr_writer :description
+    
+    def activate
+      @active = true
+    end
+    
+    def mocha_inspect
+      @description
+    end
+    
+  end
+  
+  def test_should_activate_the_given_state
+    state = FakeState.new
+    side_effect = ChangeStateSideEffect.new(state)
+    
+    side_effect.perform
+    
+    assert state.active
+  end
+  
+  def test_should_describe_itself_in_terms_of_the_activated_state
+    state = FakeState.new
+    state.description = 'the-new-state'
+    side_effect = ChangeStateSideEffect.new(state)
+    
+    assert_equal 'then the-new-state', side_effect.mocha_inspect
+  end
+  
+end