}
    }

    @Test
    @DisplayName("Test setFileInfoClass method")
    void testSetFileInfoClass() {
        request = new Smb2QueryInfoRequest(mockConfig);
        byte testFileInfoClass = (byte) 0x10;

        request.setFileInfoClass(testFileInfoClass);

        // Verify file info class was set and info type was set to SMB2_0_INFO_FILE
        try {

     * @throws SmbException if an I/O error occurs
     */
    public int read() throws SmbException {
        if (read(tmp, 0, 1) == -1) {
            return -1;
        }
        return tmp[0] & 0xFF;
    }

    /**
     * Reads bytes from the file into the specified byte array.
     *
     * @param b the byte array to read data into
     * @return the number of bytes read, or -1 if end of file is reached

     *             determine the MAC address.
     */
    byte[] getMacAddress(CIFSContext tc) throws UnknownHostException;

    /**
     * Returned the hex code associated with this name(e.g. 0x20 is for the file service)
     *
     * @return the name type
     */
    int getNameType();

    /**
     * Gets the NetBIOS name associated with this address.
     *

	var zb0001Mask uint8 /* 8 bits */
	_ = zb0001Mask
	if z.Index == nil {
		zb0001Len--
		zb0001Mask |= 0x20
	}
	if z.Checksums == nil {
		zb0001Len--
		zb0001Mask |= 0x40
	}
	if z.Error == "" {
		zb0001Len--
		zb0001Mask |= 0x80
	}
	// variable map header, size zb0001Len
	err = en.Append(0x80 | uint8(zb0001Len))
	if err != nil {
		return
	}

	// skip if no fields are to be emitted

      0xffa,
      0x1ff9,
      0x15,
      0xf8,
      0x7fa,
      0x3fa,
      0x3fb,
      0xf9,
      0x7fb,
      0xfa,
      0x16,
      0x17,
      0x18,
      0x0,
      0x1,
      0x2,
      0x19,
      0x1a,
      0x1b,
      0x1c,
      0x1d,
      0x1e,
      0x1f,
      0x5c,
      0xfb,
      0x7ffc,
      0x20,
      0xffb,
      0x3fc,
      0x1ffa,

			default:
				// This encodes bytes < 0x20 except for \t, \n and \r.
				// If escapeHTML is set, it also escapes <, >, and &
				// because they can lead to security holes when
				// user-controlled strings are rendered into JSON
				// and served to some browsers.
				dst.WriteString(`u00`)
				dst.WriteByte(hexTable[b>>4])
				dst.WriteByte(hexTable[b&0xF])
			}
			i++
			start = i
			continue
		}

        // Initialize test data
        for (int i = 0; i < 16; i++) {
            testMachineId[i] = (byte) (i + 1);
        }
        for (int i = 0; i < 32; i++) {
            testSalt[i] = (byte) (i + 0x10);
        }

        // Default configuration for SMB 3.1.1 with encryption
        when(mockConfig.isDfsDisabled()).thenReturn(false);
        when(mockConfig.isEncryptionEnabled()).thenReturn(true);

    }

    @Test
    @DisplayName("Test setCommand with various SMB command codes")
    void testSetCommandWithVariousCodes() {
        // Test common SMB command codes
        int[] commandCodes = { 0x00, 0x72, 0x73, 0x74, 0x75, 0xFF };

        for (int command : commandCodes) {
            doNothing().when(messageBlock).setCommand(command);
            when(messageBlock.getCommand()).thenReturn(command);

        this.flags0 |= fl;
    }

    @Override
    protected int writeParameterWordsWireFormat(final byte[] dst, int dstIndex) {
        final int start = dstIndex;

        dst[dstIndex] = (byte) 0x00;
        dstIndex++;
        // name length without counting null termination
        this.namelen_index = dstIndex;
        dstIndex += 2;
        SMBUtil.writeInt4(this.flags0, dst, dstIndex);

    }

    /**
     * LSA close handle message.
     */
    public static class LsarClose extends DcerpcMessage {

        @Override
        public int getOpnum() {
            return 0x00;
        }

        /** Return value from the RPC call. */
        public int retval;
        /** Policy handle to be closed. */
        public rpc.policy_handle handle;

        /**