gerrit.onap Code Review - dcaegen2/analytics/tca-gen2.git/blob

   1 package org.onap.dcae.analytics.web.dmaap;
   2
   3 import org.assertj.core.api.Assertions;
   4 import org.junit.jupiter.api.Test;
   5 import org.onap.dcae.analytics.model.AnalyticsModel;
   6 import org.onap.dcae.analytics.model.TcaModelConstants;
   7 import org.onap.dcae.analytics.web.BaseAnalyticsWebTest;
   8 import org.springframework.messaging.Message;
   9 import org.springframework.messaging.MessageHeaders;
  10 import org.springframework.messaging.support.MessageBuilder;
  11
  12 import java.util.Collection;
  13 import java.util.HashMap;
  14 import java.util.List;
  15 import java.util.Map;
  16 import java.util.stream.Collectors;
  17 import java.util.stream.IntStream;
  18 import java.util.stream.Stream;
  19
  20 import static org.junit.jupiter.api.Assertions.*;
  21 import static org.onap.dcae.analytics.model.AnalyticsHttpConstants.REQUEST_ID_HEADER_KEY;
  22 import static org.onap.dcae.analytics.model.util.json.AnalyticsModelJsonConversion.ANALYTICS_MODEL_OBJECT_MAPPER;
  23
  24 class MrMessageSplitterTest extends BaseAnalyticsWebTest {
  25
  26     @Test
  27     @SuppressWarnings("unchecked")
  28     void splitMessage() {
  29
  30         final Map<String, Object> headers = new HashMap<>();
  31         headers.put(REQUEST_ID_HEADER_KEY, "testRequestId");
  32         final MessageHeaders messageHeaders = new MessageHeaders(headers);
  33
  34         final Message<?> message =
  35                 MessageBuilder.createMessage(fromStream(TestFileLocation.CEF_JSON_MESSAGES), messageHeaders);
  36
  37         final MrMessageSplitter mrMessageSplitter = new MrMessageSplitter(ANALYTICS_MODEL_OBJECT_MAPPER, 10);
  38         final Object splitMessageStream = mrMessageSplitter.splitMessage(message);
  39
  40         final List<String> messages = ((Stream<String>) splitMessageStream).collect(Collectors.toList());
  41         Assertions.assertThat(messages.size()).isEqualTo(4);
  42     }
  43 }