require "examples/example_helper"

module RR
describe Double, "#reset", :shared => true do
  it "cleans up by removing the __rr__method" do
    @double.bind
    @object.methods.should include("__rr__foobar")

    @double.reset
    @object.methods.should_not include("__rr__foobar")
  end
end

describe Double, "#reset when method does not exist" do
  it_should_behave_like "RR::Double#reset"

  before do
    @space = Space.new
    @object = Object.new
    @method_name = :foobar
    @object.methods.should_not include(@method_name.to_s)
    @double = Double.new(@space, @object, @method_name)
  end

  it "removes the method" do
    @double.bind
    @object.methods.should include(@method_name.to_s)

    @double.reset
    @object.methods.should_not include(@method_name.to_s)
    proc {@object.foobar}.should raise_error(NoMethodError)
  end
end

describe Double, "#reset when method exists" do
  it_should_behave_like "RR::Double#reset"

  before do
    @space = Space.new
    @object = Object.new
    @method_name = :foobar
    def @object.foobar
      :original_foobar
    end
    @object.methods.should include(@method_name.to_s)
    @original_method = @object.method(@method_name)
    @double = Double.new(@space, @object, @method_name)

    @double.bind
    @object.methods.should include(@method_name.to_s)
  end

  it "rebind original method" do
    @double.reset
    @object.methods.should include(@method_name.to_s)
    @object.foobar.should == :original_foobar
  end
end

describe Double, "#reset when method with block exists" do
  it_should_behave_like "RR::Double#reset"

  before do
    @space = Space.new
    @object = Object.new
    @method_name = :foobar
    def @object.foobar
      yield(:original_argument)
    end
    @object.methods.should include(@method_name.to_s)
    @original_method = @object.method(@method_name)
    @double = Double.new(@space, @object, @method_name)

    @double.bind
    @object.methods.should include(@method_name.to_s)
  end

  it "rebinds original method with block" do
    @double.reset
    @object.methods.should include(@method_name.to_s)
    
    original_argument = nil
    @object.foobar do |arg|
      original_argument = arg
    end
    original_argument.should == :original_argument
  end
end
end