NONRTRIC - ECS updates of the NBI
[nonrtric.git] / enrichment-coordinator-service / src / test / java / org / oransc / enrichment / controller / ConsumerSimulatorController.java
1 /*-
2  * ========================LICENSE_START=================================
3  * O-RAN-SC
4  * %%
5  * Copyright (C) 2020 Nordix Foundation
6  * %%
7  * Licensed under the Apache License, Version 2.0 (the "License");
8  * you may not use this file except in compliance with the License.
9  * You may obtain a copy of the License at
10  *
11  *      http://www.apache.org/licenses/LICENSE-2.0
12  *
13  * Unless required by applicable law or agreed to in writing, software
14  * distributed under the License is distributed on an "AS IS" BASIS,
15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
16  * See the License for the specific language governing permissions and
17  * limitations under the License.
18  * ========================LICENSE_END===================================
19  */
20
21 package org.oransc.enrichment.controller;
22
23 import io.swagger.v3.oas.annotations.Operation;
24 import io.swagger.v3.oas.annotations.media.Content;
25 import io.swagger.v3.oas.annotations.media.Schema;
26 import io.swagger.v3.oas.annotations.responses.ApiResponse;
27 import io.swagger.v3.oas.annotations.responses.ApiResponses;
28 import io.swagger.v3.oas.annotations.tags.Tag;
29
30 import java.lang.invoke.MethodHandles;
31 import java.util.ArrayList;
32 import java.util.Collections;
33 import java.util.List;
34
35 import lombok.Getter;
36
37 import org.oransc.enrichment.controllers.VoidResponse;
38 import org.oransc.enrichment.controllers.a1e.A1eConsts;
39 import org.oransc.enrichment.controllers.a1e.A1eEiJobStatus;
40 import org.slf4j.Logger;
41 import org.slf4j.LoggerFactory;
42 import org.springframework.http.HttpStatus;
43 import org.springframework.http.MediaType;
44 import org.springframework.http.ResponseEntity;
45 import org.springframework.web.bind.annotation.PathVariable;
46 import org.springframework.web.bind.annotation.PostMapping;
47 import org.springframework.web.bind.annotation.RequestBody;
48 import org.springframework.web.bind.annotation.RestController;
49
50 @RestController("ConsumerSimulatorController")
51 @Tag(name = A1eConsts.CONSUMER_API_CALLBACKS_NAME)
52 public class ConsumerSimulatorController {
53
54     private final Logger logger = LoggerFactory.getLogger(MethodHandles.lookup().lookupClass());
55
56     public static class TestResults {
57
58         public List<A1eEiJobStatus> status = Collections.synchronizedList(new ArrayList<A1eEiJobStatus>());
59
60         public void reset() {
61             status.clear();
62         }
63     }
64
65     @Getter
66     private TestResults testResults = new TestResults();
67
68     public static String getJobStatusUrl(String infoJobId) {
69         return "/example_dataconsumer/info_jobs/" + infoJobId + "/status";
70     }
71
72     @PostMapping(
73         path = "/example_dataconsumer/info_jobs/{infoJobId}/status",
74         produces = MediaType.APPLICATION_JSON_VALUE)
75     @Operation(
76         summary = "Callback for changed Information Job status",
77         description = "The primitive is implemented by the data consumer and is invoked when a Information Job status has been changed.")
78     @ApiResponses(
79         value = { //
80             @ApiResponse(
81                 responseCode = "200",
82                 description = "OK", //
83                 content = @Content(schema = @Schema(implementation = VoidResponse.class))) //
84         })
85     public ResponseEntity<Object> jobStatusCallback( //
86         @PathVariable("infoJobId") String infoJobId, //
87         @RequestBody A1eEiJobStatus status) {
88         logger.info("Job status callback status: {} infoJobId: {}", status.state, infoJobId);
89         this.testResults.status.add(status);
90         return new ResponseEntity<>(HttpStatus.OK);
91     }
92
93 }