mindol1004/spring-data-mongodb
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

DATAMONGO-2330 - Polishing.

Browse Source 
Added tests.

Original Pull Request: #775
This commit is contained in:
Christoph Strobl
2019-07-31 14:12:07 +02:00
parent e6ea2e1379
commit 74325d5193
1 changed files with 23 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
spring-data-mongodb/src/test/java/org/springframework/data/mongodb/core/DefaultBulkOperationsUnitTests.java
View File

@@ -30,6 +30,7 @@ import org.bson.Document;
import org.junit.Before; import org.junit.Before;
import org.junit.Test; import org.junit.Test;
import org.junit.runner.RunWith; import org.junit.runner.RunWith;
import org.mockito.Answers;
import org.mockito.ArgumentCaptor; import org.mockito.ArgumentCaptor;
import org.mockito.Captor; import org.mockito.Captor;
import org.mockito.Mock; import org.mockito.Mock;
@@ -49,6 +50,7 @@ import org.springframework.data.mongodb.core.query.BasicQuery;
import org.springframework.data.mongodb.core.query.Collation; import org.springframework.data.mongodb.core.query.Collation;
import org.springframework.data.mongodb.core.query.Update; import org.springframework.data.mongodb.core.query.Update;
import com.mongodb.WriteConcern;
import com.mongodb.client.MongoCollection; import com.mongodb.client.MongoCollection;
import com.mongodb.client.MongoDatabase; import com.mongodb.client.MongoDatabase;
import com.mongodb.client.model.DeleteManyModel; import com.mongodb.client.model.DeleteManyModel;
@@ -67,7 +69,7 @@ public class DefaultBulkOperationsUnitTests {
MongoTemplate template; MongoTemplate template;
@Mock MongoDatabase database; @Mock MongoDatabase database;
@Mock MongoCollection<Document> collection; @Mock(answer = Answers.RETURNS_DEEP_STUBS) MongoCollection<Document> collection;
@Mock MongoDbFactory factory; @Mock MongoDbFactory factory;
@Mock DbRefResolver dbRefResolver; @Mock DbRefResolver dbRefResolver;
@Captor ArgumentCaptor<List<WriteModel<Document>>> captor; @Captor ArgumentCaptor<List<WriteModel<Document>>> captor;
@@ -156,6 +158,26 @@ public class DefaultBulkOperationsUnitTests {
assertThat(updateModel.getFilter()).isEqualTo(new Document("first_name", "danerys")); assertThat(updateModel.getFilter()).isEqualTo(new Document("first_name", "danerys"));
} }
@Test // DATAMONGO-2330
public void writeConcernNotAppliedWhenNotSet() {
ops.updateOne(new BasicQuery("{}").collation(Collation.of("de")), new Update().set("lastName", "targaryen"))
.execute();
verify(collection, never()).withWriteConcern(any());
}
@Test // DATAMONGO-2330
public void writeConcernAppliedCorrectlyWhenSet() {
ops.setDefaultWriteConcern(WriteConcern.MAJORITY);
ops.updateOne(new BasicQuery("{}").collation(Collation.of("de")), new Update().set("lastName", "targaryen"))
.execute();
verify(collection).withWriteConcern(eq(WriteConcern.MAJORITY));
}
class SomeDomainType { class SomeDomainType {
@Id String id; @Id String id;