Update UnifiedCrudHelper with new operations and arguments

JAVA-4798

Author: jyemin

driver-core/src/test/resources/unified-test-format/retryable-reads/handshakeError.json

@@ -9,7 +9,7 @@
         "sharded",
         "load-balanced"
       ],
-      "auth": true
+      "auth": false
     }
   ],
   "createEntities": [

driver-core/src/test/resources/unified-test-format/retryable-writes/handshakeError.json

@@ -208,7 +208,7 @@
               "x": 22
             }
           }
-        }
+        }                                                                 ¡¡
       ],
       "expectEvents": [
         {

driver-sync/src/test/functional/com/mongodb/client/unified/UnifiedCrudHelper.java

@@ -38,6 +38,7 @@
 import com.mongodb.client.MongoCollection;
 import com.mongodb.client.MongoCursor;
 import com.mongodb.client.MongoDatabase;
+import com.mongodb.client.MongoIterable;
 import com.mongodb.client.model.BulkWriteOptions;
 import com.mongodb.client.model.ChangeStreamPreAndPostImagesOptions;
 import com.mongodb.client.model.ClusteredIndexOptions;
@@ -169,10 +170,12 @@ OperationResult executeListDatabases(final BsonDocument operation) {
                 : client.listDatabases(session, BsonDocument.class);
 
         for (Map.Entry<String, BsonValue> cur : arguments.entrySet()) {
-            //noinspection SwitchStatementWithTooFewBranches
             switch (cur.getKey()) {
                 case "session":
                     break;
+                case "filter":
+                    iterable.filter(cur.getValue().asDocument());
+                    break;
                 default:
                     throw new UnsupportedOperationException("Unsupported argument: " + cur.getKey());
             }
@@ -182,6 +185,29 @@ OperationResult executeListDatabases(final BsonDocument operation) {
                 new BsonArray(iterable.into(new ArrayList<>())));
     }
 
+    OperationResult executeListDatabaseNames(final BsonDocument operation) {
+        MongoClient client = entities.getClient(operation.getString("object").getValue());
+
+        BsonDocument arguments = operation.getDocument("arguments", new BsonDocument());
+        ClientSession session = getSession(arguments);
+        MongoIterable<String> iterable = session == null
+                ? client.listDatabaseNames()
+                : client.listDatabaseNames(session);
+
+        for (Map.Entry<String, BsonValue> cur : arguments.entrySet()) {
+            //noinspection SwitchStatementWithTooFewBranches
+            switch (cur.getKey()) {
+                case "session":
+                    break;
+                default:
+                    throw new UnsupportedOperationException("Unsupported argument: " + cur.getKey());
+            }
+        }
+
+        return resultOf(() ->
+                new BsonArray(iterable.into(new ArrayList<>()).stream().map(BsonString::new).collect(toList())));
+    }
+
     OperationResult executeListCollections(final BsonDocument operation) {
         MongoDatabase database = entities.getDatabase(operation.getString("object").getValue());
 
@@ -209,7 +235,51 @@ OperationResult executeListCollections(final BsonDocument operation) {
                 new BsonArray(iterable.into(new ArrayList<>())));
     }
 
+    OperationResult executeListCollectionNames(final BsonDocument operation) {
+        MongoDatabase database = entities.getDatabase(operation.getString("object").getValue());
+
+        BsonDocument arguments = operation.getDocument("arguments");
+        ClientSession session = getSession(arguments);
+        MongoIterable<String> iterable = session == null
+                ? database.listCollectionNames()
+                : database.listCollectionNames(session);
+        for (Map.Entry<String, BsonValue> cur : arguments.entrySet()) {
+            switch (cur.getKey()) {
+                case "session":
+                    break;
+                case "filter":
+                    BsonDocument filter = cur.getValue().asDocument();
+                    if (!filter.isEmpty()) {
+                       throw new UnsupportedOperationException("The driver does not support filtering of collection names");
+                    }
+                    break;
+                case "batchSize":
+                    iterable.batchSize(cur.getValue().asNumber().intValue());
+                    break;
+                default:
+                    throw new UnsupportedOperationException("Unsupported argument: " + cur.getKey());
+            }
+        }
+
+        return resultOf(() ->
+                new BsonArray(iterable.into(new ArrayList<>()).stream().map(BsonString::new).collect(toList())));
+    }
+
     OperationResult executeListIndexes(final BsonDocument operation) {
+        ListIndexesIterable<BsonDocument> iterable = createListIndexesIterable(operation);
+
+        return resultOf(() ->
+                new BsonArray(iterable.into(new ArrayList<>())));
+    }
+
+    OperationResult executeListIndexNames(final BsonDocument operation) {
+        ListIndexesIterable<BsonDocument> iterable = createListIndexesIterable(operation);
+
+        return resultOf(() ->
+                new BsonArray(iterable.into(new ArrayList<>()).stream().map(document -> document.getString("name")).collect(toList())));
+    }
+
+    private ListIndexesIterable<BsonDocument> createListIndexesIterable(final BsonDocument operation) {
         MongoCollection<BsonDocument> collection = entities.getCollection(operation.getString("object").getValue());
         BsonDocument arguments = operation.getDocument("arguments", new BsonDocument());
         ClientSession session = getSession(arguments);
@@ -227,9 +297,7 @@ OperationResult executeListIndexes(final BsonDocument operation) {
                     throw new UnsupportedOperationException("Unsupported argument: " + cur.getKey());
             }
         }
-
-        return resultOf(() ->
-                new BsonArray(iterable.into(new ArrayList<>())));
+        return iterable;
     }
 
     OperationResult executeFind(final BsonDocument operation) {
@@ -238,6 +306,11 @@ OperationResult executeFind(final BsonDocument operation) {
                 new BsonArray(iterable.into(new ArrayList<>())));
     }
 
+    OperationResult executeFindOne(final BsonDocument operation) {
+        FindIterable<BsonDocument> iterable = createFindIterable(operation);
+        return resultOf(iterable::first);
+    }
+
     OperationResult createFindCursor(final BsonDocument operation) {
         FindIterable<BsonDocument> iterable = createFindIterable(operation);
         return resultOf(() -> {

driver-sync/src/test/functional/com/mongodb/client/unified/UnifiedTest.java

@@ -357,6 +357,8 @@ private OperationResult executeOperation(final UnifiedTestContext context, final
                     return crudHelper.executeAggregate(operation);
                 case "find":
                     return crudHelper.executeFind(operation);
+                case "findOne":
+                    return crudHelper.executeFindOne(operation);
                 case "distinct":
                     return crudHelper.executeDistinct(operation);
                 case "countDocuments":
@@ -371,10 +373,16 @@ private OperationResult executeOperation(final UnifiedTestContext context, final
                     return crudHelper.executeFindOneAndDelete(operation);
                 case "listDatabases":
                     return crudHelper.executeListDatabases(operation);
+                case "listDatabaseNames":
+                    return crudHelper.executeListDatabaseNames(operation);
                 case "listCollections":
                     return crudHelper.executeListCollections(operation);
+                case "listCollectionNames":
+                    return crudHelper.executeListCollectionNames(operation);
                 case "listIndexes":
                     return crudHelper.executeListIndexes(operation);
+                case "listIndexNames":
+                    return crudHelper.executeListIndexNames(operation);
                 case "dropCollection":
                     return crudHelper.executeDropCollection(operation);
                 case "createCollection":