Skip to content

Commit feda7e5

Browse files
authored
Use @VisibleForTesting instead of writing verbose comments (#835)
1 parent 8aa744d commit feda7e5

File tree

7 files changed

+26
-34
lines changed

7 files changed

+26
-34
lines changed

driver-core/src/main/com/mongodb/connection/ClusterId.java

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,9 +16,11 @@
1616

1717
package com.mongodb.connection;
1818

19+
import com.mongodb.internal.VisibleForTesting;
1920
import org.bson.types.ObjectId;
2021

2122
import static com.mongodb.assertions.Assertions.notNull;
23+
import static com.mongodb.internal.VisibleForTesting.AccessModifier.PRIVATE;
2224

2325
/**
2426
* A client-generated identifier that uniquely identifies a connection to a MongoDB cluster, which could be sharded, replica set,
@@ -48,7 +50,7 @@ public ClusterId(final String description) {
4850
this.description = description;
4951
}
5052

51-
// for testing only, as cluster identifiers should really be unique
53+
@VisibleForTesting(otherwise = PRIVATE)
5254
ClusterId(final String value, final String description) {
5355
this.value = notNull("value", value);
5456
this.description = description;

driver-core/src/main/com/mongodb/internal/Timeout.java

Lines changed: 5 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,7 @@
2323
import static com.mongodb.assertions.Assertions.assertFalse;
2424
import static com.mongodb.assertions.Assertions.assertNotNull;
2525
import static com.mongodb.assertions.Assertions.assertTrue;
26+
import static com.mongodb.internal.VisibleForTesting.AccessModifier.PRIVATE;
2627
import static java.util.concurrent.TimeUnit.MILLISECONDS;
2728
import static java.util.concurrent.TimeUnit.NANOSECONDS;
2829

@@ -135,13 +136,12 @@ private long elapsedNanos(final long currentNanos) {
135136
}
136137

137138
/**
138-
* Is package-access for the purpose of testing and must not be used for any other purpose outside of this class.
139-
* <p>
140139
* Returns 0 or a positive value.
141140
* 0 means that the timeout has expired.
142141
* <p>
143142
* Must not be called on {@linkplain #isInfinite() infinite} timeouts.
144143
*/
144+
@VisibleForTesting(otherwise = PRIVATE)
145145
long remainingNanos(final long currentNanos) {
146146
assertFalse(isInfinite() || isImmediate());
147147
long elapsedNanos = elapsedNanos(currentNanos);
@@ -262,23 +262,17 @@ public String toUserString() {
262262
}
263263
}
264264

265-
/**
266-
* Is package-access for the purpose of testing and must not be used for any other purpose outside of this class.
267-
*/
265+
@VisibleForTesting(otherwise = PRIVATE)
268266
long durationNanos() {
269267
return durationNanos;
270268
}
271269

272-
/**
273-
* Is package-access for the purpose of testing and must not be used for any other purpose outside of this class.
274-
*/
270+
@VisibleForTesting(otherwise = PRIVATE)
275271
long startNanos() {
276272
return startNanos;
277273
}
278274

279-
/**
280-
* Is package-access for the purpose of testing and must not be used for any other purpose outside of this class.
281-
*/
275+
@VisibleForTesting(otherwise = PRIVATE)
282276
static long convertRoundUp(final long nonNegativeNanos, final TimeUnit unit) {
283277
assertTrue(nonNegativeNanos >= 0);
284278
if (unit == NANOSECONDS) {

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

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,7 @@
2222
import com.mongodb.MongoServerUnavailableException;
2323
import com.mongodb.MongoTimeoutException;
2424
import com.mongodb.annotations.ThreadSafe;
25+
import com.mongodb.internal.VisibleForTesting;
2526
import com.mongodb.internal.connection.ConcurrentLinkedDeque.RemovalReportingIterator;
2627
import com.mongodb.lang.Nullable;
2728

@@ -36,6 +37,7 @@
3637
import static com.mongodb.assertions.Assertions.assertNotNull;
3738
import static com.mongodb.assertions.Assertions.assertTrue;
3839
import static com.mongodb.assertions.Assertions.notNull;
40+
import static com.mongodb.internal.VisibleForTesting.AccessModifier.PRIVATE;
3941

4042
/**
4143
* A concurrent pool implementation.
@@ -240,10 +242,9 @@ private T createNewAndReleasePermitIfFailure() {
240242
}
241243

242244
/**
243-
* Is package-access for the purpose of testing and must not be used for any other purpose outside of this class.
244-
*
245245
* @param timeout See {@link com.mongodb.internal.Timeout#startNow(long, TimeUnit)}.
246246
*/
247+
@VisibleForTesting(otherwise = PRIVATE)
247248
boolean acquirePermit(final long timeout, final TimeUnit timeUnit) {
248249
return stateAndPermits.acquirePermitFair(timeout, timeUnit);
249250
}

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

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -45,6 +45,7 @@
4545
import com.mongodb.event.ConnectionPoolReadyEvent;
4646
import com.mongodb.event.ConnectionReadyEvent;
4747
import com.mongodb.internal.Timeout;
48+
import com.mongodb.internal.VisibleForTesting;
4849
import com.mongodb.internal.async.SingleResultCallback;
4950
import com.mongodb.internal.connection.ConcurrentPool.Prune;
5051
import com.mongodb.internal.connection.SdamServerDescriptionManager.SdamIssue;
@@ -89,6 +90,7 @@
8990
import static com.mongodb.assertions.Assertions.fail;
9091
import static com.mongodb.assertions.Assertions.isTrue;
9192
import static com.mongodb.assertions.Assertions.notNull;
93+
import static com.mongodb.internal.VisibleForTesting.AccessModifier.PRIVATE;
9294
import static com.mongodb.internal.async.ErrorHandlingResultCallback.errorHandlingCallback;
9395
import static com.mongodb.internal.connection.ConcurrentPool.INFINITE_SIZE;
9496
import static com.mongodb.internal.connection.ConcurrentPool.sizeToString;
@@ -366,18 +368,15 @@ private MongoTimeoutException createTimeoutException(final Timeout timeout) {
366368
}
367369
}
368370

369-
/**
370-
* Is package-access for the purpose of testing and must not be used for any other purpose outside of this class.
371-
*/
371+
@VisibleForTesting(otherwise = PRIVATE)
372372
ConcurrentPool<UsageTrackingInternalConnection> getPool() {
373373
return pool;
374374
}
375375

376376
/**
377-
* Is package-access for the purpose of testing and must not be used for any other purpose outside of this class.
378-
* <p>
379377
* Synchronously prune idle connections and ensure the minimum pool size.
380378
*/
379+
@VisibleForTesting(otherwise = PRIVATE)
381380
void doMaintenance() {
382381
Predicate<RuntimeException> silentlyComplete = e ->
383382
e instanceof MongoInterruptedException || e instanceof MongoTimeoutException

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

Lines changed: 5 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,7 @@
3333
import com.mongodb.event.ServerClosedEvent;
3434
import com.mongodb.event.ServerListener;
3535
import com.mongodb.event.ServerOpeningEvent;
36+
import com.mongodb.internal.VisibleForTesting;
3637
import com.mongodb.internal.async.SingleResultCallback;
3738
import com.mongodb.internal.bulk.DeleteRequest;
3839
import com.mongodb.internal.bulk.InsertRequest;
@@ -50,6 +51,7 @@
5051
import static com.mongodb.assertions.Assertions.assertNotNull;
5152
import static com.mongodb.assertions.Assertions.assertTrue;
5253
import static com.mongodb.assertions.Assertions.notNull;
54+
import static com.mongodb.internal.VisibleForTesting.AccessModifier.PRIVATE;
5355
import static com.mongodb.internal.async.ErrorHandlingResultCallback.errorHandlingCallback;
5456
import static com.mongodb.internal.connection.ServerDescriptionHelper.unknownConnectingServerDescription;
5557

@@ -184,23 +186,17 @@ public void connect() {
184186
serverMonitor.connect();
185187
}
186188

187-
/**
188-
* Is package-access for the purpose of testing and must not be used for any other purpose outside of this class.
189-
*/
189+
@VisibleForTesting(otherwise = PRIVATE)
190190
ConnectionPool getConnectionPool() {
191191
return connectionPool;
192192
}
193193

194-
/**
195-
* Is package-access for the purpose of testing and must not be used for any other purpose outside of this class.
196-
*/
194+
@VisibleForTesting(otherwise = PRIVATE)
197195
SdamServerDescriptionManager sdamServerDescriptionManager() {
198196
return sdam;
199197
}
200198

201-
/**
202-
* Is package-access for the purpose of testing and must not be used for any other purpose outside of this class.
203-
*/
199+
@VisibleForTesting(otherwise = PRIVATE)
204200
ServerId serverId() {
205201
return serverId;
206202
}

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

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -34,6 +34,7 @@
3434
import com.mongodb.event.ServerDescriptionChangedEvent;
3535
import com.mongodb.event.ServerListener;
3636
import com.mongodb.event.ServerOpeningEvent;
37+
import com.mongodb.internal.VisibleForTesting;
3738
import com.mongodb.internal.async.SingleResultCallback;
3839
import com.mongodb.internal.session.SessionContext;
3940
import org.bson.types.ObjectId;
@@ -42,6 +43,7 @@
4243

4344
import static com.mongodb.assertions.Assertions.isTrue;
4445
import static com.mongodb.connection.ServerConnectionState.CONNECTING;
46+
import static com.mongodb.internal.VisibleForTesting.AccessModifier.PRIVATE;
4547
import static com.mongodb.internal.async.ErrorHandlingResultCallback.errorHandlingCallback;
4648

4749
@ThreadSafe
@@ -143,9 +145,7 @@ public int operationCount() {
143145
return -1;
144146
}
145147

146-
/**
147-
* Is package-access for the purpose of testing and must not be used for any other purpose outside of this class.
148-
*/
148+
@VisibleForTesting(otherwise = PRIVATE)
149149
ConnectionPool getConnectionPool() {
150150
return connectionPool;
151151
}

driver-core/src/main/com/mongodb/internal/operation/CommandOperationHelper.java

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,7 @@
2929
import com.mongodb.assertions.Assertions;
3030
import com.mongodb.connection.ConnectionDescription;
3131
import com.mongodb.connection.ServerDescription;
32+
import com.mongodb.internal.VisibleForTesting;
3233
import com.mongodb.internal.async.SingleResultCallback;
3334
import com.mongodb.internal.async.function.AsyncCallbackSupplier;
3435
import com.mongodb.internal.async.function.RetryState;
@@ -57,6 +58,7 @@
5758
import static com.mongodb.ReadPreference.primary;
5859
import static com.mongodb.assertions.Assertions.assertFalse;
5960
import static com.mongodb.assertions.Assertions.notNull;
61+
import static com.mongodb.internal.VisibleForTesting.AccessModifier.PRIVATE;
6062
import static com.mongodb.internal.async.ErrorHandlingResultCallback.errorHandlingCallback;
6163
import static com.mongodb.internal.operation.OperationHelper.LOGGER;
6264
import static com.mongodb.internal.operation.OperationHelper.canRetryRead;
@@ -231,9 +233,7 @@ static <D, T> T createReadCommandAndExecute(
231233

232234
/* Write Binding Helpers */
233235

234-
/**
235-
* Is package-access for the purpose of testing and must not be used for any other purpose outside of this class.
236-
*/
236+
@VisibleForTesting(otherwise = PRIVATE)
237237
static <D, T> T executeCommand(final WriteBinding binding, final String database, final BsonDocument command,
238238
final Decoder<D> decoder, final CommandWriteTransformer<D, T> transformer) {
239239
return withSourceAndConnection(binding::getWriteConnectionSource, false, (source, connection) ->

0 commit comments

Comments
 (0)