// Setup mock configuration with lenient stubbing
        lenient().when(mockContext.getConfig()).thenReturn(mockConfig);
        lenient().when(mockConfig.getWitnessHeartbeatTimeout()).thenReturn(120000L);
        lenient().when(mockConfig.getWitnessRegistrationTimeout()).thenReturn(300000L);
        lenient().when(mockConfig.getWitnessReconnectDelay()).thenReturn(1000L);

    @Mock
    SmbTreeHandleImpl tree;

    private void stubValidTree(long treeId, boolean connected, boolean smb2) {
        lenient().when(tree.acquire()).thenReturn(tree);
        lenient().when(tree.getTreeId()).thenReturn(treeId);
        lenient().when(tree.isConnected()).thenReturn(connected);
        lenient().when(tree.isSMB2()).thenReturn(smb2);
    }

    @Test

        lenient().when(handle.getTransportContext()).thenCallRealMethod();
        lenient().when(handle.getServer()).thenCallRealMethod();
        lenient().when(handle.getServerWithDfs()).thenCallRealMethod();
        lenient().when(handle.getSessionKey()).thenCallRealMethod();
        lenient().when(handle.getBinding()).thenReturn(mockDcerpcBinding);

        lenient().when(request.getSession()).thenReturn(session);
        lenient().when(request.getSession(false)).thenReturn(session);
        lenient().when(request.getSession(true)).thenReturn(session);

        // Setup default session attribute behavior
        lenient().when(session.getAttribute(anyString())).thenReturn(null);

        // Setup SmbFile mock

        // Setup mock configuration with lenient stubbing
        lenient().when(mockContext.getConfig()).thenReturn(mockConfig);
        lenient().when(mockConfig.isUseWitness()).thenReturn(true);
        lenient().when(mockConfig.getWitnessHeartbeatTimeout()).thenReturn(120000L);
        lenient().when(mockConfig.getWitnessRegistrationTimeout()).thenReturn(300000L);
        lenient().when(mockConfig.getWitnessReconnectDelay()).thenReturn(1000L);

        when(cifsContext.getNameServiceClient()).thenReturn(nameServiceClient);
        lenient().when(cifsContext.getConfig()).thenReturn(configuration);
        lenient().when(configuration.getDefaultDomain()).thenReturn("TEST_DOMAIN");
        lenient().when(nameServiceClient.getByName(anyString(), anyBoolean())).thenReturn(mock(Address.class));
        lenient().when(transportPool.getChallenge(any(), any())).thenReturn(new byte[8]);
    }

        FileEntry fe1 = mock(FileEntry.class);
        lenient().when(fe1.getName()).thenReturn("file1");
        lenient().when(fe1.getFileIndex()).thenReturn(1);

        FileEntry fe2 = mock(FileEntry.class);
        lenient().when(fe2.getName()).thenReturn("file2");
        when(fe2.getFileIndex()).thenReturn(2);

        FileEntry fe3 = mock(FileEntry.class);
        lenient().when(fe3.getName()).thenReturn("file3");

    void setUp() {
        lenient().when(mockFileEntry1.getName()).thenReturn("file1.txt");
        lenient().when(mockFileEntry2.getName()).thenReturn("file2.txt");
        lenient().when(mockFileEntry3.getName()).thenReturn("file3.txt");

        lenient().when(mockResource1.getName()).thenReturn("file1.txt");
        lenient().when(mockResource2.getName()).thenReturn("file2.txt");

            when(mockHandle.getMaxRecv()).thenReturn(maxRecv);

            // Use lenient stubbing to handle multiple calls
            lenient().doNothing().when(mockBuffer).enc_ndr_short(anyInt());
            lenient().doNothing().when(mockBuffer).enc_ndr_long(anyInt());
            lenient().doNothing().when(mockBuffer).enc_ndr_small(anyInt());

    @BeforeEach
    void setUp() {
        // Setup mock context with lenient stubbing to avoid UnnecessaryStubbingException
        lenient().when(mockContext.getConfig()).thenReturn(mockConfig);
        lenient().when(mockContext.getNameServiceClient()).thenReturn(mockNameServiceClient);
        lenient().when(mockNameServiceClient.getLocalHost()).thenReturn(mockLocalHost);
        lenient().when(mockLocalHost.getHostName()).thenReturn("localhost");