Reuse OperationContext.

Author: vbabanin

driver-core/src/test/functional/com/mongodb/connection/ConnectionSpecification.groovy

@@ -17,6 +17,7 @@
 package com.mongodb.connection
 
 import com.mongodb.OperationFunctionalSpecification
+import com.mongodb.internal.connection.OperationContext
 import com.mongodb.internal.operation.CommandReadOperation
 import org.bson.BsonDocument
 import org.bson.BsonInt32
@@ -32,8 +33,9 @@ class ConnectionSpecification extends OperationFunctionalSpecification {
 
     def 'should have id'() {
         when:
-        def source = getBinding().getReadConnectionSource(createOperationContext())
-        def connection = source.getConnection(createOperationContext())
+        def operationContext = createOperationContext()
+        def source = getBinding().getReadConnectionSource(operationContext)
+        def connection = source.getConnection(operationContext)
 
         then:
         connection.getDescription().getConnectionId() != null
@@ -45,13 +47,14 @@ class ConnectionSpecification extends OperationFunctionalSpecification {
 
     def 'should have description'() {
         when:
-        def commandResult = getHelloResult()
+        def operationContext = createOperationContext()
+        def commandResult = getHelloResult(operationContext)
         def expectedMaxMessageSize = commandResult.getNumber('maxMessageSizeBytes',
                                                              new BsonInt32(getDefaultMaxMessageSize())).intValue()
         def expectedMaxBatchCount = commandResult.getNumber('maxWriteBatchSize',
                                                             new BsonInt32(getDefaultMaxWriteBatchSize())).intValue()
-        def source = getBinding().getReadConnectionSource(createOperationContext())
-        def connection = source.getConnection(createOperationContext())
+        def source = getBinding().getReadConnectionSource(operationContext)
+        def connection = source.getConnection(operationContext)
 
         then:
         connection.description.serverAddress == source.getServerDescription().getAddress()
@@ -64,8 +67,8 @@ class ConnectionSpecification extends OperationFunctionalSpecification {
         connection?.release()
         source?.release()
     }
-   private static BsonDocument getHelloResult() {
+   private static BsonDocument getHelloResult(OperationContext operationContext) {
         new CommandReadOperation<BsonDocument>('admin', new BsonDocument(LEGACY_HELLO, new BsonInt32(1)),
-                new BsonDocumentCodec()).execute(getBinding(), createOperationContext())
+                new BsonDocumentCodec()).execute(getBinding(), operationContext)
     }
 }

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

@@ -82,7 +82,8 @@ protected void applyResponse(final BsonArray response) {
     }
 
     protected void applyApplicationError(final BsonDocument applicationError) {
-        Timeout serverSelectionTimeout = ClusterFixture.createOperationContext().getTimeoutContext().computeServerSelectionTimeout();
+        OperationContext operationContext = ClusterFixture.createOperationContext();
+        Timeout serverSelectionTimeout = operationContext.getTimeoutContext().computeServerSelectionTimeout();
         ServerAddress serverAddress = new ServerAddress(applicationError.getString("address").getValue());
         TimeoutContext timeoutContext = new TimeoutContext(TIMEOUT_SETTINGS);
         int errorGeneration = applicationError.getNumber("generation",
@@ -98,7 +99,7 @@ protected void applyApplicationError(final BsonDocument applicationError) {
         switch (type) {
             case "command":
                 exception = getCommandFailureException(applicationError.getDocument("response"), serverAddress,
-                        ClusterFixture.createOperationContext().getTimeoutContext());
+                        operationContext.getTimeoutContext());
                 break;
             case "network":
                 exception = new MongoSocketReadException("Read error", serverAddress, new IOException());

driver-core/src/test/unit/com/mongodb/internal/connection/MultiServerClusterSpecification.groovy

@@ -93,11 +93,12 @@ class MultiServerClusterSpecification extends Specification {
         def cluster = new MultiServerCluster(CLUSTER_ID, ClusterSettings.builder().hosts(Arrays.asList(firstServer)).mode(MULTIPLE).build(),
                 factory, CLIENT_METADATA)
         cluster.close()
+        def operationContext = createOperationContext()
 
         when:
         cluster.getServersSnapshot(
-                createOperationContext().getTimeoutContext().computeServerSelectionTimeout(),
-                createOperationContext().getTimeoutContext())
+                operationContext.getTimeoutContext().computeServerSelectionTimeout(),
+                operationContext.getTimeoutContext())
 
         then:
         thrown(IllegalStateException)

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

@@ -154,9 +154,10 @@ private void assertServer(final String serverName, final BsonDocument expectedSe
 
         if (expectedServerDescriptionDocument.isDocument("pool")) {
             int expectedGeneration = expectedServerDescriptionDocument.getDocument("pool").getNumber("generation").intValue();
-            Timeout serverSelectionTimeout = ClusterFixture.createOperationContext().getTimeoutContext().computeServerSelectionTimeout();
+            OperationContext operationContext = ClusterFixture.createOperationContext();
+            Timeout serverSelectionTimeout = operationContext.getTimeoutContext().computeServerSelectionTimeout();
             DefaultServer server = (DefaultServer) getCluster()
-                    .getServersSnapshot(serverSelectionTimeout, ClusterFixture.createOperationContext().getTimeoutContext())
+                    .getServersSnapshot(serverSelectionTimeout, operationContext.getTimeoutContext())
                     .getServer(new ServerAddress(serverName));
             assertEquals(expectedGeneration, server.getConnectionPool().getGeneration());
         }