Port transactions spec tests to unified format (#1310)

JAVA-4171

Author: jyemin

driver-core/src/test/functional/com/mongodb/client/test/CollectionHelper.java

@@ -19,6 +19,7 @@
 import com.mongodb.MongoClientSettings;
 import com.mongodb.MongoCommandException;
 import com.mongodb.MongoNamespace;
+import com.mongodb.MongoWriteConcernException;
 import com.mongodb.ReadPreference;
 import com.mongodb.ServerCursor;
 import com.mongodb.WriteConcern;
@@ -34,6 +35,8 @@
 import com.mongodb.internal.bulk.UpdateRequest;
 import com.mongodb.internal.bulk.WriteRequest;
 import com.mongodb.internal.client.model.AggregationLevel;
+import com.mongodb.internal.diagnostics.logging.Logger;
+import com.mongodb.internal.diagnostics.logging.Loggers;
 import com.mongodb.internal.operation.AggregateOperation;
 import com.mongodb.internal.operation.BatchCursor;
 import com.mongodb.internal.operation.CommandReadOperation;
@@ -70,6 +73,7 @@
 import static java.util.Collections.singletonList;
 
 public final class CollectionHelper<T> {
+    private static final Logger LOGGER = Loggers.getLogger("test");
 
     private final Codec<T> codec;
     private final CodecRegistry registry = MongoClientSettings.getDefaultCodecRegistry();
@@ -89,7 +93,18 @@ public static void drop(final MongoNamespace namespace) {
     }
 
     public static void drop(final MongoNamespace namespace, final WriteConcern writeConcern) {
-        new DropCollectionOperation(namespace, writeConcern).execute(getBinding());
+        // This loop is a workaround for unanticipated failures of the drop command when run on a sharded cluster < 4.2.
+        // In practice the command tends to succeed on the first attempt after a failure
+        boolean success = false;
+        while (!success) {
+            try {
+                new DropCollectionOperation(namespace, writeConcern).execute(getBinding());
+                success = true;
+            } catch (MongoWriteConcernException e) {
+                LOGGER.info("Retrying drop collection after a write concern error: " + e);
+                // repeat until success!
+            }
+        }
     }
 
     public static void dropDatabase(final String name) {
@@ -159,7 +174,23 @@ public void create(final String collectionName, final CreateCollectionOptions op
         if (validationOptions.getValidationAction() != null) {
             operation.validationAction(validationOptions.getValidationAction());
         }
-        operation.execute(getBinding());
+
+        // This loop is a workaround for unanticipated failures of the create command when run on a sharded cluster < 4.2
+        // In practice the command tends to succeed on the first attempt after a failure
+        boolean success = false;
+        while (!success) {
+            try {
+                operation.execute(getBinding());
+                success = true;
+            } catch (MongoCommandException e) {
+                if ("Interrupted".equals(e.getErrorCodeName())) {
+                    LOGGER.info("Retrying create collection after a write concern error: " + e);
+                    // repeat until success!
+                } else {
+                    throw e;
+                }
+            }
+        }
     }
 
     public void killCursor(final MongoNamespace namespace, final ServerCursor serverCursor) {
@@ -399,7 +430,7 @@ public List<BsonDocument> listIndexes(){
         return indexes;
     }
 
-    public void killAllSessions() {
+    public static void killAllSessions() {
         try {
             new CommandReadOperation<>("admin", new BsonDocument("killAllSessions", new BsonArray()),
                     new BsonDocumentCodec()).execute(getBinding());