Wait for unacknowledged write to complete (#958)

To avoid interference between tests, poll the database until
the unacknowledged write is present in the collection. This
is intended to avoid SnapshotUnavailable errors from being
reported by the server.

JAVA-4637

Author: jyemin

driver-legacy/src/test/functional/com/mongodb/MongoClientSessionSpecification.groovy

@@ -16,13 +16,16 @@
 
 package com.mongodb
 
+import com.mongodb.client.MongoCollection
+import com.mongodb.client.model.Filters
 import com.mongodb.event.CommandStartedEvent
 import com.mongodb.internal.connection.TestCommandListener
 import org.bson.BsonBinarySubType
 import org.bson.BsonDocument
 import org.bson.BsonInt32
 import org.bson.BsonTimestamp
 import org.bson.Document
+import org.bson.types.ObjectId
 import org.junit.Assert
 import spock.lang.IgnoreIf
 import util.spock.annotations.Slow
@@ -308,17 +311,18 @@ class MongoClientSessionSpecification extends FunctionalSpecification {
                 .addCommandListener(commandListener)
         def mongoClientURI = getMongoClientURI(optionsBuilder)
         def client = new MongoClient(mongoClientURI)
+        def collection = client.getDatabase(getDatabaseName()).getCollection(getCollectionName())
+        def id = new ObjectId()
 
         when:
-        client.getDatabase(getDatabaseName()).getCollection(getCollectionName())
-                .withWriteConcern(WriteConcern.UNACKNOWLEDGED)
-                .insertOne(new Document())
+        collection.withWriteConcern(WriteConcern.UNACKNOWLEDGED).insertOne(new Document('_id', id))
 
         then:
         def insertEvent = commandListener.events.get(0) as CommandStartedEvent
         !insertEvent.command.containsKey('lsid')
 
         cleanup:
+        waitForInsertAcknowledgement(collection, id)
         client?.close()
     }
 
@@ -358,4 +362,12 @@ class MongoClientSessionSpecification extends FunctionalSpecification {
         cleanup:
         session.close()
     }
+
+    void waitForInsertAcknowledgement(MongoCollection<Document> collection, ObjectId id) {
+        Document document = collection.find(Filters.eq(id)).first()
+        while (document == null) {
+            Thread.sleep(1)
+            document = collection.find(Filters.eq(id)).first()
+        }
+    }
 }

driver-sync/src/test/functional/com/mongodb/client/MongoClientSessionSpecification.groovy

@@ -23,13 +23,15 @@ import com.mongodb.ReadConcern
 import com.mongodb.ReadPreference
 import com.mongodb.TransactionOptions
 import com.mongodb.WriteConcern
+import com.mongodb.client.model.Filters
 import com.mongodb.event.CommandStartedEvent
 import com.mongodb.internal.connection.TestCommandListener
 import org.bson.BsonBinarySubType
 import org.bson.BsonDocument
 import org.bson.BsonInt32
 import org.bson.BsonTimestamp
 import org.bson.Document
+import org.bson.types.ObjectId
 import org.junit.Assert
 import spock.lang.IgnoreIf
 import util.spock.annotations.Slow
@@ -293,17 +295,18 @@ class MongoClientSessionSpecification extends FunctionalSpecification {
         def commandListener = new TestCommandListener()
         def settings = MongoClientSettings.builder(getMongoClientSettings()).commandListenerList([commandListener]).build()
         def client = MongoClients.create(settings)
+        def collection = client.getDatabase(getDatabaseName()).getCollection(getCollectionName())
+        def id = new ObjectId()
 
         when:
-        client.getDatabase(getDatabaseName()).getCollection(getCollectionName())
-                .withWriteConcern(WriteConcern.UNACKNOWLEDGED)
-                .insertOne(new Document())
+        collection.withWriteConcern(WriteConcern.UNACKNOWLEDGED).insertOne(new Document('_id', id))
 
         then:
         def insertEvent = commandListener.events.get(0) as CommandStartedEvent
         !insertEvent.command.containsKey('lsid')
 
         cleanup:
+        waitForInsertAcknowledgement(collection, id)
         client?.close()
     }
 
@@ -344,4 +347,12 @@ class MongoClientSessionSpecification extends FunctionalSpecification {
         cleanup:
         session.close()
     }
+
+    void waitForInsertAcknowledgement(MongoCollection<Document> collection, ObjectId id) {
+        Document document = collection.find(Filters.eq(id)).first()
+        while (document == null) {
+            Thread.sleep(1)
+            document = collection.find(Filters.eq(id)).first()
+        }
+    }
 }