Examples with MongoMappingContext org.springframework.data.mongodb.core.mapping.MongoMappingContext used on opensource projects

Example 61 with MongoMappingContext

use of org.springframework.data.mongodb.core.mapping.MongoMappingContext in project spring-integration by spring-projects.

the class MongoDbStoringMessageHandlerTests method validateMessageHandlingWithMongoTemplate.

@Test
@MongoDbAvailable
public void validateMessageHandlingWithMongoTemplate() throws Exception {
    MongoDbFactory mongoDbFactory = this.prepareMongoFactory();
    MappingMongoConverter converter = new TestMongoConverter(mongoDbFactory, new MongoMappingContext());
    converter.afterPropertiesSet();
    converter = spy(converter);
    MongoTemplate template = new MongoTemplate(mongoDbFactory, converter);
    MongoDbStoringMessageHandler handler = new MongoDbStoringMessageHandler(template);
    handler.setCollectionNameExpression(new LiteralExpression("foo"));
    handler.setBeanFactory(mock(BeanFactory.class));
    handler.afterPropertiesSet();
    Message<Person> message = MessageBuilder.withPayload(this.createPerson("Bob")).build();
    handler.handleMessage(message);
    MongoTemplate readingTemplate = new MongoTemplate(mongoDbFactory);
    Query query = new BasicQuery("{'name' : 'Bob'}");
    Person person = readingTemplate.findOne(query, Person.class, "foo");
    assertEquals("Bob", person.getName());
    assertEquals("PA", person.getAddress().getState());
    verify(converter, times(1)).write(Mockito.any(), Mockito.any(Bson.class));
}

Example 62 with MongoMappingContext

use of org.springframework.data.mongodb.core.mapping.MongoMappingContext in project spring-integration by spring-projects.

the class MongoDbStoringMessageHandlerTests method validateMessageHandlingWithMongoConverter.

@Test
@MongoDbAvailable
public void validateMessageHandlingWithMongoConverter() throws Exception {
    MongoDbFactory mongoDbFactory = this.prepareMongoFactory();
    MongoDbStoringMessageHandler handler = new MongoDbStoringMessageHandler(mongoDbFactory);
    handler.setCollectionNameExpression(new LiteralExpression("foo"));
    MappingMongoConverter converter = new TestMongoConverter(mongoDbFactory, new MongoMappingContext());
    converter.afterPropertiesSet();
    converter = spy(converter);
    handler.setMongoConverter(converter);
    handler.setBeanFactory(mock(BeanFactory.class));
    handler.afterPropertiesSet();
    Message<Person> message = MessageBuilder.withPayload(this.createPerson("Bob")).build();
    handler.handleMessage(message);
    MongoTemplate template = new MongoTemplate(mongoDbFactory);
    Query query = new BasicQuery("{'name' : 'Bob'}");
    Person person = template.findOne(query, Person.class, "foo");
    assertEquals("Bob", person.getName());
    assertEquals("PA", person.getAddress().getState());
    verify(converter, times(1)).write(Mockito.any(), Mockito.any(Bson.class));
}

Example 63 with MongoMappingContext

use of org.springframework.data.mongodb.core.mapping.MongoMappingContext in project spring-boot by spring-projects.

the class MongoDataConfiguration method mongoMappingContext.

@Bean
@ConditionalOnMissingBean
MongoMappingContext mongoMappingContext(ApplicationContext applicationContext, MongoProperties properties, MongoCustomConversions conversions) throws ClassNotFoundException {
    PropertyMapper mapper = PropertyMapper.get().alwaysApplyingWhenNonNull();
    MongoMappingContext context = new MongoMappingContext();
    mapper.from(properties.isAutoIndexCreation()).to(context::setAutoIndexCreation);
    context.setInitialEntitySet(new EntityScanner(applicationContext).scan(Document.class));
    Class<?> strategyClass = properties.getFieldNamingStrategy();
    if (strategyClass != null) {
        context.setFieldNamingStrategy((FieldNamingStrategy) BeanUtils.instantiateClass(strategyClass));
    }
    context.setSimpleTypeHolder(conversions.getSimpleTypeHolder());
    return context;
}

Example 64 with MongoMappingContext

use of org.springframework.data.mongodb.core.mapping.MongoMappingContext in project spring-boot by spring-projects.

the class MongoRepositoriesAutoConfigurationTests method testDefaultRepositoryConfiguration.

@Test
void testDefaultRepositoryConfiguration() {
    this.contextRunner.withUserConfiguration(TestConfiguration.class).run((context) -> {
        assertThat(context).hasSingleBean(CityRepository.class);
        assertThat(context).hasSingleBean(MongoClient.class);
        MongoMappingContext mappingContext = context.getBean(MongoMappingContext.class);
        @SuppressWarnings("unchecked") Set<? extends Class<?>> entities = (Set<? extends Class<?>>) ReflectionTestUtils.getField(mappingContext, "initialEntitySet");
        assertThat(entities).hasSize(1);
    });
}

Example 65 with MongoMappingContext

use of org.springframework.data.mongodb.core.mapping.MongoMappingContext in project spring-boot by spring-projects.

the class MongoDataAutoConfigurationTests method customFieldNamingStrategy.

@Test
void customFieldNamingStrategy() {
    this.contextRunner.withPropertyValues("spring.data.mongodb.field-naming-strategy:" + CamelCaseAbbreviatingFieldNamingStrategy.class.getName()).run((context) -> {
        MongoMappingContext mappingContext = context.getBean(MongoMappingContext.class);
        FieldNamingStrategy fieldNamingStrategy = (FieldNamingStrategy) ReflectionTestUtils.getField(mappingContext, "fieldNamingStrategy");
        assertThat(fieldNamingStrategy.getClass()).isEqualTo(CamelCaseAbbreviatingFieldNamingStrategy.class);
    });
}