update unacknowledge write test to work with 3.0.x driver

MorphiaOrg · Aug 28, 2013 · 13d11cf · 13d11cf
1 parent c6c088f
commit 13d11cf
Showing 1 changed file with 10 additions and 5 deletions.
diff --git a/morphia/src/test/java/com/google/code/morphia/TestUpdateOps.java b/morphia/src/test/java/com/google/code/morphia/TestUpdateOps.java
@@ -151,12 +151,17 @@ public void testValidationBadFieldType() throws Exception {
 
   @Test
   public void testInsertUpdatesUnsafe() throws Exception {
-    if (mongo.getVersion().startsWith("2.")) {
-      final UpdateResults<Circle> res = ds.update(ds.createQuery(Circle.class)
+    ds.getDB()
+      .requestStart();
+    try {
+      ds.update(ds.createQuery(Circle.class)
         .field("radius")
         .equal(0), ds.createUpdateOperations(Circle.class)
         .inc("radius", 1D), true, WriteConcern.NONE);
-      assertInserted(res);
+      assertEquals(1, ds.getCount(Circle.class));
+    } finally {
+      ds.getDB()
+        .requestDone();
     }
   }
 
@@ -337,7 +342,7 @@ public void testUpdateRef() throws Exception {
 
 
     //test with Key<Pic>
-    UpdateResults<ContainsPic> res = ds.updateFirst(ds.find(ContainsPic.class, "name", cp.name), ds.createUpdateOperations(
+    final UpdateResults<ContainsPic> res = ds.updateFirst(ds.find(ContainsPic.class, "name", cp.name), ds.createUpdateOperations(
       ContainsPic.class).set("pic", pic));
 
     assertEquals(1, res.getUpdatedCount());
@@ -375,7 +380,7 @@ public void testUpdateKeyRef() throws Exception {
 
 
     //test with Key<Pic>
-    UpdateResults<ContainsPicKey> res = ds.updateFirst(ds.find(ContainsPicKey.class, "name", cpk.name), ds.createUpdateOperations(
+    final UpdateResults<ContainsPicKey> res = ds.updateFirst(ds.find(ContainsPicKey.class, "name", cpk.name), ds.createUpdateOperations(
       ContainsPicKey.class).set("pic", pic));
 
     assertEquals(1, res.getUpdatedCount());