AAI-1523 Batch reformat aai-schema-ingest
[aai/aai-common.git] / aai-schema-ingest / src / test / java / org / onap / aai / restclient / MockRestClient.java
index 87e4bfe..52e6364 100644 (file)
  * limitations under the License.
  * ============LICENSE_END=========================================================
  */
+
 package org.onap.aai.restclient;
 
+import static org.junit.Assert.assertNotNull;
+import static org.springframework.test.web.client.match.MockRestRequestMatchers.*;
+import static org.springframework.test.web.client.response.MockRestResponseCreators.withStatus;
+
 import com.att.eelf.configuration.EELFLogger;
 import com.google.gson.JsonArray;
 import com.google.gson.JsonObject;
 import com.google.gson.JsonParser;
-import org.apache.commons.io.IOUtils;
-import org.springframework.core.io.Resource;
-import org.springframework.http.*;
-import org.springframework.stereotype.Component;
-import org.springframework.test.web.client.ExpectedCount;
-import org.springframework.test.web.client.MockRestServiceServer;
-import org.springframework.util.MultiValueMap;
-import org.springframework.web.client.RestTemplate;
 
 import java.io.IOException;
 import java.io.InputStream;
@@ -40,33 +37,40 @@ import java.util.Collections;
 import java.util.List;
 import java.util.Map;
 
-import static org.junit.Assert.assertNotNull;
-import static org.springframework.test.web.client.match.MockRestRequestMatchers.*;
-import static org.springframework.test.web.client.response.MockRestResponseCreators.withStatus;
+import org.apache.commons.io.IOUtils;
+import org.springframework.core.io.Resource;
+import org.springframework.http.*;
+import org.springframework.stereotype.Component;
+import org.springframework.test.web.client.ExpectedCount;
+import org.springframework.test.web.client.MockRestServiceServer;
+import org.springframework.util.MultiValueMap;
+import org.springframework.web.client.RestTemplate;
 
 @Component
 public class MockRestClient extends RestClient {
 
-private RestTemplate restTemplate;
-        private MockRestServiceServer mockRestServiceServer;
+    private RestTemplate restTemplate;
+    private MockRestServiceServer mockRestServiceServer;
 
     String fileName = "mockrequests";
 
     public MockRestClient(String fileName) {
         /*
-        List<MockRestServiceServer> mockedAAIRequests = new ArrayList<>(aaiRequests.size());
+         * List<MockRestServiceServer> mockedAAIRequests = new ArrayList<>(aaiRequests.size());
          */
         List<MockRestServiceServer> mockedAAIRequests = new ArrayList<>();
 
         restTemplate = new RestTemplate();
-           /* MockRestServiceServer server       = MockRestServiceServer
-                .bindTo(restClientFactory.getRestClient(ClientType.SchemaService).getRestTemplate())
-                .build();
-            server.expect(MockRestRequestMatchers.requestTo(url))
-                .andRespond(withSuccess("{}", MediaType.APPLICATION_JSON));*/
+        /*
+         * MockRestServiceServer server = MockRestServiceServer
+         * .bindTo(restClientFactory.getRestClient(ClientType.SchemaService).getRestTemplate())
+         * .build();
+         * server.expect(MockRestRequestMatchers.requestTo(url))
+         * .andRespond(withSuccess("{}", MediaType.APPLICATION_JSON));
+         */
 
         // RestTemplateBuilder mockBuilder = mock(RestTemplateBuilder.class);
-        //when(mockBuilder.build()).thenReturn(restTemplate);
+        // when(mockBuilder.build()).thenReturn(restTemplate);
 
         JsonObject payload = null;
         try {
@@ -77,23 +81,20 @@ private RestTemplate restTemplate;
 
         JsonArray mockUris = payload.getAsJsonArray("mock-uri");
 
-
         mockRestServiceServer = MockRestServiceServer.createServer(restTemplate);
         String url = "https://localhost:8447/aai/v14";
-        /*mockRestServiceServer.expect(requestTo(url))
-            .andRespond(withSuccess("{}", MediaType.APPLICATION_JSON));*/
-
+        /*
+         * mockRestServiceServer.expect(requestTo(url))
+         * .andRespond(withSuccess("{}", MediaType.APPLICATION_JSON));
+         */
 
         for (int i = 0; i < mockUris.size(); i++) {
             String responseFile = mockUris.get(i).getAsJsonObject().get("response-file").getAsString();
             String contentTypeValue = mockUris.get(i).getAsJsonObject().get("content").getAsString();
 
-
             String uri = mockUris.get(i).getAsJsonObject().get("aai-uri").getAsString();
 
-            InputStream inputStream = getClass()
-                .getClassLoader()
-                .getResourceAsStream(responseFile);
+            InputStream inputStream = getClass().getClassLoader().getResourceAsStream(responseFile);
             String responseBody = null;
             try {
                 responseBody = IOUtils.toString(inputStream, StandardCharsets.UTF_8);
@@ -101,57 +102,51 @@ private RestTemplate restTemplate;
                 e.printStackTrace();
             }
 
-
             mockRestServiceServer.expect(ExpectedCount.manyTimes(), requestTo(url + uri))
-                .andExpect(method(HttpMethod.GET))
-                .andExpect(content().contentType(contentTypeValue))
-                .andRespond(withStatus(HttpStatus.OK).body(responseBody.toString()).contentType(MediaType.valueOf(contentTypeValue)));
-
+                    .andExpect(method(HttpMethod.GET)).andExpect(content().contentType(contentTypeValue))
+                    .andRespond(withStatus(HttpStatus.OK).body(responseBody.toString())
+                            .contentType(MediaType.valueOf(contentTypeValue)));
 
         }
     }
 
-    public MockRestClient()  {
+    public MockRestClient() {
 
-         restTemplate = new RestTemplate();
-           /* MockRestServiceServer server       = MockRestServiceServer
-                .bindTo(restClientFactory.getRestClient(ClientType.SchemaService).getRestTemplate())
-                .build();
-            server.expect(MockRestRequestMatchers.requestTo(url))
-                .andRespond(withSuccess("{}", MediaType.APPLICATION_JSON));*/
+        restTemplate = new RestTemplate();
+        /*
+         * MockRestServiceServer server = MockRestServiceServer
+         * .bindTo(restClientFactory.getRestClient(ClientType.SchemaService).getRestTemplate())
+         * .build();
+         * server.expect(MockRestRequestMatchers.requestTo(url))
+         * .andRespond(withSuccess("{}", MediaType.APPLICATION_JSON));
+         */
 
         // RestTemplateBuilder mockBuilder = mock(RestTemplateBuilder.class);
-        //when(mockBuilder.build()).thenReturn(restTemplate);
+        // when(mockBuilder.build()).thenReturn(restTemplate);
 
         JsonObject payload = null;
         try {
-            payload = getPayload( fileName + ".json");
+            payload = getPayload(fileName + ".json");
         } catch (IOException e) {
             e.printStackTrace();
         }
 
         JsonArray mockUris = payload.getAsJsonArray("mock-uri");
 
-
-
-        mockRestServiceServer       = MockRestServiceServer.createServer(restTemplate);
-        String url="https://localhost:8447/aai/v14";
-        /*mockRestServiceServer.expect(requestTo(url))
-            .andRespond(withSuccess("{}", MediaType.APPLICATION_JSON));*/
-
-
-
+        mockRestServiceServer = MockRestServiceServer.createServer(restTemplate);
+        String url = "https://localhost:8447/aai/v14";
+        /*
+         * mockRestServiceServer.expect(requestTo(url))
+         * .andRespond(withSuccess("{}", MediaType.APPLICATION_JSON));
+         */
 
         for (int i = 0; i < mockUris.size(); i++) {
             String responseFile = mockUris.get(i).getAsJsonObject().get("response-file").getAsString();
             String contentTypeValue = mockUris.get(i).getAsJsonObject().get("content").getAsString();
 
-
             String uri = mockUris.get(i).getAsJsonObject().get("aai-uri").getAsString();
 
-            InputStream inputStream = getClass()
-                .getClassLoader()
-                .getResourceAsStream(responseFile);
+            InputStream inputStream = getClass().getClassLoader().getResourceAsStream(responseFile);
             String responseBody = null;
             try {
                 responseBody = IOUtils.toString(inputStream, StandardCharsets.UTF_8);
@@ -159,31 +154,26 @@ private RestTemplate restTemplate;
                 e.printStackTrace();
             }
 
-
             mockRestServiceServer.expect(ExpectedCount.manyTimes(), requestTo(url + uri))
-                .andExpect(method(HttpMethod.GET))
-                .andExpect(content().contentType(contentTypeValue))
-                .andRespond(withStatus(HttpStatus.OK).body(responseBody.toString()).contentType(MediaType.valueOf(contentTypeValue)));
-
+                    .andExpect(method(HttpMethod.GET)).andExpect(content().contentType(contentTypeValue))
+                    .andRespond(withStatus(HttpStatus.OK).body(responseBody.toString())
+                            .contentType(MediaType.valueOf(contentTypeValue)));
 
         }
 
-
     }
 
-    public JsonObject  getTestDetails(String fileName) throws IOException {
+    public JsonObject getTestDetails(String fileName) throws IOException {
 
-        JsonObject payload = getPayload(fileName );
+        JsonObject payload = getPayload(fileName);
 
         return payload;
     }
 
     public JsonObject getPayload(String filename) throws IOException {
-        InputStream inputStream = getClass()
-            .getClassLoader()
-            .getResourceAsStream(filename);
+        InputStream inputStream = getClass().getClassLoader().getResourceAsStream(filename);
 
-        //InputStream inputStream = new FileInputStream(filename);
+        // InputStream inputStream = new FileInputStream(filename);
 
         String result = IOUtils.toString(inputStream, StandardCharsets.UTF_8);
         String message = String.format("Unable to find the %s in src/test/resources", filename);
@@ -195,28 +185,33 @@ private RestTemplate restTemplate;
     }
 
     @Override
-        public ResponseEntity execute(String uri, HttpMethod method, Map<String,String> headers, String body) {
+    public ResponseEntity execute(String uri, HttpMethod method, Map<String, String> headers, String body) {
 
-            String url="https://localhost:8447/aai/v14/"+ uri;
+        String url = "https://localhost:8447/aai/v14/" + uri;
 
-           /* MockRestServiceServer server       = MockRestServiceServer
-                .bindTo(restClientFactory.getRestClient(ClientType.SchemaService).getRestTemplate())
-                .build();
-            server.expect(MockRestRequestMatchers.requestTo(url))
-                .andRespond(withSuccess("{}", MediaType.APPLICATION_JSON));*/
-
-           // RestTemplateBuilder mockBuilder = mock(RestTemplateBuilder.class);
-            //when(mockBuilder.build()).thenReturn(restTemplate);
+        /*
+         * MockRestServiceServer server = MockRestServiceServer
+         * .bindTo(restClientFactory.getRestClient(ClientType.SchemaService).getRestTemplate())
+         * .build();
+         * server.expect(MockRestRequestMatchers.requestTo(url))
+         * .andRespond(withSuccess("{}", MediaType.APPLICATION_JSON));
+         */
 
-        /*MockRestServiceServer server       = MockRestServiceServer.createServer(restTemplate);
-        server.expect(requestTo(url))
-            .andRespond(withSuccess("{}", MediaType.APPLICATION_JSON));
-            return new ResponseEntity("blah", HttpStatus.OK);
-        server.expect(ExpectedCount.manyTimes(), requestTo(Matchers.startsWith(aaiBaseUrl + aaiRequests.get(i).get("aai-uri").asText())))
-            .andExpect(method(HttpMethod.GET))
-            .andExpect(content().contentType(MediaType.APPLICATION_JSON))
-            .andRespond(withStatus(HttpStatus.OK).body(aaiResponses.get(i).toString()).contentType(MediaType.APPLICATION_JSON));*/
+        // RestTemplateBuilder mockBuilder = mock(RestTemplateBuilder.class);
+        // when(mockBuilder.build()).thenReturn(restTemplate);
 
+        /*
+         * MockRestServiceServer server = MockRestServiceServer.createServer(restTemplate);
+         * server.expect(requestTo(url))
+         * .andRespond(withSuccess("{}", MediaType.APPLICATION_JSON));
+         * return new ResponseEntity("blah", HttpStatus.OK);
+         * server.expect(ExpectedCount.manyTimes(), requestTo(Matchers.startsWith(aaiBaseUrl +
+         * aaiRequests.get(i).get("aai-uri").asText())))
+         * .andExpect(method(HttpMethod.GET))
+         * .andExpect(content().contentType(MediaType.APPLICATION_JSON))
+         * .andRespond(withStatus(HttpStatus.OK).body(aaiResponses.get(i).toString()).contentType(MediaType.
+         * APPLICATION_JSON));
+         */
 
         HttpHeaders headersMap = new HttpHeaders();
 
@@ -228,35 +223,40 @@ private RestTemplate restTemplate;
 
         HttpEntity httpEntity = new HttpEntity(headers);
 
-        ResponseEntity responseEntity = restTemplate.exchange(url , HttpMethod.GET, httpEntity, String.class);
+        ResponseEntity responseEntity = restTemplate.exchange(url, HttpMethod.GET, httpEntity, String.class);
 
-       // mockRestServiceServer.verify();
-       return responseEntity;
+        // mockRestServiceServer.verify();
+        return responseEntity;
     }
 
     @Override
-    public ResponseEntity executeResource(String uri, HttpMethod method, Map<String,String> headers, String body) {
+    public ResponseEntity executeResource(String uri, HttpMethod method, Map<String, String> headers, String body) {
 
-        String url="https://localhost:8447/aai/v14/"+ uri;
+        String url = "https://localhost:8447/aai/v14/" + uri;
 
-           /* MockRestServiceServer server       = MockRestServiceServer
-                .bindTo(restClientFactory.getRestClient(ClientType.SchemaService).getRestTemplate())
-                .build();
-            server.expect(MockRestRequestMatchers.requestTo(url))
-                .andRespond(withSuccess("{}", MediaType.APPLICATION_JSON));*/
+        /*
+         * MockRestServiceServer server = MockRestServiceServer
+         * .bindTo(restClientFactory.getRestClient(ClientType.SchemaService).getRestTemplate())
+         * .build();
+         * server.expect(MockRestRequestMatchers.requestTo(url))
+         * .andRespond(withSuccess("{}", MediaType.APPLICATION_JSON));
+         */
 
         // RestTemplateBuilder mockBuilder = mock(RestTemplateBuilder.class);
-        //when(mockBuilder.build()).thenReturn(restTemplate);
-
-        /*MockRestServiceServer server       = MockRestServiceServer.createServer(restTemplate);
-        server.expect(requestTo(url))
-            .andRespond(withSuccess("{}", MediaType.APPLICATION_JSON));
-            return new ResponseEntity("blah", HttpStatus.OK);
-        server.expect(ExpectedCount.manyTimes(), requestTo(Matchers.startsWith(aaiBaseUrl + aaiRequests.get(i).get("aai-uri").asText())))
-            .andExpect(method(HttpMethod.GET))
-            .andExpect(content().contentType(MediaType.APPLICATION_JSON))
-            .andRespond(withStatus(HttpStatus.OK).body(aaiResponses.get(i).toString()).contentType(MediaType.APPLICATION_JSON));*/
+        // when(mockBuilder.build()).thenReturn(restTemplate);
 
+        /*
+         * MockRestServiceServer server = MockRestServiceServer.createServer(restTemplate);
+         * server.expect(requestTo(url))
+         * .andRespond(withSuccess("{}", MediaType.APPLICATION_JSON));
+         * return new ResponseEntity("blah", HttpStatus.OK);
+         * server.expect(ExpectedCount.manyTimes(), requestTo(Matchers.startsWith(aaiBaseUrl +
+         * aaiRequests.get(i).get("aai-uri").asText())))
+         * .andExpect(method(HttpMethod.GET))
+         * .andExpect(content().contentType(MediaType.APPLICATION_JSON))
+         * .andRespond(withStatus(HttpStatus.OK).body(aaiResponses.get(i).toString()).contentType(MediaType.
+         * APPLICATION_JSON));
+         */
 
         HttpHeaders headersMap = new HttpHeaders();
 
@@ -268,7 +268,7 @@ private RestTemplate restTemplate;
 
         HttpEntity httpEntity = new HttpEntity(headers);
 
-        ResponseEntity responseEntity = restTemplate.exchange(url , HttpMethod.GET, httpEntity, Resource.class);
+        ResponseEntity responseEntity = restTemplate.exchange(url, HttpMethod.GET, httpEntity, Resource.class);
 
         // mockRestServiceServer.verify();
         return responseEntity;
@@ -280,16 +280,16 @@ private RestTemplate restTemplate;
         return restTemplate;
     }
 
-    public  String getBaseUrl(){
-            return "";
+    public String getBaseUrl() {
+        return "";
     }
 
-    protected  MultiValueMap<String,String> getHeaders(Map<String,String> headers){
-            return null;
+    protected MultiValueMap<String, String> getHeaders(Map<String, String> headers) {
+        return null;
     }
 
-    protected  EELFLogger getLogger(){
-            return null;
+    protected EELFLogger getLogger() {
+        return null;
     }
 
 }