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

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

Source:SutController.java

...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));...

initSqlScriptAndGetInsertMap

Using AI Code Generation

1package org.evomaster.client.java.controller.internal;2import org.evomaster.client.java.controller.api.dto.database.operations.DatabaseCommandDto;3import org.evomaster.client.java.controller.api.dto.database.operations.InsertionDto;4import org.evomaster.client.java.controller.api.dto.database.schema.DatabaseType;5import org.evomaster.client.java.controller.api.dto.database.schema.TableDto;6import org.evomaster.client.java.controller.api.dto.database.schema.TableIndexDto;7import org.evomaster.client.java.controller.api.dto.database.schema.TableSchemaDto;8import org.evomaster.client.java.controller.internal.db.SqlScriptRunner;9import org.evomaster.client.java.controller.internal.db.SqlScriptRunnerImpl;10import org.evomaster.client.java.controller.internal.db.schema.SqlScriptExecutor;11import org.evomaster.client.java.controller.internal.db.schema.SqlScriptExecutorImpl;12import org.evomaster.client.java.controller.internal.db.schema.SqlScriptResult;13import org.evomaster.client.java.controller.internal.db.schema.SqlScriptRunnerFactory;14import org.evomaster.client.java.controller.internal.db.schema.SqlScriptRunnerFactoryImpl;15import org.evomaster.client.java.controller.internal.db.schema.TableSchemaExtractor;16import org.evomaster.client.java.controller.internal.db.schema.TableSchemaExtractorFactory;17import org.evomaster.client.java.controller.internal.db.schema.TableSchemaExtractorFactoryImpl;18import org.evomaster.client.java.controller.internal.db.schema.TableSchemaExtractorImpl;19import org.evomaster.client.java.controller.internal.db.schema.TableSchemaInformationDto;20import org.evomaster.client.java.controller.internal.db.schema.TableSchemaInformationDtoImpl;21import org.evomaster.client.java.controller.internal.db.schema.TableSchemaInformationExtractor;22import org.evomaster.client.java.controller.internal.db.schema.TableSchemaInformationExtractorFactory;23import org.evomaster.client.java.controller.internal.db.schema.TableSchemaInformationExtractorFactoryImpl;24import org.evomaster.client.java.controller.internal.db.schema.TableSchemaInformationExtractorImpl;25import org.evomaster.client.java.controller.internal.db.schema.TableSchemaInformationExtractorPostgresImpl;26import org.evomaster.client.java.controller.internal.db.schema.TableSchemaInformationExtractorSqliteImpl;27import org.evomaster.client.java.controller.internal.db.schema.TableSchemaInformationExtractorSqlserverImpl;28import org.junit.jupiter.api.BeforeEach;29import org.junit.jupiter.api.Test;30import org.mockito.Mockito;31import java.sql.Connection;32import java.sql.SQLException;33import java.util.ArrayList;34import java.util.Arrays;35import java.util.Collections;36import java.util.List;37import java.util.Map;38import java.util.Optional;39import java.util.Set

initSqlScriptAndGetInsertMap

Using AI Code Generation

1import org.evomaster.client.java.controller.internal.SutController;2import org.evomaster.client.java.controller.db.SqlScriptRunner;3import org.evomaster.client.java.controller.db.SqlInsertionBuilder;4import java.io.File;5import java.io.IOException;6import java.io.FileWriter;7import java.io.BufferedWriter;8import java.util.Map;9import java.util.HashMap;10import java.lang.ClassLoader;11import java.net.URL;12import java.net.URLClassLoader;13import java.util.ArrayList;14import java.util.List;15import java.util.Arrays;16import java.util.Set;17import java.util.HashSet;18public class SqlScriptRunnerTest {19    public static void main(String[] args) throws Exception {20        String scriptFileName = "script.sql";21        String insertMapFileName = "insertMap.txt";22        String jarFileName = "evomaster-client-java.jar";23        String className = "org.evomaster.client.java.controller.internal.SutController";24        String methodName = "initSqlScriptAndGetInsertMap";25        String parameterName = "scriptFileName";26        String parameterName2 = "insertMapFileName";27        String jarFilePath = System.getProperty("user.dir") + File.separator + jarFileName;28        String scriptFilePath = System.getProperty("user.dir") + File.separator + scriptFileName;29        String insertMapFilePath = System.getProperty("user.dir") + File.separator

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.