require File.expand_path('../acceptance_test_helper', __FILE__)
require 'mocha_standalone'
require 'simple_counter'

class NotATestUnitAssertionFailedError < StandardError
end

class NotATestUnitTestCase

  include Mocha::API

  attr_reader :assertion_counter

  def initialize
    @assertion_counter = SimpleCounter.new
  end

  def run(test_method)
    mocha_setup
    begin
      prepare
      begin
        send(test_method)
        mocha_verify(@assertion_counter)
      rescue Mocha::ExpectationError => e
        new_error = NotATestUnitAssertionFailedError.new(e.message)
        new_error.set_backtrace(e.backtrace)
        raise new_error
      ensure
        cleanup
      end
    ensure
      mocha_teardown
    end
  end

  def prepare
  end

  def cleanup
  end

end

class SampleTest < NotATestUnitTestCase

  def mock_object_with_fulfilled_expectation
    mockee = mock()
    mockee.expects(:blah)
    mockee.blah
  end

  def mock_object_with_unfulfilled_expectation
    mockee = mock()
    mockee.expects(:blah)
  end

  def mock_object_with_unexpected_invocation
    mockee = mock()
    mockee.blah
  end

  def real_object_with_fulfilled_expectation
    stubbee = Class.new { define_method(:blah) {} }.new
    stubbee.expects(:blah)
    stubbee.blah
  end

  def real_object_with_unfulfilled_expectation
    stubbee = Class.new { define_method(:blah) {} }.new
    stubbee.expects(:blah)
  end

  def mock_object_with_matching_parameter
    mockee = mock()
    mockee.expects(:blah).with(has_key(:wibble))
    mockee.blah(:wibble => 1)
  end

  def mock_object_with_non_matching_parameter
    mockee = mock()
    mockee.expects(:blah).with(has_key(:wibble))
    mockee.blah(:wobble => 2)
  end

end

require 'test/unit'

class APITest < Test::Unit::TestCase

  attr_reader :sample_test

  include AcceptanceTest

  def setup
    @sample_test = SampleTest.new
    setup_acceptance_test
  end

  def teardown
    teardown_acceptance_test
  end

  def test_should_pass_mock_object_test
    assert_nothing_raised { sample_test.run(:mock_object_with_fulfilled_expectation) }
    assert_equal 1, sample_test.assertion_counter.count
  end

  def test_should_fail_mock_object_test_due_to_unfulfilled_exception
    assert_raises(NotATestUnitAssertionFailedError) { sample_test.run(:mock_object_with_unfulfilled_expectation) }
    assert_equal 1, sample_test.assertion_counter.count
  end

  def test_should_fail_mock_object_test_due_to_unexpected_invocation
    assert_raises(NotATestUnitAssertionFailedError) { sample_test.run(:mock_object_with_unexpected_invocation) }
    assert_equal 0, sample_test.assertion_counter.count
  end

  def test_should_pass_real_object_test
    assert_nothing_raised { sample_test.run(:real_object_with_fulfilled_expectation) }
    assert_equal 1, sample_test.assertion_counter.count
  end

  def test_should_fail_real_object_test
    assert_raises(NotATestUnitAssertionFailedError) { sample_test.run(:real_object_with_unfulfilled_expectation) }
    assert_equal 1, sample_test.assertion_counter.count
  end

  def test_should_pass_mock_object_test_with_matching_parameter
    assert_nothing_raised { sample_test.run(:mock_object_with_matching_parameter) }
    assert_equal 1, sample_test.assertion_counter.count
  end

  def test_should_fail_mock_object_test_with_non_matching_parameter
    assert_raises(NotATestUnitAssertionFailedError) { sample_test.run(:mock_object_with_non_matching_parameter) }
  end

end