jvmMain.com.apollographql.apollo3.mockserver.MockServer.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of apollo-mockserver Show documentation

Apollo Mockserver

There is a newer version: 4.0.0-beta.7

package com.apollographql.apollo3.mockserver

import okhttp3.Headers
import okhttp3.mockwebserver.MockWebServer
import okio.Buffer
import java.util.concurrent.TimeUnit

actual class MockServer : MockServerInterface {
  private val mockWebServer = MockWebServer()

  override fun enqueue(mockResponse: MockResponse) {
    mockWebServer.enqueue(
        okhttp3.mockwebserver.MockResponse()
            .setResponseCode(mockResponse.statusCode)
            .apply {
              mockResponse.headers.forEach {
                addHeader(it.key, it.value)
              }
            }.setBody(Buffer().apply { write(mockResponse.body) })
            .setHeadersDelay(mockResponse.delayMillis, TimeUnit.MILLISECONDS)
    )
  }

  override fun takeRequest(): MockRecordedRequest {
    return mockWebServer.takeRequest(10, TimeUnit.MILLISECONDS)?.let {
      MockRecordedRequest(
          method = it.method!!,
          path = it.path!!,
          version = parseRequestLine(it.requestLine).third,
          headers = it.headers.toMap(),
          body = it.body.readByteString()
      )
    } ?: error("No recorded request")
  }

  private fun Headers.toMap(): Map = (0.until(size)).map {
    name(it) to get(name(it))!!
  }.toMap()

  override suspend fun url(): String {
    return mockWebServer.url("/").toString()
  }

  override suspend fun stop() {
    mockWebServer.shutdown()
  }
}