expecting an error once

wat_spec.rb

class Wat
  attr_accessor :collaborator, :credentials

  def foo(arg)
    @collaborator.foo(arg)
  rescue
    @collaborator.set_auth_credentials(credentials)
    @collaborator.foo(arg)
  end

end

describe Wat do
  describe "#foo" do
    let(:arg) { mock }
    let(:credentials) { mock }
    let(:result) { mock }

    before do
      subject.collaborator = mock
    end

    it "delegates foo to collaborator" do
      subject.collaborator.should_receive(:foo).with(arg).and_return(result)
      subject.foo(arg).should eq(result)
    end

    it "handles the exception" do
      subject.credentials = credentials
      subject.collaborator.should_receive(:foo).once.with(arg).and_raise(StandardError.new)
      subject.collaborator.should_receive(:set_auth_credentials).once.with(credentials)
      subject.collaborator.should_receive(:foo).with(arg).and_return(result)
      subject.foo(arg).should eq(result)

    end

  end
end