import com.mongodb.*;
import org.junit.Test;
import java.net.UnknownHostException;

public class MongoDriverSortAndBatchSizeTest {
@Test
public void batchSizeAndSortBy() throws UnknownHostException {
Mongo mongo = new Mongo(new ServerAddress("localhost",
27017));
DB test = mongo.getDB("test");
DBCollection collection = test.getCollection("test");
// add 100 documents
for (int i = 0; i < 1000; i++)

{ collection.save(new BasicDBObject("sortby", i)); }

//get the cursor - batchsize = 500
DBCursor dbCursor = collection.find(new BasicDBObject())
.batchSize(500)
.sort(new BasicDBObject("sortby", 1));
int counter = 0;
try {
//consume cursor
while (dbCursor.hasNext())

{ DBObject next = dbCursor.next(); counter++; }

} finally

{ dbCursor.close(); }

assertEquals(1000, counter); //fails - counter is 500
}