gerrit.onap Code Review - dcaegen2/services/sdk.git/blob

   1 /*
   2  * ============LICENSE_START=======================================================
   3  * DCAEGEN2-SERVICES-SDK
   4  * ================================================================================
   5  * Copyright (C) 2018 NOKIA Intellectual Property. All rights reserved.
   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.onap.dcaegen2.services.sdk.rest.services.adapters.http;
  22
  23 import static org.junit.jupiter.api.Assertions.assertEquals;
  24 import static org.mockito.Mockito.mock;
  25 import static org.mockito.Mockito.when;
  26
  27 import io.netty.handler.codec.http.HttpResponseStatus;
  28 import java.util.HashMap;
  29 import java.util.Map;
  30 import java.util.UUID;
  31 import org.junit.jupiter.api.Test;
  32 import org.onap.dcaegen2.services.sdk.rest.services.model.DmaapModel;
  33 import org.onap.dcaegen2.services.sdk.rest.services.model.JsonBodyBuilder;
  34 import org.onap.dcaegen2.services.sdk.rest.services.model.logging.ImmutableRequestDiagnosticContext;
  35 import org.onap.dcaegen2.services.sdk.rest.services.model.logging.RequestDiagnosticContext;
  36 import reactor.core.publisher.Mono;
  37 import reactor.netty.DisposableServer;
  38 import reactor.netty.http.client.HttpClient;
  39 import reactor.netty.http.server.HttpServer;
  40 import reactor.netty.resources.ConnectionProvider;
  41 import reactor.test.StepVerifier;
  42 import reactor.netty.http.client.HttpClientResponse;
  43
  44 class CloudHttpClientIT {
  45
  46     private static final int MAX_CONNECTIONS = 1;
  47     private static final String SAMPLE_STRING = "sampleString";
  48     private static final String SAMPLE_URL = "/sampleURL";
  49     private static final String JSON_BODY = "{\"correlationId\":\"NOKnhfsadhff\","
  50             + "\"ipaddress-v4\":\"256.22.33.155\", "
  51             + "\"ipaddress-v6\":\"200J:0db8:85a3:0000:0000:8a2e:0370:7334\"}";
  52     private static final ConnectionProvider connectionProvider = ConnectionProvider.fixed("test", MAX_CONNECTIONS);
  53
  54     private final DmaapModel dmaapModel = mock(DmaapModel.class);
  55     private final JsonBodyBuilder<DmaapModel> jsonBodyBuilder = mock(JsonBodyBuilder.class);
  56
  57     @Test
  58     void successfulPatchResponse() {
  59         DisposableServer server = createValidServer();
  60         RxHttpClient httpClient = createHttpClientForContextWithAddress(server);
  61         CloudHttpClient cloudHttpClient = new CloudHttpClient(httpClient);
  62
  63         when(jsonBodyBuilder.createJsonBody(dmaapModel)).thenReturn(JSON_BODY);
  64         Mono<HttpResponse> content = cloudHttpClient
  65                 .patch(SAMPLE_URL, createRequestDiagnosticContext(), createCustomHeaders(),
  66                         jsonBodyBuilder, dmaapModel);
  67         HttpResponse httpClientResponse = content.block();
  68
  69         assertEquals(HttpResponseStatus.OK.code(), httpClientResponse.statusCode());
  70         server.disposeNow();
  71     }
  72
  73     @Test
  74     void errorPatchRequest() {
  75         DisposableServer server = createInvalidServer();
  76         RxHttpClient httpClient = createHttpClientForContextWithAddress(server);
  77         CloudHttpClient cloudHttpClient = new CloudHttpClient(httpClient);
  78
  79         when(jsonBodyBuilder.createJsonBody(dmaapModel)).thenReturn(JSON_BODY);
  80         Mono<HttpResponse> content = cloudHttpClient
  81                 .patch(SAMPLE_URL, createRequestDiagnosticContext(), createCustomHeaders(),
  82                         jsonBodyBuilder, dmaapModel);
  83         HttpResponse httpClientResponse = content.block();
  84
  85         assertEquals(HttpResponseStatus.INTERNAL_SERVER_ERROR.code(), httpClientResponse.statusCode());
  86         server.disposeNow();
  87     }
  88
  89     @Test
  90     void successfulPostResponse() {
  91         DisposableServer server = createValidServer();
  92         RxHttpClient httpClient = createHttpClientForContextWithAddress(server);
  93         CloudHttpClient cloudHttpClient = new CloudHttpClient(httpClient);
  94
  95         when(jsonBodyBuilder.createJsonBody(dmaapModel)).thenReturn(JSON_BODY);
  96         Mono<HttpResponse> content = cloudHttpClient
  97                 .post(SAMPLE_URL, createRequestDiagnosticContext(), createCustomHeaders(),
  98                         jsonBodyBuilder, dmaapModel);
  99         HttpResponse httpClientResponse = content.block();
 100
 101         assertEquals(HttpResponseStatus.OK.code(), httpClientResponse.statusCode());
 102         server.disposeNow();
 103     }
 104
 105     @Test
 106     void errorPostRequest() {
 107         DisposableServer server = createInvalidServer();
 108         RxHttpClient httpClient = createHttpClientForContextWithAddress(server);
 109         CloudHttpClient cloudHttpClient = new CloudHttpClient(httpClient);
 110
 111         when(jsonBodyBuilder.createJsonBody(dmaapModel)).thenReturn(JSON_BODY);
 112         Mono<HttpResponse> content = cloudHttpClient
 113                 .post(SAMPLE_URL, createRequestDiagnosticContext(), createCustomHeaders(),
 114                         jsonBodyBuilder, dmaapModel);
 115         HttpResponse httpClientResponse = content.block();
 116
 117         assertEquals(HttpResponseStatus.INTERNAL_SERVER_ERROR.code(), httpClientResponse.statusCode());
 118         server.disposeNow();
 119     }
 120
 121     @Test
 122     void successfulGetResponse() {
 123         DisposableServer server = createValidServer();
 124         RxHttpClient httpClient = createHttpClientForContextWithAddress(server);
 125         CloudHttpClient cloudHttpClient = new CloudHttpClient(httpClient);
 126
 127         when(jsonBodyBuilder.createJsonBody(dmaapModel)).thenReturn(JSON_BODY);
 128         Mono<String> content = cloudHttpClient.get(SAMPLE_URL, String.class);
 129         Mono<String> contentWithHeaders = cloudHttpClient.get(SAMPLE_URL, createRequestDiagnosticContext(),
 130                 createCustomHeaders(), String.class);
 131
 132         StepVerifier.create(content)
 133                 .expectNext(SAMPLE_STRING)
 134                 .expectComplete()
 135                 .verify();
 136         StepVerifier.create(contentWithHeaders)
 137                 .expectNext(SAMPLE_STRING)
 138                 .expectComplete()
 139                 .verify();
 140         server.disposeNow();
 141     }
 142
 143     @Test
 144     void errorGetRequest() {
 145         DisposableServer server = createInvalidServer();
 146         RxHttpClient httpClient = createHttpClientForContextWithAddress(server);
 147         CloudHttpClient cloudHttpClient = new CloudHttpClient(httpClient);
 148
 149         Mono<String> content = cloudHttpClient.get(SAMPLE_URL, String.class);
 150
 151         StepVerifier.create(content)
 152                 .expectError()
 153                 .verify();
 154         server.disposeNow();
 155     }
 156
 157     @Test
 158     void successfulPutResponse() {
 159         DisposableServer server = createValidServer();
 160         RxHttpClient httpClient = createHttpClientForContextWithAddress(server);
 161         CloudHttpClient cloudHttpClient = new CloudHttpClient(httpClient);
 162
 163         when(jsonBodyBuilder.createJsonBody(dmaapModel)).thenReturn(JSON_BODY);
 164         Mono<HttpResponse> content = cloudHttpClient
 165             .put(SAMPLE_URL, createRequestDiagnosticContext(), createCustomHeaders(),
 166                 jsonBodyBuilder, dmaapModel);
 167         HttpResponse httpClientResponse = content.block();
 168
 169         assertEquals(HttpResponseStatus.OK.code(), httpClientResponse.statusCode());
 170         server.disposeNow();
 171     }
 172
 173     @Test
 174     void errorPutRequest() {
 175         DisposableServer server = createInvalidServer();
 176         RxHttpClient httpClient = createHttpClientForContextWithAddress(server);
 177         CloudHttpClient cloudHttpClient = new CloudHttpClient(httpClient);
 178
 179         when(jsonBodyBuilder.createJsonBody(dmaapModel)).thenReturn(JSON_BODY);
 180         Mono<HttpResponse> content = cloudHttpClient
 181             .put(SAMPLE_URL, createRequestDiagnosticContext(), createCustomHeaders(),
 182                 jsonBodyBuilder, dmaapModel);
 183         HttpResponse httpClientResponse = content.block();
 184
 185         assertEquals(HttpResponseStatus.INTERNAL_SERVER_ERROR.code(), httpClientResponse.statusCode());
 186         server.disposeNow();
 187     }
 188
 189     private Map<String, String> createCustomHeaders() {
 190         Map<String, String> customHeaders = new HashMap<>();
 191         customHeaders.put("X_INVOCATION_ID", UUID.randomUUID().toString());
 192         return customHeaders;
 193     }
 194
 195     private DisposableServer createValidServer() {
 196         Mono<String> response = Mono.just(SAMPLE_STRING);
 197         return HttpServer.create()
 198                 .handle((req, resp) -> resp.sendString(response))
 199                 .wiretap(true)
 200                 .bindNow();
 201     }
 202
 203     private DisposableServer createInvalidServer() {
 204         return HttpServer.create()
 205                 .handle((req, resp) -> Mono.error(new Exception("returnError")))
 206                 .wiretap(true)
 207                 .bindNow();
 208     }
 209
 210     private RequestDiagnosticContext createRequestDiagnosticContext() {
 211         return ImmutableRequestDiagnosticContext.builder()
 212                 .invocationId(UUID.randomUUID()).requestId(UUID.randomUUID()).build();
 213     }
 214
 215     private RxHttpClient createHttpClientForContextWithAddress(DisposableServer disposableServer) {
 216         HttpClient client = HttpClient.create(connectionProvider)
 217                 .addressSupplier(disposableServer::address)
 218                 .wiretap(true);
 219         return new RxHttpClient(client);
 220     }
 221 }