How to use SqlCondition class of org.evomaster.dbconstraint.ast package

Best EvoMaster code snippet using org.evomaster.dbconstraint.ast.SqlCondition

Source:SqlConditionTranslator.java

...7import org.evomaster.dbconstraint.ast.SqlBinaryDataLiteralValue;8import org.evomaster.dbconstraint.ast.SqlBooleanLiteralValue;9import org.evomaster.dbconstraint.ast.SqlColumn;10import org.evomaster.dbconstraint.ast.SqlComparisonCondition;11import org.evomaster.dbconstraint.ast.SqlCondition;12import org.evomaster.dbconstraint.ast.SqlConditionList;13import org.evomaster.dbconstraint.ast.SqlConditionVisitor;14import org.evomaster.dbconstraint.ast.SqlInCondition;15import org.evomaster.dbconstraint.ast.SqlIsNotNullCondition;16import org.evomaster.dbconstraint.ast.SqlIsNullCondition;17import org.evomaster.dbconstraint.ast.SqlLikeCondition;18import org.evomaster.dbconstraint.ast.SqlLiteralValue;19import org.evomaster.dbconstraint.ast.SqlNullLiteralValue;20import org.evomaster.dbconstraint.ast.SqlOrCondition;21import org.evomaster.dbconstraint.ast.SqlSimilarToCondition;22import org.evomaster.dbconstraint.ast.SqlStringLiteralValue;23import java.util.ArrayList;24import java.util.Collections;25import java.util.List;26import java.util.Set;27import java.util.stream.Collectors;28import static org.evomaster.dbconstraint.ast.SqlComparisonOperator.EQUALS_TO;29public class SqlConditionTranslator implements SqlConditionVisitor<TableConstraint, Void> {30    private static final String THIS_METHOD_SHOULD_NOT_BE_INVOKED = "This method should not be directly called";31    public static final String UNEXPECTED_COMPARISON_OPERATOR_MESSAGE = "Unexpected comparison operator ";32    private final TranslationContext translationContext;33    public SqlConditionTranslator(TranslationContext translationContext) {34        this.translationContext = translationContext;35    }36    /**37     * FIXME38     * temporary workaround before major refactoring.39     * Recall that Column.getTable() is not reliable40     */41    private String getTableName(SqlColumn column) {42        String tableName = column.getTableName();43        if (tableName != null) {44            return tableName;45        } else {46            return this.translationContext.getCurrentTableName();47        }48    }49    @Override50    public TableConstraint visit(SqlAndCondition andExpression, Void argument) {51        TableConstraint left = andExpression.getLeftExpr().accept(this, null);52        TableConstraint right = andExpression.getRightExpr().accept(this, null);53        return new AndConstraint(translationContext.getCurrentTableName(), left, right);54    }55    @Override56    public TableConstraint visit(SqlComparisonCondition e, Void argument) {57        SqlCondition left = e.getLeftOperand();58        SqlCondition right = e.getRightOperand();59        if (left instanceof SqlLiteralValue && right instanceof SqlColumn) {60            SqlLiteralValue leftLiteral = (SqlLiteralValue) left;61            SqlColumn rightColumn = (SqlColumn) right;62            return visit(leftLiteral, e, rightColumn);63        } else if (left instanceof SqlColumn && right instanceof SqlLiteralValue) {64            SqlColumn leftColumn = (SqlColumn) left;65            SqlLiteralValue rightLiteral = (SqlLiteralValue) right;66            return visit(leftColumn, e, rightLiteral);67        } else if (left instanceof SqlCondition && right instanceof SqlCondition) {68            TableConstraint leftTableConstraint = e.getLeftOperand().accept(this, null);69            TableConstraint rightTableConstraint = e.getRightOperand().accept(this, null);70            if (e.getSqlComparisonOperator().equals(EQUALS_TO)) {71                return new IffConstraint(translationContext.getCurrentTableName(), leftTableConstraint, rightTableConstraint);72            }73        }74        // TODO This translation should be implemented75        throw new SqlCannotBeTranslatedException(e.toSql() + " cannot be translated yet");76    }77    private TableConstraint visit(SqlColumn leftColumn, SqlComparisonCondition e, SqlLiteralValue rightLiteral) {78        final String tableName = getTableName(leftColumn);79        final String columnName = leftColumn.getColumnName();80        if (rightLiteral instanceof SqlBigIntegerLiteralValue) {81            long value = ((SqlBigIntegerLiteralValue) rightLiteral).getBigInteger().longValue();82            switch (e.getSqlComparisonOperator()) {83                case EQUALS_TO:84                    return new RangeConstraint(tableName, columnName, value, value);85                case GREATER_THAN:86                    return new LowerBoundConstraint(tableName, columnName, value + 1);87                case GREATER_THAN_OR_EQUAL:88                    return new LowerBoundConstraint(tableName, columnName, value);89                case LESS_THAN:90                    return new UpperBoundConstraint(tableName, columnName, value - 1);91                case LESS_THAN_OR_EQUAL:92                    return new UpperBoundConstraint(tableName, columnName, value);93                default:94                    throw new UnsupportedOperationException(UNEXPECTED_COMPARISON_OPERATOR_MESSAGE + e.getSqlComparisonOperator());95            }96        } else if (rightLiteral instanceof SqlStringLiteralValue) {97            SqlStringLiteralValue stringLiteralValue = (SqlStringLiteralValue) rightLiteral;98            if (e.getSqlComparisonOperator().equals(EQUALS_TO)) {99                return new EnumConstraint(tableName, columnName, Collections.singletonList(stringLiteralValue.getStringValue()));100            } else {101                throw new UnsupportedOperationException(UNEXPECTED_COMPARISON_OPERATOR_MESSAGE + e.getSqlComparisonOperator());102            }103        } else {104            throw new UnsupportedOperationException("Unsupported literal " + rightLiteral);105        }106    }107    private TableConstraint visit(SqlLiteralValue leftLiteral, SqlComparisonCondition e, SqlColumn rightColumn) {108        if (leftLiteral instanceof SqlBigIntegerLiteralValue) {109            long value = ((SqlBigIntegerLiteralValue) leftLiteral).getBigInteger().longValue();110            final String tableName = getTableName(rightColumn);111            final String columnName = rightColumn.getColumnName();112            switch (e.getSqlComparisonOperator()) {113                case EQUALS_TO:114                    return new RangeConstraint(tableName, columnName, value, value);115                case GREATER_THAN:116                    return new UpperBoundConstraint(tableName, columnName, value - 1);117                case GREATER_THAN_OR_EQUAL:118                    return new UpperBoundConstraint(tableName, columnName, value);119                case LESS_THAN:120                    return new LowerBoundConstraint(tableName, columnName, value + 1);121                case LESS_THAN_OR_EQUAL:122                    return new LowerBoundConstraint(tableName, columnName, value);123                default:124                    throw new UnsupportedOperationException(UNEXPECTED_COMPARISON_OPERATOR_MESSAGE + e.getSqlComparisonOperator());125            }126        } else {127            throw new UnsupportedOperationException("Unsupported literal " + e.getSqlComparisonOperator());128        }129    }130    @Override131    public TableConstraint visit(SqlInCondition inExpression, Void argument) {132        SqlColumn column = inExpression.getSqlColumn();133        String tableName = getTableName(column);134        String columnName = column.getColumnName();135        SqlConditionList rightItemsList = inExpression.getLiteralList();136        List<String> stringValues = new ArrayList<>();137        for (SqlCondition expressionValue : rightItemsList.getSqlConditionExpressions()) {138            final String stringValue;139            if (expressionValue instanceof SqlStringLiteralValue) {140                stringValue = new StringValue(expressionValue.toSql()).getNotExcapedValue();141            } else {142                stringValue = expressionValue.toSql();143            }144            stringValues.add(stringValue);145        }146        return new EnumConstraint(tableName, columnName, stringValues);147    }148    @Override149    public TableConstraint visit(SqlNullLiteralValue e, Void argument) {150        throw new UnsupportedOperationException(THIS_METHOD_SHOULD_NOT_BE_INVOKED);151    }152    @Override153    public TableConstraint visit(SqlStringLiteralValue e, Void argument) {154        throw new UnsupportedOperationException(THIS_METHOD_SHOULD_NOT_BE_INVOKED);155    }156    @Override157    public TableConstraint visit(SqlConditionList e, Void argument) {158        throw new UnsupportedOperationException(THIS_METHOD_SHOULD_NOT_BE_INVOKED);159    }160    @Override161    public TableConstraint visit(SqlIsNotNullCondition e, Void argument) {162        String tableName = getTableName(e.getColumn());163        String columnName = e.getColumn().getColumnName();164        return new IsNotNullConstraint(tableName, columnName);165    }166    @Override167    public TableConstraint visit(SqlBinaryDataLiteralValue e, Void argument) {168        throw new UnsupportedOperationException(THIS_METHOD_SHOULD_NOT_BE_INVOKED);169    }170    @Override171    public TableConstraint visit(SqlSimilarToCondition e, Void argument) {...

Source:JSqlConditionParser.java

...4import net.sf.jsqlparser.parser.CCJSqlParserUtil;5import org.evomaster.dbconstraint.ConstraintDatabaseType;6import org.evomaster.dbconstraint.ast.SqlComparisonCondition;7import org.evomaster.dbconstraint.ast.SqlComparisonOperator;8import org.evomaster.dbconstraint.ast.SqlCondition;9import org.evomaster.dbconstraint.parser.SqlConditionParser;10import org.evomaster.dbconstraint.parser.SqlConditionParserException;11import java.util.regex.Matcher;12import java.util.regex.Pattern;13public class JSqlConditionParser implements SqlConditionParser {14    /**15     * JSQL does not support legal check constraints such as (x=35) = (y=32).16     * In order to support those constraints, it is possible to split the constraint into17     * two separate formulas (i.e. "x=35" and "y=32") and feed the basic formulas18     * to the JSQL parser. The pattern below allows one to split the "($1)=($2)" string19     * into those two formulas by using the Matcher.group(int) method20     */21    public static final String FORMULA_EQUALS_FORMULA_PATTERN = "\\(\\s*\\(([^<]*)\\)\\s*=\\s*\\(([^<]*)\\)\\s*\\)";22    @Override23    public SqlCondition parse(String sqlConditionStr, ConstraintDatabaseType databaseType) throws SqlConditionParserException {24        try {25            Pattern pattern = Pattern.compile(FORMULA_EQUALS_FORMULA_PATTERN);26            Matcher matcher = pattern.matcher(sqlConditionStr);27            if (matcher.find()) {28                String left = String.format("(%s)", matcher.group(1));29                SqlCondition leftCondition = parse(left, databaseType);30                String right = String.format("(%s)", matcher.group(2));31                SqlCondition rightCondition = parse(right, databaseType);32                return new SqlComparisonCondition(leftCondition, SqlComparisonOperator.EQUALS_TO, rightCondition);33            }34            String transformedSql = transformDialect(sqlConditionStr, databaseType);35            Expression expression = CCJSqlParserUtil.parseCondExpression(transformedSql, false);36            JSqlVisitor translateToSqlCondition = new JSqlVisitor();37            expression.accept(translateToSqlCondition);38            return translateToSqlCondition.getSqlCondition();39        } catch (JSQLParserException e) {40            throw new SqlConditionParserException(e);41        }42    }43    /**44     * replaces unsupported grammar of JSQLParser with equivalent supported constructs45     *46     * @param originalSqlStr47     * @return48     */49    private String transformDialect(String originalSqlStr, ConstraintDatabaseType databaseType) {50        /**51         * The JSQL parser does not properly parse the Postgresql SQL dialect function "ANY"52         * We can work aroung this limitation by replacing the "= ANY (...)" with a valid " IN (...)"53         * string54         */...

SqlCondition

Using AI Code Generation

1import org.evomaster.dbconstraint.ast.SqlCondition;2import org.evomaster.dbconstraint.ast.SqlConditionFactory;3import java.util.ArrayList;4import java.util.List;5public class SqlConditionExample {6    public static void main(String[] args) {7        List<SqlCondition> conditions = new ArrayList<>();8        SqlCondition condition1 = SqlConditionFactory.create("column1", "value1");9        SqlCondition condition2 = SqlConditionFactory.create("column2", "value2", "=");10        List<String> values = new ArrayList<>();11        values.add("value3");12        values.add("value4");13        SqlCondition condition3 = SqlConditionFactory.create("column3", values);14        SqlCondition condition4 = SqlConditionFactory.create("column4", values, "IN");15        SqlCondition condition5 = SqlConditionFactory.create("column5", values, "NOT IN");16        SqlCondition condition6 = SqlConditionFactory.create("column6", values, "BETWEEN");17        conditions.add(condition1);18        conditions.add(condition2);19        conditions.add(condition3);20        conditions.add(condition4);21        conditions.add(condition5);22        conditions.add(condition6);23        SqlCondition sqlCondition = SqlConditionFactory.and(conditions);24        System.out.println(sqlCondition.toSql());25    }26}

SqlCondition

Using AI Code Generation

1import org.evomaster.dbconstraint.ast.SqlCondition;2import org.evomaster.dbconstraint.ast.SqlConditionParser;3import org.evomaster.dbconstraint.ast.SqlRestriction;4import org.evomaster.dbconstraint.ast.SqlRestrictionParser;5class Test {6    public static void main(String[] args) {7        SqlConditionParser parser = new SqlConditionParser();8        SqlCondition condition = parser.parse("col1 = 1 AND col2 = 2");9        System.out.println(condition);10        SqlRestrictionParser restrictionParser = new SqlRestrictionParser();11        SqlRestriction restriction = restrictionParser.parse("col1 = 1 AND col2 = 2");12        System.out.println(restriction);13    }14}15import org.evomaster.dbconstraint.ast.SqlCondition;16import org.evomaster.dbconstraint.ast.SqlConditionParser;17import org.evomaster.dbconstraint.ast.SqlRestriction;18import org.evomaster.dbconstraint.ast.SqlRestrictionParser;19class Test {20    public static void main(String[] args) {21        SqlConditionParser parser = new SqlConditionParser();22        SqlCondition condition = parser.parse("col1 = 1 AND col2 = 2");23        System.out.println(condition);24        SqlRestrictionParser restrictionParser = new SqlRestrictionParser();25        SqlRestriction restriction = restrictionParser.parse("col1 = 1 AND col2 = 2");26        System.out.println(restriction);27    }28}29import org.evomaster.dbconstraint.ast.SqlCondition;30import org.evomaster.dbconstraint.ast.SqlConditionParser;31import org.evomaster.dbconstraint.ast.SqlRestriction;32import org.evomaster.dbconstraint.ast.SqlRestrictionParser;33class Test {34    public static void main(String[] args) {35        SqlConditionParser parser = new SqlConditionParser();36        SqlCondition condition = parser.parse("col1 = 1 AND col2 = 2");37        System.out.println(condition);38        SqlRestrictionParser restrictionParser = new SqlRestrictionParser();39        SqlRestriction restriction = restrictionParser.parse("col1 = 1 AND col2 = 2");40        System.out.println(restriction);41    }42}

SqlCondition

Using AI Code Generation

1import org.evomaster.dbconstraint.ast.SqlCondition;2import org.evomaster.dbconstraint.ast.SqlConditionParser;3import org.evomaster.dbconstraint.ast.SqlConditionParserException;4import org.evomaster.dbconstraint.ast.SqlConditionLexer;5import org.evomaster.dbconstraint.ast.SqlConditionLexerException;6import org.evomaster.dbconstraint.ast.SqlConditionVisitor;7import org.evomaster.dbconstraint.ast.SqlConditionVisitorException;8import org.evomaster.dbconstraint.ast.SqlConditionVisitorImpl;9import org.evomaster.dbconstraint.ast.SqlConditionVisitorImplException;10import org.evomaster.dbconstraint.ast.SqlConditionVisitorImpl;11import org.evomaster.dbconstraint.ast.SqlConditionVisitorImplException;12import org.evomaster.dbconstraint.ast.SqlConditionVisitorImpl;13import org.evomaster.dbconstraint.ast.SqlConditionVisitorImplException;14public class Main {15    public static void main(String[] args) throws SqlConditionParserException, SqlConditionLexerException, SqlConditionVisitorException, SqlConditionVisitorImplException {16        SqlConditionParser parser = new SqlConditionParser();17        SqlConditionLexer lexer = new SqlConditionLexer();18        SqlConditionVisitor visitor = new SqlConditionVisitor();19        SqlConditionVisitorImpl visitorImpl = new SqlConditionVisitorImpl();20        SqlCondition condition = parser.parse("id = 1 AND name = 'foo'");21        SqlCondition condition2 = lexer.lex("id = 1 AND name = 'foo'");22        SqlCondition condition3 = visitor.visit(condition2);23        SqlCondition condition4 = visitorImpl.visit(condition3);24        System.out.println(condition4);25    }26}27import org.evomaster.dbconstraint.*;28import org.evomaster.dbconstraint.parser.*;29import org.evomaster.dbconstraint.lexer.*;30import org.evomaster.dbconstraint.visitor.*;31import org.evomaster.dbconstraint.visitor.impl.*;32public class Main {33    public static void main(String[] args) throws SqlConditionParserException, SqlConditionLexerException, SqlConditionVisitorException, SqlConditionVisitorImplException {34        SqlConditionParser parser = new SqlConditionParser();35        SqlConditionLexer lexer = new SqlConditionLexer();36        SqlConditionVisitor visitor = new SqlConditionVisitor();37        SqlConditionVisitorImpl visitorImpl = new SqlConditionVisitorImpl();38        SqlCondition condition = parser.parse("id = 1 AND name = 'foo'");

SqlCondition

Using AI Code Generation

1import org.evomaster.dbconstraint.ast.SqlCondition;2import org.evomaster.dbconstraint.ast.SqlConditionFactory;3import org.evomaster.dbconstraint.ast.SqlConditionType;4import org.evomaster.dbconstraint.ast.SqlTable;5import org.evomaster.dbconstraint.ast.SqlColumn;6import org.evomaster.dbconstraint.ast.SqlColumnValue;7import org.evomaster.dbconstraint.ast.SqlColumnValueFactory;8import org.evomaster.dbconstraint.ast.SqlColumnValueNumeric;9import org.evomaster.dbconstraint.ast.SqlColumnValueString;10import org.evomaster.dbconstraint.ast.SqlColumnValueBoolean;11import org.evomaster.dbconstraint.ast.SqlColumnValueDate;12import org.evomaster.dbconstraint.ast.SqlColumnValueTime;13import org.evomaster.dbconstraint.ast.SqlColumnValueDateTime;14import org.evomaster.dbconstraint.ast.SqlColumnValueNull;15import org.evomaster.dbconstraint.ast.SqlColumnValueArray;16import org.evomaster.dbconstraint.ast.SqlColumnValueObject;17import org.evomaster.dbconstraint.ast.SqlColumnValueArrayElement;18import org.evomaster.dbconstraint.ast.SqlColumnValueObjectElement;19import org.evomaster.dbconstraint.ast.SqlColumnValueBinary;20import java.util.Arrays;21import java.util.List;22import java.util.Map;23import java.util.HashMap;24import java.util.ArrayList;25import java.util.Date;26import java.util.Calendar;27import java.text.SimpleDateFormat;28public class 2 {29    public static void main(String[] args) {30        String tableName1 = "table1";31        String tableName2 = "table2";32        String tableName3 = "table3";33        String columnName1 = "column1";34        String columnName2 = "column2";35        String columnName3 = "column3";36        String columnName4 = "column4";37        String columnName5 = "column5";38        String columnName6 = "column6";39        String columnName7 = "column7";40        String columnName8 = "column8";41        String columnName9 = "column9";42        String columnName10 = "column10";43        String columnName11 = "column11";44        String columnName12 = "column12";45        String columnName13 = "column13";46        String columnName14 = "column14";47        String columnName15 = "column15";48        String columnName16 = "column16";49        String columnName17 = "column17";

SqlCondition

Using AI Code Generation

1package org.evomaster.dbconstraint.ast;2import org.evomaster.dbconstraint.ast.SqlCondition;3import org.evomaster.dbconstraint.ast.SqlConditionVisitor;4import org.evomaster.dbconstraint.ast.SqlConditionVisitorImpl;5import org.evomaster.dbconstraint.ast.SqlOperator;6import org.evomaster.dbconstraint.ast.SqlPredicate;7import java.util.ArrayList;8import java.util.List;9public class SqlConditionGenerator {10    public static SqlCondition createCondition() {11        SqlConditionVisitor visitor = new SqlConditionVisitorImpl();12        SqlCondition condition = new SqlCondition();13        List<SqlPredicate> predicates = new ArrayList<>();14        SqlPredicate predicate = new SqlPredicate();15        predicate.setTable("table1");16        predicate.setColumn("column1");17        predicate.setOperator(SqlOperator.EQUALS);18        predicate.setValue("value1");19        predicates.add(predicate);20        SqlPredicate predicate2 = new SqlPredicate();21        predicate2.setTable("table2");22        predicate2.setColumn("column2");23        predicate2.setOperator(SqlOperator.EQUALS);24        predicate2.setValue("value2");25        predicates.add(predicate2);26        SqlPredicate predicate3 = new SqlPredicate();27        predicate3.setTable("table3");28        predicate3.setColumn("column3");29        predicate3.setOperator(SqlOperator.EQUALS);30        predicate3.setValue("value3");31        predicates.add(predicate3);32        condition.setPredicates(predicates);33        condition.setOperator(SqlOperator.AND);34        return condition;35    }36    public static void main(String[] args) {37        SqlCondition condition = createCondition();38        String sql = condition.accept(new SqlConditionVisitorImpl());39        System.out.println(sql);40    }41}42package org.evomaster.dbconstraint.ast;43import org.evomaster.dbconstraint.ast.SqlCondition;44import org.evomaster.dbconstraint.ast.SqlConditionVisitor;45import org.evomaster.dbconstraint.ast.SqlConditionVisitorImpl;46import org.evomaster.dbconstraint.ast.SqlOperator;47import org.evomaster.dbconstraint.ast.SqlPredicate;48import java.util.ArrayList;49import java.util.List;

SqlCondition

Using AI Code Generation

1import org.evomaster.dbconstraint.ast.SqlCondition;2import org.evomaster.dbconstraint.ast.SqlConditionVisitor;3import org.evomaster.dbconstraint.ast.SqlOperator;4import org.evomaster.dbconstraint.ast.SqlPredicate;5import org.evomaster.dbconstraint.ast.SqlTable;6import org.evomaster.dbconstraint.parser.SqlParser;7import org.evomaster.dbconstraint.parser.SqlParserFactory;8import org.evomaster.dbconstraint.parser.SqlParserType;9import org.evomaster.dbconstraint.parser.ast.SqlAnd;10import org.evomaster.dbconstraint.parser.ast.SqlConstraint;11import org.evomaster.dbconstraint.parser.ast.SqlEquals;12import org.evomaster.dbconstraint.parser.ast.SqlExists;13import org.evomaster.dbconstraint.parser.ast.SqlExpression;14import org.evomaster.dbconstraint.parser.ast.SqlIn;15import org.evomaster.dbconstraint.parser.ast.SqlIsNotNull;16import org.evomaster.dbconstraint.parser.ast.SqlIsNull;17import org.evomaster.dbconstraint.parser.ast.SqlLike;18import org.evomaster.dbconstraint.parser.ast.SqlNot;19import org.evomaster.dbconstraint.parser.ast.SqlNotEquals;20import org.evomaster.dbconstraint.parser.ast.SqlOr;21import org.evomaster.dbconstraint.parser.ast.SqlTableConstraint;22import org.evomaster.dbconstraint.parser.ast.SqlTrue;23import org.evomaster.dbconstraint.parser.ast.SqlValue;24import java.util.ArrayList;25import java.util.List;26public class SqlConditionVisitorImpl implements SqlConditionVisitor<String> {27    private List<String> sqlConstraints = new ArrayList<>();28    private List<String> sqlQueries = new ArrayList<>();29    private List<String> sqlQueriesNoFK = new ArrayList<>();30    public List<String> getSqlConstraints() {31        return sqlConstraints;32    }33    public List<String> getSqlQueries() {34        return sqlQueries;35    }36    public List<String> getSqlQueriesNoFK() {37        return sqlQueriesNoFK;38    }39    public String visit(SqlCondition condition) {40        return null;41    }42    public String visit(SqlPredicate predicate

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.