How to use getColumn method of org.evomaster.client.java.controller.internal.db.SchemaExtractor class

Best EvoMaster code snippet using org.evomaster.client.java.controller.internal.db.SchemaExtractor.getColumn

Source:SchemaExtractor.java

...228            primaryKeySequence.put(pkIndex, pkColumnName);229        }230        rsPK.close();231        tableDto.primaryKeySequence.addAll(primaryKeySequence.values());232        ResultSet columns = md.getColumns(null, schemaDto.name, tableDto.name, null);233        Set<String> columnNames = new HashSet<>();234        while (columns.next()) {235            ColumnDto columnDto = new ColumnDto();236            tableDto.columns.add(columnDto);237            columnDto.table = tableDto.name;238            columnDto.name = columns.getString("COLUMN_NAME");239            if (columnNames.contains(columnDto.name)) {240                /**241                 * Perhaps we should throw a more specific exception than IllegalArgumentException242                 */243                throw new IllegalArgumentException("Cannot handle repeated column " + columnDto.name + " in table " + tableDto.name);244            } else {245                columnNames.add(columnDto.name);246            }247            columnDto.type = columns.getString("TYPE_NAME");248            columnDto.size = columns.getInt("COLUMN_SIZE");249            columnDto.nullable = columns.getBoolean("IS_NULLABLE");250            columnDto.autoIncrement = columns.getBoolean("IS_AUTOINCREMENT");251            //columns.getString("DECIMAL_DIGITS");252            columnDto.primaryKey = pks.contains(columnDto.name);253        }254        columns.close();255        ResultSet fks = md.getImportedKeys(null, null, tableDto.name);256        while (fks.next()) {257            //TODO need to see how to handle case of multi-columns258            ForeignKeyDto fkDto = new ForeignKeyDto();259            fkDto.sourceColumns.add(fks.getString("FKCOLUMN_NAME"));260            fkDto.targetTable = fks.getString("PKTABLE_NAME");261            tableDto.foreignKeys.add(fkDto);262        }263        fks.close();264    }265    /**266     * Sets the foreignKeyToAutoIncrement field in the Column DTO267     * when a column is a foreign key to an auto increment value.268     * This information will be needed to properly handle the269     * automatically generated values in primary keys that are270     * referenced by columns in other tables271     *272     * @param schema a DTO with the database information273     */274    private static void addForeignKeyToAutoIncrement(DbSchemaDto schema) {275        for (TableDto tableDto : schema.tables) {276            for (ColumnDto columnDto : tableDto.columns) {277                if (isFKToAutoIncrementColumn(schema, tableDto, columnDto.name)) {278                    columnDto.foreignKeyToAutoIncrement = true;279                }280            }281        }282    }283    /**284     * @return  a table DTO for a particular table name285     */286    private static TableDto getTable(DbSchemaDto schema, String tableName) {287        TableDto tableDto = schema.tables.stream()288                .filter(t -> t.name.equalsIgnoreCase(tableName))289                .findFirst().orElse(null);290        return tableDto;291    }292    private static ColumnDto getColumn(TableDto table, String columnName) {293        ColumnDto columnDto = table.columns.stream()294                .filter(c -> c.name.equalsIgnoreCase(columnName))295                .findFirst().orElse(null);296        return columnDto;297    }298    /**299     * Checks if the given table/column is a foreign key to an autoincrement column.300     * This is done to be able to compute foreignKeyToAutoIncrement boolean.301     * Otherwise, we could just read that boolean.302     *303     * @return true if the given table/column is a foreign key to an autoincrement column.304     */305    private static boolean isFKToAutoIncrementColumn(DbSchemaDto schema, TableDto tableDto, String columnName) {306        Objects.requireNonNull(schema);307        Objects.requireNonNull(tableDto);308        Objects.requireNonNull(columnName);309        // is this column among the declared FKs?310        if (!tableDto.foreignKeys.stream()311                .anyMatch(fk -> fk.sourceColumns.stream()312                        .anyMatch(s -> s.equalsIgnoreCase(columnName)))) {313            return false;314        }315        ColumnDto columnDto = getColumn(tableDto, columnName);316        if (columnDto.autoIncrement == true) {317            // Assuming here that a FK cannot be auto-increment318            return false;319        }320        // check if the column belongs to a foreign key that is non printable321        for (ForeignKeyDto fk : tableDto.foreignKeys) {322            if (fk.sourceColumns.stream()323                    .anyMatch(s -> s.equalsIgnoreCase(columnName))) {324                /*325                    TODO: instead of using those positions, should have proper326                    support for multi-column PKs/FKs327                 */328                int positionInFKSequence = fk.sourceColumns.indexOf(columnName);329                TableDto targetTableDto = getTable(schema, fk.targetTable);330                String targetColumnName = targetTableDto.primaryKeySequence.get(positionInFKSequence);331                ColumnDto targetColumnDto = getColumn(targetTableDto, targetColumnName);332                /*333                    Either that target PK is auto-increment, or itself is a FK to a non-printable PK334                 */335                if (targetColumnDto.autoIncrement ||336                        isFKToAutoIncrementColumn(schema, targetTableDto, targetColumnName)) {337                    return true;338                }339            }340        }341        return false;342    }343}...

getColumn

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.internal.db.SchemaExtractor;4import org.evomaster.client.java.controller.internal.db.SqlScriptExecutor;5import org.evomaster.client.java.controller.internal.db.TableRow;6import org.evomaster.client.java.controller.internal.db.TableSchema;7import org.evomaster.client.java.controller.internal.db.h2.H2Controller;8import java.sql.Connection;9import java.sql.SQLException;10import java.util.ArrayList;11import java.util.List;12import static org.evomaster.client.java.controller.api.dto.database.schema.DatabaseType.H2;13public class H2ControllerImpl implements H2Controller {14    private final String url;15    private final String user;16    private final String password;17    public H2ControllerImpl(String url, String user, String password) {18        this.url = url;19        this.user = user;20        this.password = password;21    }22    public DatabaseType getDatabaseType() {23        return H2;24    }25    public DbSchemaDto getSchema() throws SQLException {26        try (Connection con = SqlScriptExecutor.createDatabaseConnection(url, user, password)) {27            return SchemaExtractor.extractSchema(con, H2);28        }29    }30    public List<TableRow> executeQuery(String query) throws SQLException {31        try (Connection con = SqlScriptExecutor.createDatabaseConnection(url, user, password)) {32            return SqlScriptExecutor.executeQuery(con, query);33        }34    }35    public List<String> getTableNames() throws SQLException {36        try (Connection con = SqlScriptExecutor.createDatabaseConnection(url, user, password)) {37            return SchemaExtractor.getTableNames(con, H2);38        }39    }40    public List<TableSchema> getTableSchemas() throws SQLException {41        try (Connection con = SqlScriptExecutor.createDatabaseConnection(url, user, password)) {42            return SchemaExtractor.getTableSchemas(con, H2);43        }44    }45    public List<TableSchema> getTableSchemas(List<String> tableNames) throws SQLException {46        try (Connection con = SqlScriptExecutor.createDatabaseConnection(url, user, password)) {47            return SchemaExtractor.getTableSchemas(con, H2, tableNames);48        }49    }

getColumn

Using AI Code Generation

1schemaExtractor.getColumn(table, column);2schemaExtractor.getPrimaryKey(table);3schemaExtractor.getTableNames();4schemaExtractor.getForeignKeys(table);5schemaExtractor.getExportedKeys(table);6schemaExtractor.getIndexes(table);7schemaExtractor.getUniqueConstraints(table);8schemaExtractor.getTable(table);9schemaExtractor.getTables();10schemaExtractor.getTableNames();11schemaExtractor.getTableNames();12schemaExtractor.getTableNames();13schemaExtractor.getTableNames();14schemaExtractor.getTableNames();15schemaExtractor.getTableNames();

getColumn

Using AI Code Generation

1package org.evomaster.client.java.controller.internal.db;2import org.evomaster.client.java.controller.api.dto.database.schema.DatabaseType;3import java.sql.Connection;4import java.sql.DatabaseMetaData;5import java.sql.ResultSet;6import java.sql.SQLException;7import java.util.ArrayList;8import java.util.List;9public class SchemaExtractor {10    private final String catalog;11    private final String schema;12    private final DatabaseType databaseType;13    private final Connection connection;14    public SchemaExtractor(String catalog, String schema, DatabaseType databaseType, Connection connection) {15        this.catalog = catalog;16        this.schema = schema;17        this.databaseType = databaseType;18        this.connection = connection;19    }20    public List<Table> getTables() throws SQLException {21        List<Table> tables = new ArrayList<>();22        try (ResultSet rs = connection.getMetaData().getTables(catalog, schema, null, null)) {23            while (rs.next()) {24                String tableName = rs.getString("TABLE_NAME");25                tables.add(getTable(tableName));26            }27        }28        return tables;29    }30    public Table getTable(String tableName) throws SQLException {31        List<Column> columns = new ArrayList<>();32        try (ResultSet rs = connection.getMetaData().getColumns(catalog, schema, tableName, null)) {33            while (rs.next()) {34                String columnName = rs.getString("COLUMN_NAME");35                String typeName = rs.getString("TYPE_NAME");36                int columnSize = rs.getInt("COLUMN_SIZE");37                int decimalDigits = rs.getInt("DECIMAL_DIGITS");38                int nullable = rs.getInt("NULLABLE");39                int position = rs.getInt("ORDINAL_POSITION");40                boolean primary = false;41                boolean unique = false;42                boolean autoincrement = false;43                if (databaseType == DatabaseType.POSTGRES) {44                    try (ResultSet rs2 = connection.getMetaData().getIndexInfo(catalog, schema, tableName, false, false)) {45                        while (rs2.next()) {46                            if (rs2.getString("COLUMN_NAME").equals(columnName) && rs2.getString("INDEX_NAME").equals("

getColumn

Using AI Code Generation

1package org.evomaster.client.java.controller.api.dto.database.schema;2import java.util.List;3public class Table {4    private String name;5    private List<Column> columns;6    public Table() {7    }8    public Table(String name, List<Column> columns) {9        this.name = name;10        this.columns = columns;11    }12    public String getName() {13        return name;14    }15    public void setName(String name) {16        this.name = name;17    }18    public List<Column> getColumns() {19        return columns;20    }21    public void setColumns(List<Column> columns) {22        this.columns = columns;23    }24    public String toString() {25        return "Table{" +26                '}';27    }28}29package org.evomaster.client.java.controller.api.dto.database.schema;30public class Column {31    private String name;32    private String type;33    private boolean nullable;34    private boolean primaryKey;35    private boolean autoIncrement;36    public Column() {37    }38    public Column(String name, String type, boolean nullable, boolean primaryKey, boolean autoIncrement) {39        this.name = name;40        this.type = type;41        this.nullable = nullable;42        this.primaryKey = primaryKey;43        this.autoIncrement = autoIncrement;44    }45    public String getName() {46        return name;47    }48    public void setName(String name) {49        this.name = name;50    }51    public String getType() {52        return type;53    }54    public void setType(String type) {55        this.type = type;56    }57    public boolean isNullable() {58        return nullable;59    }60    public void setNullable(boolean nullable) {61        this.nullable = nullable;62    }63    public boolean isPrimaryKey() {64        return primaryKey;65    }66    public void setPrimaryKey(boolean primaryKey) {67        this.primaryKey = primaryKey;68    }69    public boolean isAutoIncrement() {70        return autoIncrement;71    }72    public void setAutoIncrement(boolean autoIncrement)

getColumn

Using AI Code Generation

1List<String> columnNames = new ArrayList<>();2try (ResultSet rs = connection.getMetaData().getColumns(null, null, table, null)) {3    while (rs.next()) {4        columnNames.add(rs.getString("COLUMN_NAME"));5    }6}7List<String> columnNames = new ArrayList<>();8try (ResultSet rs = connection.getMetaData().getColumns(null, null, table, null)) {9    while (rs.next()) {10        columnNames.add(rs.getString("COLUMN_NAME"));11    }12}13List<String> columnNames = new ArrayList<>();14try (ResultSet rs = connection.getMetaData().getColumns(null, null, table, null)) {15    while (rs.next()) {16        columnNames.add(rs.getString("COLUMN_NAME"));17    }18}19List<String> columnNames = new ArrayList<>();20try (ResultSet rs = connection.getMetaData().getColumns(null, null, table, null)) {21    while (rs.next()) {22        columnNames.add(rs.getString("COLUMN_NAME"));23    }24}25List<String> columnNames = new ArrayList<>();26try (ResultSet rs = connection.getMetaData().getColumns(null, null, table, null)) {27    while (rs.next()) {28        columnNames.add(rs.getString("COLUMN_NAME"));29    }30}

getColumn

Using AI Code Generation

1getColumn("table_name");2getColumn("table_name", "column_name");3getColumn("table_name", "column_name", "column_name2");4getColumn("table_name", "column_name", "column_name2", "column_name3");5getColumn("table_name", "column_name", "column_name2", "column_name3", "column_name4");6getColumn("table_name", "column_name", "column_name2", "column_name3", "column_name4", "column_name5");7getColumn("table_name", "column_name", "column_name2", "column_name3", "column_name4", "column_name5", "column_name6");8getColumn("table_name", "column_name", "column_name2", "column_name3", "column_name4", "column_name5", "column_name6", "column_name7");9getColumn("table_name", "column_name", "column_name2", "column_name3", "column_name4", "column_name5", "column_name6", "column_name7", "column_name8");10getColumn("table_name", "column_name", "column_name2", "column_name3", "column_name4", "column_name5", "column_name6", "column_name7", "column_name8", "column_name9");11getColumn("table_name", "column_name", "column_name2", "column_name3", "column_name4", "column_name5", "column_name6", "column_name7", "column_name8", "column_name9", "column_name10");12getColumn("table_name", "column_name", "column_name2

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.