Search in sources :

Example 6 with MongoExceptionTranslator

use of org.springframework.data.mongodb.core.MongoExceptionTranslator in project spring-data-mongodb by spring-projects.

the class AbstractMongoQueryUnitTests method setUp.

@BeforeEach
void setUp() {
    doReturn("persons").when(persitentEntityMock).getCollection();
    doReturn(persitentEntityMock).when(mappingContextMock).getPersistentEntity(Mockito.any(Class.class));
    doReturn(persitentEntityMock).when(mappingContextMock).getRequiredPersistentEntity(Mockito.any(Class.class));
    doReturn(Person.class).when(persitentEntityMock).getType();
    MongoDatabaseFactory mongoDbFactory = mock(MongoDatabaseFactory.class);
    when(mongoDbFactory.getExceptionTranslator()).thenReturn(new MongoExceptionTranslator());
    DbRefResolver dbRefResolver = new DefaultDbRefResolver(mongoDbFactory);
    MappingMongoConverter converter = new MappingMongoConverter(dbRefResolver, mappingContextMock);
    converter.afterPropertiesSet();
    doReturn(converter).when(mongoOperationsMock).getConverter();
    doReturn(executableFind).when(mongoOperationsMock).query(any());
    doReturn(withQueryMock).when(executableFind).as(any());
    doReturn(withQueryMock).when(withQueryMock).matching(any(Query.class));
    doReturn(executableUpdate).when(mongoOperationsMock).update(any());
    doReturn(updateWithQuery).when(executableUpdate).matching(any(Query.class));
    doReturn(terminatingUpdate).when(updateWithQuery).apply(any(UpdateDefinition.class));
    when(mongoOperationsMock.remove(any(), any(), anyString())).thenReturn(deleteResultMock);
    when(mongoOperationsMock.updateMulti(any(), any(), any(), anyString())).thenReturn(updateResultMock);
}
Also used : BasicQuery(org.springframework.data.mongodb.core.query.BasicQuery) FindWithQuery(org.springframework.data.mongodb.core.ExecutableFindOperation.FindWithQuery) UpdateWithQuery(org.springframework.data.mongodb.core.ExecutableUpdateOperation.UpdateWithQuery) Query(org.springframework.data.mongodb.core.query.Query) UpdateDefinition(org.springframework.data.mongodb.core.query.UpdateDefinition) MongoDatabaseFactory(org.springframework.data.mongodb.MongoDatabaseFactory) DefaultDbRefResolver(org.springframework.data.mongodb.core.convert.DefaultDbRefResolver) MappingMongoConverter(org.springframework.data.mongodb.core.convert.MappingMongoConverter) DbRefResolver(org.springframework.data.mongodb.core.convert.DbRefResolver) DefaultDbRefResolver(org.springframework.data.mongodb.core.convert.DefaultDbRefResolver) MongoExceptionTranslator(org.springframework.data.mongodb.core.MongoExceptionTranslator) BeforeEach(org.junit.jupiter.api.BeforeEach)

Example 7 with MongoExceptionTranslator

use of org.springframework.data.mongodb.core.MongoExceptionTranslator in project spring-data-mongodb by spring-projects.

the class ReactiveMongoPersistentEntityIndexCreatorUnitTests method setUp.

@BeforeEach
@SuppressWarnings("unchecked")
void setUp() {
    when(factory.getExceptionTranslator()).thenReturn(new MongoExceptionTranslator());
    when(factory.getMongoDatabase()).thenReturn(Mono.just(db));
    when(db.getCollection(any(), any(Class.class))).thenReturn(collection);
    indexOperations = new ReactiveMongoTemplate(factory).indexOps("foo");
    keysCaptor = ArgumentCaptor.forClass(org.bson.Document.class);
    optionsCaptor = ArgumentCaptor.forClass(IndexOptions.class);
    collectionCaptor = ArgumentCaptor.forClass(String.class);
    when(collection.createIndex(keysCaptor.capture(), optionsCaptor.capture())).thenReturn(Mono.just("OK"));
}
Also used : ReactiveMongoTemplate(org.springframework.data.mongodb.core.ReactiveMongoTemplate) IndexOptions(com.mongodb.client.model.IndexOptions) Document(org.springframework.data.mongodb.core.mapping.Document) MongoExceptionTranslator(org.springframework.data.mongodb.core.MongoExceptionTranslator) BeforeEach(org.junit.jupiter.api.BeforeEach)

Example 8 with MongoExceptionTranslator

use of org.springframework.data.mongodb.core.MongoExceptionTranslator in project spring-data-mongodb by spring-projects.

the class CursorReadingTaskUnitTests method setUp.

@BeforeEach
public void setUp() {
    when(request.getRequestOptions()).thenReturn(options);
    when(request.getMessageListener()).thenReturn(listener);
    when(options.getCollectionName()).thenReturn("collection-name");
    when(template.getDb()).thenReturn(db);
    when(template.getExceptionTranslator()).thenReturn(new MongoExceptionTranslator());
    when(db.getName()).thenReturn("mock-db");
    task = new ValueCapturingTaskStub(template, request, Object.class, cursor, errorHandler);
}
Also used : MongoExceptionTranslator(org.springframework.data.mongodb.core.MongoExceptionTranslator) BeforeEach(org.junit.jupiter.api.BeforeEach)

Aggregations

BeforeEach (org.junit.jupiter.api.BeforeEach)8 MongoExceptionTranslator (org.springframework.data.mongodb.core.MongoExceptionTranslator)8 IndexOptions (com.mongodb.client.model.IndexOptions)2 DefaultDbRefResolver (org.springframework.data.mongodb.core.convert.DefaultDbRefResolver)2 MappingMongoConverter (org.springframework.data.mongodb.core.convert.MappingMongoConverter)2 Document (org.springframework.data.mongodb.core.mapping.Document)2 MongoMappingContext (org.springframework.data.mongodb.core.mapping.MongoMappingContext)2 Document (org.bson.Document)1 MongoDatabaseFactory (org.springframework.data.mongodb.MongoDatabaseFactory)1 FindWithQuery (org.springframework.data.mongodb.core.ExecutableFindOperation.FindWithQuery)1 UpdateWithQuery (org.springframework.data.mongodb.core.ExecutableUpdateOperation.UpdateWithQuery)1 MongoTemplate (org.springframework.data.mongodb.core.MongoTemplate)1 ReactiveMongoTemplate (org.springframework.data.mongodb.core.ReactiveMongoTemplate)1 DbRefResolver (org.springframework.data.mongodb.core.convert.DbRefResolver)1 BasicQuery (org.springframework.data.mongodb.core.query.BasicQuery)1 Query (org.springframework.data.mongodb.core.query.Query)1 UpdateDefinition (org.springframework.data.mongodb.core.query.UpdateDefinition)1