when(mockTreeHandle.getSendBufferSize()).thenReturn(65536);
        when(mockFileHandle.isValid()).thenReturn(false, true); // First false to trigger ensureOpen, then true
        when(mockFileHandle.getFileId()).thenReturn(new byte[16]);
        when(mockFileHandle.acquire()).thenReturn(mockFileHandle);

        // Mock for ensureOpen to reopen file

        HandleInfo info = new HandleInfo("/test/file.txt", testGuid, new byte[16], HandleType.DURABLE_V2, 120000, null);

        assertArrayEquals(new byte[16], info.getFileId());

        info.updateFileId(testFileId);
        assertArrayEquals(testFileId, info.getFileId());

        // Test invalid file ID length
        assertThrows(IllegalArgumentException.class, () -> {
            info.updateFileId(new byte[8]);
        });

            if (cr != null && cr.isReceived() && cr.getStatus() == NtStatus.NT_STATUS_SUCCESS) {
                try {
                    th.send(new Smb2CloseRequest(th.getConfig(), cr.getFileId()));
                } catch (final SmbException e2) {
                    e.addSuppressed(e2);
                }
            }

            final Smb2QueryDirectoryResponse qr = query.getResponse();

        Configuration responseConfig = (Configuration) configField.get(response);
        assertEquals(mockConfig, responseConfig);

        // Verify response has correct fileId
        assertEquals(testFileId, response.getFileId());

        // Verify response has correct fileName
        assertEquals(testFileName, response.getFileName());
    }

    @Test
    @DisplayName("size should return correct message size")

                }

                if (dh.isSMB2()) {
                    final Smb2SetInfoRequest req = new Smb2SetInfoRequest(dh.getConfig(), dfd.getFileId());
                    req.setFileInformation(new FileBasicInfo(ctime, atime, mtime, 0L, attrs));
                    dh.send(req);
                } else if (dh.hasCapability(SmbConstants.CAP_NT_SMBS)) {

        log.debug("Performing reconnection for handle: {} (type: {})", info.getPath(), info.getType());

        // Create reconnect context
        DurableHandleReconnect reconnectCtx = new DurableHandleReconnect(info.getFileId());

        // This would typically involve:
        // 1. Creating a new Smb2CreateRequest with reconnect context
        // 2. Adding lease context if needed

            do {
                r = len > this.readSize ? this.readSize : len;

                if (th.isSMB2()) {
                    final Smb2ReadRequest request = new Smb2ReadRequest(th.getConfig(), fh.getFileId(), b, off);
                    request.setOffset(this.fp);
                    request.setReadLength(r);
                    request.setRemainingBytes(len - off);
                    try {

            CommonServerMessageBlockRequest req;
            NotifyResponse resp = null;
            if (th.isSMB2()) {
                final Smb2ChangeNotifyRequest r = new Smb2ChangeNotifyRequest(th.getConfig(), this.handle.getFileId());
                r.setCompletionFilter(this.filter);
                r.setNotifyFlags(this.recursive ? Smb2ChangeNotifyRequest.SMB2_WATCH_TREE : 0);
                req = r;
            } else {

        when(handle.getTree()).thenReturn(tree);
        when(tree.isSMB2()).thenReturn(true);
        when(tree.getConfig()).thenReturn(mock(Configuration.class));
        when(handle.getFileId()).thenReturn(fileId);
        when(tree.send(any(CommonServerMessageBlockRequest.class), any(), any(), any())).thenReturn(resp);
    }

    // Prepare SMB1 flow with a given response

        resp.decode(packet, 0);
        assertEquals(Smb2IoctlRequest.FSCTL_SRV_COPYCHUNK, resp.getCtlCode());
        assertEquals(0x1234, resp.getIoctlFlags());
        assertNotNull(resp.getFileId());
        assertEquals(16, resp.getFileId().length);
        assertEquals(12, resp.getOutputLength());

        SrvCopyChunkCopyResponse od = resp.getOutputData(SrvCopyChunkCopyResponse.class);
        assertEquals(1, od.getChunksWritten());