require 'spec_helper'

module RSpec
  module Mocks
    describe "a message expectation with multiple return values and no specified count" do
      before(:each) do
        @double = double
        @return_values = [1,2,3]
        @double.should_receive(:do_something).and_return(@return_values[0],@return_values[1],@return_values[2])
      end

      it "returns values in order" do
        @double.do_something.should eq @return_values[0]
        @double.do_something.should eq @return_values[1]
        @double.do_something.should eq @return_values[2]
        @double.rspec_verify
      end

      it "falls back to a previously stubbed value" do
        @double.stub :do_something => :stub_result
        @double.do_something.should eq @return_values[0]
        @double.do_something.should eq @return_values[1]
        @double.do_something.should eq @return_values[2]
        @double.do_something.should eq :stub_result
      end

      it "fails when there are too few calls (if there is no stub)" do
        @double.do_something
        @double.do_something
        expect { @double.rspec_verify }.to raise_error
      end

      it "fails when there are too many calls (if there is no stub)" do
        @double.do_something
        @double.do_something
        @double.do_something
        @double.do_something
        expect { @double.rspec_verify }.to raise_error
      end
    end

    describe "a message expectation with multiple return values with a specified count equal to the number of values" do
      before(:each) do
        @double = double
        @return_values = [1,2,3]
        @double.should_receive(:do_something).exactly(3).times.and_return(@return_values[0],@return_values[1],@return_values[2])
      end

      it "returns values in order to consecutive calls" do
        @double.do_something.should eq @return_values[0]
        @double.do_something.should eq @return_values[1]
        @double.do_something.should eq @return_values[2]
        @double.rspec_verify
      end
    end

    describe "a message expectation with multiple return values specifying at_least less than the number of values" do
      before(:each) do
        @double = double
        @double.should_receive(:do_something).at_least(:twice).with(no_args).and_return(11, 22)
      end

      it "uses the last return value for subsequent calls" do
        @double.do_something.should equal(11)
        @double.do_something.should equal(22)
        @double.do_something.should equal(22)
        @double.rspec_verify
      end

      it "fails when called less than the specified number" do
        @double.do_something.should equal(11)
        expect { @double.rspec_verify }.to raise_error(RSpec::Mocks::MockExpectationError)
      end

      context "when method is stubbed too" do
        before { @double.stub(:do_something).and_return :stub_result }

        it "uses the last value for subsequent calls" do
          @double.do_something.should equal(11)
          @double.do_something.should equal(22)
          @double.do_something.should equal(22)
          @double.rspec_verify
        end

        it "fails when called less than the specified number" do
          @double.do_something.should equal(11)
          expect { @double.rspec_verify }.to raise_error(RSpec::Mocks::MockExpectationError)
        end
      end
    end

    describe "a message expectation with multiple return values with a specified count larger than the number of values" do
      before(:each) do
        @double = RSpec::Mocks::Mock.new("double")
        @double.should_receive(:do_something).exactly(3).times.and_return(11, 22)
      end

      it "uses the last return value for subsequent calls" do
        @double.do_something.should equal(11)
        @double.do_something.should equal(22)
        @double.do_something.should equal(22)
        @double.rspec_verify
      end

      it "fails when called less than the specified number" do
        @double.do_something
        @double.do_something
        expect { @double.rspec_verify }.to raise_error
      end

      it "fails fast when called greater than the specified number" do
        @double.do_something
        @double.do_something
        @double.do_something
        expect { @double.do_something }.to raise_error
      end
    end
  end
end