Examples with QueryMapper org.springframework.data.mongodb.core.convert.QueryMapper used on opensource projects

Example 1 with QueryMapper

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

the class DefaultBulkOperationsUnitTests method bulkInsertInvokesEntityCallbacks.

// DATAMONGO-2261, DATAMONGO-2479
@Test
void bulkInsertInvokesEntityCallbacks() {
    BeforeConvertPersonCallback beforeConvertCallback = spy(new BeforeConvertPersonCallback());
    BeforeSavePersonCallback beforeSaveCallback = spy(new BeforeSavePersonCallback());
    AfterSavePersonCallback afterSaveCallback = spy(new AfterSavePersonCallback());
    ops = new DefaultBulkOperations(template, "collection-1", new BulkOperationContext(BulkMode.ORDERED, Optional.of(mappingContext.getPersistentEntity(Person.class)), new QueryMapper(converter), new UpdateMapper(converter), null, EntityCallbacks.create(beforeConvertCallback, beforeSaveCallback, afterSaveCallback)));
    Person entity = new Person("init");
    ops.insert(entity);
    ArgumentCaptor<Person> personArgumentCaptor = ArgumentCaptor.forClass(Person.class);
    verify(beforeConvertCallback).onBeforeConvert(personArgumentCaptor.capture(), eq("collection-1"));
    verifyNoInteractions(beforeSaveCallback);
    ops.execute();
    verify(beforeSaveCallback).onBeforeSave(personArgumentCaptor.capture(), any(), eq("collection-1"));
    verify(afterSaveCallback).onAfterSave(personArgumentCaptor.capture(), any(), eq("collection-1"));
    assertThat(personArgumentCaptor.getAllValues()).extracting("firstName").containsExactly("init", "before-convert", "before-convert");
    verify(collection).bulkWrite(captor.capture(), any());
    InsertOneModel<Document> updateModel = (InsertOneModel<Document>) captor.getValue().get(0);
    assertThat(updateModel.getDocument()).containsEntry("firstName", "after-save");
}

Example 2 with QueryMapper

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

the class DefaultBulkOperationsUnitTests method bulkReplaceOneEmitsEventsCorrectly.

// DATAMONGO-2290
@Test
void bulkReplaceOneEmitsEventsCorrectly() {
    ApplicationEventPublisher eventPublisher = mock(ApplicationEventPublisher.class);
    ops = new DefaultBulkOperations(template, "collection-1", new BulkOperationContext(BulkMode.ORDERED, Optional.of(mappingContext.getPersistentEntity(Person.class)), new QueryMapper(converter), new UpdateMapper(converter), eventPublisher, null));
    ops.replaceOne(query(where("firstName").is("danerys")), new SomeDomainType());
    verify(eventPublisher).publishEvent(any(BeforeConvertEvent.class));
    verify(eventPublisher, never()).publishEvent(any(BeforeSaveEvent.class));
    verify(eventPublisher, never()).publishEvent(any(AfterSaveEvent.class));
    ops.execute();
    verify(eventPublisher).publishEvent(any(BeforeSaveEvent.class));
    verify(eventPublisher).publishEvent(any(AfterSaveEvent.class));
}

Example 3 with QueryMapper

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

the class DefaultBulkOperationsUnitTests method bulkInsertEmitsEventsCorrectly.

// DATAMONGO-2290
@Test
void bulkInsertEmitsEventsCorrectly() {
    ApplicationEventPublisher eventPublisher = mock(ApplicationEventPublisher.class);
    ops = new DefaultBulkOperations(template, "collection-1", new BulkOperationContext(BulkMode.ORDERED, Optional.of(mappingContext.getPersistentEntity(Person.class)), new QueryMapper(converter), new UpdateMapper(converter), eventPublisher, null));
    ops.insert(new SomeDomainType());
    verify(eventPublisher).publishEvent(any(BeforeConvertEvent.class));
    verify(eventPublisher, never()).publishEvent(any(BeforeSaveEvent.class));
    verify(eventPublisher, never()).publishEvent(any(AfterSaveEvent.class));
    ops.execute();
    verify(eventPublisher).publishEvent(any(BeforeSaveEvent.class));
    verify(eventPublisher).publishEvent(any(AfterSaveEvent.class));
}

Example 4 with QueryMapper

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

the class DefaultReactiveIndexOperationsTests method shouldFavorExplicitMappingHintViaClass.

// DATAMONGO-1682, DATAMONGO-2198
@Test
public void shouldFavorExplicitMappingHintViaClass() {
    IndexDefinition id = new Index().named("partial-with-inheritance").on("k3y", Direction.ASC).partial(of(where("age").gte(10)));
    indexOps = new DefaultReactiveIndexOperations(template, this.template.getCollectionName(DefaultIndexOperationsIntegrationTestsSample.class), new QueryMapper(template.getConverter()), MappingToSameCollection.class);
    indexOps.ensureIndex(id).as(StepVerifier::create).expectNextCount(1).verifyComplete();
    // 
    indexOps.getIndexInfo().filter(this.indexByName("partial-with-inheritance")).as(StepVerifier::create).consumeNextWith(indexInfo -> {
        assertThat(Document.parse(indexInfo.getPartialFilterExpression())).isEqualTo(Document.parse("{ \"a_g_e\" : { \"$gte\" : 10 } }"));
    }).verifyComplete();
}

Example 5 with QueryMapper

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

the class AggregationUnitTests method shouldNotConvertIncludeExcludeValuesForProjectOperation.

// GH-3898
@Test
void shouldNotConvertIncludeExcludeValuesForProjectOperation() {
    MongoMappingContext mappingContext = new MongoMappingContext();
    RelaxedTypeBasedAggregationOperationContext context = new RelaxedTypeBasedAggregationOperationContext(WithRetypedIdField.class, mappingContext, new QueryMapper(new MappingMongoConverter(NoOpDbRefResolver.INSTANCE, mappingContext)));
    Document document = project(WithRetypedIdField.class).toDocument(context);
    assertThat(document).isEqualTo(new Document("$project", new Document("_id", 1).append("renamed-field", 1)));
}