Class AbstractTest

compat/src/test/java/io/questdb/compat/AbstractTest.java:51–143

import java.util.concurrent.TimeUnit;

public class AbstractTest {
    @ClassRule
    public static final TemporaryFolder temp = new TemporaryFolder();
    protected static final Log LOG = LogFactory.getLog(AbstractTest.class);
    protected static String root;
    @Rule
    public final TestName testName = new TestName();
    protected final StringSink sink = new StringSink();

    public static void assertEventually(Runnable assertion) {
        assertEventually(assertion, 30);
    }

    public static void assertEventually(Runnable assertion, int timeoutSeconds) {
        long maxSleepingTimeMillis = 1000;
        long nextSleepingTimeMillis = 10;
        long startTime = System.nanoTime();
        long deadline = startTime + TimeUnit.SECONDS.toNanos(timeoutSeconds);
        for (; ; ) {
            try {
                assertion.run();
                return;
            } catch (AssertionError error) {
                if (System.nanoTime() >= deadline) {
                    throw error;
                }
            }
            Os.sleep(nextSleepingTimeMillis);
            nextSleepingTimeMillis = Math.min(maxSleepingTimeMillis, nextSleepingTimeMillis << 1);
        }
    }

    public static void createTestPath() {
        final Path path = Path.getThreadLocal(root);
        if (Files.exists(path.$())) {
            return;
        }
        Files.mkdirs(path.of(root).slash(), 509);
    }

    public static Rnd generateRandom(Log log) {
        return generateRandom(log, System.nanoTime(), System.currentTimeMillis());
    }

    public static Rnd generateRandom(Log log, long s0, long s1) {
        if (log != null) {
            log.info().$("random seeds: ").$(s0).$("L, ").$(s1).$('L').$();
        }
        System.out.printf("random seeds: %dL, %dL%n", s0, s1);
        return new Rnd(s0, s1);
    }

    public static void removeTestPath() {
        final Path path = Path.getThreadLocal(root);
        FilesFacade ff = FilesFacadeImpl.INSTANCE;
        path.slash$();
        Assert.assertTrue("Test dir cleanup error", !ff.exists(path.$()) || Files.rmdir(path.slash(), true));
    }