result.onRequestStart(CloseSocket())
    }

    SocketPolicy.DISCONNECT_AFTER_REQUEST -> {
      result.onResponseStart(CloseSocket())
    }

    SocketPolicy.DISCONNECT_DURING_REQUEST_BODY -> {
      result.onRequestBody(CloseSocket())
    }

    SocketPolicy.DISCONNECT_DURING_RESPONSE_BODY -> {
      result.onResponseBody(CloseSocket())
    }

import kotlin.test.assertFailsWith
import mockwebserver3.Dispatcher
import mockwebserver3.MockResponse
import mockwebserver3.MockWebServer
import mockwebserver3.RecordedRequest
import mockwebserver3.SocketEffect.CloseSocket
import mockwebserver3.SocketEffect.Stall
import mockwebserver3.junit5.StartStop
import okhttp3.EventRecorder
import okhttp3.Interceptor
import okhttp3.OkHttpClient
import okhttp3.OkHttpClientTestRule

import kotlin.time.Duration
import kotlin.time.Duration.Companion.milliseconds
import kotlin.time.measureTime
import mockwebserver3.MockResponse
import mockwebserver3.MockWebServer
import mockwebserver3.SocketEffect.CloseSocket
import mockwebserver3.SocketEffect.ShutdownConnection
import mockwebserver3.junit5.StartStop
import okhttp3.Headers.Companion.headersOf
import okhttp3.HttpUrl.Companion.toHttpUrl

import java.util.concurrent.CountDownLatch
import java.util.concurrent.TimeUnit
import kotlin.test.assertFailsWith
import mockwebserver3.MockResponse
import mockwebserver3.MockWebServer
import mockwebserver3.SocketEffect.CloseSocket
import mockwebserver3.junit5.StartStop
import okhttp3.CallEvent.CacheConditionalHit
import okhttp3.CallEvent.CacheHit
import okhttp3.CallEvent.CacheMiss
import okhttp3.CallEvent.CallEnd

  @Test
  fun serverShutdownInput() {
    testServerClosesOutput(
      MockResponse
        .Builder()
        .body("This connection won't pool properly")
        .onResponseEnd(
          CloseSocket(
            closeSocket = false,
            shutdownInput = true,
          ),
        ).build(),
    )
  }

  @Test
  fun serverShutdownOutput() {
    testServerClosesOutput(
      MockResponse

          if (peek().onResponseStart is CloseSocket) {
            requestFinished.await()
          }
          return super.dispatch(request)
        }
      }
    dispatcher.enqueue(MockResponse(body = "seed connection pool"))
    dispatcher.enqueue(MockResponse.Builder().onResponseStart(CloseSocket()).build())
    dispatcher.enqueue(MockResponse(body = "retry success"))