io.zeebe.test.util.TestUtil Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of zb-test-util Show documentation
There is a newer version: 0.16.4
/*
 * Copyright © 2017 camunda services GmbH ([email protected])
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package io.zeebe.test.util;

import static org.assertj.core.api.Assertions.assertThat;

import java.util.concurrent.Callable;
import java.util.function.BooleanSupplier;
import java.util.function.Function;

public class TestUtil
{

    public static final int MAX_RETRIES = 100;

    public static  Invocation doRepeatedly(Callable callable)
    {
        return new Invocation<>(callable);
    }

    public static Invocation doRepeatedly(Runnable runnable)
    {
        return new Invocation<>(() ->
        {
            runnable.run();
            return null;
        });
    }

    public static void waitUntil(final BooleanSupplier condition)
    {
        doRepeatedly(() -> null).until((r) -> condition.getAsBoolean());
    }

    public static void waitUntil(final BooleanSupplier condition, final String message, final Object... args)
    {
        doRepeatedly(() -> null).until((r) -> condition.getAsBoolean(), message, args);
    }

    public static void waitUntil(final BooleanSupplier condition, final int retries)
    {
        doRepeatedly(() -> null).until((r) -> condition.getAsBoolean(), retries);
    }

    public static void waitUntil(final BooleanSupplier condition, final int retries, final String message, final Object... args)
    {
        doRepeatedly(() -> null).until((r) -> condition.getAsBoolean(), retries, message, args);
    }

    public static class Invocation
    {
        protected Callable callable;

        public Invocation(Callable callable)
        {
            this.callable = callable;
        }

        public T until(Function resultCondition)
        {
            return until(resultCondition, (e) -> false);
        }

        public T until(Function resultCondition, final String message, final Object... args)
        {
            return until(resultCondition, (e) -> false, message, args);
        }

        public T until(Function resultCondition, final int retries)
        {
            return until(resultCondition, (e) -> false, retries);
        }

        public T until(Function resultCondition, final int retries, final String message, final Object... args)
        {
            return until(resultCondition, (e) -> false, retries, message, args);
        }

        public T until(final Function resultCondition, Function exceptionCondition)
        {
            final T result = whileConditionHolds((t) -> !resultCondition.apply(t), (e) -> !exceptionCondition.apply(e));

            assertThat(resultCondition.apply(result)).isTrue();

            return result;
        }

        public T until(final Function resultCondition, Function exceptionCondition, final String message, final Object... args)
        {
            final T result = whileConditionHolds((t) -> !resultCondition.apply(t), (e) -> !exceptionCondition.apply(e));

            assertThat(resultCondition.apply(result))
                .withFailMessage(message, args)
                .isTrue();

            return result;
        }

        public T until(final Function resultCondition, Function exceptionCondition, final int retries)
        {
            final T result = whileConditionHolds((t) -> !resultCondition.apply(t), (e) -> !exceptionCondition.apply(e), retries);

            assertThat(resultCondition.apply(result)).isTrue();

            return result;
        }

        public T until(final Function resultCondition, Function exceptionCondition, final int retries, final String message, final Object... args)
        {
            final T result = whileConditionHolds((t) -> !resultCondition.apply(t), (e) -> !exceptionCondition.apply(e), retries);

            assertThat(resultCondition.apply(result))
                .withFailMessage(message, args)
                .isTrue();

            return result;
        }

        public T whileConditionHolds(Function resultCondition)
        {
            return whileConditionHolds(resultCondition, (e) -> true);
        }

        public T whileConditionHolds(Function resultCondition, final int retires)
        {
            return whileConditionHolds(resultCondition, (e) -> true, retires);
        }

        public T whileConditionHolds(Function resultCondition, Function exceptionCondition)
        {
            return whileConditionHolds(resultCondition, exceptionCondition, MAX_RETRIES);
        }

        public T whileConditionHolds(Function resultCondition, Function exceptionCondition, final int retries)
        {
            int numTries = 0;

            T result;

            do
            {
                result = null;

                try
                {
                    if (numTries > 0)
                    {
                        Thread.sleep(100L);
                    }

                    result = callable.call();
                }
                catch (Exception e)
                {
                    if (!exceptionCondition.apply(e))
                    {
                        throw new RuntimeException("Unexpected exception while checking condition", e);
                    }
                }

                numTries++;
            }
            while (numTries < retries && resultCondition.apply(result));

            return result;
        }
    }
}