#!/usr/bin/env ruby

require 'test/unit'
require 'flexmock'

module A
  class B
    def sample
      :ab
    end
  end
end

class Bark
  def listen
    :woof
  end
  def Bark.identify
    :barking
  end
end

class Meow
  def listen
    :meow
  end
end

class Dog
  def bark
    Bark.new
  end
  def wag
    A::B.new
  end
  def bark_id
    Bark.identify
  end
end

class TestClassInterception < Test::Unit::TestCase
  include FlexMock::TestCase

  def test_original_functionality
    dog = Dog.new
    assert_equal :woof, dog.bark.listen
    assert_equal :ab, dog.wag.sample
  end

  def test_interception
    interceptor = intercept(Bark).in(Dog).with(Meow)
    assert_kind_of FlexMock::ClassProxy, Dog::Bark
    assert_equal Meow, Dog::Bark.proxied_class
    dog = Dog.new
    assert_equal :meow, dog.bark.listen
  ensure
    interceptor.restore if interceptor
  end

  # Since interception leaves a proxy class behind, we want to make
  # sure interception works twice on the same object.  So we repeat
  # this test in full.  We don't know which tests will _actually_ run
  # first, but it doesn't matter as long as it is done twice.
  def test_interception_repeated
    test_interception
  end

  def test_interception_with_strings
    interceptor = intercept("Bark").in("Dog").with("Meow")
    assert_equal :meow, Dog.new.bark.listen
  ensure
    interceptor.restore if interceptor
  end

  def test_interception_with_symbols
    interceptor = intercept(:Bark).in(:Dog).with(:Meow)
    assert_equal :meow, Dog.new.bark.listen
  ensure
    interceptor.restore if interceptor
  end

  def test_interception_with_mixed_identifiers
    interceptor = intercept(:Bark).in("Dog").with(Meow)
    assert_equal :meow, Dog.new.bark.listen
  ensure
    interceptor.restore if interceptor
  end

  def test_interception_proxy_forward_other_methods
    d = Dog.new
    assert_equal :barking, d.bark_id

    interceptor = intercept(:Bark).in("Dog").with(Meow)
    interceptor.restore

    d = Dog.new
    assert_equal :barking, d.bark_id
  end

end

class TestMockFactory < Test::Unit::TestCase
  def test_mock_returns_factory
    m = FlexMock.new("m")
    f = m.mock_factory
    assert_equal m, f.new
  end
end