Class AbstractTest

core/src/test/java/io/questdb/test/AbstractTest.java:60–208

import java.util.concurrent.atomic.AtomicInteger;

@SuppressWarnings("ClassEscapesDefinedScope")
@OrderWith(RandomOrder.class)
public class AbstractTest {
    public static final Set<QuietCloseable> CLOSEABLE = Collections.newSetFromMap(new ConcurrentHashMap<>());

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

    @BeforeClass
    public static void setUpStatic() throws Exception {
        TestOs.init();
        Zip.init();
        // it is necessary to initialize logger before tests start
        // logger doesn't relinquish memory until JVM stops,
        // which causes memory leak detector to fail should logger be
        // created mid-test
        LOG.info().$("setup logger").$();
        root = temp.newFolder("dbRoot").getAbsolutePath();
    }

    @AfterClass
    public static void tearDownStatic() {
        TestUtils.removeTestPath(root);
    }

    @Before
    public void setUp() {
        LOG.info().$("Starting test ").$safe(getClass().getSimpleName()).$('#').$safe(testName.getMethodName()).$();
        TestUtils.createTestPath(root);
        Metrics.ENABLED.clear();
        SqlCodeGenerator.ALLOW_FUNCTION_MEMOIZATION = false;
    }

    @After
    public void tearDown() throws Exception {
        LOG.info().$("Finished test ").$safe(getClass().getSimpleName()).$('#').$safe(testName.getMethodName()).$();
        TestUtils.removeTestPath(root);
        OFF_POOL_READER_ID.set(0);
        SqlCodeGenerator.ALLOW_FUNCTION_MEMOIZATION = false;
    }

    /**
     * Builds a {@link QueryAssertion} for tests that have no ambient {@link CairoEngine} (e.g. those
     * that drive a {@link TestServerMain}). The caller MUST supply the engine and execution context
     * via {@link QueryAssertion#withEngine} / {@link QueryAssertion#withContext} before the terminal,
     * typically from {@code serverMain.getEngine()} / {@code serverMain.getSqlExecutionContext()}.
     * {@link AbstractCairoTest} overrides this with a variant bound to its own static engine.
     */
    protected QueryAssertion assertQuery(CharSequence query) {
        return new QueryAssertion(null, null, () -> {
        }, query);
    }