org.http4k.connect.amazon.sqs.FakeSQS.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of http4k-connect-amazon-sqs-fake Show documentation

null

There is a newer version: 5.35.2.0

Show newest version

package org.http4k.connect.amazon.sqs

import org.http4k.aws.AwsCredentials
import org.http4k.chaos.ChaoticHttpHandler
import org.http4k.chaos.start
import org.http4k.connect.amazon.AmazonRestfulFake
import org.http4k.connect.amazon.core.model.AwsAccount
import org.http4k.connect.amazon.core.model.AwsService
import org.http4k.connect.amazon.core.model.Region
import org.http4k.connect.amazon.sqs.model.SQSMessage
import org.http4k.connect.storage.InMemory
import org.http4k.connect.storage.Storage
import org.http4k.core.Method.POST
import org.http4k.routing.bind
import org.http4k.routing.routes

class FakeSQS(
    private val queues: Storage> = Storage.InMemory(),
    awsAccount: AwsAccount = AwsAccount.of("1234567890"),
    private val region: Region = Region.of("ldn-north-1")
) : ChaoticHttpHandler() {

    private val api = AmazonRestfulFake(SqsMoshi, AwsService.of("sqs"), region, awsAccount)

    override val app = routes(
        "/" bind POST to routes(
            api.deleteMessage(queues),
            api.deleteQueue(queues),
            api.deleteMessageBatch(queues),
            api.receiveMessage(queues),
            api.createQueue(queues, awsAccount),
            api.getQueueAttributes(queues),
            api.sendMessage(queues),
            api.sendMessageBatch(queues),
            api.listQueues(region, awsAccount, queues)
        )
    )

    /**
     * Convenience function to get a SQS client
     */
    fun client() = SQS.Http(region, { AwsCredentials("accessKey", "secret") }, this)
}

fun main() {
    FakeSQS().start()
}