Examples with EsPublisher com.netflix.titus.supplementary.taskspublisher.es.EsPublisher used on opensource projects

Example 1 with EsPublisher

use of com.netflix.titus.supplementary.taskspublisher.es.EsPublisher in project titus-control-plane by Netflix.

the class TaskEventsGeneratorTest method checkPublisherState.

@Test
public void checkPublisherState() {
    int numTasks = 5;
    final TaskEventsGenerator taskEventsGenerator = new TaskEventsGenerator(mockTitusClient(numTasks), Collections.emptyMap(), titusRuntime);
    try {
        EsPublisher esPublisher = new EsPublisher(taskEventsGenerator, mockElasticSearchClient(), mockEsPublisherConfiguration(), new DefaultRegistry());
        esPublisher.activate();
        final CountDownLatch latch = new CountDownLatch(1);
        Flux.interval(Duration.ofSeconds(1), Schedulers.elastic()).take(1).doOnNext(i -> {
            final int numTasksUpdated = esPublisher.getNumTasksPublished();
            final int numErrors = esPublisher.getNumErrorsInPublishing();
            assertThat(numErrors).isEqualTo(0);
            assertThat(numTasksUpdated).isGreaterThanOrEqualTo(numTasks);
            latch.countDown();
        }).subscribe();
        try {
            latch.await(2, TimeUnit.MINUTES);
        } catch (InterruptedException e) {
            fail("Timeout in checkPublisherState ", e);
        }
    } finally {
        taskEventsGenerator.shutdown();
    }
}