KaneAI - World’s First E2E Software Testing Agent.Signup for Private Beta Join TestMu Conference

How to use cannotHandle method of org.evomaster.client.java.controller.internal.db.constraint.PostgresConstraintExtractor class

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

Source:PostgresConstraintExtractor.java

...22     * Logs that a constraint could not be handled by the extractor.23     *24     * @param constraintType25     */26    private static void cannotHandle(String constraintType) {27        SimpleLogger.uniqueWarn("WARNING, EvoMaster cannot extract Postgres constraints with type '" + constraintType);28    }29    private static DbTableUniqueConstraint getDbTableUniqueConstraint(Connection connectionToPostgres, String tableSchema, String tableName, Integer[] columnIds) throws SQLException {30        List<String> uniqueColumnNames = new ArrayList<>();31        for (int columnId : columnIds) {32            String qry = String.format("SELECT att.* " +33                    " FROM pg_catalog.pg_attribute att " +34                    " INNER JOIN pg_catalog.pg_class rel\n " +35                    "    ON rel.oid = att.attrelid\n " +36                    " INNER JOIN pg_catalog.pg_namespace nsp\n " +37                    "    ON nsp.oid = rel.relnamespace\n " +38                    " WHERE nsp.nspname = '%s'\n" +39                    "   AND rel.relname = '%s' \n" +40                    "   AND att.attnum =  %s;", tableSchema, tableName, columnId);41            try (Statement stmt = connectionToPostgres.createStatement()) {42                try (ResultSet rs = stmt.executeQuery(qry)) {43                    boolean hasRows = rs.next();44                    if (!hasRows) {45                        throw new IllegalStateException("Unexpected missing pg_catalog.pg_attribute data");46                    }47                    String uniqueColumnName = rs.getString("attname");48                    uniqueColumnNames.add(uniqueColumnName);49                }50            }51        }52        return new DbTableUniqueConstraint(tableName, uniqueColumnNames);53    }54    public List<DbTableConstraint> extract(Connection connectionToPostgres, DbSchemaDto schemaDto) throws SQLException {55        String tableSchema = schemaDto.name;56        List<DbTableConstraint> constraints = new ArrayList<>();57        for (TableDto tableDto : schemaDto.tables) {58            try (Statement statement = connectionToPostgres.createStatement()) {59                String tableName = tableDto.name;60                String query = String.format("SELECT con.*\n" +61                        "       FROM pg_catalog.pg_constraint con\n" +62                        "            INNER JOIN pg_catalog.pg_class rel\n" +63                        "                       ON rel.oid = con.conrelid\n" +64                        "            INNER JOIN pg_catalog.pg_namespace nsp\n" +65                        "                       ON nsp.oid = connamespace\n" +66                        "       WHERE nsp.nspname = '%s'\n" +67                        "             AND rel.relname = '%s';", tableSchema, tableName);68                try (ResultSet columns = statement.executeQuery(query)) {69                    while (columns.next()) {70                        String checkConstraint = columns.getString(CONSRC);71                        Array array = columns.getArray(CONKEY);72                        String constraintType = columns.getString(CONTYPE);73                        DbTableConstraint constraint;74                        switch (constraintType) {75                            case CONSTRAINT_TYPE_CHECK:76                                constraint = new DbTableCheckExpression(tableName, checkConstraint);77                                constraints.add(constraint);78                                break;79                            case CONSTRAINT_TYPE_UNIQUE:80                                Integer[] uniqueColumnIds = (Integer[]) array.getArray();81                                constraint = getDbTableUniqueConstraint(connectionToPostgres, tableSchema, tableName, uniqueColumnIds);82                                constraints.add(constraint);83                                break;84                            case CONSTRAINT_TYPE_FOREIGN_KEY:85                            case CONSTRAINT_TYPE_PRIMARY_KEY:86                                /**87                                 * These types of constraints are already handled by88                                 * JDBC Metadata89                                 **/90                                break;91                            case CONSTRAINT_TYPE_TRIGGER:92                                cannotHandle("TRIGGER CONSTRAINT");93                                break;94                            case CONSTRAINT_TYPE_EXCLUSION:95                                cannotHandle("EXCLUSION CONSTRAINT");96                                break;97                            default:98                                cannotHandle("Unknown constraint type " + constraintType);99                                break;100                        }101                    }102                }103            }104        }105        return constraints;106    }107}...

cannotHandle

Using AI Code Generation

1public class TestSuiteExample extends EMTest {2    public void initController() throws Exception {3        controller = new org.evomaster.client.java.controller.api.EMTestController();4        controller.init();5    }6    public void initTestState() throws Exception {7        testState = new org.evomaster.client.java.controller.api.TestState();8    }9    public void resetStateOfSUT() {

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.