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

Example 71 with MongoMappingContext

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

the class MongoTemplateUnitTests method beforeEach.

@BeforeEach
void beforeEach() {
    when(findIterable.iterator()).thenReturn(cursor);
    when(factory.getMongoDatabase()).thenReturn(db);
    when(factory.getExceptionTranslator()).thenReturn(exceptionTranslator);
    when(factory.getCodecRegistry()).thenReturn(MongoClientSettings.getDefaultCodecRegistry());
    when(db.getCollection(any(String.class), eq(Document.class))).thenReturn(collection);
    when(db.runCommand(any(), any(Class.class))).thenReturn(commandResultDocument);
    when(collection.find(any(org.bson.Document.class), any(Class.class))).thenReturn(findIterable);
    when(collection.mapReduce(any(), any(), eq(Document.class))).thenReturn(mapReduceIterable);
    when(collection.countDocuments(any(Bson.class), any(CountOptions.class))).thenReturn(1L);
    when(collection.estimatedDocumentCount(any())).thenReturn(1L);
    when(collection.getNamespace()).thenReturn(new MongoNamespace("db.mock-collection"));
    when(collection.aggregate(any(List.class), any())).thenReturn(aggregateIterable);
    when(collection.withReadPreference(any())).thenReturn(collection);
    when(collection.replaceOne(any(), any(), any(ReplaceOptions.class))).thenReturn(updateResult);
    when(collection.withWriteConcern(any())).thenReturn(collectionWithWriteConcern);
    when(collection.distinct(anyString(), any(Document.class), any())).thenReturn(distinctIterable);
    when(collectionWithWriteConcern.deleteOne(any(Bson.class), any())).thenReturn(deleteResult);
    when(findIterable.projection(any())).thenReturn(findIterable);
    when(findIterable.sort(any(org.bson.Document.class))).thenReturn(findIterable);
    when(findIterable.collation(any())).thenReturn(findIterable);
    when(findIterable.limit(anyInt())).thenReturn(findIterable);
    when(mapReduceIterable.collation(any())).thenReturn(mapReduceIterable);
    when(mapReduceIterable.sort(any())).thenReturn(mapReduceIterable);
    when(mapReduceIterable.iterator()).thenReturn(cursor);
    when(mapReduceIterable.filter(any())).thenReturn(mapReduceIterable);
    when(mapReduceIterable.collectionName(any())).thenReturn(mapReduceIterable);
    when(mapReduceIterable.databaseName(any())).thenReturn(mapReduceIterable);
    when(mapReduceIterable.action(any())).thenReturn(mapReduceIterable);
    when(aggregateIterable.collation(any())).thenReturn(aggregateIterable);
    when(aggregateIterable.allowDiskUse(any())).thenReturn(aggregateIterable);
    when(aggregateIterable.batchSize(anyInt())).thenReturn(aggregateIterable);
    when(aggregateIterable.map(any())).thenReturn(aggregateIterable);
    when(aggregateIterable.maxTime(anyLong(), any())).thenReturn(aggregateIterable);
    when(aggregateIterable.into(any())).thenReturn(Collections.emptyList());
    when(distinctIterable.collation(any())).thenReturn(distinctIterable);
    when(distinctIterable.map(any())).thenReturn(distinctIterable);
    when(distinctIterable.into(any())).thenReturn(Collections.emptyList());
    this.mappingContext = new MongoMappingContext();
    mappingContext.setAutoIndexCreation(true);
    mappingContext.setSimpleTypeHolder(new MongoCustomConversions(Collections.emptyList()).getSimpleTypeHolder());
    mappingContext.afterPropertiesSet();
    this.converter = spy(new MappingMongoConverter(new DefaultDbRefResolver(factory), mappingContext));
    converter.afterPropertiesSet();
    this.template = new MongoTemplate(factory, converter);
}

Example 72 with MongoMappingContext

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

the class AbstractMongoConfigurationUnitTests method lifecycleCallbacksAreInvokedInAppropriateOrder.

// DATAMONGO-717
@Test
public void lifecycleCallbacksAreInvokedInAppropriateOrder() {
    AbstractApplicationContext context = new AnnotationConfigApplicationContext(SampleMongoConfiguration.class);
    MongoMappingContext mappingContext = context.getBean(MongoMappingContext.class);
    MongoPersistentEntity<?> entity = mappingContext.getRequiredPersistentEntity(Entity.class);
    EvaluationContextProvider provider = (EvaluationContextProvider) ReflectionTestUtils.getField(entity, "evaluationContextProvider");
    assertThat(provider).isInstanceOf(ExtensionAwareEvaluationContextProvider.class);
    context.close();
}

Example 73 with MongoMappingContext

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

the class AbstractReactiveMongoConfigurationUnitTests method returnsUninitializedMappingContext.

// DATAMONGO-1444
@Test
public void returnsUninitializedMappingContext() throws Exception {
    SampleMongoConfiguration configuration = new SampleMongoConfiguration();
    MongoMappingContext context = configuration.mongoMappingContext(configuration.customConversions());
    assertThat(context.getPersistentEntities()).isEmpty();
    context.initialize();
    assertThat(context.getPersistentEntities()).isNotEmpty();
}

Example 74 with MongoMappingContext

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

the class CountQueryUnitTests method setUp.

@BeforeEach
void setUp() {
    this.context = new MongoMappingContext();
    this.converter = new MappingMongoConverter(NoOpDbRefResolver.INSTANCE, context);
    this.converter.afterPropertiesSet();
    this.mapper = new QueryMapper(converter);
}

Example 75 with MongoMappingContext

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

the class DefaultBulkOperationsUnitTests method setUp.

@BeforeEach
void setUp() {
    when(factory.getMongoDatabase()).thenReturn(database);
    when(factory.getExceptionTranslator()).thenReturn(new NullExceptionTranslator());
    when(database.getCollection(anyString(), eq(Document.class))).thenReturn(collection);
    mappingContext = new MongoMappingContext();
    mappingContext.afterPropertiesSet();
    converter = new MappingMongoConverter(dbRefResolver, mappingContext);
    template = new MongoTemplate(factory, converter);
    ops = new DefaultBulkOperations(template, "collection-1", new BulkOperationContext(BulkMode.ORDERED, Optional.of(mappingContext.getPersistentEntity(SomeDomainType.class)), new QueryMapper(converter), new UpdateMapper(converter), null, null));
}