rest-services/http-client/src/main/java/org/onap/dcaegen2/services/sdk/rest/services/adapters/http/test/DummyHttpServer.java

   1 /*
   2  * ============LICENSE_START====================================
   3  * DCAEGEN2-SERVICES-SDK
   4  * =========================================================
   5  * Copyright (C) 2019 Nokia. 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.test;
  22
  23 import io.vavr.CheckedFunction0;
  24 import java.net.URL;
  25 import java.nio.file.Files;
  26 import java.nio.file.Paths;
  27 import java.util.concurrent.atomic.AtomicInteger;
  28 import java.util.function.Consumer;
  29 import org.reactivestreams.Publisher;
  30 import org.slf4j.Logger;
  31 import org.slf4j.LoggerFactory;
  32 import reactor.core.publisher.Mono;
  33 import reactor.netty.DisposableServer;
  34 import reactor.netty.http.server.HttpServer;
  35 import reactor.netty.http.server.HttpServerResponse;
  36 import reactor.netty.http.server.HttpServerRoutes;
  37
  38 /**
  39  * @author <a href="mailto:piotr.jaszczyk@nokia.com">Piotr Jaszczyk</a>
  40  * @since February 2019
  41  */
  42 public class DummyHttpServer {
  43
  44     private static final Logger LOGGER = LoggerFactory.getLogger(DummyHttpServer.class);
  45     private final DisposableServer server;
  46
  47     private DummyHttpServer(DisposableServer server) {
  48         this.server = server;
  49     }
  50
  51     public static DummyHttpServer start(Consumer<HttpServerRoutes> routes) {
  52         LOGGER.info("Starting dummy server");
  53         final DisposableServer server = HttpServer.create()
  54                 .host("127.0.0.1")
  55                 .route(routes)
  56                 .bind()
  57                 .block();
  58         LOGGER.info("Server started");
  59         return new DummyHttpServer(server);
  60     }
  61
  62     public static Publisher<Void> sendInOrder(AtomicInteger state, Publisher<Void>... responses) {
  63         return responses[state.getAndIncrement()];
  64     }
  65
  66     public static Publisher<Void> sendResource(HttpServerResponse httpServerResponse, String resourcePath) {
  67         return sendString(httpServerResponse, Mono.fromCallable(() -> readResource(resourcePath)));
  68     }
  69
  70     public static Publisher<Void> sendError(HttpServerResponse httpServerResponse, int statusCode, String message){
  71         return sendString(httpServerResponse.status(statusCode), Mono.just(message));
  72     }
  73
  74     public static Publisher<Void> sendString(HttpServerResponse httpServerResponse, Publisher<String> content) {
  75         return httpServerResponse.sendString(content);
  76     }
  77
  78     public void close() {
  79         server.disposeNow();
  80     }
  81
  82     public String host() {
  83         return server.host();
  84     }
  85
  86     public int port() {
  87         return server.port();
  88     }
  89
  90     private static String readResource(String resourcePath) {
  91         try {
  92             return CheckedFunction0.constant(resourcePath)
  93                     .andThen(DummyHttpServer.class::getResource)
  94                     .andThen(URL::toURI)
  95                     .andThen(Paths::get)
  96                     .andThen(Files::readAllBytes)
  97                     .andThen(String::new)
  98                     .apply();
  99         } catch (Throwable throwable) {
 100             throw new RuntimeException(throwable);
 101         }
 102     }
 103 }