doNothing().when(trans1).close();

        // Return mocked transports
        doReturn(trans1).when(poolSpy).getSmbTransport(eq(ctx), eq(addr1), anyInt(), anyBoolean(), anyBoolean());
        doReturn(trans2).when(poolSpy).getSmbTransport(eq(ctx), eq(addr2), anyInt(), anyBoolean(), anyBoolean());

        // When: Get transport by name
        SmbTransportImpl result = poolSpy.getSmbTransport(ctx, "test.server", 445, false, false);

        lenient().when(handle.getBinding()).thenReturn(mockDcerpcBinding);
        lenient().doCallRealMethod().when(handle).doSendReceiveFragment(any(byte[].class), anyInt(), anyInt(), any(byte[].class));
        lenient().doCallRealMethod().when(handle).doSendFragment(any(byte[].class), anyInt(), anyInt());
        lenient().doCallRealMethod().when(handle).doReceiveFragment(any(byte[].class));
        lenient().doCallRealMethod().when(handle).close();

    }

    @Test
    @DisplayName("readn should handle multiple read calls")
    void testReadnMultipleCalls() throws IOException {
        when(mockInputStream.read(any(byte[].class), anyInt(), anyInt())).thenReturn(2) // First read returns 2 bytes
                .thenReturn(2) // Second read returns 2 bytes
                .thenReturn(1); // Third read returns 1 byte

        byte[] buffer = new byte[10];

import static org.junit.jupiter.api.Assertions.assertTrue;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.ArgumentMatchers.anyBoolean;
import static org.mockito.ArgumentMatchers.anyInt;
import static org.mockito.ArgumentMatchers.eq;
import static org.mockito.ArgumentMatchers.isNull;
import static org.mockito.Mockito.doThrow;
import static org.mockito.Mockito.times;
import static org.mockito.Mockito.verify;

            // Given
            when(mockResource.openInputStream()).thenReturn(mockInputStream);
            when(mockResource.openInputStream(anyInt())).thenReturn(mockInputStream);
            when(mockResource.openInputStream(anyInt(), anyInt(), anyInt())).thenReturn(mockInputStream);

            // When
            InputStream is1 = mockResource.openInputStream();

        // Set the deferred field directly since it's a public field, not a method
        mockNdrBuffer.deferred = mockDeferredBuffer;
        lenient().when(mockNdrBuffer.derive(anyInt())).thenReturn(mockDeferredBuffer);
        lenient().when(mockDeferredBuffer.derive(anyInt())).thenReturn(mockDeferredBuffer);
    }

    @Test
    void testGetSyntax() {
        String syntax = srvsvc.getSyntax();

            CreateContextResponse response = mock(CreateContextResponse.class);
            byte[] buffer = new byte[100];
            when(response.decode(any(byte[].class), anyInt(), anyInt())).thenReturn(10);

            int result = response.decode(buffer, 0, 10);
            assertEquals(10, result);
            verify(response, times(1)).decode(buffer, 0, 10);
        }

        @Test

        void shouldHandleMockedDecodeMethod() throws jcifs.internal.SMBProtocolDecodingException {
            // Given
            byte[] buffer = new byte[100];
            when(mockAllocInfo.decode(any(byte[].class), anyInt(), anyInt())).thenReturn(42);

            // When
            int result = mockAllocInfo.decode(buffer, 10, 50);

            // Then
            assertEquals(42, result);

        when(mockFileHandle.acquire()).thenReturn(mockFileHandle);

        // Mock for ensureOpen to reopen file
        when(mockFile.openUnshared(anyInt(), anyInt(), anyInt(), anyInt(), anyInt())).thenReturn(mockFileHandle);

        when(mockTreeHandle.send(any(Smb2WriteRequest.class), any())).thenReturn(mockWriteResponse);
        when(mockWriteResponse.getCount()).thenReturn(1);

        when(tree.acquire()).thenReturn(tree);
        when(tree.isSMB2()).thenReturn(true);

        // SMB2 branch in ensureOpen uses openUnshared(String, ...)
        when(pipe.openUnshared(anyString(), anyInt(), anyInt(), anyInt(), anyInt(), anyInt())).thenReturn(fileHandle);
        when(fileHandle.acquire()).thenReturn(fileHandle);
        when(fileHandle.isValid()).thenReturn(true);
        when(fileHandle.getTree()).thenReturn(tree);