when(transport.disconnect(anyBoolean(), anyBoolean())).thenAnswer(inv -> {
            boolean h = inv.getArgument(0);
            boolean u = inv.getArgument(1);
            // Arbitrary behavior mapping for test purposes
            return (h && !u) || (!h && u);
        });

        boolean result = transport.disconnect(hard, inuse);

        ArgumentCaptor<MsrpcSamrCloseHandle> closeCaptor = ArgumentCaptor.forClass(MsrpcSamrCloseHandle.class);
        doAnswer(invocation -> {
            MsrpcSamrCloseHandle rpc = invocation.getArgument(0);
            rpc.retval = 0; // Simulate success
            return null;
        }).when(mockHandle).sendrecv(closeCaptor.capture());

            // For LM compatibility 3-5, only 16 bytes are used
            doAnswer(invocation -> {
                byte[] dest = invocation.getArgument(2);
                int offset = invocation.getArgument(3);
                System.arraycopy(new byte[16], 0, dest, offset, 16);
                return null;

                    when(mock.read(any(byte[].class), any(int.class), any(int.class))).thenAnswer(inv -> {
                        byte[] buffer = inv.getArgument(0);
                        int offset = inv.getArgument(1);
                        int length = inv.getArgument(2);
                        int remaining = includeContent.length - index[0];
                        if (remaining <= 0) {
                            return -1;

        when(mockProvider.registerMemory(any(ByteBuffer.class), any())).thenAnswer(invocation -> {
            ByteBuffer buffer = invocation.getArgument(0);
            EnumSet<RdmaAccess> access = invocation.getArgument(1);
            return new TcpMemoryRegion(buffer, access);
        });

        bufferManager = new RdmaBufferManager(mockProvider);
    }

    @AfterEach

            when(mockMd5.digest(buffer, 0, expectedOutput.length)).thenAnswer(invocation -> {
                byte[] b = invocation.getArgument(0);
                int off = invocation.getArgument(1);
                int l = invocation.getArgument(2);
                System.arraycopy(expectedOutput, 0, b, off, l);
                return l;
            });

            HMACT64 hmac = new HMACT64(TEST_KEY);

        when(mockClearScrollBuilder.addScrollId(any(String.class))).thenReturn(mockClearScrollBuilder);
        doAnswer(invocation -> {
            ActionListener<ClearScrollResponse> listener = invocation.getArgument(0);
            listener.onResponse(mock(ClearScrollResponse.class));
            return null;
        }).when(mockClearScrollBuilder).execute(any(ActionListener.class));

        doNothing().when(transport).connect();

        // Setup response for tree connect
        doAnswer(invocation -> {
            ServerMessageBlock request = invocation.getArgument(0);
            ServerMessageBlock response = invocation.getArgument(1);
            if (request instanceof SmbComTreeConnectAndX && response instanceof SmbComTreeConnectAndXResponse) {

        when(config.getPid()).thenReturn(1234);

        // Stub send to populate the provided response instance via reflection
        Mockito.doAnswer(inv -> {
            Object resp = inv.getArgument(1);
            // Set private fields: status on SmbComTransactionResponse and available on TransPeekNamedPipeResponse

        lenient().when(servletConfig.getInitParameterNames()).thenReturn(Collections.enumeration(initParams.keySet()));
        lenient().when(servletConfig.getInitParameter(anyString())).thenAnswer(invocation -> initParams.get(invocation.getArgument(0)));

        // Mock HTTP method for request - this is required for HttpServlet.service()
        lenient().when(request.getMethod()).thenReturn("GET");

        // Setup CIFSContext configuration mock