}

    @Test
    @DisplayName("Test constructor with FS_SIZE_INFO")
    void testConstructorWithFsSizeInfo() {
        // Create instance with FS_SIZE_INFO
        trans2QueryFSInfo = new Trans2QueryFSInformation(config, FileSystemInformation.FS_SIZE_INFO);

        // Verify initialization
        assertNotNull(trans2QueryFSInfo);

        void shouldReturnCorrectFileSystemInformationClass() {
            // Test with default FS_SIZE_INFO
            TestAllocInfo defaultInfo = new TestAllocInfo(1000L, 500L);
            assertEquals(FileSystemInformation.FS_SIZE_INFO, defaultInfo.getFileSystemInformationClass());

            // Test with SMB_INFO_ALLOCATION

    }

    @Test
    void testGetInformationLevel() {
        // Test getInformationLevel method
        response = new Trans2QueryFSInformationResponse(config, FileSystemInformation.FS_SIZE_INFO);

        assertEquals(FileSystemInformation.FS_SIZE_INFO, response.getInformationLevel());
    }

    @Test
    void testGetInfo() {
        // Test getInfo method when info is null

    /**
     * {@inheritDoc}
     *
     * @see jcifs.internal.fscc.FileSystemInformation#getFileSystemInformationClass()
     */
    @Override
    public byte getFileSystemInformationClass() {
        return FS_SIZE_INFO;
    }

    @Override
    public long getCapacity() {
        return this.alloc * this.sectPerAlloc * this.bytesPerSect;
    }

    @Override
    public long getFree() {

    }

    @Test
    @DisplayName("Test readBytesWireFormat with FILESYSTEM info type - FS_SIZE_INFO")
    void testReadBytesWireFormatFilesystemSizeInfo() throws Exception {
        response = new Smb2QueryInfoResponse(mockConfig, Smb2Constants.SMB2_0_INFO_FILESYSTEM, FileSystemInformation.FS_SIZE_INFO);

        byte[] buffer = new byte[1024];
        int bufferIndex = 0;

        // Set structure size to 9

     * @return
     */
    private static int mapInformationLevel(final int il) {
        switch (il) {
        case FileSystemInformation.SMB_INFO_ALLOCATION:
            return 0x1;
        case FileSystemInformation.FS_SIZE_INFO:
            return 0x103;
        }
        throw new IllegalArgumentException("Unhandled information level");
    }

    @Override
    protected int writeDataWireFormat(final byte[] dst, final int dstIndex) {

        return switch (infoClass) {
        case FileSystemInformation.FS_FULL_SIZE_INFO -> new FileFsFullSizeInformation();
        case FileSystemInformation.FS_SIZE_INFO -> new FileFsSizeInformation();
        default -> throw new SMBProtocolDecodingException("Unknown filesystem info class " + infoClass);
        };
    }

    /**
     * @param infoClass
     * @return

        }

        @Test
        @DisplayName("Should return correct file system information class")
        void shouldReturnCorrectFileSystemInformationClass() {
            assertEquals(FileSystemInformation.FS_SIZE_INFO, fileFsSizeInfo.getFileSystemInformationClass());
        }
    }

    @Nested
    @DisplayName("Decode Method Tests")
    class DecodeMethodTests {

        @Test

    private AllocInfo fetchAllocationInfo(final SmbTreeHandleImpl th) throws CIFSException, SmbException {
        AllocInfo ai;
        try {
            ai = queryFSInformation(th, AllocInfo.class, FileSystemInformation.FS_SIZE_INFO);
        } catch (final SmbException ex) {
            log.debug("getDiskFreeSpace", ex);
            switch (ex.getNtStatus()) {
            case NtStatus.NT_STATUS_INVALID_INFO_CLASS: