aboutsummaryrefslogtreecommitdiffstats
path: root/sources/hv-collector-ct/src/test/kotlin/org/onap/dcae/collectors/veshv/tests/fakes/configuration.kt
blob: a398967dcd7bcb7d56ff062793cb2683fdad98eb (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
/*
 * ============LICENSE_START=======================================================
 * dcaegen2-collectors-veshv
 * ================================================================================
 * Copyright (C) 2018-2019 NOKIA
 * ================================================================================
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 * ============LICENSE_END=========================================================
 */
package org.onap.dcae.collectors.veshv.tests.fakes

import org.onap.dcae.collectors.veshv.boundary.ConfigurationProvider
import org.onap.dcae.collectors.veshv.domain.VesEventDomain.HEARTBEAT
import org.onap.dcae.collectors.veshv.domain.VesEventDomain.MEASUREMENT
import org.onap.dcae.collectors.veshv.domain.VesEventDomain.PERF3GPP
import org.onap.dcaegen2.services.sdk.model.streams.dmaap.KafkaSink
import org.onap.dcaegen2.services.sdk.model.streams.dmaap.ImmutableKafkaSink
import reactor.core.publisher.FluxProcessor
import reactor.core.publisher.UnicastProcessor
import reactor.retry.RetryExhaustedException


const val PERF3GPP_TOPIC = "HV_VES_PERF3GPP"
const val MEASUREMENTS_FOR_VF_SCALING_TOPIC = "HV_VES_MEAS_FOR_VF_SCALING"
const val ALTERNATE_PERF3GPP_TOPIC = "HV_VES_PERF3GPP_ALTERNATIVE"
const val SAMPLE_BOOTSTRAP_SERVERS = "dmaap-mr-kafka-0:6060,dmaap-mr-kafka-1:6060"

val configWithBasicRouting = sequenceOf(
        ImmutableKafkaSink.builder()
                .name(PERF3GPP.domainName)
                .topicName(PERF3GPP_TOPIC)
                .bootstrapServers(SAMPLE_BOOTSTRAP_SERVERS)
                .build()
)

val configWithTwoDomainsToOneTopicRouting = sequenceOf(
        ImmutableKafkaSink.builder()
                .name(PERF3GPP.domainName)
                .topicName(PERF3GPP_TOPIC)
                .bootstrapServers(SAMPLE_BOOTSTRAP_SERVERS)
                .build(),
        ImmutableKafkaSink.builder()
                .name(HEARTBEAT.domainName)
                .topicName(PERF3GPP_TOPIC)
                .bootstrapServers(SAMPLE_BOOTSTRAP_SERVERS)
                .build(),
        ImmutableKafkaSink.builder()
                .name(MEASUREMENT.domainName)
                .topicName(MEASUREMENTS_FOR_VF_SCALING_TOPIC)
                .bootstrapServers(SAMPLE_BOOTSTRAP_SERVERS)
                .build()
)

val configWithDifferentRouting = sequenceOf(
        ImmutableKafkaSink.builder()
                .name(PERF3GPP.domainName)
                .topicName(ALTERNATE_PERF3GPP_TOPIC)
                .bootstrapServers(SAMPLE_BOOTSTRAP_SERVERS)
                .build()
)

val configWithEmptyRouting = emptySequence<KafkaSink>()


class FakeConfigurationProvider : ConfigurationProvider {
    private var shouldThrowException = false
    private val configStream: FluxProcessor<Sequence<KafkaSink>, Sequence<KafkaSink>> = UnicastProcessor.create()

    fun updateConfiguration(kafkaSinkSequence: Sequence<KafkaSink>) =
            if (shouldThrowException) {
                configStream.onError(RetryExhaustedException("I'm so tired"))
            } else {
                configStream.onNext(kafkaSinkSequence)
            }


    fun shouldThrowExceptionOnConfigUpdate(shouldThrowException: Boolean) {
        this.shouldThrowException = shouldThrowException
    }

    override fun invoke() = configStream
}