se.fortnox.reactivewizard.test.WaitingTestUtils Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of reactivewizard-test Show documentation
There is a newer version: 24.6.0
package se.fortnox.reactivewizard.test;

import org.junit.jupiter.api.Assertions;
import reactor.core.Disposable;
import reactor.core.publisher.Flux;
import reactor.core.scheduler.Schedulers;

import java.util.concurrent.CountDownLatch;
import java.util.concurrent.TimeUnit;
import java.util.function.BooleanSupplier;

import static java.time.Duration.ofMillis;

/**
 * Small class for verifying things in a certain time.
 */
public abstract class WaitingTestUtils {

    private static final String DEFAULT_ERROR_MESSAGE = "Condition not met within time";
    private static final int DEFAULT_WAITING_SECONDS = 5;
    private static final TimeUnit DEFAULT_WAITING_TIME_UNIT = TimeUnit.SECONDS;

    private WaitingTestUtils() {

    }

    /**
     * Assert that supplier returns true within five seconds.
     * @param condition condition to evaluate
     */
    public static void assertConditionIsTrueWithinDefaultTime(BooleanSupplier condition) {
        assertConditionIsTrueWithinTime(DEFAULT_WAITING_SECONDS, DEFAULT_WAITING_TIME_UNIT, condition, DEFAULT_ERROR_MESSAGE);
    }

    /**
     * Assert that supplier returns true within five seconds.
     * @param condition condition to evaluate
     * @param message message to show if assert fails
     */
    public static void assertConditionIsTrueWithinDefaultTime(BooleanSupplier condition, String message) {
        assertConditionIsTrueWithinTime(DEFAULT_WAITING_SECONDS, DEFAULT_WAITING_TIME_UNIT, condition, message);
    }

    /**
     * Assert that supplier returns true.
     * @param wait time to wait
     * @param waitingUnit unit
     * @param condition condition to evaluate
     */
    public static void assertConditionIsTrueWithinTime(int wait, TimeUnit waitingUnit, BooleanSupplier condition) {
        assertConditionIsTrueWithinTime(wait, waitingUnit, condition, DEFAULT_ERROR_MESSAGE);
    }

    /**
     * Assert that supplier returns true.
     * @param wait time to wait
     * @param waitingUnit unit
     * @param condition condition to evaluate
     * @param message message to show if assert fails
     */
    public static void assertConditionIsTrueWithinTime(int wait, TimeUnit waitingUnit, BooleanSupplier condition, String message) {
        CountDownLatch countDownLatch = new CountDownLatch(1);

        final Disposable subscribe = Flux
            .interval(ofMillis(100))
            .publishOn(Schedulers.boundedElastic())
            .takeUntil(aLong -> {
                final boolean result = condition.getAsBoolean();

                if (result) {
                    countDownLatch.countDown();
                }

                return result;
            })
            .subscribe();

        try {
            final boolean await = countDownLatch.await(wait, waitingUnit);
            if (!await) {
                Assertions.fail(message);
            }
        } catch (InterruptedException e) {
            Thread.currentThread().interrupt();
            Assertions.fail();
        } finally {
            subscribe.dispose();
        }
    }
}