require 'spec_helper'

describe Sources::DB do
  
  before(:each) do
    @type       = stub :type, :name => 'some_type_name'
    
    @connection = stub :connection
    @adapter    = stub :adapter, :connection => @connection
    
    @select_statement = stub :statement
    
    @source = Sources::DB.new @select_statement, :option => :some_options
    
    @source.stub! :database => @adapter
    @source.stub! :connect_backend
  end
  
  describe "get_data" do
    it "delegates" do
      type  = stub :type, :name => :some_type
      field = stub :field, :name => :some_field
      
      @connection.should_receive(:execute).once.with 'SELECT indexed_id, some_field FROM some_type_type_index st WHERE st.id > some_offset LIMIT 25000'
      
      @source.get_data type, field, :some_offset
    end
  end
  
  describe "configure" do
    it "works" do
      lambda { @source.configure({}) }.should_not raise_error
    end
    context "with file" do
      it "opens the config file relative to root" do
        File.should_receive(:open).once.with 'some/search/root/app/bla.yml'
        
        @source.configure :file => 'app/bla.yml'
      end
    end
  end
  
  describe "connect_backend" do
    it "works" do
      lambda { @source.connect_backend }.should_not raise_error
    end
  end
  
  describe "chunksize" do
    it "should be a specific size" do
      @source.chunksize.should == 25_000
    end
  end
  
  describe "count" do
    it "should get the count through the public DB connection" do
      result = stub(:result, :to_i => 12_345)
      @connection.should_receive(:select_value).once.and_return(result)
      
      @source.count @type
    end
    it "should get the id count" do
      result = stub(:result, :to_i => 12_345)
      @connection.should_receive(:select_value).once.with("SELECT COUNT(id) FROM some_type_name_type_index")
      
      @source.count @type
    end
  end
  
  # TODO Redo.
  #
  # describe "harvest" do
  #   before(:each) do
  #     @source.stub! :harvest_statement_with_offset
  #   end
  #   context 'expectations' do
  #     before(:each) do
  #       @connection.stub! :execute => []
  #       @connection.stub! :select_value
  #     end
  #     after(:each) do
  #       @source.harvest :type_name, :some_field
  #     end
  #     context "with WHERE" do
  #       before(:each) do
  #         @source.stub! :select_statement => 'bla WHERE blu'
  #       end
  #       it "should connect" do
  #         @source.should_receive(:connect_backend).once.with
  #       end
  #       it "should call the harvest statement with an offset" do
  #         @source.should_receive(:harvest_statement_with_offset).once.with :some_type, :some_field, :some_offset
  #       end
  #     end
  #     context "without WHERE" do
  #       it "should connect" do
  #         @adapter.should_receive(:connect).once.with
  #       end
  #       it "should call the harvest statement with an offset" do
  #         @source.should_receive(:harvest_statement_with_offset).once.with :some_type, :some_field, :some_offset
  #       end
  #     end
  #   end
  #   context 'returns' do
  #     it "should return whatever the execute statement returns" do
  #       @connection.stub! :execute => :some_result
  #       
  #       @source.harvest(:some_type, :some_field).should == :some_result
  #     end
  #   end
  # end

  describe "harvest_statement_with_offset" do
    before(:each) do
      @category = stub :category, :name => :some_category
      @type     = stub :type,     :name => :some_type
    end
    it "should get a harvest statement and the chunksize to put the statement together" do
      @source.should_receive(:harvest_statement).once.with(@type, @category).and_return 'some_example_statement'
      @source.harvest_statement_with_offset(@type, @category, :some_offset)
    end
    it "should add an AND if it already contains a WHERE statement" do
      @source.should_receive(:harvest_statement).and_return 'WHERE'
      @source.harvest_statement_with_offset(@type, @category, :some_offset).should == "WHERE AND st.id > some_offset LIMIT 25000"
    end
    it "should add a WHERE if it doesn't already contain one" do
      @source.should_receive(:harvest_statement).and_return 'some_statement'
      @source.harvest_statement_with_offset(@type, @category, :some_offset).should == "some_statement WHERE st.id > some_offset LIMIT 25000"
    end
  end
  
end