void testFlagsProperty() {
            testBlock.setFlags((byte) 0x12);
            assertEquals((byte) 0x12, testBlock.getFlags());
        }

        @Test
        @DisplayName("Test flags2 property")
        void testFlags2Property() {
            testBlock.setFlags2(0x1234);
            assertEquals(0x1234, testBlock.getFlags2());

            testBlock.addFlags2(0x0001);

     * Sets the SMB message flags
     * @param flags
     *            the flags to set
     */
    public final void setFlags(final byte flags) {
        this.flags = flags;
    }

    /**
     * Gets the SMB message flags2 field
     * @return the flags2
     */
    public final int getFlags2() {
        return this.flags2;
    }

    /**
     * Sets the SMB message flags2 field
     * @param fl

            this.andx.headerStart = this.headerStart;
            this.andx.setCommand(this.andxCommand);
            this.andx.setErrorCode(getErrorCode());
            this.andx.setFlags(getFlags());
            this.andx.setFlags2(getFlags2());
            this.andx.setTid(getTid());
            this.andx.setPid(getPid());
            this.andx.setUid(getUid());
            this.andx.setMid(getMid());

        // Test setting and getting flags
        int testFlags = 0x12345678;
        message.setFlags(testFlags);
        assertEquals(testFlags, message.getFlags(), "Flags should be set and retrieved correctly.");
    }

    @Test
    void testGetFlag() {
        // Test getting individual flags
        message.setFlags(0b1010); // Set flags: 8 (0x8) and 2 (0x2)

    }

    @Test
    @DisplayName("Should set and get flags")
    void testFlags() {
        // Given
        int flags = 0x0001; // Encrypted flag

        // When
        transformHeader.setFlags(flags);

        // Then
        assertEquals(flags, transformHeader.getFlags());
    }

    @Test
    @DisplayName("Should set and get session ID")
    void testSessionId() {
        // When

     */
    public int getFlags() {
        return flags;
    }

    /**
     * Sets the flags for this message.
     *
     * @param flags The flags for this message.
     */
    public void setFlags(final int flags) {
        this.flags = flags;
    }

    /**
     * Returns the status of the specified flag.
     *
     * @param flag The flag to test (i.e., <code>NTLMSSP_NEGOTIATE_OEM</code>).

        assertEquals(WitnessRegistration.WITNESS_REGISTER_IP_NOTIFICATION, registration.getFlags());

        registration.setFlags(WitnessRegistration.WITNESS_REGISTER_NONE);
        assertEquals(WitnessRegistration.WITNESS_REGISTER_NONE, registration.getFlags());

        registration.setFlags(WitnessRegistration.WITNESS_REGISTER_IP_NOTIFICATION);

        assertNotNull(response);
        verify(mockContext).getConfig();
    }

    @Test
    @DisplayName("Test setFlags method")
    void testSetFlags() {
        setupMockConfig();
        Smb2IoctlRequest request = new Smb2IoctlRequest(mockConfig, TEST_CONTROL_CODE);

        request.setFlags(Smb2IoctlRequest.SMB2_O_IOCTL_IS_FSCTL);

        assertNotNull(request);
    }

    @Test

     * @param suppliedWorkstation The supplied workstation name.
     */
    public Type1Message(final int flags, final String suppliedDomain, String suppliedWorkstation) {
        setFlags(getDefaultFlags() | flags);
        setSuppliedDomain(suppliedDomain);
        if (suppliedWorkstation == null) {
            suppliedWorkstation = getDefaultWorkstation();
        }

        SMB1SigningDigest signDigest = new SMB1SigningDigest(testMacSigningKey, false, 10);
        byte[] data = new byte[100];

        // Prepare mock request
        when(mockRequest.getFlags2()).thenReturn(SmbConstants.FLAGS2_SECURITY_SIGNATURES);
        when(mockRequest.getSignSeq()).thenReturn(10);
        when(mockRequest.getCommand()).thenReturn((int) ServerMessageBlock.SMB_COM_WRITE);