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.mockito.Mockito.mock;
  24 import static org.mockito.Mockito.when;
  25
  26 import io.netty.handler.codec.http.HttpResponseStatus;
  27 import java.util.HashMap;
  28 import java.util.Map;
  29 import java.util.UUID;
  30 import org.junit.jupiter.api.Test;
  31 import org.onap.dcaegen2.services.sdk.rest.services.model.DmaapModel;
  32 import org.onap.dcaegen2.services.sdk.rest.services.model.JsonBodyBuilder;
  33 import org.onap.dcaegen2.services.sdk.rest.services.model.logging.ImmutableRequestDiagnosticContext;
  34 import org.onap.dcaegen2.services.sdk.rest.services.model.logging.RequestDiagnosticContext;
  35 import reactor.core.publisher.Mono;
  36 import reactor.netty.DisposableServer;
  37 import reactor.netty.http.client.HttpClient;
  38 import reactor.netty.http.server.HttpServer;
  39 import reactor.netty.resources.ConnectionProvider;
  40 import reactor.test.StepVerifier;
  41
  42 class CloudHttpClientIT {
  43     private static final int MAX_CONNECTIONS = 1;
  44     private static final String SAMPLE_STRING = "sampleString";
  45     private static final String SAMPLE_URL = "/sampleURL";
  46     private static final String JSON_BODY = "{\"correlationId\":\"NOKnhfsadhff\","
  47         + "\"ipaddress-v4\":\"256.22.33.155\", "
  48         + "\"ipaddress-v6\":\"200J:0db8:85a3:0000:0000:8a2e:0370:7334\"}";
  49     private static final ConnectionProvider connectionProvider = ConnectionProvider.fixed("test", MAX_CONNECTIONS);
  50
  51     private DmaapModel dmaapModel = mock(DmaapModel.class);
  52     private JsonBodyBuilder<DmaapModel> jsonBodyBuilder = mock(JsonBodyBuilder.class);
  53
  54     @Test
  55     void successfulPatchResponse() {
  56         DisposableServer server = createValidServer();
  57         HttpClient httpClient = createHttpClientForContextWithAddress(server, connectionProvider);
  58         CloudHttpClient cloudHttpClient = new CloudHttpClient(httpClient);
  59
  60         when(jsonBodyBuilder.createJsonBody(dmaapModel)).thenReturn(JSON_BODY);
  61         Mono<Integer> content = cloudHttpClient.patch(SAMPLE_URL, createRequestDiagnosticContext(), createCustomHeaders(),
  62             jsonBodyBuilder, dmaapModel);
  63
  64         StepVerifier.create(content)
  65             .expectNext(HttpResponseStatus.OK.code())
  66             .expectComplete()
  67             .verify();
  68         server.disposeNow();
  69     }
  70
  71     @Test
  72     void errorPatchRequest() {
  73         DisposableServer server = createInvalidServer();
  74         HttpClient httpClient = createHttpClientForContextWithAddress(server, connectionProvider);
  75         CloudHttpClient cloudHttpClient = new CloudHttpClient(httpClient);
  76
  77         when(jsonBodyBuilder.createJsonBody(dmaapModel)).thenReturn(JSON_BODY);
  78         Mono<Integer> content = cloudHttpClient.patch(SAMPLE_URL, createRequestDiagnosticContext(), createCustomHeaders(),
  79             jsonBodyBuilder, dmaapModel);
  80
  81         StepVerifier.create(content)
  82             .expectNext(HttpResponseStatus.INTERNAL_SERVER_ERROR.code())
  83             .expectComplete()
  84             .verify();
  85         server.disposeNow();
  86     }
  87
  88     @Test
  89     void successfulGetResponse() {
  90         DisposableServer server = createValidServer();
  91         HttpClient httpClient = createHttpClientForContextWithAddress(server, connectionProvider);
  92         CloudHttpClient cloudHttpClient = new CloudHttpClient(httpClient);
  93
  94         Mono<String> content = cloudHttpClient.get(SAMPLE_URL, String.class);
  95
  96         StepVerifier.create(content)
  97             .expectNext(SAMPLE_STRING)
  98             .expectComplete()
  99             .verify();
 100         server.disposeNow();
 101     }
 102     @Test
 103     void errorGetRequest() {
 104         DisposableServer server = createInvalidServer();
 105         HttpClient httpClient = createHttpClientForContextWithAddress(server, connectionProvider);
 106         CloudHttpClient cloudHttpClient = new CloudHttpClient(httpClient);
 107
 108         Mono<String> content = cloudHttpClient.get(SAMPLE_URL, String.class);
 109
 110         StepVerifier.create(content)
 111             .expectError()
 112             .verify();
 113         server.disposeNow();
 114     }
 115
 116     private Map<String, String> createCustomHeaders() {
 117         Map<String, String> customHeaders = new HashMap<>();
 118         customHeaders.put("X_INVOCATION_ID", UUID.randomUUID().toString());
 119
 120         return customHeaders;
 121     }
 122
 123     private DisposableServer createValidServer() {
 124         Mono<String> response = Mono.just(SAMPLE_STRING);
 125         return HttpServer.create()
 126             .handle((req, resp) -> resp.sendString(response))
 127             .wiretap(true)
 128             .bindNow();
 129     }
 130
 131     private DisposableServer createInvalidServer() {
 132         return HttpServer.create()
 133             .handle((req, resp) -> Mono.error(new Exception("returnError")))
 134             .wiretap(true)
 135             .bindNow();
 136     }
 137
 138     private RequestDiagnosticContext createRequestDiagnosticContext() {
 139         return ImmutableRequestDiagnosticContext.builder()
 140             .invocationId(UUID.randomUUID()).requestId(UUID.randomUUID()).build();
 141     }
 142
 143     private HttpClient createHttpClientForContextWithAddress(DisposableServer disposableServer,
 144         ConnectionProvider connectionProvider) {
 145         HttpClient client = connectionProvider == null? HttpClient.create() : HttpClient.create(connectionProvider);
 146         return client.addressSupplier(disposableServer::address).wiretap(true);
 147     }
 148 }