How to use DbTableCheckExpression class of org.evomaster.client.java.controller.internal.db.constraint package

Best EvoMaster code snippet using org.evomaster.client.java.controller.internal.db.constraint.DbTableCheckExpression

Source:MySQLConstraintExtractor.java

...44                            case MYSQL_CONSTRAINT_TYPE_FOREIGN_KEY:45                                break;46                            case MYSQL_CONSTRAINT_TYPE_CHECK:47                                String constraintName = columns.getString(MYSQL_CONSTRAINT_NAME);48                                DbTableCheckExpression check = getCheckConstraint(connectionToMySQL, tableName, constraintName);49                                constraints.add(check);50                                break;51                            case MYSQL_CONSTRAINT_TYPE_UNIQUE:52                                String uniqueConstraintName = columns.getString(MYSQL_CONSTRAINT_NAME);53                                DbTableUniqueConstraint uniqueConstraint = getUniqueConstraint(connectionToMySQL, tableSchema, tableName, uniqueConstraintName);54                                constraints.add(uniqueConstraint);55                                break;56                            default:57                                cannotHandle("Unknown constraint type " + type);58                        }59                    }60                }61            }62            // handle enum column63            for (ColumnDto column: tableDto.columns){64                if (column.type.equalsIgnoreCase("enum")){65                    DbTableCheckExpression enumConstraint = handleEnum(connectionToMySQL, tableSchema, tableName, column.name);66                    constraints.add(enumConstraint);67                }68            }69        }70        return constraints;71    }72    private DbTableUniqueConstraint getUniqueConstraint(Connection connectionToMySQL, String tableSchema, String tableName, String constraintName) throws SQLException{73        String query = String.format("SELECT %s \n" +74                "   FROM information_schema.KEY_COLUMN_USAGE\n" +75                "   WHERE TABLE_SCHEMA = '%s'\n" +76                "       AND TABLE_NAME = '%s'\n" +77                "       AND CONSTRAINT_NAME='%s';\n", MYSQL_COLUMN_NAME, tableSchema, tableName, constraintName);78        try (Statement stmt = connectionToMySQL.createStatement()) {79            try (ResultSet columns = stmt.executeQuery(query)) {80                List<String> uniqueColumnNames = new ArrayList<>();81                while(columns.next()){82                    uniqueColumnNames.add(columns.getString(MYSQL_COLUMN_NAME));83                }84                if (uniqueColumnNames.isEmpty()) {85                    throw new IllegalStateException("Unexpected missing column names");86                }87                return new DbTableUniqueConstraint(tableName, uniqueColumnNames);88            }89        }90    }91    private DbTableCheckExpression getCheckConstraint(Connection connectionToMySQL, String tableName, String constraintName) throws SQLException{92        String query = String.format("SELECT %s \n" +93                "FROM information_schema.CHECK_CONSTRAINTS\n" +94                "WHERE CONSTRAINT_NAME='%s';\n", MYSQL_CHECK_CLAUSE, constraintName);95        try (Statement stmt = connectionToMySQL.createStatement()) {96            try (ResultSet check = stmt.executeQuery(query)) {97                boolean hasChecks = check.next();98                if (!hasChecks) {99                    throw new IllegalStateException("Unexpected missing check scripts");100                }101                String check_clause = postCheckConstraintHandling(check.getString(MYSQL_CHECK_CLAUSE));102                return new DbTableCheckExpression(tableName, check_clause);103            }104        }105    }106    // Man: Shall I move this into TableCheckExpressionDto e.g., formatCheckExpression107    private String postCheckConstraintHandling(String check_exp){108        return check_exp109                .replaceAll("`", "")110                .replaceAll("_utf8mb4", "") // mysql https://dev.mysql.com/doc/refman/8.0/en/charset-unicode-utf8mb4.html111                .replaceAll("\\\\'","'");112    }113    private DbTableCheckExpression handleEnum(Connection connectionToMySQL, String schemaName, String tableName, String columnName) throws SQLException{114        String query = String.format("SELECT %s\n" +115                "       FROM information_schema.COLUMNS\n" +116                "       WHERE TABLE_SCHEMA='%s'\n" +117                "           AND TABLE_NAME='%s'\n" +118                "           AND COLUMN_NAME='%s';", MYSQL_ENUM_COLUMN_TYPE, schemaName, tableName, columnName);119        try (Statement stmt = connectionToMySQL.createStatement()) {120            try (ResultSet literals = stmt.executeQuery(query)) {121                boolean hasLiterals = literals.next();122                if (!hasLiterals) {123                    throw new IllegalStateException("Unexpected missing literals of enum");124                }125                String literalsValue = literals.getString(MYSQL_ENUM_COLUMN_TYPE);126                return new DbTableCheckExpression(tableName, String.format(127                        "%s %s", columnName, literalsValue128                ));129            }130        }131    }132}...

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.