rest-services/dmaap-client/src/test/java/org/onap/dcaegen2/services/sdk/rest/services/dmaap/client/MessageRouterTestsUtils.java

   1 /*
   2  * ============LICENSE_START====================================
   3  * DCAEGEN2-SERVICES-SDK
   4  * =========================================================
   5  * Copyright (C) 2019-2021 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.dmaap.client;
  22
  23 import com.google.gson.Gson;
  24 import com.google.gson.JsonElement;
  25 import com.google.gson.JsonObject;
  26 import com.google.gson.JsonParser;
  27 import com.google.gson.JsonPrimitive;
  28 import io.vavr.collection.List;
  29 import org.onap.dcaegen2.services.sdk.model.streams.dmaap.ImmutableMessageRouterSink;
  30 import org.onap.dcaegen2.services.sdk.model.streams.dmaap.ImmutableMessageRouterSource;
  31 import org.onap.dcaegen2.services.sdk.rest.services.dmaap.client.api.MessageRouterPublisher;
  32 import org.onap.dcaegen2.services.sdk.rest.services.dmaap.client.api.MessageRouterSubscriber;
  33 import org.onap.dcaegen2.services.sdk.rest.services.dmaap.client.model.ImmutableMessageRouterPublishRequest;
  34 import org.onap.dcaegen2.services.sdk.rest.services.dmaap.client.model.ImmutableMessageRouterPublishResponse;
  35 import org.onap.dcaegen2.services.sdk.rest.services.dmaap.client.model.ImmutableMessageRouterSubscribeRequest;
  36 import org.onap.dcaegen2.services.sdk.rest.services.dmaap.client.model.ImmutableMessageRouterSubscribeResponse;
  37 import org.onap.dcaegen2.services.sdk.rest.services.dmaap.client.model.MessageRouterPublishRequest;
  38 import org.onap.dcaegen2.services.sdk.rest.services.dmaap.client.model.MessageRouterPublishResponse;
  39 import org.onap.dcaegen2.services.sdk.rest.services.dmaap.client.model.MessageRouterSubscribeRequest;
  40 import org.onap.dcaegen2.services.sdk.rest.services.dmaap.client.model.MessageRouterSubscribeResponse;
  41 import org.onap.dcaegen2.services.sdk.rest.services.dmaap.client.model.config.ImmutableDmaapTimeoutConfig;
  42 import reactor.core.publisher.Flux;
  43
  44 import java.time.Duration;
  45
  46
  47 public final class MessageRouterTestsUtils {
  48     private MessageRouterTestsUtils() {
  49     }
  50
  51     public static MessageRouterPublishRequest createPublishRequest(String topicUrl) {
  52         return createPublishRequest(topicUrl, ContentType.APPLICATION_JSON);
  53     }
  54
  55     public static MessageRouterPublishRequest createPublishRequest(String topicUrl, Duration timeout) {
  56         return ImmutableMessageRouterPublishRequest.builder()
  57                 .sinkDefinition(createMessageRouterSink(topicUrl))
  58                 .contentType(ContentType.APPLICATION_JSON)
  59                 .timeoutConfig(ImmutableDmaapTimeoutConfig.builder()
  60                         .timeout(timeout)
  61                         .build())
  62                 .build();
  63     }
  64
  65     public static MessageRouterPublishRequest createPublishRequest(String topicUrl, ContentType contentType) {
  66         return ImmutableMessageRouterPublishRequest.builder()
  67                 .sinkDefinition(createMessageRouterSink(topicUrl))
  68                 .contentType(contentType)
  69                 .build();
  70     }
  71
  72     public static MessageRouterSubscribeRequest createMRSubscribeRequest(String topicUrl,
  73                                                                          String consumerGroup, String consumerId) {
  74
  75         return ImmutableMessageRouterSubscribeRequest
  76                 .builder()
  77                 .sourceDefinition(getImmutableMessageRouterSource(topicUrl))
  78                 .consumerGroup(consumerGroup)
  79                 .consumerId(consumerId)
  80                 .build();
  81     }
  82
  83     public static MessageRouterSubscribeRequest createMRSubscribeRequest(String topicUrl,
  84                                                                          String consumerGroup, String consumerId,
  85                                                                          Duration timeout) {
  86
  87         return ImmutableMessageRouterSubscribeRequest
  88                 .builder()
  89                 .timeoutConfig(ImmutableDmaapTimeoutConfig.builder()
  90                         .timeout(timeout)
  91                         .build())
  92                 .sourceDefinition(getImmutableMessageRouterSource(topicUrl))
  93                 .consumerGroup(consumerGroup)
  94                 .consumerId(consumerId)
  95                 .build();
  96     }
  97
  98     private static ImmutableMessageRouterSource getImmutableMessageRouterSource(String topicUrl) {
  99         return ImmutableMessageRouterSource.builder()
 100                 .name("the topic")
 101                 .topicUrl(topicUrl)
 102                 .build();
 103     }
 104
 105     public static List<JsonElement> getAsJsonElements(List<String> messages) {
 106         return messages.map(JsonParser::parseString);
 107     }
 108
 109     public static List<JsonObject> getAsJsonObjects(List<String> messages) {
 110         return getAsJsonElements(messages).map(JsonElement::getAsJsonObject);
 111     }
 112
 113     public static List<JsonPrimitive> getAsJsonPrimitives(List<String> messages) {
 114         return getAsJsonElements(messages).map(JsonElement::getAsJsonPrimitive);
 115     }
 116
 117     public static JsonObject getAsJsonObject(String item) {
 118         return new Gson().fromJson(item, JsonObject.class);
 119     }
 120
 121     public static Flux<JsonElement> plainBatch(List<String> messages) {
 122         return Flux.fromIterable(getAsJsonElements(messages));
 123     }
 124
 125     public static Flux<JsonObject> jsonBatch(List<String> messages) {
 126         return Flux.fromIterable(getAsJsonObjects(messages));
 127     }
 128
 129     public static MessageRouterSubscribeResponse errorSubscribeResponse(String failReasonFormat, Object... formatArgs) {
 130         String failReason = formatArgs.length == 0 ? failReasonFormat : String.format(failReasonFormat, formatArgs);
 131         return ImmutableMessageRouterSubscribeResponse
 132                 .builder()
 133                 .failReason(failReason)
 134                 .build();
 135     }
 136
 137     public static MessageRouterSubscribeResponse successSubscribeResponse(List<JsonElement> items) {
 138         return ImmutableMessageRouterSubscribeResponse
 139                 .builder()
 140                 .items(items)
 141                 .build();
 142     }
 143
 144     public static MessageRouterPublishResponse errorPublishResponse(String failReasonFormat, Object... formatArgs) {
 145         String failReason = formatArgs.length == 0 ? failReasonFormat : String.format(failReasonFormat, formatArgs);
 146         return ImmutableMessageRouterPublishResponse
 147                 .builder()
 148                 .failReason(failReason)
 149                 .build();
 150     }
 151
 152     public static MessageRouterPublishResponse successPublishResponse(List<JsonElement> items) {
 153         return ImmutableMessageRouterPublishResponse
 154                 .builder()
 155                 .items(items)
 156                 .build();
 157     }
 158
 159     public static void registerTopic(MessageRouterPublisher publisher, MessageRouterPublishRequest publishRequest,
 160                                      MessageRouterSubscriber subscriber, MessageRouterSubscribeRequest subscribeRequest) {
 161         final List<String> sampleJsonMessages = List.of("{\"message\":\"message1\"}",
 162                 "{\"differentMessage\":\"message2\"}");
 163         final Flux<JsonObject> jsonMessageBatch = MessageRouterTestsUtils.jsonBatch(sampleJsonMessages);
 164
 165         publisher.put(publishRequest, jsonMessageBatch).blockLast();
 166         subscriber.get(subscribeRequest).block();
 167     }
 168
 169     private static ImmutableMessageRouterSink createMessageRouterSink(String topicUrl) {
 170         return ImmutableMessageRouterSink.builder()
 171                 .name("the topic")
 172                 .topicUrl(topicUrl)
 173                 .build();
 174     }
 175 }