planetiler/core/src/test/java/com/onthegomap/flatmap/monitoring/ProgressLoggersTest.java

package com.onthegomap.flatmap.monitoring;

import static org.junit.jupiter.api.Assertions.assertEquals;

import com.onthegomap.flatmap.worker.Topology;
import java.time.Duration;
import java.util.concurrent.CountDownLatch;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.Timeout;

public class ProgressLoggersTest {

  @Test
  @Timeout(10)
  public void testLogTopology() {
    var latch = new CountDownLatch(1);
    var topology = Topology.start("topo", new Stats.InMemory())
      .fromGenerator("reader", next -> latch.await())
      .addBuffer("reader_queue", 10)
      .addWorker("worker", 2, (a, b) -> latch.await())
      .addBuffer("writer_queue", 10)
      .sinkTo("writer", 2, a -> latch.await());

    var loggers = new ProgressLoggers("prefix")
      .addTopologyStats(topology);

    String log;
    while ((log = loggers.getLog()).split("%").length < 6) {
      // spin waiting for threads to start
    }

    assertEquals("[prefix] | reader( 0%) ->    (0/10) -> worker( 0%  0%) ->    (0/10) -> writer( 0%  0%)",
      log.replaceAll("[ 0-9][0-9]%", " 0%"));
    latch.countDown();
    topology.awaitAndLog(loggers, Duration.ofSeconds(10));
    assertEquals("[prefix] | reader( -%) ->    (0/10) -> worker( -%  -%) ->    (0/10) -> writer( -%  -%)",
      loggers.getLog());
  }
}