when(tree.send(any(Trans2FindFirst2.class), any(Trans2FindFirst2Response.class))).thenAnswer((InvocationOnMock inv) -> {
            Trans2FindFirst2Response resp = inv.getArgument(1);
            // First response content
            FE[] res = java.util.Arrays.stream(batches.get(0)).map(FE::new).toArray(FE[]::new);

        final int[] queryCount = { 0 };

        // Mock send(create) to wire a query response with fe1, fe2
        doAnswer(inv -> {
            Object arg = inv.getArgument(0);
            if (arg instanceof Smb2CreateRequest) {
                Smb2CreateRequest create = (Smb2CreateRequest) arg;
                // Initialize response chain (also for chained query)

            // Mock initial receive with valid PDU header
            when(mockSmbPipeHandleInternal.recv(buf, 0, buf.length)).thenAnswer(invocation -> {
                byte[] buffer = invocation.getArgument(0);
                buffer[0] = 5; // Valid PDU version
                buffer[1] = 0; // Valid PDU type
                Encdec.enc_uint16le((short) 50, buffer, 8); // Fragment length

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

        // Mock both sends and return response
        when(tree.send(any(TransWaitNamedPipe.class), any(TransWaitNamedPipeResponse.class))).thenAnswer(inv -> inv.getArgument(1));
        when(tree.send(any(TransCallNamedPipe.class), any(TransCallNamedPipeResponse.class))).thenAnswer(inv -> inv.getArgument(1));

        int n = handle.sendrecv(new byte[3], 0, 3, new byte[8], 64);
        assertEquals(0, n, "Default response length is 0 unless protocol fills it");