run.qontract.stub.KafkaStub.kt Maven / Gradle / Ivy

Go to download
package run.qontract.stub

import org.apache.kafka.clients.producer.Producer
import org.apache.kafka.clients.producer.ProducerRecord
import run.qontract.core.Feature
import run.qontract.core.value.KafkaMessage
import run.qontract.mock.ScenarioStub

fun stubKafkaContracts(kafkaStubs: List, bootstrapServers: String, createTopics: (List, String) -> Unit, createProducer: (String) -> Producer) {
    createTopics(kafkaStubs.map { it.kafkaMessage.topic }, bootstrapServers)

    createProducer(bootstrapServers).use { producer ->
        for(stub in kafkaStubs) {
            val producerRecord = producerRecord(stub.kafkaMessage)
            val future = producer.send(producerRecord)
            future.get()
        }
    }
}

fun producerRecord(kafkaMessage: KafkaMessage): ProducerRecord {
    return when(val key = kafkaMessage.key) {
        null -> ProducerRecord(kafkaMessage.topic, kafkaMessage.value.toStringValue())
        else -> ProducerRecord(kafkaMessage.topic, key.toStringValue(), kafkaMessage.value.toStringValue())
    }
}

fun contractInfoToKafkaExpectations(contractInfo: List>>): List {
    return contractInfo.flatMap { (_, mocks) ->
        mocks.mapNotNull { it.kafkaMessage }.fold(emptyList()) { innerStubs, kafkaMessage ->
            innerStubs.plus(KafkaStubData(kafkaMessage))
        }
    }
}