import okio.Buffer
import okio.ByteString

internal fun Dispatcher.wrap(): mockwebserver3.Dispatcher {
  if (this is QueueDispatcher) return this.delegate

  val delegate = this
  return object : mockwebserver3.Dispatcher() {
    override fun dispatch(request: mockwebserver3.RecordedRequest): mockwebserver3.MockResponse = delegate.dispatch(request.unwrap()).wrap()

    override fun peek(): mockwebserver3.MockResponse = delegate.peek().wrap()

        // Configure the mock to throw DcerpcException when unwrap is called
        doThrow(new DcerpcException("Test unwrap exception")).when(dcerpcSecurityProvider).unwrap(mockNdrBuffer);

        // Assert that calling unwrap throws DcerpcException
        assertThrows(DcerpcException.class, () -> dcerpcSecurityProvider.unwrap(mockNdrBuffer));

        // Verify that the unwrap method was called exactly once

  }

  @Throws(InterruptedException::class)
  fun takeRequest(): RecordedRequest = delegate.takeRequest().unwrap()

  @Throws(InterruptedException::class)
  fun takeRequest(
    timeout: Long,
    unit: TimeUnit,
  ): RecordedRequest? = delegate.takeRequest(timeout, unit)?.unwrap()

  @JvmName("-deprecated_requestCount")
  fun getRequestCount(): Int = delegate.requestCount

  public void testWrap() {
    assertThat(Primitives.wrap(int.class)).isSameInstanceAs(Integer.class);
    assertThat(Primitives.wrap(Integer.class)).isSameInstanceAs(Integer.class);
    assertThat(Primitives.wrap(String.class)).isSameInstanceAs(String.class);
  }

  public void testUnwrap() {
    assertThat(Primitives.unwrap(Integer.class)).isSameInstanceAs(int.class);
    assertThat(Primitives.unwrap(int.class)).isSameInstanceAs(int.class);

     * @throws DcerpcException if the wrapping operation fails
     */
    void wrap(NdrBuffer outgoing) throws DcerpcException;

    /**
     * Unwraps incoming DCERPC message data after security processing
     * @param incoming the buffer containing data to be unwrapped
     * @throws DcerpcException if the unwrapping operation fails
     */
    void unwrap(NdrBuffer incoming) throws DcerpcException;

        when(mockTransport.unwrap(SmbTransportInternal.class)).thenReturn(internal);
        when(internal.getSmbSession(eq(ctx), eq("server.test"), isNull())).thenReturn(session);
        when(session.unwrap(SmbSessionInternal.class)).thenReturn(session);
        when(session.getSmbTree(eq("IPC$"), isNull())).thenReturn(tree);
        when(tree.unwrap(SmbTreeInternal.class)).thenReturn(tree);

                assertTrue(SmbSession.class.isAssignableFrom((Class<?>) unwrapMethod.getReturnType()),
                        "unwrap method should return SmbSession or subtype");
            }, "unwrap(Class) method should be properly defined");
        }

        @Test
        @DisplayName("Should handle unwrap with SmbSession type")

    }

    @Override
    Collection<V2> createValues() {
      return Collections2.transform(fromTable.values(), function);
    }

    @Override
    public Map<R, Map<C, V2>> rowMap() {
      return Maps.transformValues(fromTable.rowMap(), row -> Maps.transformValues(row, function));
    }

    @Override
    public Map<C, Map<R, V2>> columnMap() {
      return Maps.transformValues(

     * @throws DcerpcException if the wrapping operation fails
     */
    void wrap(NdrBuffer outgoing) throws DcerpcException;

    /**
     * Unwraps incoming DCERPC message data after security processing
     * @param incoming the buffer containing data to be unwrapped
     * @throws DcerpcException if the unwrapping operation fails
     */
    void unwrap(NdrBuffer incoming) throws DcerpcException;

     */
    runConcurrentlyMutatedTest(elements(), ops(add(1), add(2)), wrap);

    runConcurrentlyMutatedTest(elements(), ops(add(1), nop()), wrap);

    runConcurrentlyMutatedTest(elements(), ops(add(1), remove()), wrap);

    runConcurrentlyMutatedTest(elements(), ops(nop(), add(1)), wrap);

    runConcurrentlyMutatedTest(elements(1), ops(remove(), nop()), wrap);

    runConcurrentlyMutatedTest(elements(1), ops(remove(), add(2)), wrap);