Add OIDC test for infinite server selection timeout.

Author: vbabanin

driver-core/src/main/com/mongodb/internal/connection/OidcAuthenticator.java

@@ -26,7 +26,6 @@
 import com.mongodb.MongoSecurityException;
 import com.mongodb.ServerAddress;
 import com.mongodb.ServerApi;
-import com.mongodb.assertions.Assertions;
 import com.mongodb.connection.ClusterConnectionMode;
 import com.mongodb.connection.ConnectionDescription;
 import com.mongodb.internal.Locks;
@@ -47,6 +46,7 @@
 import java.nio.file.Files;
 import java.nio.file.Paths;
 import java.time.Duration;
+import java.time.temporal.ChronoUnit;
 import java.util.Arrays;
 import java.util.Collections;
 import java.util.List;
@@ -131,10 +131,9 @@ private Duration getCallbackTimeout(final TimeoutContext timeoutContext) {
 
         if (timeoutContext.hasTimeoutMS()) {
             return assertNotNull(timeoutContext.getTimeout()).call(TimeUnit.MILLISECONDS,
-                    () -> {
-                        Assertions.fail();
-                        return null;
-                    },
+                    () ->
+                          // we can get here if server selection timeout was set to infinite.
+                          ChronoUnit.FOREVER.getDuration(),
                     (renamingMs) -> Duration.ofMillis(renamingMs),
                     () -> throwMongoTimeoutException());
 

driver-sync/src/test/functional/com/mongodb/internal/connection/OidcAuthenticationProseTests.java

@@ -41,11 +41,11 @@
 import org.bson.Document;
 import org.junit.jupiter.api.AfterEach;
 import org.junit.jupiter.api.BeforeEach;
-import org.junit.jupiter.api.DisplayName;
 import org.junit.jupiter.api.Test;
 import org.junit.jupiter.params.ParameterizedTest;
 import org.junit.jupiter.params.provider.Arguments;
 import org.junit.jupiter.params.provider.MethodSource;
+import org.junit.jupiter.params.provider.ValueSource;
 
 import java.io.IOException;
 import java.lang.reflect.Field;
@@ -54,6 +54,7 @@
 import java.nio.file.Path;
 import java.nio.file.Paths;
 import java.time.Duration;
+import java.time.temporal.ChronoUnit;
 import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.Collections;
@@ -208,10 +209,12 @@ public void test2p1ValidCallbackInputs() {
     }
 
     // Not a prose test
-    @ParameterizedTest
+    @ParameterizedTest(name = "{testName}. "
+            + "Parameters: timeoutMs={1}, "
+            + "serverSelectionTimeoutMS={2},"
+            + " expectedTimeoutThreshold={3}")
     @MethodSource
-    @DisplayName("{testName}")
-    void testValidCallbackInputsTimeout(final String testName,
+    void testValidCallbackInputsTimeoutWhenTimeoutMsIsSet(final String testName,
                                         final int timeoutMs,
                                         final int serverSelectionTimeoutMS,
                                         final int expectedTimeoutThreshold) {
@@ -247,7 +250,7 @@ void testValidCallbackInputsTimeout(final String testName,
         }
     }
 
-    private static Stream<Arguments> testValidCallbackInputsTimeout() {
+    private static Stream<Arguments> testValidCallbackInputsTimeoutWhenTimeoutMsIsSet() {
         return Stream.of(
                 Arguments.of("serverSelectionTimeoutMS honored for oidc callback if it's lower than timeoutMS",
                         1000, // timeoutMS
@@ -264,6 +267,32 @@ private static Stream<Arguments> testValidCallbackInputsTimeout() {
         );
     }
 
+    // Not a prose test
+    @ParameterizedTest(name = "test callback timeout when server selection timeout is "
+            + "infinite and timeoutMs is set to {0}")
+    @ValueSource(ints = {0, 100})
+    void testCallbackTimeoutWhenServerSelectionTimeoutIsInfiniteTimeoutMsIsSet(final int timeoutMs) {
+        TestCallback callback1 = createCallback();
+
+        OidcCallback callback2 = (context) -> {
+            assertEquals(context.getTimeout(), ChronoUnit.FOREVER.getDuration());
+            return callback1.onRequest(context);
+        };
+
+        MongoClientSettings clientSettings = MongoClientSettings.builder(createSettings(callback2))
+                .applyToClusterSettings(builder ->
+                        builder.serverSelectionTimeout(
+                                -1, // -1 means infinite
+                                TimeUnit.MILLISECONDS))
+                .timeout(timeoutMs, TimeUnit.MILLISECONDS)
+                .build();
+
+        try (MongoClient mongoClient = createMongoClient(clientSettings)) {
+            performFind(mongoClient);
+            assertEquals(1, callback1.getInvocations());
+        }
+    }
+
     @Test
     public void test2p2RequestCallbackReturnsNull() {
         //noinspection ConstantConditions