when(mockAllocInfo.getFree()).thenReturn(400L);

        assertEquals(1000L, mockAllocInfo.getCapacity(), "capacity should match stubbed value");
        assertEquals(400L, mockAllocInfo.getFree(), "free space should match stubbed value");
        verify(mockAllocInfo, times(1)).getCapacity();
        verify(mockAllocInfo, times(1)).getFree();
    }

    /**

    }

    /**
     * Test that getTree() returns the correct SmbTreeHandle.
     */
    @Test
    void testGetTree() {
        when(smbFileHandle.getTree()).thenReturn(smbTreeHandle);
        SmbTreeHandle result = smbFileHandle.getTree();
        assertEquals(smbTreeHandle, result, "getTree should return the mocked SmbTreeHandle.");
        verify(smbFileHandle, times(1)).getTree();
    }

    /**

        SmbInfoAllocation info = (SmbInfoAllocation) response.getInfo();
        assertEquals(8L * 1000000L * 512L, info.getCapacity());
        assertEquals(8L * 500000L * 512L, info.getFree());
    }

    @Test
    void testReadDataWireFormat_FileFsSizeInformation() throws Exception {
        // Test reading FS_SIZE_INFO data

        } else if (requestType.equals(MetadataResolutionRequestTypeEnum.versionedGraph)) {
            return new MetadataGraph(getTree(), true, false);
        } else if (requestType.equals(MetadataResolutionRequestTypeEnum.scopedGraph)) {
            return new MetadataGraph(getTree(), true, true);
        }
        return null;
    }
    // ----------------------------------------------------------------------------

        SmbPipeInputStream stream = newStreamWithMinimalStubs(true);

        // Arrange mocks for available()
        when(handle.ensureOpen()).thenReturn(fd);
        when(fd.getTree()).thenReturn(tree);
        when(tree.isSMB2()).thenReturn(true);
        when(tree.getConfig()).thenReturn(config);
        when(config.getTransactionBufferSize()).thenReturn(65535);

    // Prepare SMB2 flow with a given response
    private void setupSmb2(NotifyResponse resp, byte[] fileId) throws Exception {
        when(handle.isValid()).thenReturn(true);
        when(handle.getTree()).thenReturn(tree);
        when(tree.isSMB2()).thenReturn(true);
        when(tree.getConfig()).thenReturn(mock(Configuration.class));
        when(handle.getFileId()).thenReturn(fileId);

    }

    @Test
    @DisplayName("getTree() acquires and returns same tree instance")
    void getTree_acquiresAndReturnsTree() {
        when(cfg.isTraceResourceUsage()).thenReturn(false);
        stubValidTree(7L, true, true);

        SmbFileHandleImpl h = new SmbFileHandleImpl(cfg, 123, tree, "//server/share", 0, 0, 0, 0, 0L);

        SmbTreeHandleImpl t1 = h.getTree();
        SmbTreeHandleImpl t2 = h.getTree();

        assertEquals(512, info.bytesPerSect);
        assertEquals(1000L * 100 * 512, info.getCapacity(), "Capacity calculation should be correct");
        assertEquals(500L * 100 * 512, info.getFree(), "Free space calculation should be correct");
    }

    /**
     * Tests the readDataWireFormat method with the SMB_QUERY_FS_SIZE_INFO information level.
     */
    @Test

        when(fileHandle.acquire()).thenReturn(fileHandle);
        when(fileHandle.isValid()).thenReturn(true);
        when(fileHandle.getTree()).thenReturn(tree);

        // Initially not open (isStale returns false when handle is null, not true)
        assertFalse(target.isOpen());
        assertFalse(target.isStale());

            assertEquals(20, bytesDecoded);
            assertEquals(8L * 4096L * 512L, smbInfoAllocation.getCapacity());
            assertEquals(8L * 2048L * 512L, smbInfoAllocation.getFree());
        }

        @Test
        @DisplayName("Should calculate capacity correctly")
        void testCapacityCalculation() throws SMBProtocolDecodingException {
            byte[] buffer = new byte[24];