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

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

Source:SutController.java

...293                        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                }408            });409        }410        return schemaDto;411    }412    /**413     *414     * @return a map from the name of interface to extracted interface415     */416    public final Map<String, InterfaceSchema> getRPCSchema(){417        return rpcInterfaceSchema;418    }419    /**420     *421     * @return a map of auth local method422     */423    public Map<Integer, LocalAuthSetupSchema> getLocalAuthSetupSchemaMap() {424        return localAuthSetupSchemaMap;425    }426    /**427     * extract endpoints info of the RPC interface by reflection based on the specified service interface name428     */429    @Override430    public final void extractRPCSchema(){431        if (objectMapper == null)432            objectMapper = new ObjectMapper();433        if (!rpcInterfaceSchema.isEmpty())434            return;435        if (!(getProblemInfo() instanceof RPCProblem)){436            SimpleLogger.error("Problem ("+getProblemInfo().getClass().getSimpleName()+") is not RPC but request RPC schema.");437            return;438        }439        try {440            RPCEndpointsBuilder.validateCustomizedValueInRequests(getCustomizedValueInRequests());441            RPCEndpointsBuilder.validateCustomizedNotNullAnnotationForRPCDto(specifyCustomizedNotNullAnnotation());442            RPCProblem rpcp = (RPCProblem) getProblemInfo();443            for (String interfaceName: rpcp.getMapOfInterfaceAndClient()){444                InterfaceSchema schema = RPCEndpointsBuilder.build(interfaceName, rpcp.getType(), rpcp.getClient(interfaceName),445                        rpcp.skipEndpointsByName!=null? rpcp.skipEndpointsByName.get(interfaceName):null,446                        rpcp.skipEndpointsByAnnotation!=null?rpcp.skipEndpointsByAnnotation.get(interfaceName):null,447                        rpcp.involveEndpointsByName!=null? rpcp.involveEndpointsByName.get(interfaceName):null,448                        rpcp.involveEndpointsByAnnotation!=null? rpcp.involveEndpointsByAnnotation.get(interfaceName):null,449                        getInfoForAuthentication(),450                        getCustomizedValueInRequests(),451                        specifyCustomizedNotNullAnnotation());452                rpcInterfaceSchema.put(interfaceName, schema);453            }454            localAuthSetupSchemaMap.clear();455            Map<Integer, LocalAuthSetupSchema> local = RPCEndpointsBuilder.buildLocalAuthSetup(getInfoForAuthentication());456            if (local!=null && !local.isEmpty())457                localAuthSetupSchemaMap.putAll(local);458        }catch (Exception e){459//            SimpleLogger.error("Failed to extract the RPC Schema: " + e.getMessage());460            throw new RuntimeException("Failed to extract the RPC Schema: " + e.getMessage());461        }462    }463    /**464     * parse seeded tests for RPC465     * @return a list of tests, and each test is a list of RCPActionDto466     */467    public List<List<RPCActionDto>> handleSeededTests(){468        if (seedRPCTests() == null || seedRPCTests().isEmpty()) return null;469        if (rpcInterfaceSchema.isEmpty())470            throw new IllegalStateException("empty RPC interface: The RPC interface schemas are not extracted yet");471        List<List<RPCActionDto>> results = new ArrayList<>();472        for (SeededRPCTestDto dto: seedRPCTests()){473            if (dto.rpcFunctions != null && !dto.rpcFunctions.isEmpty()){474                List<RPCActionDto> test = new ArrayList<>();475                for (SeededRPCActionDto actionDto : dto.rpcFunctions){476                    InterfaceSchema schema = rpcInterfaceSchema.get(actionDto.interfaceName);477                    if (schema != null){478                        EndpointSchema actionSchema = schema.getOneEndpointWithSeededDto(actionDto);479                        if (actionSchema != null){480                            EndpointSchema copy = actionSchema.copyStructure();481                            for (int i = 0; i < copy.getRequestParams().size(); i++){482                                // TODO need to check if generic type could be handled with jackson483                                NamedTypedValue p = copy.getRequestParams().get(i);484                                try {485                                    String stringValue = actionDto.inputParams.get(i);486//                                    Object value = objectMapper.readValue(stringValue, p.getType().getClazz());487                                    p.setValueBasedOnInstanceOrJson(stringValue);488                                } catch (JsonProcessingException e) {489                                    throw new IllegalStateException(490                                            String.format("Seeded Test Error: cannot parse the seeded test %s at the parameter %d with error msg: %s", actionDto, i, e.getMessage()));491                                }492                            }493                            test.add(copy.getDto());494                        }else {495                            throw new IllegalStateException("Seeded Test Error: cannot find the action "+actionDto.functionName);496                        }497                    } else {498                        throw new IllegalStateException("Seeded Test Error: cannot find the interface "+ actionDto.interfaceName);499                    }500                }501                results.add(test);502            } else {503                SimpleLogger.warn("Seeded Test: empty RPC function calls for the test "+ dto.testName);504            }505        }506        return results;507    }508    /**509     * Either there is no connection, or, if there is, then it must have P6Spy configured.510     * But this might not apply to all kind controllers511     *512     * @return false if the verification failed513     */514    @Deprecated515    public final boolean verifySqlConnection(){516        return true;517//        Connection connection = getConnection();518//        if(connection == null519//                //check does not make sense for External520//                || !(this instanceof EmbeddedSutController)){521//            return true;522//        }523//524//        /*525//            bit hacky/brittle, but seems there is no easy way to check if a connection is526//            using P6Spy.527//            However, the name of driver's package would appear when doing a toString on it528//         */529//        String info = connection.toString();530//531//        return info.contains("p6spy");532    }533    /**534     * Re-initialize all internal data to enable a completely new search phase535     * which should be independent from previous ones536     */537    public abstract void newSearch();538    /**539     * Re-initialize some internal data needed before running a new test540     */541    public final void newTest() {542        actionIndex = -1;543        resetExtraHeuristics();544        extras.clear();545        //clean all accessed table in a test546        accessedTables.clear();547        newTestSpecificHandler();548        // set executingAction state false for newTest549        setExecutingAction(false);550    }551    /**552     * As some heuristics are based on which action (eg HTTP call, or click of button)553     * in the test sequence is executed, and their order, we need to keep track of which554     * action does cover what.555     *556     * @param dto the DTO with the information about the action (eg its index in the test)557     */558    public final void newAction(ActionDto dto) {559        if (dto.index > extras.size()) {560            extras.add(computeExtraHeuristics());561        }562        this.actionIndex = dto.index;563        resetExtraHeuristics();564        newActionSpecificHandler(dto);565    }566    public final void executeHandleLocalAuthenticationSetup(RPCActionDto dto, ActionResponseDto responseDto){567        LocalAuthSetupSchema endpointSchema = new LocalAuthSetupSchema();568        endpointSchema.setValue(dto);569        handleLocalAuthenticationSetup(endpointSchema.getAuthenticationInfo());570        if (dto.responseVariable != null && dto.doGenerateTestScript){571            responseDto.testScript = endpointSchema.newInvocationWithJava(dto.responseVariable, dto.controllerVariable,dto.clientVariable);572        }573    }574    /**575     * execute a RPC request based on the specified dto576     * @param dto is the action DTO to be executed577     */578    public final void executeAction(RPCActionDto dto, ActionResponseDto responseDto) {579        EndpointSchema endpointSchema = getEndpointSchema(dto);580        if (dto.responseVariable != null && dto.doGenerateTestScript){581            try{582                responseDto.testScript = endpointSchema.newInvocationWithJava(dto.responseVariable, dto.controllerVariable,dto.clientVariable);583            }catch (Exception e){584                SimpleLogger.warn("Fail to generate test script"+e.getMessage());585            }586            if (responseDto.testScript ==null)587                SimpleLogger.warn("Null test script for action "+dto.actionName);588        }589        Object response;590        try {591            response = executeRPCEndpoint(dto, false);592        } catch (Exception e) {593            throw new RuntimeException("ERROR: target exception should be caught, but "+ e.getMessage());594        }595        //handle exception596        if (response instanceof Exception){597            try{598                RPCExceptionHandler.handle(response, responseDto, endpointSchema, getRPCType(dto));599                return;600            } catch (Exception e){601                SimpleLogger.error("ERROR: fail to handle exception instance to dto "+ e.getMessage());602                //throw new RuntimeException("ERROR: fail to handle exception instance to dto "+ e.getMessage());603            }604        }605        if (endpointSchema.getResponse() != null){606            if (response != null){607                try{608                    // successful execution609                    NamedTypedValue resSchema = endpointSchema.getResponse().copyStructureWithProperties();610                    resSchema.setValueBasedOnInstance(response);611                    responseDto.rpcResponse = resSchema.getDto();612                    if (dto.doGenerateAssertions && dto.responseVariable != null)613                        responseDto.assertionScript = resSchema.newAssertionWithJava(dto.responseVariable, dto.maxAssertionForDataInCollection);614                    else615                        responseDto.jsonResponse = objectMapper.writeValueAsString(response);616                } catch (Exception e){617                    SimpleLogger.error("ERROR: fail to set successful response instance value to dto "+ e.getMessage());618                    //throw new RuntimeException("ERROR: fail to set successful response instance value to dto "+ e.getMessage());619                }620                try {621                    responseDto.customizedCallResultCode = categorizeBasedOnResponse(response);622                } catch (Exception e){623                    SimpleLogger.error("ERROR: fail to categorize result with implemented categorizeBasedOnResponse "+ e.getMessage());624                    //throw new RuntimeException("ERROR: fail to categorize result with implemented categorizeBasedOnResponse "+ e.getMessage());625                }626            }627        }628    }629    private Object executeRPCEndpoint(RPCActionDto dto, boolean throwTargetException) throws Exception {630        Object client = ((RPCProblem)getProblemInfo()).getClient(dto.interfaceId);631        EndpointSchema endpointSchema = getEndpointSchema(dto);632        return executeRPCEndpointCatchTargetException(client, endpointSchema, throwTargetException);633    }634    private Object executeRPCEndpointCatchTargetException(Object client, EndpointSchema endpoint, boolean throwTargetException) throws Exception {635        Object res;636        try {637            res = executeRPCEndpoint(client, endpoint);638        } catch (ClassNotFoundException | NoSuchMethodException | IllegalAccessException e) {639            throw new RuntimeException("EM RPC REQUEST EXECUTION ERROR: fail to process a RPC request with "+ e.getMessage());640        } catch (InvocationTargetException e) {641            if (throwTargetException)642                throw (Exception) e.getTargetException();643            else644                res = e.getTargetException();645        } catch (Exception e){646            SimpleLogger.error("ERROR: other exception exists "+ e.getMessage());647            if (throwTargetException) throw e;648            else res = e;649        }650        return res;651    }652    @Override653    public Object executeRPCEndpoint(String json) throws Exception{654        try {655            RPCActionDto dto = objectMapper.readValue(json, RPCActionDto.class);656            return executeRPCEndpoint(dto, true);657        } catch (JsonProcessingException e) {658            SimpleLogger.error("Failed to extract the json: " + e.getMessage());659        }660        return null;661    }662    /**663     * execute a RPC request with specified client664     * @param client is the client to execute the endpoint665     * @param endpoint is the endpoint to be executed666     */667    private final Object executeRPCEndpoint(Object client, EndpointSchema endpoint) throws NoSuchMethodException, InvocationTargetException, IllegalAccessException, ClassNotFoundException {668        if (endpoint.getRequestParams().isEmpty()){669            Method method = client.getClass().getDeclaredMethod(endpoint.getName());670            return method.invoke(client);671        }672        Object[] params = new Object[endpoint.getRequestParams().size()];673        Class<?>[] types = new Class<?>[endpoint.getRequestParams().size()];674        try{675            for (int i = 0; i < params.length; i++){676                NamedTypedValue param = endpoint.getRequestParams().get(i);677                params[i] = param.newInstance();678                types[i] = param.getType().getClazz();679            }680        } catch (Exception e){681            throw new RuntimeException("ERROR: fail to instance value of input parameters based on dto/schema, msg error:"+e.getMessage());682        }683        Method method = client.getClass().getDeclaredMethod(endpoint.getName(), types);684        return method.invoke(client, params);685    }686    private EndpointSchema getEndpointSchema(RPCActionDto dto){687        InterfaceSchema interfaceSchema = rpcInterfaceSchema.get(dto.interfaceId);688        EndpointSchema endpointSchema = interfaceSchema.getOneEndpoint(dto).copyStructure();689        endpointSchema.setValue(dto);690        return endpointSchema;691    }692    private RPCType getRPCType(RPCActionDto dto){693        return rpcInterfaceSchema.get(dto.interfaceId).getRpcType();694    }695    public abstract void newTestSpecificHandler();696    public abstract void newActionSpecificHandler(ActionDto dto);697    /**698     * Check if bytecode instrumentation is on.699     *700     * @return true if the instrumentation is on701     */702    public abstract boolean isInstrumentationActivated();703    /**704     * <p>705     * Check if the system under test (SUT) is running and fully initialized706     * </p>707     *708     * <p>709     * How to implement this method depends on the library/framework used710     * to build the application.711     * In Spring applications, this can be done with something like:712     * {@code ctx != null && ctx.isRunning()}, where {@code ctx} is a field where713     * {@code ConfigurableApplicationContext} should be stored when starting714     * the application.715     * </p>716     * @return true if the SUT is running717     */718    public abstract boolean isSutRunning();719    /**720     * <p>721     * A "," separated list of package prefixes or class names.722     * For example, "com.foo.,com.bar.Bar".723     * This is used to specify for which classes we want to measure724     * code coverage.725     * </p>726     *727     * <p>728     * Note: be careful of using something as general as "com."729     * or "org.", as most likely ALL your third-party libraries730     * would be instrumented as well, which could have a severe731     * impact on performance.732     * </p>733     *734     * @return a String representing the packages to cover735     */736    public abstract String getPackagePrefixesToCover();737    /**738     * <p>739     * If the application uses some sort of authentication, these details740     * need to be provided here.741     * Even if EvoMaster can have access to the database, it would not be able742     * to recover hashed passwords.743     * </p>744     *745     * <p>746     * To test the application, there is the need to provide auth for at least 1 user747     * (and more if they have different authorization roles).748     * When EvoMaster generates test cases, it can decide to use the credential of749     * any user provided by this method.750     * </p>751     *752     * <p>753     * What type of info to provide here depends on the auth mechanism, e.g.,754     * Basic or cookie-based (using {@link CookieLoginDto}).755     * To simplify the creation of these DTOs with auth info, you can look756     * at {@link org.evomaster.client.java.controller.AuthUtils}.757     * </p>758     *759     * <p>760     * If the credential are stored in a database, be careful on how the761     * method {@code resetStateOfSUT} is implemented.762     * If you delete all data with {@link DbCleaner}, then you will need as well to763     * recreate the auth details.764     * This can be put in a script, executed then with {@link SqlScriptRunner}.765     * </p>766     *767     * @return a list of valid authentication credentials, or {@code null} if768     *      * none is necessary769     */770    public abstract List<AuthenticationDto> getInfoForAuthentication();771    /**772     * <p>773     * If the system under test (SUT) uses a SQL database, we need to have a774     * configured connection to access it.775     * </p>776     *777     * <p>778     * This method is related to {@link SutHandler#resetStateOfSUT}.779     * When accessing a {@code Connection} object to reset the state of780     * the application, we suggest to save it to field (eg when starting the781     * application), and return such field here, e.g., {@code return connection;}.782     * This connection object will be used by EvoMaster to analyze the state of783     * the database to create better test cases.784     * </p>785     *786     * @return {@code null} if the SUT does not use any SQL database787     */788    @Deprecated789    public Connection getConnection(){790        throw new IllegalStateException("This deprecated method should never be called");791    }792    /**793     * If the system under test (SUT) uses a SQL database, we need to specify794     * the driver used to connect, eg. {@code org.h2.Driver}.795     * This is needed for when we intercept SQL commands with P6Spy796     *797     * @return {@code null} if the SUT does not use any SQL database798     * @deprecated this method is no longer needed799     */800    @Deprecated801    public String getDatabaseDriverName(){802        throw new IllegalStateException("This deprecated method should never be called");803    }804    public abstract List<TargetInfo> getTargetInfos(Collection<Integer> ids);805    /**806     * @return additional info for each action in the test.807     * The list is ordered based on the action index.808     */809    public abstract List<AdditionalInfo> getAdditionalInfoList();810    /**811     * <p>812     * Depending of which kind of SUT we are dealing with (eg, REST, GraphQL or SPA frontend),813     * there is different info that must be provided.814     * For example, in a RESTful API, you need to speficy where the OpenAPI/Swagger schema815     * is located.816     * </p>817     *818     * <p>819     * The interface {@link ProblemInfo} provides different implementations, like820     * {@code RestProblem}.821     * You will need to instantiate one of such classes, and return it here in this method.822     * </p>823     * @return an instance of object with all the needed data for the specific addressed problem824     */825    public abstract ProblemInfo getProblemInfo();826    /**827     * Test cases could be outputted in different language (e.g., Java and Kotlin),828     * using different testing libraries (e.g., JUnit 4 or 5).829     * Here, need to specify the default option.830     *831     * @return the format in which the test cases should be generated832     */833    public abstract SutInfoDto.OutputFormat getPreferredOutputFormat();834    public abstract UnitsInfoDto getUnitsInfoDto();835    public abstract void setKillSwitch(boolean b);836    public abstract void setExecutingInitSql(boolean executingInitSql);837    public abstract void setExecutingAction(boolean executingAction);838    public abstract BootTimeInfoDto getBootTimeInfoDto();839    protected BootTimeInfoDto getBootTimeInfoDto(BootTimeObjectiveInfo info){840        if (info == null)841            return null;842        BootTimeInfoDto infoDto = new BootTimeInfoDto();843        infoDto.targets = info.getObjectiveCoverageAtSutBootTime()844                .entrySet().stream().map(e-> new TargetInfoDto(){{845                    descriptiveId = e.getKey();846                    value = e.getValue();847                }}).collect(Collectors.toList());848        infoDto.externalServicesDto = info.getExternalServiceInfo().stream()849                .map(e -> new ExternalServiceInfoDto(e.getProtocol(), e.getHostname(), e.getRemotePort()))850                .collect(Collectors.toList());851        return infoDto;852    }853    public abstract String getExecutableFullPath();854    protected UnitsInfoDto getUnitsInfoDto(UnitsInfoRecorder recorder){855        if(recorder == null){856            return null;857        }858        UnitsInfoDto dto = new UnitsInfoDto();859        dto.numberOfBranches = recorder.getNumberOfBranches();860        dto.numberOfLines = recorder.getNumberOfLines();861        dto.numberOfReplacedMethodsInSut = recorder.getNumberOfReplacedMethodsInSut();862        dto.numberOfReplacedMethodsInThirdParty = recorder.getNumberOfReplacedMethodsInThirdParty();863        dto.numberOfTrackedMethods = recorder.getNumberOfTrackedMethods();864        dto.unitNames = recorder.getUnitNames();865        dto.parsedDtos = recorder.getParsedDtos();866        dto.numberOfInstrumentedNumberComparisons = recorder.getNumberOfInstrumentedNumberComparisons();867        return dto;868    }869    @Override870    public Object getRPCClient(String interfaceName) {871        if (!(getProblemInfo() instanceof RPCProblem))872            throw new RuntimeException("ERROR: the problem should be RPC but it is "+ getProblemInfo().getClass().getSimpleName());873        Object client = ((RPCProblem) getProblemInfo()).getClient(interfaceName);874        if (client == null)875            throw new RuntimeException("ERROR: cannot find any client with the name :"+ interfaceName);876        return client;877    }878    @Override879    public CustomizedCallResultCode categorizeBasedOnResponse(Object response) {880        return null;881    }882    @Override883    public List<CustomizedRequestValueDto> getCustomizedValueInRequests() {884        return null;885    }886    @Override887    public List<CustomizedNotNullAnnotationForRPCDto> specifyCustomizedNotNullAnnotation() {888        return null;889    }890    @Override891    public List<SeededRPCTestDto> seedRPCTests() {892        return null;893    }894    @Override895    public void resetDatabase(List<String> tablesToClean) {896        if (getDbSpecifications()!= null && !getDbSpecifications().isEmpty()){897            getDbSpecifications().forEach(spec->{898                if (spec==null || spec.connection == null || !spec.employSmartDbClean){899                    return;900                }901                if (spec.schemaNames == null || spec.schemaNames.isEmpty())902                    DbCleaner.clearDatabase(spec.connection, null, null, tablesToClean, spec.dbType);903                else904                    spec.schemaNames.forEach(sp-> DbCleaner.clearDatabase(spec.connection, sp, null, tablesToClean, spec.dbType));905                try {906                    handleInitSql(tablesToClean, spec);907                } catch (SQLException e) {908                    throw new RuntimeException("Fail to execute the specified initSqlScript "+e);909                }910            });911        }912    }913}...

handleInitSql

Using AI Code Generation

1    public static void handleInitSql(String sql) {2        try {3            Class<?> clazz = Class.forName("org.evomaster.client.java.controller.internal.SutController");4            Method method = clazz.getDeclaredMethod("handleInitSql", String.class);5            method.invoke(null, sql);6        } catch (Exception e) {7            throw new IllegalStateException("Failed to call handleInitSql method of org.evomaster.client.java.controller.internal.SutController class", e);8        }9    }10    public static void handleInitSql(String sql) {11            try (Statement statement = connection.createStatement()) {12                statement.execute(sql);13            }14        } catch (SQLException e) {15            throw new IllegalStateException("Failed to execute sql", e);16        }17    }18    public static void handleInitSql(String sql) {19            try (Statement statement = connection.createStatement()) {20                statement.execute(sql);21            }22        } catch (SQLException e) {23            throw new IllegalStateException("Failed to execute sql", e);24        }25    }26    public static void handleInitSql(String sql) {27            try (Statement statement = connection.createStatement()) {28                statement.execute(sql);29            }30        } catch (SQLException e) {31            throw new IllegalStateException("Failed to execute sql", e);32        }33    }34    public static void handleInitSql(String sql) {35            try (Statement statement = connection.createStatement()) {36                statement.execute(sql);37            }38        } catch (SQLException e) {39            throw new IllegalStateException("Failed to execute sql", e);40        }41    }42    public static void handleInitSql(String sql) {43            try (Statement statement = connection.createStatement()) {44                statement.execute(sql);45            }46        } catch (SQLException e) {47            throw new IllegalStateException("Failed to execute sql", e);

handleInitSql

Using AI Code Generation

1    public void testInitSql() throws Exception {2        String sql = "INSERT INTO table1 VALUES (1, 'a', 'b'), (2, 'c', 'd'), (3, 'e', 'f');";3        sutController.handleInitSql(sql);4        assertEquals(3, sutController.getDatabase().getNumberOfRows("table1"));5        assertEquals(3, sutController.getDatabase().getNumberOfRows("table1", "id", 1));6        assertEquals(1, sutController.getDatabase().getNumberOfRows("table1", "id", 2));7        assertEquals(1, sutController.getDatabase().getNumberOfRows("table1", "id", 3));8        assertEquals(1, sutController.getDatabase().getNumberOfRows("table1", "name", "a"));9        assertEquals(1, sutController.getDatabase().getNumberOfRows("table1", "name", "c"));10        assertEquals(1, sutController.getDatabase().getNumberOfRows("table1", "name", "e"));11        assertEquals(1, sutController.getDatabase().getNumberOfRows("table1", "surname", "b"));12        assertEquals(1, sutController.getDatabase().getNumberOfRows("table1", "surname", "d"));13        assertEquals(1, sutController.getDatabase().getNumberOfRows("table1", "surname", "f"));14    }15}16The testInitSql() method uses the handleInitSql() method of the SutController class to execute the SQL statement. The SutController class is part of

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.