gerrit.onap Code Review - ccsdk/cds.git/blob

   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
  33     @Test
  34     fun testMessagingHeaders() {
  35         val messageLoggerService = MessageLoggerService()
  36         val commonHeader = CommonHeader().apply {
  37             requestId = "1234"
  38             subRequestId = "1234-12"
  39             originatorId = "cds-test"
  40         }
  41
  42         val consumerRecord = mockk<ConsumerRecord<*, *>>()
  43         every { consumerRecord.headers() } returns null
  44         every { consumerRecord.key() } returns "1234"
  45         every { consumerRecord.offset() } returns 12345
  46         every { consumerRecord.topic() } returns "sample-topic"
  47         every { consumerRecord.timestamp() } returns System.currentTimeMillis()
  48         messageLoggerService.messageConsuming(commonHeader, consumerRecord)
  49         assertEquals(commonHeader.requestId, MDC.get("RequestID"))
  50         assertEquals(commonHeader.subRequestId, MDC.get("InvocationID"))
  51
  52         val mockHeaders = RecordHeaders()
  53         messageLoggerService.messageProducing(mockHeaders)
  54         val map = mockHeaders.toMap()
  55         assertEquals("1234-12", map[BluePrintConstants.ONAP_REQUEST_ID])
  56
  57         messageLoggerService.messageConsumingExisting()
  58
  59     }
  60
  61 }