* limitations under the License.
 */
package mockwebserver3.junit5.internal

import okhttp3.ExperimentalOkHttpApi

@ExperimentalOkHttpApi
annotation class MockWebServerInstance(
  val name: String,

  var defaultInstancePort: Int = -1
  var instanceAPort: Int = -1
  var instanceBPort: Int = -1

  @BeforeEach
  fun setup(
    defaultInstance: MockWebServer,
    @MockWebServerInstance("A") instanceA: MockWebServer,
    @MockWebServerInstance("B") instanceB: MockWebServer,
  ) {
    defaultInstancePort = defaultInstance.port
    instanceAPort = instanceA.port
    instanceBPort = instanceB.port

  fun test() {
    ...
  }
}
```

Multiple instances can be obtained by naming additional ones:

```
class MyTest(
  private val server: MockWebServer,
  @MockWebServerInstance("server2") private val server2: MockWebServer,
  @MockWebServerInstance("server3") private val server3: MockWebServer
) {
  @Test
  fun test() {
    ...
  }
}
```

Requirements
------------

 *
 * There are 3 ids for instances
 * - The test instance default (passed into constructor)
 * - The test lifecycle default (passed into test method, plus @BeforeEach, @AfterEach)
 * - named instances with @MockWebServerInstance.
 */
@ExperimentalOkHttpApi
class MockWebServerExtension :
  BeforeEachCallback, AfterEachCallback, ParameterResolver, BeforeAllCallback, AfterAllCallback {
  private val ExtensionContext.resource: ServersForTest

import java.net.Proxy
import java.net.SocketTimeoutException
import mockwebserver3.MockResponse
import mockwebserver3.MockWebServer
import mockwebserver3.SocketPolicy.ResetStreamAtStart
import mockwebserver3.junit5.internal.MockWebServerInstance
import okhttp3.internal.http.RecordingProxySelector
import okhttp3.internal.http2.ErrorCode
import okhttp3.testing.PlatformRule
import org.junit.jupiter.api.BeforeEach
import org.junit.jupiter.api.Test

  private lateinit var client: OkHttpClient
  private lateinit var cache: Cache
  private val cookieManager = CookieManager()

  @BeforeEach
  fun setUp(
    @MockWebServerInstance(name = "1") server: MockWebServer,
    @MockWebServerInstance(name = "2") server2: MockWebServer,
  ) {
    this.server = server
    this.server2 = server2
    platform.assumeNotOpenJSSE()
    server.protocolNegotiationEnabled = false

import mockwebserver3.SocketPolicy.FailHandshake
import mockwebserver3.SocketPolicy.ShutdownInputAtEnd
import mockwebserver3.SocketPolicy.ShutdownOutputAtEnd
import mockwebserver3.junit5.internal.MockWebServerInstance
import okhttp3.Credentials.basic
import okhttp3.Headers.Companion.headersOf
import okhttp3.HttpUrl.Companion.toHttpUrl
import okhttp3.HttpUrl.Companion.toHttpUrlOrNull
import okhttp3.MediaType.Companion.toMediaType

import mockwebserver3.SocketPolicy.HalfCloseAfterRequest
import mockwebserver3.SocketPolicy.NoResponse
import mockwebserver3.SocketPolicy.StallSocketAtStart
import mockwebserver3.junit5.internal.MockWebServerInstance
import okhttp3.CallEvent.CallEnd
import okhttp3.CallEvent.ConnectStart
import okhttp3.CallEvent.ConnectionAcquired
import okhttp3.CallEvent.ConnectionReleased
import okhttp3.CallEvent.ResponseFailed