9ece90ffc176b15c26ab13237c261793bc75e4e3
[ccsdk/cds.git] /
1 /*
2  * Copyright © 2018-2019 AT&T Intellectual Property.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *     http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16
17 package org.onap.ccsdk.cds.blueprintsprocessor.message
18
19 import org.apache.kafka.streams.StreamsConfig
20 import org.junit.Test
21 import org.onap.ccsdk.cds.controllerblueprints.core.BluePrintTypes
22 import org.onap.ccsdk.cds.controllerblueprints.core.dsl.relationshipTypeConnectsTo
23 import org.onap.ccsdk.cds.controllerblueprints.core.dsl.serviceTemplate
24 import kotlin.test.assertEquals
25 import kotlin.test.assertNotNull
26
27 class MessagePropertiesDSLTest {
28
29     @Test
30     fun testMessageProducerDSL() {
31         val serviceTemplate = serviceTemplate("message-properties-test", "1.0.0", "xxx.@xx.com", "message") {
32             topologyTemplate {
33                 relationshipTemplateMessageProducer("sample-basic-auth", "Message Producer") {
34                     kafkaBasicAuth {
35                         bootstrapServers("sample-bootstrapServers")
36                         clientId("sample-client-id")
37                         acks("all")
38                         retries(3)
39                         enableIdempotence(true)
40                         topic("sample-topic")
41                     }
42                 }
43             }
44             relationshipTypes(
45                 arrayListOf(
46                     BluePrintTypes.relationshipTypeConnectsToMessageProducer(),
47                     BluePrintTypes.relationshipTypeConnectsTo()
48                 )
49             )
50         }
51         assertNotNull(serviceTemplate, "failed to create service template")
52         val relationshipTemplates = serviceTemplate.topologyTemplate?.relationshipTemplates
53         assertNotNull(relationshipTemplates, "failed to get relationship templates")
54         assertEquals(1, relationshipTemplates.size, "relationshipTemplates doesn't match")
55         assertNotNull(relationshipTemplates["sample-basic-auth"], "failed to get sample-basic-auth")
56         // println(serviceTemplate.asJsonString(true))
57     }
58
59     @Test
60     fun testMessageConsumerDSL() {
61         val serviceTemplate = serviceTemplate("message-properties-test", "1.0.0", "xxx.@xx.com", "message") {
62             topologyTemplate {
63                 relationshipTemplateMessageConsumer("sample-basic-auth", "Message Consumer") {
64                     kafkaBasicAuth {
65                         bootstrapServers("sample-bootstrapServers")
66                         clientId("sample-client-id")
67                         groupId("sample-group-id")
68                         topic("sample-topic")
69                         autoCommit(false)
70                         autoOffsetReset("latest")
71                         pollMillSec(5000)
72                         pollRecords(20)
73                     }
74                 }
75                 relationshipTemplateMessageConsumer("sample-stream-basic-auth", "Message Consumer") {
76                     kafkaStreamsBasicAuth {
77                         bootstrapServers("sample-bootstrapServers")
78                         applicationId("sample-application-id")
79                         autoOffsetReset("latest")
80                         processingGuarantee(StreamsConfig.EXACTLY_ONCE)
81                         topic("sample-streaming-topic")
82                     }
83                 }
84             }
85             relationshipTypes(
86                 arrayListOf(
87                     BluePrintTypes.relationshipTypeConnectsToMessageConsumer(),
88                     BluePrintTypes.relationshipTypeConnectsTo()
89                 )
90             )
91         }
92
93         assertNotNull(serviceTemplate, "failed to create service template")
94         val relationshipTemplates = serviceTemplate.topologyTemplate?.relationshipTemplates
95         assertNotNull(relationshipTemplates, "failed to get relationship templates")
96         assertEquals(2, relationshipTemplates.size, "relationshipTemplates doesn't match")
97         assertNotNull(relationshipTemplates["sample-basic-auth"], "failed to get sample-basic-auth")
98         assertNotNull(relationshipTemplates["sample-stream-basic-auth"], "failed to get sample-stream-basic-auth")
99         // println(serviceTemplate.asJsonString(true))
100     }
101 }