require 'spec_helper'

module RSpec
  module Mocks
    describe "A chained method stub" do
      before(:each) do
        @subject = Object.new
      end


      context "with one method in chain" do
        context "using and_return" do
          it "returns expected value from chaining only one method call" do
            @subject.stub_chain(:msg1).and_return(:return_value)
            @subject.msg1.should equal(:return_value)
          end
        end

        context "using a block" do
          it "returns the correct value" do
            @subject.stub_chain(:msg1) { :return_value }
            @subject.msg1.should equal(:return_value)
          end
        end

        context "using a hash" do
          it "returns the value of the key/value pair" do
            @subject.stub_chain(:msg1 => :return_value)
            @subject.msg1.should equal(:return_value)
          end
        end
      end

      context "with two methods in chain" do
        context "using and_return" do
          it "returns expected value from chaining two method calls" do
            @subject.stub_chain(:msg1, :msg2).and_return(:return_value)
            @subject.msg1.msg2.should equal(:return_value)
          end
        end

        context "using a block" do
          it "returns the correct value" do
            @subject.stub_chain(:msg1, :msg2) { :return_value }
            @subject.msg1.msg2.should equal(:return_value)
          end
        end

        context "using a hash" do
          it "returns the value of the key/value pair" do
            @subject.stub_chain(:msg1, :msg2 => :return_value)
            @subject.msg1.msg2.should equal(:return_value)
          end
        end
      end

      context "with four methods in chain" do
        context "using and_return" do
          it "returns expected value from chaining two method calls" do
            @subject.stub_chain(:msg1, :msg2, :msg3, :msg4).and_return(:return_value)
            @subject.msg1.msg2.msg3.msg4.should equal(:return_value)
          end
        end

        context "using a block" do
          it "returns the correct value" do
            @subject.stub_chain(:msg1, :msg2, :msg3, :msg4) { :return_value }
            @subject.msg1.msg2.msg3.msg4.should equal(:return_value)
          end
        end

        context "using a hash" do
          it "returns the value of the key/value pair" do
            @subject.stub_chain(:msg1, :msg2, :msg3, :msg4 => :return_value)
            @subject.msg1.msg2.msg3.msg4.should equal(:return_value)
          end
        end

        context "using a hash with a string key" do
          it "returns the value of the key/value pair" do
            @subject.stub_chain("msg1.msg2.msg3.msg4" => :return_value)
            @subject.msg1.msg2.msg3.msg4.should equal(:return_value)
          end
        end
      end

      it "returns expected value from chaining four method calls" do
        @subject.stub_chain(:msg1, :msg2, :msg3, :msg4).and_return(:return_value)
        @subject.msg1.msg2.msg3.msg4.should equal(:return_value)
      end

      it "returns expected value from chaining four method calls twice with some shared" do
        @subject.stub_chain(:msg1, :msg2, :msg3, :msg4).and_return(:first)
        @subject.stub_chain(:msg5, :msg2, :msg3, :msg4).and_return(:second)

        @subject.msg1.msg2.msg3.msg4.should equal(:first)
        @subject.msg5.msg2.msg3.msg4.should equal(:second)
      end

      it "returns expected value when chain is a dot separated string, like stub_chain('msg1.msg2.msg3')" do
        @subject.stub_chain("msg1.msg2.msg3").and_return(:return_value)
        @subject.msg1.msg2.msg3.should equal(:return_value)
      end

      it "returns expected value from two chains with shared messages at the beginning" do
        @subject.stub_chain(:msg1, :msg2, :msg3, :msg4).and_return(:first)
        @subject.stub_chain(:msg1, :msg2, :msg3, :msg5).and_return(:second)

        @subject.msg1.msg2.msg3.msg4.should equal(:first)
        @subject.msg1.msg2.msg3.msg5.should equal(:second)
      end
    end
  end
end