when(mockTransportPool.getSmbTransport(any(), anyString(), anyInt(), anyBoolean(), anyBoolean())).thenReturn(mockTransport);
        when(mockTransport.unwrap(any())).thenReturn(mockTransport);
        when(mockTransport.getDfsReferrals(any(), anyString(), anyString(), anyString(), anyInt())).thenReturn(null);
        assertNull(dfsResolver.getDc(mockContext, "anyDomain"));
    }

    }

    @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];

        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();

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

        // Setup for ensureOpen
        when(tree.isSMB2()).thenReturn(true);
        when(pipe.openUnshared(anyString(), anyInt(), anyInt(), anyInt(), anyInt(), anyInt())).thenReturn(fh);
        when(fh.acquire()).thenReturn(fh);
        lenient().when(fh.isValid()).thenReturn(true);
        lenient().when(fh.getTree()).thenReturn(tree);

        when(tree.getSendBufferSize()).thenReturn(1024);
        when(tree.hasCapability(anyInt())).thenAnswer(inv -> ntSmbsCap && inv.getArgument(0).equals(SmbConstants.CAP_NT_SMBS));
        when(tree.areSignaturesActive()).thenReturn(false);
        when(tree.isSMB2()).thenReturn(smb2);

        when(file.openUnshared(anyInt(), anyInt(), anyInt(), anyInt(), anyInt())).thenReturn(fh);
        when(fh.acquire()).thenReturn(fh);

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();

            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

        // 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();