Skip to content

Commit c4e904b

Browse files
committed
Support Convert CSFLE spec tests to unified test format changes
JAVA-5674
1 parent 554fe79 commit c4e904b

File tree

3 files changed

+35
-9
lines changed

3 files changed

+35
-9
lines changed

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

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -179,6 +179,11 @@ public void create(final WriteConcern writeConcern, final BsonDocument createOpt
179179
case "encryptedFields":
180180
createCollectionOptions.encryptedFields(createOptions.getDocument("encryptedFields"));
181181
break;
182+
case "validator":
183+
ValidationOptions validationOptions = new ValidationOptions();
184+
validationOptions.validator(createOptions.getDocument("validator"));
185+
createCollectionOptions.validationOptions(validationOptions);
186+
break;
182187
default:
183188
throw new UnsupportedOperationException("Unsupported create collection option: " + option);
184189
}
@@ -198,6 +203,10 @@ public void create(final String collectionName, final CreateCollectionOptions op
198203
if (indexOptionDefaults.getStorageEngine() != null) {
199204
operation.indexOptionDefaults(new BsonDocument("storageEngine", toBsonDocument(indexOptionDefaults.getStorageEngine())));
200205
}
206+
Bson encryptedFields = options.getEncryptedFields();
207+
if (encryptedFields != null) {
208+
operation.encryptedFields(encryptedFields.toBsonDocument());
209+
}
201210
ValidationOptions validationOptions = options.getValidationOptions();
202211
if (validationOptions.getValidator() != null) {
203212
operation.validator(toBsonDocument(validationOptions.getValidator()));

driver-core/src/test/unit/com/mongodb/internal/connection/TestClusterListener.java

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,6 @@
3232
import java.util.concurrent.locks.ReentrantLock;
3333
import java.util.function.Predicate;
3434

35-
import static com.mongodb.assertions.Assertions.isTrue;
3635
import static com.mongodb.assertions.Assertions.notNull;
3736
import static com.mongodb.internal.Locks.withLock;
3837

@@ -48,15 +47,17 @@ public final class TestClusterListener implements ClusterListener {
4847

4948
@Override
5049
public void clusterOpening(final ClusterOpeningEvent event) {
51-
isTrue("clusterOpeningEvent is null", clusterOpeningEvent == null);
52-
clusterOpeningEvent = event;
50+
if(clusterOpeningEvent == null) {
51+
clusterOpeningEvent = event;
52+
}
5353
}
5454

5555
@Override
5656
public void clusterClosed(final ClusterClosedEvent event) {
57-
isTrue("clusterClosingEvent is null", clusterClosingEvent == null);
58-
closedLatch.countDown();
59-
clusterClosingEvent = event;
57+
if(clusterClosingEvent == null) {
58+
closedLatch.countDown();
59+
clusterClosingEvent = event;
60+
}
6061
}
6162

6263
@Override

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

Lines changed: 19 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -57,6 +57,7 @@
5757
import com.mongodb.client.model.DeleteManyModel;
5858
import com.mongodb.client.model.DeleteOneModel;
5959
import com.mongodb.client.model.DeleteOptions;
60+
import com.mongodb.client.model.DropCollectionOptions;
6061
import com.mongodb.client.model.DropIndexOptions;
6162
import com.mongodb.client.model.EstimatedDocumentCountOptions;
6263
import com.mongodb.client.model.FindOneAndDeleteOptions;
@@ -77,6 +78,7 @@
7778
import com.mongodb.client.model.UpdateManyModel;
7879
import com.mongodb.client.model.UpdateOneModel;
7980
import com.mongodb.client.model.UpdateOptions;
81+
import com.mongodb.client.model.ValidationOptions;
8082
import com.mongodb.client.model.WriteModel;
8183
import com.mongodb.client.model.bulk.ClientBulkWriteOptions;
8284
import com.mongodb.client.model.bulk.ClientBulkWriteResult;
@@ -1364,12 +1366,21 @@ public OperationResult executeDropCollection(final BsonDocument operation) {
13641366
BsonDocument arguments = operation.getDocument("arguments", new BsonDocument());
13651367
String collectionName = arguments.getString("collection").getValue();
13661368

1367-
if (operation.getDocument("arguments").size() > 1) {
1368-
throw new UnsupportedOperationException("Unexpected arguments " + operation.get("arguments"));
1369+
DropCollectionOptions dropCollectionOptions = new DropCollectionOptions();
1370+
for (Map.Entry<String, BsonValue> entry : arguments.entrySet()) {
1371+
switch (entry.getKey()) {
1372+
case "collection":
1373+
break;
1374+
case "encryptedFields":
1375+
dropCollectionOptions.encryptedFields(entry.getValue().asDocument());
1376+
break;
1377+
default:
1378+
throw new UnsupportedOperationException("Unsupported drop collections option: " + entry.getKey());
1379+
}
13691380
}
13701381

13711382
return resultOf(() -> {
1372-
database.getCollection(collectionName).drop();
1383+
database.getCollection(collectionName).drop(dropCollectionOptions);
13731384
return null;
13741385
});
13751386
}
@@ -1429,6 +1440,11 @@ public OperationResult executeCreateCollection(final BsonDocument operation) {
14291440
case "clusteredIndex":
14301441
options.clusteredIndexOptions(createClusteredIndexOptions(cur.getValue().asDocument()));
14311442
break;
1443+
case "validator":
1444+
ValidationOptions validationOptions = new ValidationOptions();
1445+
validationOptions.validator(cur.getValue().asDocument());
1446+
options.validationOptions(validationOptions);
1447+
break;
14321448
default:
14331449
throw new UnsupportedOperationException("Unsupported argument: " + cur.getKey());
14341450
}

0 commit comments

Comments
 (0)