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

Copied to Clipboard!

How to use getConnectionIfExist method of org.evomaster.client.java.controller.internal.SutController class

Best EvoMaster code snippet using org.evomaster.client.java.controller.internal.SutController.getConnectionIfExist

Source:SutController.java

...163        this.controllerHost = controllerHost;164    }165    @Override166    public InsertionResultsDto execInsertionsIntoDatabase(List<InsertionDto> insertions, InsertionResultsDto... previous) {167        Connection connection = getConnectionIfExist();168        if (connection == null) {169            throw new IllegalStateException("No connection to database");170        }171        try {172            return SqlScriptRunner.execInsert(connection, insertions, previous);173        } catch (SQLException e) {174            throw new RuntimeException(e);175        }176    }177    public int getActionIndex(){178        return actionIndex;179    }180    /**181     * Calculate heuristics based on intercepted SQL commands182     *183     * @param sql command as a string184     */185    @Deprecated186    public final void handleSql(String sql) {187        Objects.requireNonNull(sql);188        sqlHandler.handle(sql);189    }190    public final void enableComputeSqlHeuristicsOrExtractExecution(boolean enableSqlHeuristics, boolean enableSqlExecution){191        sqlHandler.setCalculateHeuristics(enableSqlHeuristics);192        sqlHandler.setExtractSqlExecution(enableSqlHeuristics || enableSqlExecution);193    }194    /**195     * This is needed only during test generation (not execution),196     * and it is automatically called by the EM controller after197     * the SUT is started.198     */199    public final void initSqlHandler() {200        sqlHandler.setConnection(getConnectionIfExist());201        sqlHandler.setSchema(getSqlDatabaseSchema());202    }203    /**204     * TODO further handle multiple connections205     * @return sql connection if there exists206     */207    public final Connection getConnectionIfExist(){208        return (getDbSpecifications() == null209                || getDbSpecifications().isEmpty())? null: getDbSpecifications().get(0).connection;210    }211    /**212     *213     * @return whether to employ smart db clean214     */215    public final boolean doEmploySmartDbClean(){216        return getDbSpecifications() != null217                && !getDbSpecifications().isEmpty() && getDbSpecifications().get(0).employSmartDbClean;218    }219    public final void resetExtraHeuristics() {220        sqlHandler.reset();221    }222    public final List<ExtraHeuristicsDto> getExtraHeuristics() {223        if (extras.size() == actionIndex) {224            extras.add(computeExtraHeuristics());225        }226        return new ArrayList<>(extras);227    }228    public final ExtraHeuristicsDto computeExtraHeuristics() {229        ExtraHeuristicsDto dto = new ExtraHeuristicsDto();230        if(sqlHandler.isCalculateHeuristics() || sqlHandler.isExtractSqlExecution()){231            /*232                TODO refactor, once we move SQL analysis into Core233             */234            List<AdditionalInfo> list = getAdditionalInfoList();235            if(!list.isEmpty()) {236                AdditionalInfo last = list.get(list.size() - 1);237                last.getSqlInfoData().stream().forEach(it -> {238//                    String sql = it.getCommand();239                    try {240                        sqlHandler.handle(it);241                    } catch (Exception e){242                        SimpleLogger.error("FAILED TO HANDLE SQL COMMAND: " + it.getCommand());243                        assert false; //we should try to handle all cases in our tests244                    }245                });246            }247        }248        if(sqlHandler.isCalculateHeuristics()) {249            sqlHandler.getDistances().stream()250                    .map(p ->251                            new HeuristicEntryDto(252                                    HeuristicEntryDto.Type.SQL,253                                    HeuristicEntryDto.Objective.MINIMIZE_TO_ZERO,254                                    p.sqlCommand,255                                    p.distance256                            ))257                    .forEach(h -> dto.heuristics.add(h));258        }259        if (sqlHandler.isCalculateHeuristics() || sqlHandler.isExtractSqlExecution()){260            ExecutionDto executionDto = sqlHandler.getExecutionDto();261            dto.databaseExecutionDto = executionDto;262            // set accessed table263            if (executionDto != null){264                accessedTables.addAll(executionDto.deletedData);265                accessedTables.addAll(executionDto.insertedData.keySet());266//                accessedTables.addAll(executionDto.queriedData.keySet());267                accessedTables.addAll(executionDto.insertedData.keySet());268                accessedTables.addAll(executionDto.updatedData.keySet());269            }270        }271        return dto;272    }273    /**274     * perform smart db clean by cleaning the data in accessed table275     */276    public final void cleanAccessedTables(){277        if (getDbSpecifications() == null || getDbSpecifications().isEmpty()) return;278        if (getDbSpecifications().size() > 1)279            throw new RuntimeException("Error: DO NOT SUPPORT MULTIPLE SQL CONNECTION YET");280        DbSpecification emDbClean = getDbSpecifications().get(0);281        if (getConnectionIfExist() == null || !emDbClean.employSmartDbClean) return;282        try {283            setExecutingInitSql(true);284            // clean accessed tables285            Set<String> tableDataToInit = null;286            if (!accessedTables.isEmpty()){287                List<String> tablesToClean = new ArrayList<>();288                getTableToClean(accessedTables, tablesToClean);289                if (!tablesToClean.isEmpty()){290                    if (emDbClean.schemaNames != null && !emDbClean.schemaNames.isEmpty()){291                        emDbClean.schemaNames.forEach(sch-> DbCleaner.clearDatabase(getConnectionIfExist(), sch,  null, tablesToClean, emDbClean.dbType));292                    }else293                        DbCleaner.clearDatabase(getConnectionIfExist(), null,  null, tablesToClean, emDbClean.dbType);294                    tableDataToInit = tablesToClean.stream().filter(a-> tableInitSqlMap.keySet().stream().anyMatch(t-> t.equalsIgnoreCase(a))).collect(Collectors.toSet());295                }296            }297            handleInitSql(tableDataToInit, emDbClean);298        }catch (SQLException e) {299            throw new RuntimeException("SQL Init Execution Error: fail to execute "+e);300        }finally {301            setExecutingInitSql(false);302        }303    }304    private void handleInitSql(Collection<String> tableDataToInit, DbSpecification spec) throws SQLException {305        // init db script306        boolean initAll = initSqlScriptAndGetInsertMap(getConnectionIfExist(), spec);307        if (!initAll && tableDataToInit!= null &&!tableDataToInit.isEmpty()){308            tableDataToInit.forEach(a->{309                tableInitSqlMap.keySet().stream().filter(t-> t.equalsIgnoreCase(a)).forEach(t->{310                    tableInitSqlMap.get(t).forEach(c->{311                        try {312                            SqlScriptRunner.execCommand(getConnectionIfExist(), c);313                        } catch (SQLException e) {314                            throw new RuntimeException("SQL Init Execution Error: fail to execute "+ c + " with error "+e);315                        }316                    });317                });318            });319        }320    }321    /**322     * collect info about what table are manipulated by evo in order to generate data directly into it323     * @param tables a list of name of tables324     */325    public void addTableToInserted(List<String> tables){326        accessedTables.addAll(tables);327    }328    private void getTableToClean(List<String> accessedTables, List<String> tablesToClean){329        for (String t: accessedTables){330            if (!findInCollectionIgnoreCase(t, tablesToClean).isPresent()){331                if (findInMapIgnoreCase(t, fkMap).isPresent()){332                    tablesToClean.add(t);333                    List<String> fk = fkMap.entrySet().stream().filter(e->334                            findInCollectionIgnoreCase(t, e.getValue()).isPresent()335                                    && !findInCollectionIgnoreCase(e.getKey(), tablesToClean).isPresent()).map(Map.Entry::getKey).collect(Collectors.toList());336                    if (!fk.isEmpty())337                        getTableToClean(fk, tablesToClean);338                }else {339                    SimpleLogger.uniqueWarn("Cannot find the table "+t+" in ["+String.join(",", fkMap.keySet())+"]");340                }341            }342        }343    }344    private Optional<String> findInCollectionIgnoreCase(String name, Collection<String> list){345        return list.stream().filter(i-> i.equalsIgnoreCase(name)).findFirst();346    }347    private Optional<? extends Map.Entry<String, ?>> findInMapIgnoreCase(String name, Map<String, ?> list){348        return list.entrySet().stream().filter(x-> x.getKey().equalsIgnoreCase(name)).findFirst();349    }350    /**351     *352     * @param dbSpecification contains info of the db connection353     * @return whether the init script is executed354     */355    private boolean initSqlScriptAndGetInsertMap(Connection connection, DbSpecification dbSpecification) throws SQLException {356        if (dbSpecification.initSqlOnResourcePath == null357                && dbSpecification.initSqlScript == null) return false;358        // TODO to handle initSqlMap for multiple connections359        if (tableInitSqlMap.isEmpty()){360            List<String> all = new ArrayList<>();361            if (dbSpecification.initSqlOnResourcePath != null){362               all.addAll(SqlScriptRunnerCached.extractSqlScriptFromResourceFile(dbSpecification.initSqlOnResourcePath));363            }364            if (dbSpecification.initSqlScript != null){365                all.addAll(SqlScriptRunner.extractSql(dbSpecification.initSqlScript));366            }367            if (!all.isEmpty()){368                // collect insert sql commands map, key is table name, and value is a list sql insert commands369                tableInitSqlMap.putAll(SqlScriptRunner.extractSqlTableMap(all));370                // execute all commands371                SqlScriptRunner.runCommands(connection, all);372                return true;373            }374        }375        return false;376    }377    /**378     * Extra information about the SQL Database Schema, if any is present.379     * Note: this is extracted by querying the database itself.380     * So the database must be up and running.381     *382     * @return a DTO with the schema information383     * @see SutHandler#getDbSpecifications384     */385    public final DbSchemaDto getSqlDatabaseSchema() {386        if (schemaDto != null) {387            return schemaDto;388        }389        if (getDbSpecifications() == null || getDbSpecifications().isEmpty()) {390            return null;391        }392        try {393            schemaDto = SchemaExtractor.extract(getConnectionIfExist());394            Objects.requireNonNull(schemaDto);395            schemaDto.employSmartDbClean = doEmploySmartDbClean();396        } catch (Exception e) {397            SimpleLogger.error("Failed to extract the SQL Database Schema: " + e.getMessage());398            return null;399        }400        if (fkMap.isEmpty()){401            schemaDto.tables.forEach(t->{402                fkMap.putIfAbsent(t.name, new ArrayList<>());403                if (t.foreignKeys!=null && !t.foreignKeys.isEmpty()){404                    t.foreignKeys.forEach(f->{405                        fkMap.get(t.name).add(f.targetTable.toUpperCase());406                    });407                }...

Source:EMController.java

...531            sutController.setExecutingInitSql(true);532            // collect info about tables to insert533            noKillSwitch(()-> sutController.addTableToInserted(dto.insertions.stream().map(x-> x.targetTable).collect(Collectors.toList())));534            SimpleLogger.debug("Received database command");535            Connection connection = noKillSwitch(() -> sutController.getConnectionIfExist());536            if (connection == null) {537                String msg = "No active database connection";538                SimpleLogger.warn(msg);539                return Response.status(400).entity(WrappedResponseDto.withError(msg)).build();540            }541            if (dto.command == null && (dto.insertions == null || dto.insertions.isEmpty())) {542                String msg = "No input command";543                SimpleLogger.warn(msg);544                return Response.status(400).entity(WrappedResponseDto.withError(msg)).build();545            }546            if (dto.command != null && dto.insertions != null && !dto.insertions.isEmpty()) {547                String msg = "Only 1 command can be specified";548                SimpleLogger.warn(msg);549                return Response.status(400).entity(WrappedResponseDto.withError(msg)).build();...

getConnectionIfExist

Using AI Code Generation

1import org.evomaster.client.java.controller.api.dto.SutInfoDto;2import org.evomaster.client.java.controller.api.dto.TestResultsDto;3import org.evomaster.client.java.controller.api.dto.database.operations.DatabaseCommandDto;4import org.evomaster.client.java.controller.api.dto.database.operations.InsertionDto;5import org.evomaster.client.java.controller.api.dto.database.operations.SqlScriptDto;6import org.evomaster.client.java.controller.internal.SutController;7import org.evomaster.client.java.controller.internal.db.SqlScriptRunner;8import org.evomaster.client.java.controller.internal.db.SqlScriptRunnerImpl;9import org.evomaster.client.java.controller.internal.db.h2.H2Handler;10import org.evomaster.client.java.controller.internal.db.h2.H2Initializer;11import org.evomaster.client.java.controller.internal.db.h2.H2Wrapper;12import org.evomaster.client.java.controller.internal.db.h2.H2WrapperImpl;13import org.evomaster.client.java.controller.internal.db.schema.DatabaseType;14import org.evomaster.client.java.controller.internal.db.schema.Table;15import org.evomaster.client.java.controller.internal.db.schema.TableColumn;16import org.evomaster.client.java.controller.internal.db.schema.TableSchema;17import org.evomaster.client.java.controller.internal.db.schema.TableSchemaExtractor;18import org.evomaster.client.java.controller.internal.db.schema.TableSchemaExtractorImpl;19import org.evomaster.client.java.controller.internal.db.schema.TableSchemaExtractorNoCache;20import org.evomast

getConnectionIfExist

Using AI Code Generation

1package org.evomaster.client.java.controller.internal;2import org.evomaster.client.java.controller.api.dto.database.operations.DatabaseCommandDto;3import java.sql.Connection;4import java.sql.SQLException;5import java.util.List;6public class SutController {7    private static SutController instance = null;8    private final DBInitializer dbInitializer;9    private final DBController dbController;10    private final RestController restController;11    private final SutHandler sutHandler;12    private SutController() {13        dbInitializer = new DBInitializer();14        dbController = new DBController();15        restController = new RestController();16        sutHandler = new SutHandler();17    }18    public static SutController getInstance() {19        if (instance == null) {20            instance = new SutController();21        }22        return instance;23    }24    public void resetStateOfSUT() {25        dbController.resetStateOfSUT();26        restController.resetStateOfSUT();27        sutHandler.resetStateOfSUT();28    }29    public void startSut() {30        sutHandler.startSut();31    }32    public void stopSut() {33        sutHandler.stopSut();34    }35    public void initializeDatabase() {36        dbInitializer.initializeDatabase();37    }38    public void resetStateOfDatabase() {39        dbController.resetStateOfDatabase();40    }41    public void handleInsertion(String id, List<DatabaseCommandDto> statements) {42        dbController.handleInsertion(id, statements);43    }44    public void handleDeletion(String id, List<DatabaseCommandDto> statements) {45        dbController.handleDeletion(id, statements);46    }47    public Connection getConnectionIfExist() throws SQLException {48        return dbController.getConnectionIfExist();49    }50    public Connection getConnection() throws SQLException {51        return dbController.getConnection();52    }53    public void executeStatements(List<DatabaseCommandDto> statements) {54        dbController.executeStatements(statements);55    }56    public void verifyTableHasExpectedData(String tableName, List<DatabaseCommandDto> statements) {57        dbController.verifyTableHasExpectedData(tableName, statements);58    }59    public void verifyTableHasExpectedData(String tableName, List<DatabaseCommandDto> statements, boolean allowExtraColumns) {60        dbController.verifyTableHasExpectedData(tableName, statements, allowExtraColumns);61    }62    public void verifyTableHasNotExpectedData(String tableName, List<DatabaseCommandDto> statements) {

getConnectionIfExist

Using AI Code Generation

1import org.evomaster.client.java.controller.api.dto.database.operations.DatabaseCommandDto;2import org.evomaster.client.java.controller.api.dto.database.operations.InsertionDto;3import org.evomaster.client.java.controller.api.dto.database.operations.SelectionDto;4import org.evomaster.client.java.controller.api.dto.database.operations.SqlScriptDto;5import org.evomaster.client.java.controller.internal.SutController;6import org.evomaster.client.java.controller.internal.db.SqlScriptRunner;7import org.evomaster.client.java.controller.internal.db.SqlScriptRunnerImpl;8import org.evomaster.client.java.controller.internal.db.schema.SqlSchema;9import org.evomaster.client.java.controller.internal.db.schema.Table;10import org.evomaster.client.java.controller.internal.db.schema.TableColumn;11import org.evomaster.client.java.controller.internal.db.schema.TableIndex;12import org.evomaster.client.java.controller.internal.db.schema.TableRow;13import org.evomaster.client.java.controller.internal.db.schema.View;14import org.evomaster.client.java.controller.internal.db.schema.ViewColumn;15import org.evomaster.client.java.controller.internal.db.schema.ViewRow;16import org.evomaster.client.java.controller.internal.db.schema.ViewTable;17import org.evomaster.client.java.controller.internal.db.schema.ViewTableColumn;18import org.evomaster.client.java.controller.internal.db.schema.ViewTableRow;19import org.evomaster.client.java.controller.internal.db.schema.Views;20import org.evomaster.client.java.controller.internal.db.schema.structure.Schema;21import org.evomaster.client.java.controller.internal.db.schema.structure.TableInfo;22import org.evomaster.client.java.controller.internal.db.schema.structure.TableRowInfo;23import org.evomaster.client.java.controller.internal.db.schema.structure.ViewInfo;24import org.evomaster.client.java.controller.internal.db.schema.structure.ViewRowInfo;25import org.evomaster.client.java.controller.internal.db.schema.structure.ViewTableInfo;26import org.evomaster.client.java.controller.internal.db.schema.structure.ViewTableRowInfo;27import org.evomaster.client.java.controller.internal.db.schema.structure.ViewsInfo;28import org.evomaster.client.java.controller.internal.db.schema.structure.ViewsStructure;29import org.evomaster.client.java.controller.internal.db.schema.structure.ViewsStructureFactory;30import org.evomaster.client.java.controller.internal.db.schema.structure.ViewsStructureFactoryImpl;31import org.evomaster.client.java.controller.internal.db.schema.structure.ViewsStructureFactoryImplTest;32import org.evomaster.client.java.controller.internal.db.schema.structure.ViewsStructureImpl;33import org.evomaster.client.java.controller.internal.db.schema.structure.ViewsStructure

getConnectionIfExist

Using AI Code Generation

1import org.evomaster.client.java.controller.api.dto.SutInfoDto;2import org.evomaster.client.java.controller.internal.SutController;3public class 3 {4    public static void main(String[] args) {5        SutInfoDto sutInfo = new SutInfoDto();6        sutInfo.setSutName("com.example:example:0.0.1-SNAPSHOT");7        sutInfo.setSutControllerPort(40000);8        sutInfo.setPackagePrefixes(new String[]{"com.example"});9        sutInfo.setSutVersion("0.0.1-SNAPSHOT");10        sutInfo.setSutChecksum("c4d6d0c6f1f6b4c6e4f4d4e0d0e4c4f6d0e4c4d6f0c4c4e0e4f0f4d4f6d0e0c4c6f1f6b4c6e4f4d4e0d0e4c4f6d0e4c4d6f0c4c4e0e4f0f4d4f6d0e0c4c6f1f6b4c6e4f4d4e0d0e4c4f6d0e4c4d6f0c4c4e0e4f0f4d4f6d0e0c4c6f1f6b4c6e4f4d4e0d0e4c4f6d0e4c4d6f0c4c4e0e4f0f4d4f6d0e0c4c6f1f6b4c6e4f4d4e0d0e4c4f6d0e4c4d6f0c4c4e0e4f0f4d4f6d0e0c4c6f1f6b4c6e4f4d4e0d0e4c4f6d0e4c4d6f0c4c4e0e4f0f4

getConnectionIfExist

Using AI Code Generation

1package org.evomaster.client.java.controller;2import org.evomaster.client.java.controller.api.dto.database.operations.DatabaseCommandDto;3import org.evomaster.client.java.controller.api.dto.database.operations.DatabaseResultDto;4import org.evomaster.client.java.controller.api.dto.database.operations.InsertionDto;5import org.evomaster.client.java.controller.api.dto.database.operations.SqlScriptDto;6import org.evomaster.client.java.controller.api.dto.database.schema.DbSchemaDto;7import org.evomaster.client.java.controller.api.dto.database.schema.TableDto;8import org.evomaster.client.java.controller.internal.SutController;9import org.evomaster.client.java.controller.problem.ProblemInfo;10import org.evomaster.client.java.controller.problem.RestProblem;11import org.evomaster.client.java.controller.problem.RestResourceCalls;12import org.evomaster.client.java.controller.problem.RestResourceInfo;13import java.sql.Connection;14import java.sql.SQLException;15import java.util.ArrayList;16import java.util.List;17public class Main {18    public static void main(String[] args) throws SQLException {19        SutController controller = SutController.getInstance();20        ProblemInfo problemInfo = new RestProblem("localhost", 8080, "/api");21        controller.initForProblem(problemInfo);22        Connection connection = controller.getConnectionIfExist();23        if (connection == null) {24            System.out.println("No connection available");25        } else {26            System.out.println("Connection available");27        }28    }29}30package org.evomaster.client.java.controller;31import org.evomaster.client.java.controller.api.dto.database.operations.DatabaseCommandDto;32import org.evomaster.client.java.controller.api.dto.database.operations.DatabaseResultDto;33import org.evomaster.client.java.controller.api.dto.database.operations.InsertionDto;34import org.evomaster.client.java.controller.api.dto.database.operations.SqlScriptDto;35import org.evomaster.client.java.controller.api.dto.database.schema.DbSchemaDto;36import org.evomaster.client.java.controller.api.dto.database.schema.TableDto;37import org.evomaster.client.java.controller.internal.SutController;38import org.evomaster.client.java.controller.problem.ProblemInfo;39import org.evomaster.client.java.controller.problem.RestProblem;40import org.evomaster.client.java.controller.problem.RestResourceCalls;41import org.evomaster.client.java.controller

getConnectionIfExist

Using AI Code Generation

1import org.evomaster.client.java.controller.api.dto.database.operations.DatabaseCommandDto;2import org.evomaster.client.java.controller.api.dto.database.operations.InsertionDto;3import org.evomaster.client.java.controller.api.dto.database.operations.SqlScriptDto;4import org.evomaster.client.java.controller.internal.SutController;5import org.evomaster.client.java.controller.internal.db.SqlScriptRunner;6import org.evomaster.client.java.controller.internal.db.SqlScriptWriter;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;12public class 3 {13    public static void main(String[] args) throws SQLException {14        SutController sutController = new SutController("localhost", 40100);15        H2Controller h2Controller = new H2Controller();16        Connection connection = h2Controller.getConnectionIfExist();17        SqlScriptWriter sqlScriptWriter = new SqlScriptWriter();18        SqlScriptRunner sqlScriptRunner = new SqlScriptRunner();19        SqlScriptDto sqlScriptDto = new SqlScriptDto();20        List<DatabaseCommandDto> commands = new ArrayList<>();21        InsertionDto insertionDto = new InsertionDto();22        insertionDto.setTable("test");23        insertionDto.setValues("1, 'test'");24        commands.add(insertionDto);25        sqlScriptDto.setCommands(commands);26        sqlScriptWriter.write(sqlScriptDto);27        sqlScriptRunner.run(sqlScriptDto, connection);28    }29}30import org.evomaster.client.java.controller.api.dto.database.operations.DatabaseCommandDto;31import org.evomaster.client.java.controller.api.dto.database.operations.InsertionDto;32import org.evomaster.client.java.controller.api.dto.database.operations.SqlScriptDto;33import org.evomaster.client.java.controller.internal.SutController;34import org.evomaster.client.java.controller.internal.db.SqlScriptRunner;35import org.evomaster.client.java.controller.internal.db.SqlScriptWriter;36import org.evomaster.client.java.controller.internal.db.h2.H2Controller;37import java.sql.Connection;38import java.sql.SQLException;39import java.util.ArrayList;40import java.util.List;41public class 4 {42    public static void main(String[] args)

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.