gerrit.onap Code Review - dcaegen2/collectors/hv-ves.git/blob

   1 /*
   2  * ============LICENSE_START=======================================================
   3  * dcaegen2-collectors-veshv
   4  * ================================================================================
   5  * Copyright (C) 2018 NOKIA
   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 package org.onap.dcae.collectors.veshv.ves.message.generator.impl
  21
  22 import com.google.protobuf.ByteString
  23 import com.google.protobuf.InvalidProtocolBufferException
  24 import org.assertj.core.api.Assertions.assertThat
  25 import org.assertj.core.api.Assertions.assertThatExceptionOfType
  26 import org.jetbrains.spek.api.Spek
  27 import org.jetbrains.spek.api.dsl.describe
  28 import org.jetbrains.spek.api.dsl.given
  29 import org.jetbrains.spek.api.dsl.it
  30 import org.jetbrains.spek.api.dsl.on
  31 import org.onap.dcae.collectors.veshv.domain.ByteData
  32 import org.onap.dcae.collectors.veshv.domain.WireFrameMessage
  33 import org.onap.dcae.collectors.veshv.domain.VesEventDomain.PERF3GPP
  34 import org.onap.dcae.collectors.veshv.domain.VesEventDomain.FAULT
  35 import org.onap.dcae.collectors.veshv.domain.VesEventDomain.HEARTBEAT
  36 import org.onap.dcae.collectors.veshv.tests.utils.commonHeader
  37 import org.onap.dcae.collectors.veshv.ves.message.generator.api.MessageGenerator
  38 import org.onap.dcae.collectors.veshv.ves.message.generator.api.MessageParameters
  39 import org.onap.dcae.collectors.veshv.ves.message.generator.api.MessageType
  40 import org.onap.ves.VesEventOuterClass.CommonEventHeader
  41 import org.onap.ves.VesEventOuterClass.VesEvent
  42 import reactor.test.test
  43
  44 /**
  45  * @author Jakub Dudycz <jakub.dudycz@nokia.com>
  46  * @since June 2018
  47  */
  48 object MessageGeneratorImplTest : Spek({
  49     describe("message factory") {
  50         val maxPayloadSizeBytes = 1024
  51         val generator = MessageGeneratorImpl(PayloadGenerator(), maxPayloadSizeBytes)
  52         given("single message parameters") {
  53             on("messages amount not specified in parameters") {
  54                 it("should create infinite flux") {
  55                     val limit = 1000L
  56                     generator
  57                             .createMessageFlux(listOf(MessageParameters(
  58                                     commonHeader(PERF3GPP),
  59                                     MessageType.VALID
  60                             )))
  61                             .take(limit)
  62                             .test()
  63                             .expectNextCount(limit)
  64                             .verifyComplete()
  65                 }
  66             }
  67             on("messages amount specified in parameters") {
  68                 it("should create message flux of specified size") {
  69                     generator
  70                             .createMessageFlux(listOf(MessageParameters(
  71                                     commonHeader(PERF3GPP),
  72                                     MessageType.VALID,
  73                                     5
  74                             )))
  75                             .test()
  76                             .expectNextCount(5)
  77                             .verifyComplete()
  78                 }
  79             }
  80             on("message type requesting valid message") {
  81                 it("should create flux of valid messages with given domain") {
  82                     generator
  83                             .createMessageFlux(listOf(MessageParameters(
  84                                     commonHeader(FAULT),
  85                                     MessageType.VALID,
  86                                     1
  87                             )))
  88                             .test()
  89                             .assertNext {
  90                                 assertThat(it.isValid()).isTrue()
  91                                 assertThat(it.payloadSize).isLessThan(maxPayloadSizeBytes)
  92                                 assertThat(extractCommonEventHeader(it.payload).domain).isEqualTo(FAULT.domainName)
  93                             }
  94                             .verifyComplete()
  95                 }
  96             }
  97             on("message type requesting too big payload") {
  98                 it("should create flux of messages with given domain and payload exceeding threshold") {
  99
 100                     generator
 101                             .createMessageFlux(listOf(MessageParameters(
 102                                     commonHeader(PERF3GPP),
 103                                     MessageType.TOO_BIG_PAYLOAD,
 104                                     1
 105                             )))
 106                             .test()
 107                             .assertNext {
 108                                 assertThat(it.isValid()).isTrue()
 109                                 assertThat(it.payloadSize).isGreaterThan(maxPayloadSizeBytes)
 110                                 assertThat(extractCommonEventHeader(it.payload).domain).isEqualTo(PERF3GPP.domainName)
 111                             }
 112                             .verifyComplete()
 113                 }
 114             }
 115             on("message type requesting invalid GPB data ") {
 116                 it("should create flux of messages with invalid payload") {
 117                     generator
 118                             .createMessageFlux(listOf(MessageParameters(
 119                                     commonHeader(PERF3GPP),
 120                                     MessageType.INVALID_GPB_DATA,
 121                                     1
 122                             )))
 123                             .test()
 124                             .assertNext {
 125                                 assertThat(it.isValid()).isTrue()
 126                                 assertThat(it.payloadSize).isLessThan(maxPayloadSizeBytes)
 127                                 assertThatExceptionOfType(InvalidProtocolBufferException::class.java)
 128                                         .isThrownBy { extractCommonEventHeader(it.payload) }
 129                             }
 130                             .verifyComplete()
 131                 }
 132             }
 133             on("message type requesting invalid wire frame ") {
 134                 it("should create flux of messages with invalid version") {
 135                     generator
 136                             .createMessageFlux(listOf(MessageParameters(
 137                                     commonHeader(PERF3GPP),
 138                                     MessageType.INVALID_WIRE_FRAME,
 139                                     1
 140                             )))
 141                             .test()
 142                             .assertNext {
 143                                 assertThat(it.isValid()).isFalse()
 144                                 assertThat(it.payloadSize).isLessThan(maxPayloadSizeBytes)
 145                                 assertThat(extractCommonEventHeader(it.payload).domain).isEqualTo(PERF3GPP.domainName)
 146                                 assertThat(it.versionMajor).isNotEqualTo(WireFrameMessage.SUPPORTED_VERSION_MINOR)
 147                             }
 148                             .verifyComplete()
 149                 }
 150             }
 151             on("message type requesting fixed payload") {
 152                 it("should create flux of valid messages with fixed payload") {
 153                     generator
 154                             .createMessageFlux(listOf(MessageParameters(
 155                                     commonHeader(FAULT),
 156                                     MessageType.FIXED_PAYLOAD,
 157                                     1
 158                             )))
 159                             .test()
 160                             .assertNext {
 161                                 assertThat(it.isValid()).isTrue()
 162                                 assertThat(it.payloadSize).isLessThan(maxPayloadSizeBytes)
 163                                 assertThat(extractEventFields(it.payload).size()).isEqualTo(MessageGenerator.FIXED_PAYLOAD_SIZE)
 164                                 assertThat(extractCommonEventHeader(it.payload).domain).isEqualTo(FAULT.domainName)
 165                             }
 166                             .verifyComplete()
 167                 }
 168             }
 169         }
 170         given("list of message parameters") {
 171             it("should create concatenated flux of messages") {
 172                 val singleFluxSize = 5L
 173                 val messageParameters = listOf(
 174                         MessageParameters(commonHeader(PERF3GPP), MessageType.VALID, singleFluxSize),
 175                         MessageParameters(commonHeader(FAULT), MessageType.TOO_BIG_PAYLOAD, singleFluxSize),
 176                         MessageParameters(commonHeader(HEARTBEAT), MessageType.VALID, singleFluxSize)
 177                 )
 178                 generator.createMessageFlux(messageParameters)
 179                         .test()
 180                         .assertNext {
 181                             assertThat(it.payloadSize).isLessThan(maxPayloadSizeBytes)
 182                             assertThat(extractCommonEventHeader(it.payload).domain).isEqualTo(PERF3GPP.domainName)
 183                         }
 184                         .expectNextCount(singleFluxSize - 1)
 185                         .assertNext {
 186                             assertThat(it.payloadSize).isGreaterThan(maxPayloadSizeBytes)
 187                             assertThat(extractCommonEventHeader(it.payload).domain).isEqualTo(FAULT.domainName)
 188                         }
 189                         .expectNextCount(singleFluxSize - 1)
 190                         .assertNext {
 191                             assertThat(it.payloadSize).isLessThan(maxPayloadSizeBytes)
 192                             assertThat(extractCommonEventHeader(it.payload).domain).isEqualTo(HEARTBEAT.domainName)
 193                         }
 194                         .expectNextCount(singleFluxSize - 1)
 195                         .verifyComplete()
 196             }
 197         }
 198     }
 199 })
 200
 201 fun extractCommonEventHeader(bytes: ByteData): CommonEventHeader =
 202         VesEvent.parseFrom(bytes.unsafeAsArray()).commonEventHeader
 203
 204
 205 fun extractEventFields(bytes: ByteData): ByteString =
 206         VesEvent.parseFrom(bytes.unsafeAsArray()).eventFields
 207