diff options
Diffstat (limited to 'src')
3 files changed, 91 insertions, 8 deletions
diff --git a/src/main/kotlin/dev/dnpm/etl/processor/config/AppKafkaConfiguration.kt b/src/main/kotlin/dev/dnpm/etl/processor/config/AppKafkaConfiguration.kt index 1ff5e58..3799762 100644 --- a/src/main/kotlin/dev/dnpm/etl/processor/config/AppKafkaConfiguration.kt +++ b/src/main/kotlin/dev/dnpm/etl/processor/config/AppKafkaConfiguration.kt @@ -98,9 +98,10 @@ class AppKafkaConfiguration { @Bean @ConditionalOnProperty(value = ["app.kafka.input-topic"]) fun kafkaInputListener( - requestProcessor: RequestProcessor + requestProcessor: RequestProcessor, + objectMapper: ObjectMapper ): KafkaInputListener { - return KafkaInputListener(requestProcessor) + return KafkaInputListener(requestProcessor, objectMapper) } @Bean diff --git a/src/main/kotlin/dev/dnpm/etl/processor/input/KafkaInputListener.kt b/src/main/kotlin/dev/dnpm/etl/processor/input/KafkaInputListener.kt index f3c0ab4..63bf60a 100644 --- a/src/main/kotlin/dev/dnpm/etl/processor/input/KafkaInputListener.kt +++ b/src/main/kotlin/dev/dnpm/etl/processor/input/KafkaInputListener.kt @@ -19,6 +19,7 @@ package dev.dnpm.etl.processor.input +import com.fasterxml.jackson.databind.ObjectMapper import de.ukw.ccc.bwhc.dto.Consent import de.ukw.ccc.bwhc.dto.MtbFile import dev.dnpm.etl.processor.services.RequestProcessor @@ -27,17 +28,19 @@ import org.slf4j.LoggerFactory import org.springframework.kafka.listener.MessageListener class KafkaInputListener( - private val requestProcessor: RequestProcessor -) : MessageListener<String, MtbFile> { + private val requestProcessor: RequestProcessor, + private val objectMapper: ObjectMapper +) : MessageListener<String, String> { private val logger = LoggerFactory.getLogger(KafkaInputListener::class.java) - override fun onMessage(data: ConsumerRecord<String, MtbFile>) { - if (data.value().consent.status == Consent.Status.ACTIVE) { + override fun onMessage(data: ConsumerRecord<String, String>) { + val mtbFile = objectMapper.readValue(data.value(), MtbFile::class.java) + if (mtbFile.consent.status == Consent.Status.ACTIVE) { logger.debug("Accepted MTB File for processing") - requestProcessor.processMtbFile(data.value()) + requestProcessor.processMtbFile(mtbFile) } else { logger.debug("Accepted MTB File and process deletion") - requestProcessor.processDeletion(data.value().patient.id) + requestProcessor.processDeletion(mtbFile.patient.id) } } }
\ No newline at end of file diff --git a/src/test/kotlin/dev/dnpm/etl/processor/input/KafkaInputListenerTest.kt b/src/test/kotlin/dev/dnpm/etl/processor/input/KafkaInputListenerTest.kt new file mode 100644 index 0000000..cf5ba39 --- /dev/null +++ b/src/test/kotlin/dev/dnpm/etl/processor/input/KafkaInputListenerTest.kt @@ -0,0 +1,79 @@ +/* + * This file is part of ETL-Processor + * + * Copyright (c) 2024 Comprehensive Cancer Center Mainfranken, Datenintegrationszentrum Philipps-Universität Marburg and Contributors + * + * This program is free software: you can redistribute it and/or modify + * it under the terms of the GNU Affero General Public License as published + * by the Free Software Foundation, either version 3 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU Affero General Public License for more details. + * + * You should have received a copy of the GNU Affero General Public License + * along with this program. If not, see <https://www.gnu.org/licenses/>. + */ + +package dev.dnpm.etl.processor.input + +import com.fasterxml.jackson.databind.ObjectMapper +import de.ukw.ccc.bwhc.dto.Consent +import de.ukw.ccc.bwhc.dto.MtbFile +import de.ukw.ccc.bwhc.dto.Patient +import dev.dnpm.etl.processor.services.RequestProcessor +import org.apache.kafka.clients.consumer.ConsumerRecord +import org.junit.jupiter.api.BeforeEach +import org.junit.jupiter.api.Test +import org.junit.jupiter.api.extension.ExtendWith +import org.mockito.ArgumentMatchers.anyString +import org.mockito.Mock +import org.mockito.junit.jupiter.MockitoExtension +import org.mockito.kotlin.any +import org.mockito.kotlin.times +import org.mockito.kotlin.verify + +@ExtendWith(MockitoExtension::class) +class KafkaInputListenerTest { + + private lateinit var requestProcessor: RequestProcessor + private lateinit var objectMapper: ObjectMapper + private lateinit var kafkaInputListener: KafkaInputListener + + @BeforeEach + fun setup( + @Mock requestProcessor: RequestProcessor + ) { + this.requestProcessor = requestProcessor + this.objectMapper = ObjectMapper() + + this.kafkaInputListener = KafkaInputListener(requestProcessor, objectMapper) + } + + @Test + fun shouldProcessMtbFileRequest() { + val mtbFile = MtbFile.builder() + .withPatient(Patient.builder().withId("DUMMY_12345678").build()) + .withConsent(Consent.builder().withStatus(Consent.Status.ACTIVE).build()) + .build() + + kafkaInputListener.onMessage(ConsumerRecord("testtopic", 0, 0, "", this.objectMapper.writeValueAsString(mtbFile))) + + verify(requestProcessor, times(1)).processMtbFile(any()) + } + + @Test + fun shouldProcessDeleteRequest() { + val mtbFile = MtbFile.builder() + .withPatient(Patient.builder().withId("DUMMY_12345678").build()) + .withConsent(Consent.builder().withStatus(Consent.Status.REJECTED).build()) + .build() + + kafkaInputListener.onMessage(ConsumerRecord("testtopic", 0, 0, "", this.objectMapper.writeValueAsString(mtbFile))) + + verify(requestProcessor, times(1)).processDeletion(anyString()) + } + +}
\ No newline at end of file |