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.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

import java.util.concurrent.TimeUnit
import javax.net.ssl.SSLException
import kotlin.test.assertFailsWith
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.MediaType.Companion.toMediaType

          stream.connection.shutdown(ErrorCode.NO_ERROR)
        } else {
          socket.close()
        }
      }

      is CloseSocket -> {
        if (effect.shutdownInput) socket.shutdownInput()
        if (effect.shutdownOutput) socket.shutdownOutput()
        if (effect.closeSocket) socket.close()
      }

      Stall -> {
        // Sleep until the socket is closed.

  @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"))