require File.expand_path(File.dirname(__FILE__) + '/spec_helper')

describe "MongoDoc::Cursor" do
  before do
    @mongo_cursor = stub('cursor')
    @cursor = MongoDoc::Cursor.new(@mongo_cursor)
  end

  it "is Enumerable" do
    Enumerable.should === @cursor
  end

  it ".new wraps a Mongo::Cursor" do
    @cursor._cursor.should == @mongo_cursor
  end

  context "with the underlying cursor" do
    %w(close closed? count explain limit query_options_hash query_opts skip sort).each do |delegated_method|
      it "delegates #{delegated_method} to the Mongo::Cursor" do
        @mongo_cursor.should_receive(delegated_method)
        @cursor.send(delegated_method)
      end
    end
  end

  context "#each" do
    it "delegates to the cursor" do
      @mongo_cursor.should_receive(:each)
      @cursor.each
    end

    it "decodes the return from the delegate" do
      bson = stub('bson')
      @cursor.stub(:_cursor).and_return([bson])
      MongoDoc::BSON.should_receive(:decode).with(bson)
      @cursor.each {}
    end

    it "calls the block with the decoded return" do
      result = stub('bson')
      @cursor.stub(:_cursor).and_return([result])
      MongoDoc::BSON.stub(:decode).and_return(result)
      @cursor.each {|obj| @obj = obj}
      @obj.should == result
    end
  end

  context "#next_document" do
    it "delegates to the cursor" do
      @mongo_cursor.should_receive(:next_document)
      @cursor.next_document
    end

    it "decodes the return from the delegate" do
      bson = stub('bson')
      @mongo_cursor.stub(:next_document).and_return(bson)
      MongoDoc::BSON.should_receive(:decode).with(bson)
      @cursor.next_document
    end

    it "returns nil if the delegate returns nil" do
      @mongo_cursor.stub(:next_document)
      @cursor.next_document.should be_nil
    end
  end

  context "#to_a" do
    it "delegates to the cursor" do
      @mongo_cursor.should_receive(:to_a)
      @cursor.to_a
    end

    it "decodes the return from the delegate" do
      array = stub('array')
      @mongo_cursor.stub(:to_a).and_return(array)
      MongoDoc::BSON.should_receive(:decode).with(array)
      @cursor.to_a
    end

    it "returns [] if the delegate returns []" do
      @mongo_cursor.stub(:to_a).and_return([])
      @cursor.to_a.should == []
    end
  end
end