5ccb483a2bd9e41ede5f640efa17ac9bcddcf42d
[dcaegen2/collectors/hv-ves.git] /
1 /*
2  * ============LICENSE_START=======================================================
3  * dcaegen2-collectors-veshv
4  * ================================================================================
5  * Copyright (C) 2019 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.kafkaconsumer.state
21
22 import com.nhaarman.mockitokotlin2.mock
23 import com.nhaarman.mockitokotlin2.verify
24 import org.apache.kafka.common.TopicPartition
25 import org.jetbrains.spek.api.Spek
26 import org.jetbrains.spek.api.dsl.describe
27 import org.jetbrains.spek.api.dsl.given
28 import org.jetbrains.spek.api.dsl.it
29 import org.jetbrains.spek.api.dsl.on
30 import org.onap.dcae.collectors.veshv.kafkaconsumer.metrics.Metrics
31
32 object OffsetConsumerTest : Spek({
33     describe("OffsetConsumer with metrics") {
34         val mockedMetrics = mock<Metrics>()
35         val offsetConsumer = OffsetConsumer(mockedMetrics)
36         given("topicName with partition"){
37             val topicPartition = TopicPartition(topicName, partitionNumber)
38
39             on("new update method call") {
40                 offsetConsumer.update(topicPartition, newOffset)
41
42                 it("should notify message newOffset metric") {
43                     verify(mockedMetrics).notifyOffsetChanged(newOffset, topicPartition)
44                 }
45             }
46         }
47     }
48 })
49
50 private const val partitionNumber = 1
51 private const val newOffset: Long = 99
52 private const val topicName = "sample-topicName"