Examples with UpdateWithQuery org.springframework.data.mongodb.core.ExecutableUpdateOperation.UpdateWithQuery used on opensource projects

Example 1 with UpdateWithQuery

use of org.springframework.data.mongodb.core.ExecutableUpdateOperation.UpdateWithQuery 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);
}