src/sqlancer/arangodb/test/ArangoDBQueryPartitioningBase.java - master - NikolayIvkin/sqlancer2

src/sqlancer/arangodb/test/ArangoDBQueryPartitioningBase.java
67 строк3 KB
Patrick
Enable computation values in filter expressions for ArangoDB
23 мар 2021, 01:03
23 мар 2021, 01:036de066e
package sqlancer.arangodb.test;

import java.util.ArrayList;
import java.util.List;

import sqlancer.Randomly;
import sqlancer.arangodb.ArangoDBProvider;
import sqlancer.arangodb.ArangoDBSchema;
import sqlancer.arangodb.ast.ArangoDBExpression;
import sqlancer.arangodb.ast.ArangoDBSelect;
import sqlancer.arangodb.gen.ArangoDBComputedExpressionGenerator;
import sqlancer.arangodb.gen.ArangoDBFilterExpressionGenerator;
import sqlancer.common.ast.newast.Node;
import sqlancer.common.gen.ExpressionGenerator;
import sqlancer.common.oracle.TernaryLogicPartitioningOracleBase;
import sqlancer.common.oracle.TestOracle;

public class ArangoDBQueryPartitioningBase
        extends TernaryLogicPartitioningOracleBase<Node<ArangoDBExpression>, ArangoDBProvider.ArangoDBGlobalState>
        implements TestOracle {

    protected ArangoDBSchema schema;
    protected List<ArangoDBSchema.ArangoDBColumn> targetColumns;
    protected ArangoDBFilterExpressionGenerator expressionGenerator;
    protected ArangoDBSelect<ArangoDBExpression> select;
    protected int numberComputedColumns;

    protected ArangoDBQueryPartitioningBase(ArangoDBProvider.ArangoDBGlobalState state) {
        super(state);
    }

    @Override
    protected ExpressionGenerator<Node<ArangoDBExpression>> getGen() {
        return expressionGenerator;
    }

    @Override
    public void check() throws Exception {
        numberComputedColumns = state.getRandomly().getInteger(0, 4);
        schema = state.getSchema();
        generateTargetColumns();
        expressionGenerator = new ArangoDBFilterExpressionGenerator(state).setColumns(targetColumns);
        expressionGenerator.setNumberOfComputedVariables(numberComputedColumns);
        initializeTernaryPredicateVariants();
        select = new ArangoDBSelect<>();
        select.setFromColumns(targetColumns);
        select.setProjectionColumns(Randomly.nonEmptySubset(targetColumns));
        generateComputedClause();
    }

    private void generateComputedClause() {
        List<Node<ArangoDBExpression>> computedColumns = new ArrayList<>();
        ArangoDBComputedExpressionGenerator generator = new ArangoDBComputedExpressionGenerator(state);
        generator.setColumns(targetColumns);
        for (int i = 0; i < numberComputedColumns; i++) {
            computedColumns.add(generator.generateExpression());
        }
        select.setComputedClause(computedColumns);
    }

    private void generateTargetColumns() {
        ArangoDBSchema.ArangoDBTables targetTables;
        targetTables = schema.getRandomTableNonEmptyTables();
        List<ArangoDBSchema.ArangoDBColumn> allColumns = targetTables.getColumns();
        targetColumns = Randomly.nonEmptySubset(allColumns);
    }
}