added setOptioins, resetOptions and getOptions to DBCursor, JAVA-222

daspilker · erh · commit efb4b8164176 · 2010-11-30T23:46:13.000+08:00
diff --git a/src/main/com/mongodb/DBCursor.java b/src/main/com/mongodb/DBCursor.java
@@ -262,6 +262,18 @@ public DBCursor addOption( int option ){
         return this;
     }
 
+    public void setOptions( int options ){
+        _options = options;
+    }
+
+    public void resetOptions(){
+        _options = 0;
+    }
+
+    public int getOptions(){
+        return _options;
+    }
+
     // ----  internal stuff ------
 
     private void _check()
diff --git a/src/test/com/mongodb/DBCursorTest.java b/src/test/com/mongodb/DBCursorTest.java
@@ -66,6 +66,20 @@ public void testSnapshot() {
         assertEquals( 50 , c.find().snapshot().limit(50).toArray().size() );
     }
 
+    @Test(groups = {"basic"})
+    public void testOptions() {
+        DBCollection c = _db.getCollection("test");
+        DBCursor dbCursor = c.find();
+
+        assertEquals(0, dbCursor.getOptions());
+        dbCursor.setOptions(Bytes.QUERYOPTION_TAILABLE);
+        assertEquals(Bytes.QUERYOPTION_TAILABLE, dbCursor.getOptions());
+        dbCursor.addOption(Bytes.QUERYOPTION_SLAVEOK);
+        assertEquals(Bytes.QUERYOPTION_TAILABLE | Bytes.QUERYOPTION_SLAVEOK, dbCursor.getOptions());
+        dbCursor.resetOptions();
+        assertEquals(0, dbCursor.getOptions());
+    }
+
     @Test
     public void testBig(){
         DBCollection c = _db.getCollection("big1");
