Skip to content

Commit 145ae9e

Browse files
committed
feat: add MongoDB driver handshake metadata to MongoDBResourceManager
1 parent d6ab992 commit 145ae9e

2 files changed

Lines changed: 13 additions & 1 deletion

File tree

it/mongodb/src/main/java/org/apache/beam/it/mongodb/MongoDBResourceManager.java

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@
2020
import static org.apache.beam.it.mongodb.MongoDBResourceManagerUtils.checkValidCollectionName;
2121
import static org.apache.beam.it.mongodb.MongoDBResourceManagerUtils.generateDatabaseName;
2222

23+
import com.mongodb.MongoDriverInformation;
2324
import com.mongodb.client.FindIterable;
2425
import com.mongodb.client.MongoClient;
2526
import com.mongodb.client.MongoClients;
@@ -57,6 +58,10 @@ public class MongoDBResourceManager extends TestContainerResourceManager<MongoDB
5758

5859
private static final String DEFAULT_MONGODB_CONTAINER_NAME = "mongo";
5960

61+
@VisibleForTesting
62+
static final MongoDriverInformation DRIVER_INFO =
63+
MongoDriverInformation.builder().driverName("Apache Beam").build();
64+
6065
// A list of available MongoDB Docker image tags can be found at
6166
// https://hub.docker.com/_/mongo/tags
6267
private static final String DEFAULT_MONGODB_CONTAINER_TAG = "4.0.18";
@@ -88,7 +93,8 @@ private MongoDBResourceManager(Builder builder) {
8893
usingStaticDatabase ? builder.databaseName : generateDatabaseName(builder.testId);
8994
this.connectionString =
9095
String.format("mongodb://%s:%d", this.getHost(), this.getPort(MONGODB_INTERNAL_PORT));
91-
this.mongoClient = mongoClient == null ? MongoClients.create(connectionString) : mongoClient;
96+
this.mongoClient =
97+
mongoClient == null ? MongoClients.create(connectionString, DRIVER_INFO) : mongoClient;
9298
}
9399

94100
public static Builder builder(String testId) {

it/mongodb/src/test/java/org/apache/beam/it/mongodb/MongoDBResourceManagerTest.java

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@
2727
import static org.mockito.Mockito.verify;
2828
import static org.mockito.Mockito.when;
2929

30+
import com.mongodb.MongoDriverInformation;
3031
import com.mongodb.MongoBulkWriteException;
3132
import com.mongodb.client.MongoClient;
3233
import com.mongodb.client.MongoCollection;
@@ -71,6 +72,11 @@ public void setUp() {
7172
new MongoDBResourceManager(mongoClient, container, MongoDBResourceManager.builder(TEST_ID));
7273
}
7374

75+
@Test
76+
public void testDriverInfoHasExpectedName() {
77+
assertThat(MongoDBResourceManager.DRIVER_INFO.getDriverNames()).contains("Apache Beam");
78+
}
79+
7480
@Test
7581
public void testCreateResourceManagerBuilderReturnsMongoDBResourceManager() {
7682
assertThat(

0 commit comments

Comments
 (0)