require File.expand_path('spec_helper', File.dirname(__FILE__))

describe Sunspot::SessionProxy::Retry5xxSessionProxy do
  
  before :each do
    Sunspot::Session.connection_class = Mock::ConnectionFactory.new
    @sunspot_session = Sunspot.session
    @proxy = Sunspot::SessionProxy::Retry5xxSessionProxy.new(@sunspot_session)
    Sunspot.session = @proxy
  end

  after :each do
    Sunspot::Session.connection_class = nil
    Sunspot.reset!(true)
  end

  class FakeRSolrErrorHttp < RSolr::Error::Http
    def backtrace
      []
    end
  end

  let :fake_rsolr_request do
    {:uri => 'http://solr.test/uri'}
  end

  def fake_rsolr_response(status)
    {:status => status.to_s}
  end

  let :post do
    Post.new(:title => 'test')
  end

  it "should behave normally without a stubbed exception" do
    @sunspot_session.should_receive(:index).and_return(mock)
    Sunspot.index(post)
  end

  it "should be successful with a single exception followed by a sucess" do
    e = FakeRSolrErrorHttp.new(fake_rsolr_request, fake_rsolr_response(503))
    @sunspot_session.should_receive(:index).and_return do
      @sunspot_session.should_receive(:index).and_return(mock)
      raise e
    end
    Sunspot.index(post)
  end

  it "should return the error response after two exceptions" do
    fake_response = fake_rsolr_response(503)
    e = FakeRSolrErrorHttp.new(fake_rsolr_request, fake_response)
    fake_success = mock('success')

    @sunspot_session.should_receive(:index).and_return do
      @sunspot_session.should_receive(:index).and_return do
        @sunspot_session.stub!(:index).and_return(fake_success)
        raise e
      end
      raise e
    end

    response = Sunspot.index(post)
    response.should_not == fake_success
    response.should == fake_response
  end

  it "should not retry a 4xx" do
    e = FakeRSolrErrorHttp.new(fake_rsolr_request, fake_rsolr_response(400))
    @sunspot_session.should_receive(:index).and_raise(e)
    lambda { Sunspot.index(post) }.should raise_error
  end

  # TODO: try against more than just Sunspot.index? but that's just testing the
  # invocation of delegate, so probably not important. -nz 11Apr12

  it_should_behave_like 'session proxy'
  
end