How to use trackRequestSource method of org.evomaster.client.java.controller.internal.EMController class

Best EvoMaster code snippet using org.evomaster.client.java.controller.internal.EMController.trackRequestSource

Source:EMController.java

...43    private static final Set<String> connectedClientsSoFar = new CopyOnWriteArraySet<>();44    public EMController(SutController sutController) {45        this.sutController = Objects.requireNonNull(sutController);46    }47    private boolean trackRequestSource(HttpServletRequest request){48        String source = request.getRemoteAddr() + ":" + request.getRemotePort();49        connectedClientsSoFar.add(source);50        return true;51    }52    /**53     * Only used for debugging/testing54     *55     * @return host:port of all clients connected so far56     */57    public static Set<String> getConnectedClientsSoFar() {58        return connectedClientsSoFar;59    }60    /**61     * Only used debugging/testing62     */63    public static void resetConnectedClientsSoFar(){64        connectedClientsSoFar.clear();65    }66    @Path(ControllerConstants.INFO_SUT_PATH)67    @GET68    public Response getSutInfo(@Context HttpServletRequest httpServletRequest) {69        String connectionHeader = httpServletRequest.getHeader("Connection");70        if( connectionHeader == null71                || !connectionHeader.equalsIgnoreCase("keep-alive")){72            return Response.status(400).entity(WrappedResponseDto73                    .withError("Requests should always contain a 'Connection: keep-alive'")).build();74        }75        assert trackRequestSource(httpServletRequest);76        if(! sutController.verifySqlConnection()){77            String msg = "SQL drivers are misconfigured. You must use a 'p6spy' wrapper when you " +78                    "run the SUT. For example, a database connection URL like 'jdbc:h2:mem:testdb' " +79                    "should be changed into 'jdbc:p6spy:h2:mem:testdb'. " +80                    "See documentation on how to configure P6Spy.";81            SimpleLogger.error(msg);82            return Response.status(500).entity(WrappedResponseDto.withError(msg)).build();83        }84        SutInfoDto dto = new SutInfoDto();85        dto.isSutRunning = sutController.isSutRunning();86        dto.baseUrlOfSUT = baseUrlOfSUT;87        dto.infoForAuthentication = sutController.getInfoForAuthentication();88        dto.sqlSchemaDto = sutController.getSqlDatabaseSchema();89        dto.defaultOutputFormat = sutController.getPreferredOutputFormat();90        ProblemInfo info = sutController.getProblemInfo();91        if (info == null) {92            String msg = "Undefined problem type in the EM Controller";93            SimpleLogger.error(msg);94            return Response.status(500).entity(WrappedResponseDto.withError(msg)).build();95        } else if (info instanceof RestProblem) {96            RestProblem rp = (RestProblem) info;97            dto.restProblem = new RestProblemDto();98            dto.restProblem.swaggerJsonUrl = rp.getSwaggerJsonUrl();99            dto.restProblem.endpointsToSkip = rp.getEndpointsToSkip();100        } else {101            String msg = "Unrecognized problem type: " + info.getClass().getName();102            SimpleLogger.error(msg);103            return Response.status(500).entity(WrappedResponseDto.withError(msg)).build();104        }105        dto.unitsInfoDto = sutController.getUnitsInfoDto();106        if(dto.unitsInfoDto == null){107            String msg = "Failed to extract units info";108            SimpleLogger.error(msg);109            return Response.status(500).entity(WrappedResponseDto.withError(msg)).build();110        }111        return Response.status(200).entity(WrappedResponseDto.withData(dto)).build();112    }113    @Path(ControllerConstants.CONTROLLER_INFO)114    @GET115    public Response getControllerInfoDto(@Context HttpServletRequest httpServletRequest) {116        assert trackRequestSource(httpServletRequest);117        ControllerInfoDto dto = new ControllerInfoDto();118        dto.fullName = sutController.getClass().getName();119        dto.isInstrumentationOn = sutController.isInstrumentationActivated();120        return Response.status(200).entity(WrappedResponseDto.withData(dto)).build();121    }122    @Path(ControllerConstants.NEW_SEARCH)123    @POST124    public Response newSearch(@Context HttpServletRequest httpServletRequest) {125        assert trackRequestSource(httpServletRequest);126        sutController.newSearch();127        return Response.status(201).entity(WrappedResponseDto.withNoData()).build();128    }129    @Path(ControllerConstants.RUN_SUT_PATH)130    @PUT131    @Consumes(Formats.JSON_V1)132    public Response runSut(SutRunDto dto, @Context HttpServletRequest httpServletRequest) {133        assert trackRequestSource(httpServletRequest);134        try {135            if (dto.run == null) {136                String msg = "Invalid JSON: 'run' field is required";137                SimpleLogger.warn(msg);138                return Response.status(400).entity(WrappedResponseDto.withError(msg)).build();139            }140            boolean sqlHeuristics = dto.calculateSqlHeuristics != null && dto.calculateSqlHeuristics;141            boolean sqlExecution = dto.extractSqlExecutionInfo != null && dto.extractSqlExecutionInfo;142            sutController.enableComputeSqlHeuristicsOrExtractExecution(sqlHeuristics, sqlExecution);143            boolean doReset = dto.resetState != null && dto.resetState;144            synchronized (this) {145                if (!dto.run) {146                    if (doReset) {147                        String msg = "Invalid JSON: cannot reset state and stop service at same time";148                        SimpleLogger.warn(msg);149                        return Response.status(400).entity(WrappedResponseDto.withError(msg)).build();150                    }151                    //if on, we want to shut down the server152                    if (sutController.isSutRunning()) {153                        sutController.stopSut();154                        baseUrlOfSUT = null;155                    }156                } else {157                    /*158                        If SUT is not up and running, let's start it159                     */160                    if (!sutController.isSutRunning()) {161                        baseUrlOfSUT = sutController.startSut();162                        if (baseUrlOfSUT == null) {163                            //there has been an internal failure in starting the SUT164                            String msg = "Internal failure: cannot start SUT based on given configuration";165                            SimpleLogger.warn(msg);166                            return Response.status(500).entity(WrappedResponseDto.withError(msg)).build();167                        }168                        sutController.initSqlHandler();169                    } else {170                        //TODO as starting should be blocking, need to check171                        //if initialized, and wait if not172                    }173                    /*174                        regardless of where it was running or not, need to reset state.175                        this is controlled by a boolean, although most likely we ll always176                        want to do it177                     */178                    if (dto.resetState != null && dto.resetState) {179                        sutController.resetStateOfSUT();180                        sutController.newTest();181                    }182                    /*183                        Note: here even if we start the SUT, the starting of a "New Search"184                        cannot be done here, as in this endpoint we also deal with the reset185                        of state. When we reset state for a new test run, we do not want to186                        reset all the other data regarding the whole search187                     */188                }189            }190        } catch (RuntimeException e) {191            /*192                FIXME: ideally, would not need to do a try/catch on each single endpoint,193                as could configure Jetty/Jackson to log all errors.194                But even after spending hours googling it, haven't managed to configure it195             */196            String msg = e.getMessage();197            SimpleLogger.error(msg, e);198            return Response.status(500).entity(WrappedResponseDto.withError(msg)).build();199        }200        return Response.status(204).entity(WrappedResponseDto.withNoData()).build();201    }202    @Path(ControllerConstants.TEST_RESULTS)203    @GET204    public Response getTestResults(205            @QueryParam("ids")206            @DefaultValue("")207                    String idList,208            @Context HttpServletRequest httpServletRequest) {209        assert trackRequestSource(httpServletRequest);210        try {211            TestResultsDto dto = new TestResultsDto();212            Set<Integer> ids;213            try {214                ids = Arrays.stream(idList.split(","))215                        .filter(s -> !s.trim().isEmpty())216                        .map(Integer::parseInt)217                        .collect(Collectors.toSet());218            } catch (NumberFormatException e) {219                String msg = "Invalid parameter 'ids': " + e.getMessage();220                SimpleLogger.warn(msg);221                return Response.status(400).entity(WrappedResponseDto.withError(msg)).build();222            }223            List<TargetInfo> targetInfos = sutController.getTargetInfos(ids);224            if (targetInfos == null) {225                String msg = "Failed to collect target information for " + ids.size() + " ids";226                SimpleLogger.error(msg);227                return Response.status(500).entity(WrappedResponseDto.withError(msg)).build();228            }229            targetInfos.forEach(t -> {230                TargetInfoDto info = new TargetInfoDto();231                info.id = t.mappedId;232                info.value = t.value;233                info.descriptiveId = t.descriptiveId;234                info.actionIndex = t.actionIndex;235                dto.targets.add(info);236            });237            List<AdditionalInfo> additionalInfos = sutController.getAdditionalInfoList();238            if (additionalInfos != null) {239                additionalInfos.forEach(a -> {240                    AdditionalInfoDto info = new AdditionalInfoDto();241                    info.queryParameters = new HashSet<>(a.getQueryParametersView());242                    info.headers = new HashSet<>(a.getHeadersView());243                    info.lastExecutedStatement = a.getLastExecutedStatement();244                    info.stringSpecializations = new HashMap<>();245                    for(Map.Entry<String, Set<StringSpecializationInfo>> entry :246                            a.getStringSpecializationsView().entrySet()){247                        assert ! entry.getValue().isEmpty();248                        List<StringSpecializationInfoDto> list = entry.getValue().stream()249                                .map(it -> new StringSpecializationInfoDto(250                                        it.getStringSpecialization().toString(),251                                        it.getValue(),252                                        it.getType().toString()))253                                .collect(Collectors.toList());254                        info.stringSpecializations.put(entry.getKey(), list);255                    }256                    dto.additionalInfoList.add(info);257                });258            } else {259                String msg = "Failed to collect additional info";260                SimpleLogger.error(msg);261                return Response.status(500).entity(WrappedResponseDto.withError(msg)).build();262            }263            dto.extraHeuristics = sutController.getExtraHeuristics();264            return Response.status(200).entity(WrappedResponseDto.withData(dto)).build();265        } catch (RuntimeException e) {266            /*267                FIXME: ideally, would not need to do a try/catch on each single endpoint,268                as could configure Jetty/Jackson to log all errors.269                But even after spending hours googling it, haven't managed to configure it270             */271            String msg = "Thrown exception: " + e.getMessage();272            SimpleLogger.error(msg, e);273            return Response.status(500).entity(WrappedResponseDto.withError(msg)).build();274        }275    }276    @Path(ControllerConstants.NEW_ACTION)277    @Consumes(MediaType.APPLICATION_JSON)278    @PUT279    public Response newAction(ActionDto dto, @Context HttpServletRequest httpServletRequest) {280        assert trackRequestSource(httpServletRequest);281        sutController.newAction(dto);282        return Response.status(204).entity(WrappedResponseDto.withNoData()).build();283    }284    @Path(ControllerConstants.DATABASE_COMMAND)285    @Consumes(Formats.JSON_V1)286    @POST287    public Response executeDatabaseCommand(DatabaseCommandDto dto, @Context HttpServletRequest httpServletRequest) {288        assert trackRequestSource(httpServletRequest);289        try {290            Connection connection = sutController.getConnection();291            if (connection == null) {292                String msg = "No active database connection";293                SimpleLogger.warn(msg);294                return Response.status(400).entity(WrappedResponseDto.withError(msg)).build();295            }296            if (dto.command == null && (dto.insertions == null || dto.insertions.isEmpty())) {297                String msg = "No input command";298                SimpleLogger.warn(msg);299                return Response.status(400).entity(WrappedResponseDto.withError(msg)).build();300            }301            if (dto.command != null && dto.insertions != null && !dto.insertions.isEmpty()) {302                String msg = "Only 1 command can be specified";...

trackRequestSource

Using AI Code Generation

1import org.evomaster.client.java.controller.api.EMTestUtils;2import org.evomaster.client.java.controller.api.dto.SutInfoDto;3import org.evomaster.client.java.controller.api.dto.TestResultsDto;4import org.evomaster.client.java.controller.internal.EMController;5import org.evomaster.client.java.controller.internal.SutHandler;6import org.evomaster.client.java.controller.internal.SutInfo;7import org.evomaster.client.java.controller.internal.SutStateHandler;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.SqlScriptWriter;11import org.evomaster.client.java.controller.internal.db.SqlScriptWriterImpl;12import org.evomaster.client.java.controller.internal.db.h2.H2Controller;13import org.evomaster.client.java.controller.internal.db.h2.H2EmbeddedServer;14import org.evomaster.client.java.controller.internal.db.h2.H2JdbcUrlParser;15import org.evomaster.client.java.controller.internal.db.h2.H2Schema;16import org.evomaster.client.java.controller.internal.db.sql.SqlScriptExecutor;17import org.evomaster.client.java.controller.internal.db.sql.SqlScriptExecutorImpl;18import org.evomaster.client.java.controller.internal.db.sql.SqlScriptExtractor;19import org.evomaster.client.java.controller.internal.db.sql.SqlScriptExtractorImpl;20import org.evomaster.client.java.controller.internal.db.sql.schema.SchemaExtractor;21import org.evomaster.client.java.controller.internal.db.sql.schema.SchemaExtractorImpl;22import org.evomaster.client.java.controller.internal.db.sql.schema.SchemaExtractorJdbc;23import org.evomaster.client.java.controller.internal.db.sql.schema.SchemaExtractorJdbcImpl;24import org.evomaster.client.java.controller.internal.db.sql.schema.SchemaExtractorPostgres;25import org.evomaster.client.java.controller.internal.db.sql.schema.SchemaExtractorPostgresImpl;26import org.evomaster.client.java.controller.internal.db.sql.schema.SchemaExtractorSqlite;27import org.evomaster.client.java.controller.internal.db.sql.schema.SchemaExtractorSqliteImpl;28import org.evomaster.client.java.controller.internal.db.sql.schema.SchemaExtractorSqlServer;29import org.evomaster.client.java.controller.internal.db.sql.schema.SchemaExtractorSqlServerImpl;30import org.evomaster.client.java.controller.internal.db.sql.schema.SchemaExtractorSybase;31import org.evomaster.client.java.controller

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.