assertFalse(message.getFlag(0b0100), "Flag 0b0100 should not be set.");
        assertFalse(message.getFlag(0b0001), "Flag 0b0001 should not be set.");

        message.setFlags(0); // No flags set
        assertFalse(message.getFlag(0b1000), "No flags should be set.");
    }

    @Test
    void testSetFlag() {
        // Test setting a flag
        message.setFlags(0);
        message.setFlag(0b0001, true);

            assertTrue(message.isFlagSet(DcerpcConstants.RPC_C_PF_BROADCAST));
        }

        @Test
        @DisplayName("setFlag should add specified flags")
        void testSetFlag() {
            // Test setting a new flag
            message.flags = 0;
            message.setFlag(DcerpcConstants.RPC_C_PF_BROADCAST);
            assertTrue(message.isFlagSet(DcerpcConstants.RPC_C_PF_BROADCAST));

        reconnect();
        if (message == null) {
            message = new Type1Message();
            if (LM_COMPATIBILITY > 2) {
                message.setFlag(NtlmFlags.NTLMSSP_REQUEST_TARGET, true);
            }
        } else {
            String domain = DEFAULT_DOMAIN;
            String user = Type3Message.getDefaultUser();

            flags = NTLMSSP_NEGOTIATE_NTLM | NTLMSSP_NEGOTIATE_OEM;
            setFlags(flags);
            charset = getOEMEncoding();
            end = true;
        } else {
            setEncryptedSessionKey(readSecurityBuffer(material, pos));
            pos += 8;

            flags = readULong(material, pos);
            setFlags(flags);
            pos += 4;

            if (message == null) {
                message = new Type1Message(this.transportContext);
                if (this.transportContext.getConfig().getLanManCompatibility() > 2) {
                    message.setFlag(NtlmFlags.NTLMSSP_REQUEST_TARGET, true);
                }
            } else if (this.transportContext.getCredentials() instanceof NtlmPasswordAuthentication) {

    public Type2Message(final CIFSContext tc, final Type1Message type1, final byte[] challenge, final String target) {
        this(tc, getDefaultFlags(tc, type1), challenge,
                type1 != null && target == null && type1.getFlag(NTLMSSP_REQUEST_TARGET) ? tc.getConfig().getDefaultDomain() : target);
    }

    /**
     * Creates a Type-2 message with the specified parameters.
     *
     * @param tc
     *            context to use

     */
    public Type2Message(final Type1Message type1, final byte[] challenge, final String target) {
        this(getDefaultFlags(type1), challenge,
                type1 != null && target == null && type1.getFlag(NTLMSSP_REQUEST_TARGET) ? getDefaultDomain() : target);
    }

    /**
     * Creates a Type-2 message with the specified parameters.
     *
     * @param flags The flags to apply to this message.

        }

        @Test
        @DisplayName("Test isResponse method")
        void testIsResponse() {
            testBlock.setFlags((byte) 0);
            assertFalse(testBlock.isResponse());

            testBlock.setFlags((byte) SmbConstants.FLAGS_RESPONSE);
            assertTrue(testBlock.isResponse());
        }

        @Test
        @DisplayName("Test hashCode method")

        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

    }

    @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