Unlock 30% off on Manual Testing Annual Plans this Holiday Season.

Copied to Clipboard!

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

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

Source:SchemaExtractor.java

...273     */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}...

isFKToAutoIncrementColumn

Using AI Code Generation

1package org.evomaster.client.java.controller.internal.db;2import org.evomaster.client.java.controller.api.dto.database.schema.DatabaseType;3import org.evomaster.client.java.controller.api.dto.database.schema.SchemaDto;4import org.evomaster.client.java.controller.api.dto.database.schema.TableDto;5import org.evomaster.client.java.controller.api.dto.database.schema.TableIndexDto;6import org.evomaster.client.java.controller.api.dto.database.schema.TableIndexType;7import org.evomaster.client.java.controller.api.dto.database.schema.ViewDto;8import java.util.ArrayList;9import java.util.List;10import java.util.stream.Collectors;11public class SchemaExtractor {12    private final SchemaDto schema;13    private final DatabaseType databaseType;14    public SchemaExtractor(SchemaDto schema, DatabaseType databaseType) {15        this.schema = schema;16        this.databaseType = databaseType;17    }18    public List<TableDto> getTables() {19        return schema.getTables();20    }21    public List<ViewDto> getViews() {22        return schema.getViews();23    }24    public List<TableIndexDto> getIndexes() {25        return schema.getIndexes();26    }27    public List<TableIndexDto> getUniqueIndexes() {28        return schema.getIndexes().stream()29                .filter(i -> i.getType() == TableIndexType.UNIQUE)30                .collect(Collectors.toList());31    }32    public List<TableIndexDto> getIndexesForTable(String table) {33        return schema.getIndexes().stream()34                .filter(i -> i.getTable().equals(table))35                .collect(Collectors.toList());36    }37    public List<TableIndexDto> getUniqueIndexesForTable(String table) {38        return schema.getIndexes().stream()39                .filter(i -> i.getType() == TableIndexType.UNIQUE)40                .filter(i -> i.getTable().equals(table))41                .collect(Collectors.toList());42    }43    public TableDto getTable(String name) {44        return schema.getTables().stream()45                .filter(t -> t.getName().equals(name))46                .findFirst()47                .orElse(null);48    }49    public ViewDto getView(String name) {50        return schema.getViews().stream()51                .filter(t -> t.getName().equals(name))52                .findFirst()53                .orElse(null);54    }55    public TableIndexDto getIndex(String name) {56        return schema.getIndexes().stream()57                .filter(i -> i.getName().equals(name))58                .findFirst()59                .orElse(null);60    }

isFKToAutoIncrementColumn

Using AI Code Generation

1import org.evomaster.client.java.controller.internal.db.SchemaExtractor2import org.evomaster.client.java.controller.db.DbAction3import org.evomaster.client.java.controller.db.SqlInsertBuilder4import org.evomaster.client.java.controller.db.SqlScriptExecutor5import org.evomaster.client.java.controller.db.Table6import org.evomaster.client.java.controller.api.dto.database.schema.DatabaseType7import org.evomaster.client.java.controller.api.dto.database.operations.InsertionDto8import org.evomaster.client.java.controller.api.dto.database.operations.DatabaseCommandDto9import org.evomaster.client.java.controller.api.dto.database.operations.SqlScriptDto10import org.evomaster.client.java.controller.api.dto.database.schema.TableDto11import org.evomaster.client.java.controller.api.dto.database.schema.ColumnDto12import org.evomaster.client.java.controller.api.dto.database.schema.ForeignKeyDto13import org.evomaster.client.java.controller.api.dto.database.schema.SchemaDto14import org.evomaster.client.java.controller.api.dto.database.operations.DatabaseCommandDto15import org.evomaster.client.java.controller.api.dto.database.operations.InsertionDto16import org.evomaster.client.java.controller.api.dto.database.operations.SqlScriptDto17import org.evomaster.client.java.controller.api.dto.database.schema.ColumnDto18import org.evomaster.client.java.controller.api.dto.database.schema.DatabaseType19import org.evomaster.client.java.controller.api.dto.database.schema.ForeignKeyDto20import org.evomaster.client.java.controller.api.dto.database.schema.SchemaDto21import org.evomaster.client.java.controller.api.dto.database.schema.TableDto22import org.evomaster.client.java.controller.db.DbAction23import org.evomaster.client.java.controller.db.SqlInsertBuilder24import org.evomaster.client.java.controller.db.SqlScriptExecutor25import org.evomaster.client.java.controller.db.Table26import org.evomaster.client.java.controller.internal.db.SchemaExtractor27import org.evomaster.client.java.controller.problem.ProblemInfo28import org.evomaster.client.java.controller.problem.RestProblem29import org.evomaster.client.java.controller.problem.RestProblemException30import org.evomaster.client.java.controller.problem.RestProblemFactory31import org.evomaster.client.java.controller.api.dto.database.operations.SqlScriptDto32import org.evomaster.client.java.controller.api.dto.database.operations.DatabaseCommandDto33import org.evomaster.client.java.controller.api.dto.database.operations.InsertionDto34import org.evomaster.client.java.controller.api.dto.database.schema.ColumnDto35import org.evomaster.client.java.controller.api.dto.database.schema.DatabaseType

isFKToAutoIncrementColumn

Using AI Code Generation

1def extractSchema = { String url, String username, String password ->2    def schema = SchemaExtractor.extractSchema(url, username, password)3    def tables = schema.getTables()4    tables.each { table ->5        println "Table: " + table.getName()6        table.getColumns().each { column ->7            println "Column: " + column.getName() + " is FK: " + column.isForeignKey()8        }9    }10}11extractSchema(url, username, password)12def extractSchema = { String url, String username, String password ->13    def schema = SchemaExtractor.extractSchema(url, username, password)14    def tables = schema.getTables()15    tables.each { table ->16        println "Table: " + table.getName()17        table.getColumns().each { column ->18            println "Column: " + column.getName() + " is FK: " + column.isForeignKey() + " and FK is to autoincrement column: " + column.isFKToAutoIncrementColumn()19        }20    }21}22extractSchema(url, username, password)

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.