Examples with EnableIfReplicaSetAvailable org.springframework.data.mongodb.test.util.EnableIfReplicaSetAvailable used on opensource projects

Example 6 with EnableIfReplicaSetAvailable

use of org.springframework.data.mongodb.test.util.EnableIfReplicaSetAvailable in project spring-data-mongodb by spring-projects.

the class DefaultMessageListenerContainerTests method shouldStartReceivingMessagesWhenContainerStarts.

// DATAMONGO-1803
@Test
@EnableIfReplicaSetAvailable
public void shouldStartReceivingMessagesWhenContainerStarts() throws InterruptedException {
    MessageListenerContainer container = new DefaultMessageListenerContainer(template);
    Subscription subscription = container.register(new ChangeStreamRequest(messageListener, options()), Document.class);
    collection.insertOne(new Document("_id", "id-1").append("value", "foo"));
    Thread.sleep(200);
    container.start();
    awaitSubscription(subscription);
    Document expected = new Document("_id", "id-2").append("value", "bar");
    collection.insertOne(expected);
    awaitMessages(messageListener);
    container.stop();
    assertThat(messageListener.getMessages().stream().map(Message::getBody).collect(Collectors.toList())).containsExactly(expected);
}

Example 7 with EnableIfReplicaSetAvailable

use of org.springframework.data.mongodb.test.util.EnableIfReplicaSetAvailable in project spring-data-mongodb by spring-projects.

the class DefaultMessageListenerContainerTests method shouldNotifyErrorHandlerOnErrorInListener.

// DATAMONGO-2322
@Test
@EnableIfReplicaSetAvailable
public void shouldNotifyErrorHandlerOnErrorInListener() throws InterruptedException {
    ErrorHandler errorHandler = mock(ErrorHandler.class);
    MessageListenerContainer container = new DefaultMessageListenerContainer(template);
    AtomicBoolean thrownException = new AtomicBoolean();
    Subscription subscription = container.register(new ChangeStreamRequest(message -> {
        try {
            if (thrownException.compareAndSet(false, true)) {
                throw new IllegalStateException("Boom!");
            }
        } finally {
            messageListener.onMessage(message);
        }
    }, options()), Person.class, errorHandler);
    container.start();
    awaitSubscription(subscription, TIMEOUT);
    collection.insertOne(new Document("_id", "id-1").append("firstname", "foo"));
    collection.insertOne(new Document("_id", "id-2").append("firstname", "bar"));
    awaitMessages(messageListener, 2, TIMEOUT);
    verify(errorHandler, atLeast(1)).handleError(any(IllegalStateException.class));
    assertThat(messageListener.getTotalNumberMessagesReceived()).isEqualTo(2);
}

Example 8 with EnableIfReplicaSetAvailable

use of org.springframework.data.mongodb.test.util.EnableIfReplicaSetAvailable in project spring-data-mongodb by spring-projects.

the class ReactiveMongoTemplateTests method resumesAtBsonTimestampCorrectly.

// DATAMONGO-2115
@Test
@EnableIfMongoServerVersion(isGreaterThanEqual = "4.0")
@EnableIfReplicaSetAvailable
void resumesAtBsonTimestampCorrectly() throws InterruptedException {
    template.createCollection(Person.class).as(StepVerifier::create).expectNextCount(1).verifyComplete();
    BlockingQueue<ChangeStreamEvent<Person>> documents = new LinkedBlockingQueue<>(100);
    Disposable disposable = template.changeStream("person", ChangeStreamOptions.empty(), Person.class).doOnNext(documents::add).subscribe();
    // just give it some time to link to the collection.
    Thread.sleep(500);
    Person person1 = new Person("Spring", 38);
    Person person2 = new Person("Data", 37);
    Person person3 = new Person("MongoDB", 39);
    // 
    template.save(person1).delayElement(Duration.ofSeconds(1)).as(// 
    StepVerifier::create).expectNextCount(// 
    1).verifyComplete();
    // 
    template.save(person2).as(// 
    StepVerifier::create).expectNextCount(// 
    1).verifyComplete();
    // skip first
    documents.take();
    // take 2nd
    BsonTimestamp resumeAt = documents.take().getBsonTimestamp();
    disposable.dispose();
    template.save(person3).as(StepVerifier::create).expectNextCount(1).verifyComplete();
    template.changeStream("person", ChangeStreamOptions.builder().resumeAt(resumeAt).build(), Person.class).map(// 
    ChangeStreamEvent::getBody).buffer(// 
    2).as(// 
    StepVerifier::create).consumeNextWith(actual -> {
        assertThat(actual).containsExactly(person2, person3);
    }).thenCancel().verify();
}

Example 9 with EnableIfReplicaSetAvailable

use of org.springframework.data.mongodb.test.util.EnableIfReplicaSetAvailable in project spring-data-mongodb by spring-projects.

the class ReactiveMongoTemplateTests method changeStreamEventsShouldBeResumedCorrectly.

// DATAMONGO-1803
@Test
@Disabled("Heavily relying on timing assumptions. Cannot test message resumption properly. Too much race for too little time in between.")
@EnableIfReplicaSetAvailable
void changeStreamEventsShouldBeResumedCorrectly() throws InterruptedException {
    template.createCollection(Person.class).as(StepVerifier::create).expectNextCount(1).verifyComplete();
    BlockingQueue<ChangeStreamEvent<Person>> documents = new LinkedBlockingQueue<>(100);
    Disposable disposable = template.changeStream("person", ChangeStreamOptions.empty(), Person.class).doOnNext(documents::add).subscribe();
    // just give it some time to link to the collection.
    Thread.sleep(500);
    Person person1 = new Person("Spring", 38);
    Person person2 = new Person("Data", 37);
    Person person3 = new Person("MongoDB", 39);
    // 
    Flux.merge(template.insert(person1), template.insert(person2), template.insert(person3)).as(// 
    StepVerifier::create).expectNextCount(// 
    3).verifyComplete();
    // just give it some time to link receive all events
    Thread.sleep(500);
    disposable.dispose();
    BsonDocument resumeToken = documents.take().getRaw().getResumeToken();
    BlockingQueue<ChangeStreamEvent<Person>> resumeDocuments = new LinkedBlockingQueue<>(100);
    template.changeStream("person", ChangeStreamOptions.builder().resumeToken(resumeToken).build(), Person.class).doOnNext(resumeDocuments::add).subscribe();
    // just give it some time to link receive all events
    Thread.sleep(500);
    try {
        assertThat(resumeDocuments.stream().map(ChangeStreamEvent::getBody).collect(Collectors.toList())).containsExactly(person2, person3);
    } finally {
        disposable.dispose();
    }
}

Example 10 with EnableIfReplicaSetAvailable

use of org.springframework.data.mongodb.test.util.EnableIfReplicaSetAvailable in project spring-data-mongodb by spring-projects.

the class ReactiveMongoTemplateTests method mapsReservedWordsCorrectly.

// DATAMONGO-1803
@Test
@EnableIfReplicaSetAvailable
void mapsReservedWordsCorrectly() throws InterruptedException {
    template.dropCollection(Person.class).onErrorResume(it -> Mono.empty()).as(StepVerifier::create).verifyComplete();
    template.createCollection(Person.class).as(StepVerifier::create).expectNextCount(1).verifyComplete();
    BlockingQueue<ChangeStreamEvent<Person>> documents = new LinkedBlockingQueue<>(100);
    Disposable disposable = template.changeStream("person", ChangeStreamOptions.builder().filter(newAggregation(Person.class, match(where("operationType").is("replace")))).build(), Person.class).doOnNext(documents::add).subscribe();
    // just give it some time to link to the collection.
    Thread.sleep(500);
    Person person1 = new Person("Spring", 38);
    Person person2 = new Person("Data", 37);
    // 
    Flux.merge(template.insert(person1), template.insert(person2)).as(// 
    StepVerifier::create).expectNextCount(// 
    2).verifyComplete();
    Person replacement = new Person(person2.getId(), "BDognoM");
    replacement.setAge(person2.getAge());
    // 
    template.save(replacement).as(// 
    StepVerifier::create).expectNextCount(// 
    1).verifyComplete();
    // just give it some time to link receive all events
    Thread.sleep(500);
    try {
        assertThat(documents.stream().map(ChangeStreamEvent::getBody).collect(Collectors.toList())).containsExactly(replacement);
    } finally {
        disposable.dispose();
    }
}