io.javalin.mock.servlet.InMemoryPart.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.servlet

import jakarta.servlet.http.Part
import java.io.InputStream
import java.nio.file.Files
import java.nio.file.Paths

// @formatter:off
class InMemoryPart(private val state: PartState) : Part {

    class PartState @JvmOverloads constructor(
        @JvmField val name: String,
        @JvmField val fileName: String,
        @JvmField val content: ByteArray,
        @JvmField val size: Long = content.size.toLong(),
        @JvmField val contentType: String = "application/octet-stream",
        @JvmField val headers: Map> = emptyMap()
    )

    override fun getInputStream(): InputStream = state.content.inputStream()
    override fun getContentType(): String = state.contentType
    override fun getName(): String = state.name
    override fun getSubmittedFileName(): String = state.fileName
    override fun getSize(): Long = state.size
    override fun write(fileName: String) { Files.write(Paths.get(fileName), state.content) }
    override fun delete() {}
    override fun getHeader(name: String): String? = state.headers[name]?.firstOrNull()
    override fun getHeaders(name: String): Collection = state.headers[name] ?: emptyList()
    override fun getHeaderNames(): Collection = state.headers.keys

}
// @formatter:on