}

        @Test
        @DisplayName("Should allow close method to be called multiple times")
        void shouldAllowMultipleCloseCallsOnMock() throws Exception {
            SmbSession mockSession = mock(SmbSession.class);
            doNothing().when(mockSession).close();

            assertDoesNotThrow(() -> {
                mockSession.close();
                mockSession.close();

}

type pooledFlateWriter struct {
	mu sync.Mutex // guards Close and Write
	fw *flate.Writer
}

func (w *pooledFlateWriter) Write(p []byte) (n int, err error) {
	w.mu.Lock()
	defer w.mu.Unlock()
	if w.fw == nil {
		return 0, errors.New("Write after Close")
	}
	return w.fw.Write(p)
}

func (w *pooledFlateWriter) Close() error {
	w.mu.Lock()
	defer w.mu.Unlock()
	var err error
	if w.fw != nil {

    // Confirm all sent messages were received, followed by a client-initiated close.
    val server = serverListener.assertOpen()
    for (i in 0 until messageCount) {
      serverListener.assertBinaryMessage(message)
    }
    serverListener.assertClosing(1001, "")

    // When the server acknowledges the close the connection shuts down gracefully.
    server.close(1000, null)
    clientListener.assertClosing(1000, "")

      @Override
      public void flush() throws IOException {
        delegate.flush();
      }

      @Override
      public void close() throws IOException {
        delegate.close();
      }
    };
  }

  static final class SeparatedBaseEncoding extends BaseEncoding {
    private final BaseEncoding delegate;
    private final String separator;
    private final int afterEveryChars;

        }).when(mockDcerpcHandle).sendrecv(any(MsrpcLsarClose.class));

        // Close once
        handle.close();

        // Act
        // Try to close again
        handle.close();

        // Assert
        // Verify sendrecv was called once for open and only once for close
        verify(mockDcerpcHandle, times(1)).sendrecv(any(MsrpcLsarOpenPolicy2.class));

        try (SmbFileHandleImpl fh = ensureOpen()) {}
    }

    @Override
    public synchronized void close() throws SmbException {
        try {
            if (this.handle != null) {
                try {
                    this.handle.close();
                } catch (final CIFSException e) {
                    throw SmbException.wrap(e);
                }

            verify(transportPool).close();
        }

        @Test
        @DisplayName("Should handle exception during close")
        void testClosePoolException() throws CIFSException {
            // Given
            when(transportPool.close()).thenThrow(new CIFSException("Close failed"));

            // When & Then
            assertThrows(CIFSException.class, () -> transportPool.close());
        }

        @Test

            return invocation.getArgument(1);
        });

        // When: Create iterator and close multiple times
        NetServerEnumIterator iterator = new NetServerEnumIterator(parent, treeHandle, "*", 0, null);

        iterator.close();
        iterator.close(); // Second close should be safe

        // Then: Tree handle should be released only once
        verify(treeHandle, times(1)).release();

              public Closeable call(DeferredCloser closer) throws Exception {
                return closer.eventuallyClose(mockCloseable, executor);
              }
            },
            executor);
    closingFuture.finishToValueAndCloser(
        new NoOpValueAndCloserConsumer<>(), finishToValueAndCloserExecutor);
    assertThrows(
        IllegalStateException.class,
        () ->

    SampleImpl target = new SampleImpl(9999);
    Sample proxy = service.newProxy(target, Sample.class, NOT_ENOUGH_MS, MILLISECONDS);
    Stopwatch stopwatch = Stopwatch.createStarted();

    assertThrows(UncheckedTimeoutException.class, () -> proxy.sleepThenReturnInput("x"));

    assertThat(stopwatch.elapsed(MILLISECONDS)).isIn(Range.closed(NOT_ENOUGH_MS, DELAY_MS * 2));
    // Is it still computing away anyway?