moved logic of ExponentialBackoff cleanup to correct class

Author: nhachicha

driver-core/src/test/unit/com/mongodb/internal/time/ExponentialBackoffTest.java

@@ -70,7 +70,7 @@ void testCustomJitter() {
         // Test with jitter = 0, all backoffs should be 0
         ExponentialBackoff.setTestJitterSupplier(() -> 0.0);
         try {
-            for (int attemptNumber = 1; attemptNumber < expectedBackoffs.length; attemptNumber++) {
+            for (int attemptNumber = 1; attemptNumber < EXPECTED_BACKOFFS_MAX_VALUES.length; attemptNumber++) {
                 long backoff = ExponentialBackoff.calculateTransactionBackoffMs(attemptNumber);
                 assertEquals(0, backoff, "With jitter=0, backoff should always be 0 ms");
             }

driver-reactive-streams/src/test/functional/com/mongodb/reactivestreams/client/ClientSideOperationTimeoutProseTest.java

@@ -26,7 +26,6 @@
 import com.mongodb.client.model.changestream.FullDocument;
 import com.mongodb.event.CommandFailedEvent;
 import com.mongodb.event.CommandStartedEvent;
-import com.mongodb.internal.time.ExponentialBackoff;
 import com.mongodb.reactivestreams.client.gridfs.GridFSBucket;
 import com.mongodb.reactivestreams.client.gridfs.GridFSBuckets;
 import com.mongodb.reactivestreams.client.syncadapter.SyncGridFSBucket;
@@ -579,15 +578,13 @@ private void assertOnlyOneCommandTimeoutFailure(final String command) {
     public void setUp() {
         super.setUp();
         SyncMongoClient.enableSleepAfterSessionClose(postSessionCloseSleep());
-        ExponentialBackoff.setTestJitterSupplier(() -> 0);
     }
 
     @Override
     @AfterEach
     public void tearDown() throws InterruptedException {
         super.tearDown();
         SyncMongoClient.disableSleep();
-        ExponentialBackoff.clearTestJitterSupplier();
     }
 
     @Override

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

@@ -19,6 +19,7 @@
 import com.mongodb.MongoClientSettings;
 import com.mongodb.client.gridfs.GridFSBucket;
 import com.mongodb.client.gridfs.GridFSBuckets;
+import com.mongodb.internal.time.ExponentialBackoff;
 
 
 /**
@@ -36,6 +37,22 @@ protected GridFSBucket createGridFsBucket(final MongoDatabase mongoDatabase, fin
         return GridFSBuckets.create(mongoDatabase, bucketName);
     }
 
+    @Override
+    public void setUp() {
+        super.setUp();
+        ExponentialBackoff.setTestJitterSupplier(() -> 0);
+    }
+
+    @Override
+    public void tearDown() throws InterruptedException {
+        super.tearDown();
+        try {
+            super.tearDown();
+        } finally {
+            ExponentialBackoff.clearTestJitterSupplier();
+        }
+    }
+
     @Override
     protected boolean isAsync() {
         return false;