Class QueryAssertion

core/src/test/java/io/questdb/test/QueryAssertion.java:64–2009

Standalone fluent query-assertion builder. It is parameterized at construction with the CairoEngine, default SqlExecutionContext and a pre-assertion hook, and carries its own copies of the cursor-assertion and factory-memory-leak primitives so it does not depend on the test base clas

 * construction-time collaborators for a single assertion.
 */
public class QueryAssertion {
    private static final double EPSILON = 0.000001;
    private static final boolean[] FACTORY_TAGS = new boolean[MemoryTag.SIZE];
    private static final Log LOG = LogFactory.getLog(QueryAssertion.class);
    private final long[] SNAPSHOT = new long[MemoryTag.SIZE];
    private final Runnable prepareHook;
    private final CharSequence query;
    private final LongList rows = new LongList();
    private final StringSink sink = new StringSink();
    private Class<?> baseFactoryClass;
    private SqlCompiler compiler;
    private SqlExecutionContext context;
    private CharSequence ddl;
    private CharSequence ddl2;
    private ObjList<CharSequence> ddl2More;
    private ObjList<CharSequence> ddlMore;
    private CairoEngine engine;
    private boolean expectSize;
    private IntList expectedColumnTypes;
    private CharSequence expectedPlan;
    private CharSequence expectedTimestamp;
    private TimestampOrder expectedTimestampOrder = TimestampOrder.ASC;
    private RecordCursorFactory factory;
    private boolean fullFatJoins;
    private boolean isRandomAccessInferred;
    private boolean isTimestampInferred;
    private boolean leakCheck = true;
    private long memoryUsage = -1;
    private boolean memoryUsageCheck = true;
    private boolean overridden;
    private ObjList<CharSequence> planFragments;
    private ObjList<CharSequence> planFragmentsAbsent;
    private boolean sizeCanBeVariable;
    private boolean supportsRandomAccess = true;

    public QueryAssertion(CairoEngine engine, SqlExecutionContext context, Runnable prepareHook, CharSequence query) {
        this.engine = engine;
        this.context = context;
        this.prepareHook = prepareHook;
        this.query = query;
    }

    /**
     * External-factory mode. The builder asserts {@code factory}, which the caller built and owns, instead
     * of compiling SQL. It does NOT run the prepare hook, DDL, leak check, factory free, or the shared
     * memory-snapshot accounting, and it asserts each cursor through caller-local scratch buffers, so it is
     * safe to call concurrently from multiple worker threads. Pass the per-thread context via
     * {@link #withContext(SqlExecutionContext)}.
     */
    public QueryAssertion(CairoEngine engine, RecordCursorFactory factory) {
        this.engine = engine;
        this.context = null;
        this.prepareHook = null;
        this.query = null;
        this.factory = factory;
        this.leakCheck = false;
    }