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

   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.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.model.streams.dmaap.MessageRouterSink;
  32 import org.onap.dcaegen2.services.sdk.rest.services.dmaap.client.api.MessageRouterPublisher;
  33 import org.onap.dcaegen2.services.sdk.rest.services.dmaap.client.api.MessageRouterSubscriber;
  34 import org.onap.dcaegen2.services.sdk.rest.services.dmaap.client.model.ImmutableMessageRouterPublishRequest;
  35 import org.onap.dcaegen2.services.sdk.rest.services.dmaap.client.model.ImmutableMessageRouterPublishResponse;
  36 import org.onap.dcaegen2.services.sdk.rest.services.dmaap.client.model.ImmutableMessageRouterSubscribeRequest;
  37 import org.onap.dcaegen2.services.sdk.rest.services.dmaap.client.model.ImmutableMessageRouterSubscribeResponse;
  38 import org.onap.dcaegen2.services.sdk.rest.services.dmaap.client.model.MessageRouterPublishRequest;
  39 import org.onap.dcaegen2.services.sdk.rest.services.dmaap.client.model.MessageRouterPublishResponse;
  40 import org.onap.dcaegen2.services.sdk.rest.services.dmaap.client.model.MessageRouterSubscribeRequest;
  41 import org.onap.dcaegen2.services.sdk.rest.services.dmaap.client.model.MessageRouterSubscribeResponse;
  42 import reactor.core.publisher.Flux;
  43
  44
  45 public final class MessageRouterTestsUtils {
  46     private static final JsonParser parser = new JsonParser();
  47     private MessageRouterTestsUtils() {}
  48
  49     public static MessageRouterPublishRequest createPublishRequest(String topicUrl){
  50         return createPublishRequest(topicUrl, ContentType.APPLICATION_JSON);
  51     }
  52
  53     public static MessageRouterPublishRequest createPublishRequest(String topicUrl, ContentType contentType){
  54         MessageRouterSink sinkDefinition = ImmutableMessageRouterSink.builder()
  55                 .name("the topic")
  56                 .topicUrl(topicUrl)
  57                 .build();
  58
  59         return ImmutableMessageRouterPublishRequest.builder()
  60                 .sinkDefinition(sinkDefinition)
  61                 .contentType(contentType)
  62                 .build();
  63     }
  64
  65     public static MessageRouterSubscribeRequest createMRSubscribeRequest(String topicUrl,
  66             String consumerGroup, String consumerId) {
  67         ImmutableMessageRouterSource sourceDefinition = ImmutableMessageRouterSource.builder()
  68                 .name("the topic")
  69                 .topicUrl(topicUrl)
  70                 .build();
  71
  72         return ImmutableMessageRouterSubscribeRequest
  73                 .builder()
  74                 .sourceDefinition(sourceDefinition)
  75                 .consumerGroup(consumerGroup)
  76                 .consumerId(consumerId)
  77                 .build();
  78     }
  79
  80     public static List<JsonElement> getAsJsonElements(List<String> messages){
  81         return messages.map(parser::parse);
  82     }
  83
  84     public static List<JsonObject> getAsJsonObjects(List<String> messages){
  85         return getAsJsonElements(messages).map(JsonElement::getAsJsonObject);
  86     }
  87
  88     public static List<JsonPrimitive> getAsJsonPrimitives(List<String> messages){
  89         return getAsJsonElements(messages).map(JsonElement::getAsJsonPrimitive);
  90     }
  91
  92     public static JsonObject getAsJsonObject(String item){
  93         return new Gson().fromJson(item, JsonObject.class);
  94     }
  95
  96     public static Flux<JsonElement> plainBatch(List<String> messages){
  97         return Flux.fromIterable(getAsJsonElements(messages));
  98     }
  99
 100     public static Flux<JsonObject> jsonBatch(List<String> messages){
 101         return Flux.fromIterable(getAsJsonObjects(messages));
 102     }
 103
 104     public static MessageRouterSubscribeResponse errorSubscribeResponse(String failReasonFormat, Object... formatArgs){
 105         return ImmutableMessageRouterSubscribeResponse
 106                 .builder()
 107                 .failReason(String.format(failReasonFormat, formatArgs))
 108                 .build();
 109     }
 110
 111     public static MessageRouterSubscribeResponse successSubscribeResponse(List<JsonElement> items){
 112         return ImmutableMessageRouterSubscribeResponse
 113                 .builder()
 114                 .items(items)
 115                 .build();
 116     }
 117
 118     public static MessageRouterPublishResponse errorPublishResponse(String failReasonFormat, Object... formatArgs){
 119         return ImmutableMessageRouterPublishResponse
 120                 .builder()
 121                 .failReason(String.format(failReasonFormat, formatArgs))
 122                 .build();
 123     }
 124
 125     public static MessageRouterPublishResponse successPublishResponse(List<JsonElement> items){
 126         return ImmutableMessageRouterPublishResponse
 127                 .builder()
 128                 .items(items)
 129                 .build();
 130     }
 131
 132     public static void registerTopic(MessageRouterPublisher publisher, MessageRouterPublishRequest publishRequest,
 133             MessageRouterSubscriber subscriber, MessageRouterSubscribeRequest subscribeRequest) {
 134         final List<String> sampleJsonMessages = List.of("{\"message\":\"message1\"}",
 135                 "{\"differentMessage\":\"message2\"}");
 136         final Flux<JsonObject> jsonMessageBatch = MessageRouterTestsUtils.jsonBatch(sampleJsonMessages);
 137
 138         publisher.put(publishRequest, jsonMessageBatch).blockLast();
 139         subscriber.get(subscribeRequest).block();
 140     }
 141 }