when(request.getNext()).thenReturn(nextRequest);
        when(nextRequest.getNext()).thenReturn(thirdRequest);
        when(thirdRequest.getNext()).thenReturn(null);
        when(request.allowChain(nextRequest)).thenReturn(true);
        when(nextRequest.allowChain(thirdRequest)).thenReturn(true);

        // When

        SmbTreeHandleImpl dh = mock(SmbTreeHandleImpl.class);
        when(sh.isSMB2()).thenReturn(true);
        when(dh.isSMB2()).thenReturn(true);
        when(sh.isSameTree(dh)).thenReturn(true);

        // Source open returns a handle that reports size 0
        SmbFileHandleImpl sfd = mock(SmbFileHandleImpl.class);
        when(sfd.getInitialSize()).thenReturn(0L);

        when(mockData.getServer()).thenReturn("SERVER");
        when(mockData.getShare()).thenReturn("share");
        when(mockData.getPath()).thenReturn("path");
        when(mockData.getExpiration()).thenReturn(System.currentTimeMillis() + 60000);
        when(mockData.getPathConsumed()).thenReturn(0);
        when(mockData.getDomain()).thenReturn("DOMAIN");
        return (DfsReferralDataImpl) base.combine(mockData);

        filter = new NtlmHttpFilter();

        // Setup lenient stubs for common mock interactions
        lenient().when(request.getSession()).thenReturn(httpSession);
        lenient().when(request.getSession(anyBoolean())).thenReturn(httpSession);
        lenient().when(response.getOutputStream()).thenReturn(servletOutputStream);
    }

    @Test
    void testInit_success() throws ServletException {

        jcifs.SID mockSid2 = mock(jcifs.SID.class);
        jcifs.SID mockSid3 = mock(jcifs.SID.class);
        when(mockSid1.unwrap(sid_t.class)).thenReturn(mockSidT);
        when(mockSid2.unwrap(sid_t.class)).thenReturn(mockSidT);
        when(mockSid3.unwrap(sid_t.class)).thenReturn(mockSidT);
        testSids = new jcifs.SID[] { mockSid1, mockSid2, mockSid3 };

        // Act

        // Only set up parent mocks when they're actually needed for URL creation
        when(parent.getContext()).thenReturn(this.ctx);
        when(parent.getLocator()).thenReturn(parentLocator);
        // Make the parent appear as a workgroup to use the simpler URL code-path
        when(parentLocator.isWorkgroup()).thenReturn(true);
    }

    /** Simple implementation of FileEntry for controlled inputs in tests. */

        lenient().when(mockNdrBuffer.derive(anyInt())).thenReturn(mockDeferredNdrBuffer);
        lenient().when(mockDeferredNdrBuffer.derive(anyInt())).thenReturn(mockDeferredNdrBuffer);

        // Set up lenient stubs for common operations to avoid unnecessary stubbing exceptions
        lenient().when(mockNdrBuffer.getIndex()).thenReturn(0);
        lenient().when(mockDeferredNdrBuffer.getIndex()).thenReturn(0);
        // Mock the index field access

        });
    }

    @Test
    @DisplayName("Should get server")
    void testGetServer() {
        // Given
        String server = "testserver";
        when(mockReferralData.getServer()).thenReturn(server);

        // When
        String result = mockReferralData.getServer();

        // Then
        assertEquals(server, result);
        verify(mockReferralData).getServer();
    }

    @Test

        when(mockConfig.isDfsDisabled()).thenReturn(false);
        when(mockConfig.isEncryptionEnabled()).thenReturn(true);
        when(mockConfig.getMinimumVersion()).thenReturn(DialectVersion.SMB202);
        when(mockConfig.getMaximumVersion()).thenReturn(DialectVersion.SMB311);
        when(mockConfig.getTransactionBufferSize()).thenReturn(65536);
        when(mockConfig.getReceiveBufferSize()).thenReturn(65536);

        when(mockChannelManager.getHealthyChannels()).thenReturn(Collections.singletonList(channel1));

        ChannelInfo selected = loadBalancer.selectChannel(mockMessage);
        assertEquals(channel1, selected);
    }

    @Test
    void testNoChannelsAvailable() {
        when(mockChannelManager.getHealthyChannels()).thenReturn(Collections.emptyList());