io.javalin.mock.Body.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of javalin-context-mock Show documentation
There is a newer version: 6.3.0
package io.javalin.mock

import io.javalin.http.ContentType
import io.javalin.json.JsonMapper
import io.javalin.json.toJsonString
import java.io.InputStream

interface Body {

    fun init(mockConfig: ContextMockConfig) {}
    fun toInputStream(): InputStream
    fun getContentType(): String?
    fun getContentLength(): Long?

    companion object {

        private abstract class AbstractBody(private val contentType: String?, private val contentLength: Long? = null) : Body {
            override fun getContentType(): String? = contentType
            override fun getContentLength(): Long? = contentLength
        }

        @JvmStatic
        @JvmOverloads
        fun ofString(
            body: String,
            contentType: String? = ContentType.PLAIN,
            contentLength: Long? = body.length.toLong()
        ): Body =
            object : AbstractBody(contentType = contentType, contentLength = contentLength) {
                override fun toInputStream(): InputStream = body.byteInputStream()
            }

        @JvmStatic
        @JvmOverloads
        fun ofInputStream(
            body: InputStream,
            contentType: String? = ContentType.OCTET_STREAM,
            contentLength: Long? = null
        ): Body =
            object : AbstractBody(contentType = contentType) {
                val content = body.use { it.readAllBytes() }
                override fun toInputStream(): InputStream = content.inputStream()
                override fun getContentLength(): Long = contentLength ?: content.size.toLong()
            }

        @JvmStatic
        @JvmOverloads
        fun ofObject(
            body: Any,
            contentType: String? = ContentType.JSON,
            contentLength: Long? = null
        ): Body =
            object : AbstractBody(contentType = contentType, contentLength = contentLength) {
                private lateinit var content: ByteArray
                override fun init(mockConfig: ContextMockConfig) {
                    this.content = mockConfig.javalinConfig.pvt.jsonMapper.value.toJsonString(body).toByteArray()
                }
                override fun toInputStream(): InputStream = content.inputStream()
                override fun getContentLength(): Long = content.size.toLong()
            }

    }

}