sqlancer.postgres.ast.PostgresInOperation Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of sqlancer Show documentation

SQLancer finds logic bugs in Database Management Systems through automatic testing

There is a newer version: 2.0.0

package sqlancer.postgres.ast;

import java.util.List;

import sqlancer.postgres.PostgresSchema.PostgresDataType;

public class PostgresInOperation implements PostgresExpression {

    private final PostgresExpression expr;
    private final List listElements;
    private final boolean isTrue;

    public PostgresInOperation(PostgresExpression expr, List listElements, boolean isTrue) {
        this.expr = expr;
        this.listElements = listElements;
        this.isTrue = isTrue;
    }

    public PostgresExpression getExpr() {
        return expr;
    }

    public List getListElements() {
        return listElements;
    }

    @Override
    public PostgresConstant getExpectedValue() {
        PostgresConstant leftValue = expr.getExpectedValue();
        if (leftValue == null) {
            return null;
        }
        if (leftValue.isNull()) {
            return PostgresConstant.createNullConstant();
        }
        boolean isNull = false;
        for (PostgresExpression expr : getListElements()) {
            PostgresConstant rightExpectedValue = expr.getExpectedValue();
            if (rightExpectedValue == null) {
                return null;
            }
            if (rightExpectedValue.isNull()) {
                isNull = true;
            } else if (rightExpectedValue.isEquals(this.expr.getExpectedValue()).isBoolean()
                    && rightExpectedValue.isEquals(this.expr.getExpectedValue()).asBoolean()) {
                return PostgresConstant.createBooleanConstant(isTrue);
            }
        }

        if (isNull) {
            return PostgresConstant.createNullConstant();
        } else {
            return PostgresConstant.createBooleanConstant(!isTrue);
        }
    }

    public boolean isTrue() {
        return isTrue;
    }

    @Override
    public PostgresDataType getExpressionType() {
        return PostgresDataType.BOOLEAN;
    }
}