cps-ncmp-service/src/test/groovy/org/onap/cps/ncmp/api/kafka/MessagingBaseSpec.groovy

   1 /*
   2  * ============LICENSE_START=======================================================
   3  * Copyright (c) 2023 Nordix Foundation.
   4  *  ================================================================================
   5  *  Licensed under the Apache License, Version 2.0 (the "License");
   6  *  you may not use this file except in compliance with the License.
   7  *  You may obtain a copy of the License at
   8  *
   9  *        http://www.apache.org/licenses/LICENSE-2.0
  10  *
  11  *  Unless required by applicable law or agreed to in writing, software
  12  *  distributed under the License is distributed on an 'AS IS' BASIS,
  13  *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14  *  See the License for the specific language governing permissions and
  15  *  limitations under the License.
  16  *
  17  *  SPDX-License-Identifier: Apache-2.0
  18  *  ============LICENSE_END=========================================================
  19  */
  20
  21 package org.onap.cps.ncmp.api.kafka
  22
  23 import io.cloudevents.CloudEvent
  24 import io.cloudevents.kafka.CloudEventSerializer
  25 import org.apache.kafka.common.serialization.StringDeserializer
  26 import org.apache.kafka.common.serialization.StringSerializer
  27 import org.spockframework.spring.SpringBean
  28 import org.springframework.kafka.core.DefaultKafkaProducerFactory
  29 import org.springframework.kafka.core.KafkaTemplate
  30 import org.springframework.kafka.support.serializer.JsonSerializer
  31 import org.springframework.test.context.DynamicPropertyRegistry
  32 import org.springframework.test.context.DynamicPropertySource
  33 import org.testcontainers.containers.KafkaContainer
  34 import org.testcontainers.utility.DockerImageName
  35 import spock.lang.Specification
  36
  37 class MessagingBaseSpec extends Specification {
  38
  39     def setupSpec() {
  40         kafkaTestContainer.start()
  41     }
  42
  43     def cleanupSpec() {
  44         kafkaTestContainer.stop()
  45     }
  46
  47     static kafkaTestContainer = new KafkaContainer(DockerImageName.parse('registry.nordix.org/onaptest/confluentinc/cp-kafka:6.2.1').asCompatibleSubstituteFor('confluentinc/cp-kafka'))
  48
  49     @SpringBean
  50     KafkaTemplate legacyEventKafkaTemplate = new KafkaTemplate<>(new DefaultKafkaProducerFactory<Integer, String>(eventProducerConfigProperties(JsonSerializer)))
  51
  52     @SpringBean
  53     KafkaTemplate cloudEventKafkaTemplate = new KafkaTemplate<>(new DefaultKafkaProducerFactory<String, CloudEvent>(eventProducerConfigProperties(CloudEventSerializer)))
  54
  55     @DynamicPropertySource
  56     static void registerKafkaProperties(DynamicPropertyRegistry dynamicPropertyRegistry) {
  57         dynamicPropertyRegistry.add('spring.kafka.bootstrap-servers', kafkaTestContainer::getBootstrapServers)
  58     }
  59
  60     def eventProducerConfigProperties(valueSerializer) {
  61         return [('bootstrap.servers'): kafkaTestContainer.getBootstrapServers().split(',')[0],
  62                 ('retries')          : 0,
  63                 ('batch-size')       : 16384,
  64                 ('linger.ms')        : 1,
  65                 ('buffer.memory')    : 33554432,
  66                 ('key.serializer')   : StringSerializer,
  67                 ('value.serializer') : valueSerializer]
  68     }
  69
  70     def eventConsumerConfigProperties(consumerGroupId, valueSerializer) {
  71         return [('bootstrap.servers') : kafkaTestContainer.getBootstrapServers().split(',')[0],
  72                 ('key.deserializer')  : StringDeserializer,
  73                 ('value.deserializer'): valueSerializer,
  74                 ('auto.offset.reset') : 'earliest',
  75                 ('group.id')          : consumerGroupId
  76         ]
  77     }
  78 }