24
24
import static org .junit .jupiter .api .Assertions .assertThrows ;
25
25
import static org .junit .jupiter .api .Assertions .assertTrue ;
26
26
import static org .mockito .ArgumentMatchers .any ;
27
- import static org .mockito .ArgumentMatchers .eq ;
28
27
import static org .mockito .ArgumentMatchers .anyList ;
29
28
import static org .mockito .ArgumentMatchers .anyString ;
29
+ import static org .mockito .ArgumentMatchers .eq ;
30
30
import static org .mockito .Mockito .doReturn ;
31
- import static org .mockito .Mockito .verify ;
32
31
import static org .mockito .Mockito .times ;
32
+ import static org .mockito .Mockito .verify ;
33
33
34
34
import com .mongodb .bulk .BulkWriteResult ;
35
35
import com .mongodb .client .ClientSession ;
36
- import com .mongodb .client .MongoDatabase ;
37
36
import com .mongodb .client .MongoCollection ;
37
+ import com .mongodb .client .MongoDatabase ;
38
38
import com .mongodb .client .model .DeleteManyModel ;
39
39
import com .mongodb .client .model .DeleteOneModel ;
40
40
import com .mongodb .client .model .InsertOneModel ;
@@ -247,9 +247,6 @@ private static Stream<Arguments> getMongoPreparedStatementMethodInvocationsImpac
247
247
@ Nested
248
248
class BatchTests {
249
249
250
- @ Mock
251
- private MongoDatabase mongoDatabase ;
252
-
253
250
@ Mock
254
251
private MongoCollection <BsonDocument > mongoCollection ;
255
252
@@ -275,8 +272,7 @@ class BatchTests {
275
272
void testSuccess () throws SQLException {
276
273
277
274
// given
278
- doReturn (mongoDatabase ).when (mongoClient ).getDatabase (anyString ());
279
- doReturn (mongoCollection ).when (mongoDatabase ).getCollection (anyString (), eq (BsonDocument .class ));
275
+ doReturn (mongoCollection ).when (mongoDatabase ).getCollection (eq ("books" ), eq (BsonDocument .class ));
280
276
doReturn (bulkWriteResult ).when (mongoCollection ).bulkWrite (eq (clientSession ), anyList ());
281
277
282
278
// when
@@ -362,7 +358,6 @@ void testClearBatch() throws SQLException {
362
358
void testBulkWriteModels (String mql , List <? extends WriteModel <BsonDocument >> expectedWriteModels )
363
359
throws SQLException {
364
360
// given
365
- doReturn (mongoDatabase ).when (mongoClient ).getDatabase (anyString ());
366
361
doReturn (mongoCollection ).when (mongoDatabase ).getCollection (anyString (), eq (BsonDocument .class ));
367
362
doReturn (bulkWriteResult ).when (mongoCollection ).bulkWrite (eq (clientSession ), anyList ());
368
363
0 commit comments