ms/blueprintsprocessor/modules/commons/message-lib/src/test/kotlin/org/onap/ccsdk/cds/blueprintsprocessor/message/service/MessageLoggerServiceTest.kt

   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.service
  18
  19 import io.mockk.every
  20 import io.mockk.mockk
  21 import org.apache.kafka.clients.consumer.ConsumerRecord
  22 import org.apache.kafka.common.header.internals.RecordHeaders
  23 import org.junit.Test
  24 import org.onap.ccsdk.cds.blueprintsprocessor.core.api.data.CommonHeader
  25 import org.onap.ccsdk.cds.blueprintsprocessor.message.toMap
  26 import org.onap.ccsdk.cds.controllerblueprints.core.BlueprintConstants
  27 import org.slf4j.MDC
  28 import kotlin.test.assertEquals
  29
  30 class MessageLoggerServiceTest {
  31
  32     @Test
  33     fun testMessagingHeaders() {
  34         val messageLoggerService = MessageLoggerService()
  35         val commonHeader = CommonHeader().apply {
  36             requestId = "1234"
  37             subRequestId = "1234-12"
  38             originatorId = "cds-test"
  39         }
  40
  41         val consumerRecord = mockk<ConsumerRecord<*, *>>()
  42         every { consumerRecord.headers() } returns null
  43         every { consumerRecord.key() } returns "1234"
  44         every { consumerRecord.offset() } returns 12345
  45         every { consumerRecord.topic() } returns "sample-topic"
  46         every { consumerRecord.timestamp() } returns System.currentTimeMillis()
  47         messageLoggerService.messageConsuming(commonHeader, consumerRecord)
  48         assertEquals(commonHeader.requestId, MDC.get("RequestID"))
  49         assertEquals(commonHeader.subRequestId, MDC.get("InvocationID"))
  50
  51         val mockHeaders = RecordHeaders()
  52         messageLoggerService.messageProducing(mockHeaders)
  53         val map = mockHeaders.toMap()
  54         assertEquals("1234-12", map[BlueprintConstants.ONAP_REQUEST_ID])
  55
  56         messageLoggerService.messageConsumingExisting()
  57     }
  58 }