How to use extract method of org.evomaster.client.java.controller.internal.db.constraint.H2ConstraintExtractor class

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

Source:H2ConstraintExtractor.java

...26     * @param connectionToH2 a connection to a H2 database27     * @param schemaDto      a DTO schema with retrieved information from the JBDC metadata28     * @throws SQLException  if the connection to the H2 database fails29     */30    public List<DbTableConstraint> extract(Connection connectionToH2, DbSchemaDto schemaDto) throws SQLException {31        List<DbTableConstraint> columnConstraints = extractColumnConstraints(connectionToH2, schemaDto);32        List<DbTableConstraint> tableCheckExpressions = extractTableConstraints(connectionToH2, schemaDto);33        List<DbTableConstraint> allConstraints = new ArrayList<>();34        allConstraints.addAll(columnConstraints);35        allConstraints.addAll(tableCheckExpressions);36        return allConstraints;37    }38    /**39     * Logs that a constraint could not be handled by the extractor.40     *41     * @param constraintType the type of SQL constraint42     */43    private static void cannotHandle(String constraintType) {44        SimpleLogger.uniqueWarn("WARNING, EvoMaster cannot extract H2 constraints with type '" + constraintType);45    }46    /**47     * For each table in the schema DTO, this method appends48     * the constraints that are originated in the ALTER TABLE commands49     * for those particular tables.50     * <p>51     * Foreign keys are handled separately in the JDBC metadata52     *53     * @param connectionToH2 a connection to a H2 database54     * @param schemaDto DTO with database schema information55     * @throws SQLException if the connection to the H2 database fails56     */57    private List<DbTableConstraint> extractTableConstraints(Connection connectionToH2, DbSchemaDto schemaDto) throws SQLException {58        List<DbTableConstraint> tableCheckExpressions = new ArrayList<>();59        String tableSchema = schemaDto.name;60        for (TableDto tableDto : schemaDto.tables) {61            String tableName = tableDto.name;62            try (Statement statement = connectionToH2.createStatement()) {63                final String query = String.format("Select * From INFORMATION_SCHEMA.CONSTRAINTS\n" +64                        " where CONSTRAINTS.TABLE_SCHEMA='%s' \n"65                        + " and CONSTRAINTS.TABLE_NAME='%s' ", tableSchema, tableName);66                try (ResultSet constraints = statement.executeQuery(query)) {67                    while (constraints.next()) {68                        String constraintType = constraints.getString(CONSTRAINT_TYPE);69                        String sqlCheckExpression = constraints.getString(CHECK_EXPRESSION);70                        String columnList = constraints.getString(COLUMN_LIST);71                        DbTableConstraint constraint;72                        switch (constraintType) {73                            case UNIQUE:74                                List<String> uniqueColumnNames = Arrays.stream(columnList.split(",")).map(String::trim).collect(Collectors.toList());75                                constraint = new DbTableUniqueConstraint(tableName, uniqueColumnNames);76                                tableCheckExpressions.add(constraint);77                                break;78                            case PRIMARY_KEY:79                            case PRIMARY_KEY_BLANK:80                            case REFERENTIAL:81                                /**82                                 * This type of constraint is already handled by83                                 * JDBC Metadata84                                 **/85                                break;86                            case CHECK:87                                constraint = new DbTableCheckExpression(tableName, sqlCheckExpression);88                                tableCheckExpressions.add(constraint);89                                break;90                            default:91                                cannotHandle(constraintType);92                        }93                    }94                }95            }96        }97        return tableCheckExpressions;98    }99    /**100     * For each table in the schema DTO, this method appends101     * the constraints that are originated in the CREATE TABLE commands102     * for those particular tables.103     * <p>104     * Unique constraints and Foreign keys are handled separately in the JDBC metadata105     *106     * @param connectionToH2 a connection to a H2 database107     * @param schemaDto DTO with database schema information108     * @throws SQLException if the connection to the database fails109     */110    private List<DbTableConstraint> extractColumnConstraints(Connection connectionToH2, DbSchemaDto schemaDto) throws SQLException {111        String tableSchema = schemaDto.name;112        List<DbTableConstraint> columnConstraints = new ArrayList<>();113        for (TableDto tableDto : schemaDto.tables) {114            String tableName = tableDto.name;115            try (Statement statement = connectionToH2.createStatement()) {116                final String query = String.format("Select * From INFORMATION_SCHEMA.COLUMNS where COLUMNS.TABLE_SCHEMA='%s' and COLUMNS.TABLE_NAME='%s' ", tableSchema, tableName);117                try (ResultSet columns = statement.executeQuery(query)) {118                    while (columns.next()) {119                        String sqlCheckExpression = columns.getString("CHECK_CONSTRAINT");120                        if (sqlCheckExpression != null && !sqlCheckExpression.equals("")) {121                            DbTableCheckExpression constraint = new DbTableCheckExpression(tableName, sqlCheckExpression);122                            columnConstraints.add(constraint);123                        }124                    }...

extract

Using AI Code Generation

1import org.evomaster.client.java.controller.api.dto.database.schema.DbSchemaDto;2import org.evomaster.client.java.controller.api.dto.database.schema.TableDto;3import org.evomaster.client.java.controller.internal.db.constraint.ConstraintExtractor;4import org.evomaster.client.java.controller.internal.db.constraint.H2ConstraintExtractor;5import java.sql.Connection;6import java.sql.DriverManager;7import java.sql.SQLException;8import java.util.List;9public class H2ConstraintExtractorTest {10    public static void main(String[] args) throws SQLException {11        String url = "jdbc:h2:mem:test;DB_CLOSE_DELAY=-1";12        Connection connection = DriverManager.getConnection(url, "sa", "");13        ConstraintExtractor constraintExtractor = new H2ConstraintExtractor();14        DbSchemaDto schema = constraintExtractor.extractSchema(connection);15        List<TableDto> tables = schema.getTables();16        for (TableDto table : tables) {17            System.out.println(table.getName());18            System.out.println(table.getColumns());19        }20    }21}22[TableColumnDto{name='id', type='INT', nullable=false}, TableColumnDto{name='name', type='VARCHAR', nullable=true}]23[TableColumnDto{name='id', type='INT', nullable=false}, TableColumnDto{name='name', type='VARCHAR', nullable=true}]24[TableColumnDto{name='id', type='INT', nullable=false}, TableColumnDto{name='name', type='VARCHAR', nullable=true}]25[TableColumnDto{name='id', type='INT', nullable=false}, TableColumnDto{name='name', type='VARCHAR', nullable=true}]26[TableColumnDto{name='id', type='INT', nullable=false}, TableColumnDto{name='name', type='VARCHAR', nullable=true}]27import org.evomaster.client.java.controller.api.dto.database.schema.DbSchemaDto;28import org.evomaster.client.java.controller.internal.db.constraint.ConstraintExtractor;29import org.evomaster.client.java.controller.internal.db.constraint.H2ConstraintExtractor;30import java.sql.Connection;31import java.sql.DriverManager;32import java.sql.SQLException;33public class ConstraintExtractorTest {

extract

Using AI Code Generation

1package org.evomaster.client.java.controller.internal.db.constraint;2import java.sql.*;3import java.util.*;4import java.util.stream.Collectors;5public class H2ConstraintExtractor implements ConstraintExtractor {6    public List<String> extract(String tableName, Connection connection) throws SQLException {7        List<String> constraints = new ArrayList<>();8        try (PreparedStatement ps = connection.prepareStatement("SELECT * FROM INFORMATION_SCHEMA.CONSTRAINTS WHERE TABLE_NAME = ?")) {9            ps.setString(1, tableName);10            try (ResultSet rs = ps.executeQuery()) {11                while (rs.next()) {12                    String constraintName = rs.getString("CONSTRAINT_NAME");13                    String constraintType = rs.getString("CONSTRAINT_TYPE");14                    String constraintTable = rs.getString("TABLE_NAME");15                    constraints.add(constraintName);16                }17            }18        }19        return constraints;20    }21}22public List<String> getConstraints(String tableName) throws SQLException {23        List<String> constraints = new ArrayList<>();24        try (PreparedStatement ps = connection.prepareStatement("SELECT * FROM INFORMATION_SCHEMA.CONSTRAINTS WHERE TABLE_NAME = ?")) {25            ps.setString(1, tableName);26            try (ResultSet rs = ps.executeQuery()) {27                while (rs.next()) {28                    String constraintName = rs.getString("CONSTRAINT_NAME");29                    String constraintType = rs.getString("CONSTRAINT_TYPE");30                    String constraintTable = rs.getString("TABLE_NAME");31                    constraints.add(constraintNa

extract

Using AI Code Generation

1public class H2ConstraintExtractorTest {2    public void testExtract() {3        H2ConstraintExtractor extractor = new H2ConstraintExtractor();4        List<TableConstraint> result = extractor.extract("jdbc:h2:mem:test", "sa", "");5        Assertions.assertEquals(2, result.size());6        Assertions.assertEquals("test", result.get(0).getTableName());7        Assertions.assertEquals("name", result.get(0).getColumnName());8        Assertions.assertEquals("NOT NULL", result.get(0).getConstraintName());9        Assertions.assertEquals("test", result.get(1).getTableName());10        Assertions.assertEquals("age", result.get(1).getColumnName());11        Assertions.assertEquals("NOT NULL", result.get(1).getConstraintName());12    }13}

extract

Using AI Code Generation

1import org.evomaster.client.java.controller.api.dto.database.schema.DatabaseType;2import org.evomaster.client.java.controller.api.dto.database.schema.DbSchemaDto;3import org.evomaster.client.java.controller.api.dto.database.schema.TableDto;4import org.evomaster.client.java.controller.internal.db.constraint.H2ConstraintExtractor;5import org.evomaster.client.java.controller.internal.db.schema.H2SchemaExtractor;6import java.sql.Connection;7import java.sql.DriverManager;8import java.sql.SQLException;9import java.util.List;10public class ExtractConstraintsFromH2 {11    private static final String JDBC_URL = "jdbc:h2:mem:test;DB_CLOSE_DELAY=-1";12    public static void main(String[] args) throws SQLException {13        try (Connection connection = DriverManager.getConnection(JDBC_URL)) {14            H2SchemaExtractor extractor = new H2SchemaExtractor(connection);15            DbSchemaDto schema = extractor.extract();16            H2ConstraintExtractor constraintExtractor = new H2ConstraintExtractor(connection);17            for (TableDto table : schema.getTables()) {18                List<String> constraints = constraintExtractor.extract(table);19                System.out.println("constraints:");20                for (String constraint : constraints) {21                    System.out.println("  - " + constraint);22                }23            }24        }25    }26}27  - "check (id > 0)"28  - "check (name <> '')"29  - "check (salary > 0)"30  - "check (id > 0)"31  - "check (name <> '')"32  - "check (salary > 0)"

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.