Skip to content

Commit 51c736e

Browse files
authored
Fix test cleanup to prevent conflicts between tests. (#1608)
JAVA-5761
1 parent 75cb0e6 commit 51c736e

File tree

2 files changed

+18
-5
lines changed

2 files changed

+18
-5
lines changed

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

+8-3
Original file line numberDiff line numberDiff line change
@@ -20,9 +20,11 @@
2020
import com.mongodb.ClientEncryptionSettings;
2121
import com.mongodb.MongoClientSettings;
2222
import com.mongodb.MongoCredential;
23+
import com.mongodb.MongoNamespace;
2324
import com.mongodb.MongoSecurityException;
2425
import com.mongodb.WriteConcern;
2526
import com.mongodb.client.model.vault.EncryptOptions;
27+
import com.mongodb.client.test.CollectionHelper;
2628
import com.mongodb.client.vault.ClientEncryption;
2729
import com.mongodb.client.vault.ClientEncryptions;
2830
import org.bson.BsonBinary;
@@ -57,6 +59,7 @@ public class ClientSideEncryptionExternalKeyVaultTest {
5759
private MongoClient client, clientEncrypted;
5860
private ClientEncryption clientEncryption;
5961
private final boolean withExternalKeyVault;
62+
private static final MongoNamespace NAMESPACE = new MongoNamespace("db", ClientSideEncryptionExternalKeyVaultTest.class.getName());
6063

6164
public ClientSideEncryptionExternalKeyVaultTest(final boolean withExternalKeyVault) {
6265
this.withExternalKeyVault = withExternalKeyVault;
@@ -84,7 +87,7 @@ public void setUp() throws IOException, URISyntaxException {
8487
+ "UN3YkQ5aXRRMkhGRGdQV09wOGVNYUMxT2k3NjZKelhaQmRCZGJkTXVyZG9uSjFk");
8588
localMasterkey.put("key", localMasterkeyBytes);
8689
kmsProviders.put("local", localMasterkey);
87-
schemaMap.put("db.coll", bsonDocumentFromPath("external-schema.json"));
90+
schemaMap.put(NAMESPACE.getFullName(), bsonDocumentFromPath("external-schema.json"));
8891

8992
AutoEncryptionSettings.Builder autoEncryptionSettingsBuilder = AutoEncryptionSettings.builder()
9093
.keyVaultNamespace("keyvault.datakeys")
@@ -123,8 +126,8 @@ public void setUp() throws IOException, URISyntaxException {
123126
public void testExternal() {
124127
boolean authExceptionThrown = false;
125128
MongoCollection<BsonDocument> coll = clientEncrypted
126-
.getDatabase("db")
127-
.getCollection("coll", BsonDocument.class);
129+
.getDatabase(NAMESPACE.getDatabaseName())
130+
.getCollection(NAMESPACE.getCollectionName(), BsonDocument.class);
128131
try {
129132
coll.insertOne(new BsonDocument().append("encrypted", new BsonString("test")));
130133
} catch (MongoSecurityException mse) {
@@ -169,5 +172,7 @@ public void after() {
169172
// ignore
170173
}
171174
}
175+
176+
CollectionHelper.drop(NAMESPACE);
172177
}
173178
}

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

+10-2
Original file line numberDiff line numberDiff line change
@@ -34,6 +34,7 @@
3434
import com.mongodb.client.model.bulk.ClientBulkWriteOptions;
3535
import com.mongodb.client.model.bulk.ClientBulkWriteResult;
3636
import com.mongodb.client.model.bulk.ClientNamespacedWriteModel;
37+
import com.mongodb.client.test.CollectionHelper;
3738
import com.mongodb.event.CommandStartedEvent;
3839
import com.mongodb.internal.connection.TestCommandListener;
3940
import org.bson.BsonArray;
@@ -47,6 +48,7 @@
4748
import org.bson.RawBsonDocument;
4849
import org.bson.codecs.configuration.CodecRegistry;
4950
import org.bson.codecs.pojo.PojoCodecProvider;
51+
import org.junit.jupiter.api.AfterAll;
5052
import org.junit.jupiter.api.DisplayName;
5153
import org.junit.jupiter.api.Test;
5254
import org.junit.jupiter.params.ParameterizedTest;
@@ -96,7 +98,7 @@
9698
* <a href="https://github.com/mongodb/specifications/blob/master/source/crud/tests/README.md#prose-tests">CRUD Prose Tests</a>.
9799
*/
98100
public class CrudProseTest {
99-
private static final MongoNamespace NAMESPACE = new MongoNamespace("db", "coll");
101+
private static final MongoNamespace NAMESPACE = new MongoNamespace("db", CrudProseTest.class.getName());
100102

101103
@DisplayName("1. WriteConcernError.details exposes writeConcernError.errInfo")
102104
@Test
@@ -367,7 +369,8 @@ private void testBulkWriteSplitsWhenExceedingMaxMessageSizeBytesDueToNsInfo(
367369
Document helloResponse = droppedDatabase(client).runCommand(new Document("hello", 1));
368370
int maxBsonObjectSize = helloResponse.getInteger("maxBsonObjectSize");
369371
int maxMessageSizeBytes = helloResponse.getInteger("maxMessageSizeBytes");
370-
int opsBytes = maxMessageSizeBytes - 1122;
372+
// By the spec test, we have to subtract only 1122, however, we have different collection name.
373+
int opsBytes = maxMessageSizeBytes - 1118 - NAMESPACE.getCollectionName().length();
371374
int numModels = opsBytes / maxBsonObjectSize;
372375
int remainderBytes = opsBytes % maxBsonObjectSize;
373376
List<ClientNamespacedWriteModel> models = new ArrayList<>(nCopies(
@@ -613,4 +616,9 @@ private static ClientBulkWriteResult runInTransaction(final ClientSession sessio
613616
throw throwable;
614617
}
615618
}
619+
620+
@AfterAll
621+
public static void cleanUp() {
622+
CollectionHelper.drop(NAMESPACE);
623+
}
616624
}

0 commit comments

Comments
 (0)