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.simulators.dcaeapp.impl
  21
  22 import org.apache.kafka.clients.consumer.ConsumerRecord
  23 import org.assertj.core.api.Assertions.assertThat
  24 import org.jetbrains.spek.api.Spek
  25 import org.jetbrains.spek.api.dsl.describe
  26 import org.jetbrains.spek.api.dsl.it
  27 import org.jetbrains.spek.api.dsl.on
  28 import reactor.kafka.receiver.ReceiverRecord
  29
  30
  31 /**
  32  * @author Piotr Jaszczyk <piotr.jaszczyk@nokia.com>
  33  * @since August 2018
  34  */
  35 internal class ConsumerTest : Spek({
  36
  37     lateinit var cut: Consumer
  38
  39     beforeEachTest {
  40         cut = Consumer()
  41     }
  42
  43     describe("Consumer which holds the state of received Kafka records") {
  44         it("should contain empty state in the beginning") {
  45             assertEmptyState(cut)
  46         }
  47
  48         describe("update") {
  49             val value = byteArrayOf(2)
  50
  51             beforeEachTest {
  52                 cut.update(receiverRecord(
  53                         topic = "topic",
  54                         key = byteArrayOf(1),
  55                         value = value
  56                 )).unsafeRunSync()
  57             }
  58
  59             it("should contain one message if it was updated once") {
  60                 assertState(cut, value)
  61             }
  62
  63             it("should contain empty state message if it was reset after update") {
  64                 cut.reset().unsafeRunSync()
  65                 assertEmptyState(cut)
  66             }
  67         }
  68     }
  69 })
  70
  71 fun assertEmptyState(cut: Consumer) {
  72     assertState(cut)
  73 }
  74
  75 fun assertState(cut: Consumer, vararg values: ByteArray) {
  76     assertThat(cut.currentState().consumedMessages)
  77             .containsOnly(*values)
  78     assertThat(cut.currentState().messagesCount)
  79             .isEqualTo(values.size)
  80 }
  81
  82 fun receiverRecord(topic: String, key: ByteArray, value: ByteArray) =
  83         ReceiverRecord(ConsumerRecord(topic, 1, 100L, key, value), null)