void testInitialState() {
            assertNull(buffer.getReferrals());
            assertEquals(0, buffer.getPathConsumed());
            assertEquals(0, buffer.getNumReferrals());
            assertEquals(0, buffer.getTflags());
        }
    }

    @Nested
    @DisplayName("Decode Tests")
    class DecodeTests {

        @Test
        @DisplayName("Should decode buffer with no referrals")

        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);

        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);

    }

    @Override
    public byte[] toByteArray() throws IOException {
        int size = 48;
        int flags = getFlags();
        final String targetName = getTarget();
        final byte[] targetInformationBytes = getTargetInformation();
        byte[] targetBytes = {};

        if (getFlag(NTLMSSP_REQUEST_TARGET)) {
            if (targetName != null && targetName.length() != 0) {

        pos += 8;
        final int skOff = writeSecurityBuffer(type3, 52, sessionKeyBytes);
        pos += 8;

        writeULong(type3, pos, getFlags());
        pos += 4;

        if (getFlag(NTLMSSP_NEGOTIATE_VERSION)) {
            System.arraycopy(NTLMSSP_VERSION, 0, type3, pos, NTLMSSP_VERSION.length);
            pos += NTLMSSP_VERSION.length;
        } else if (micBytes != null) {

            final byte[] challenge = getChallenge();
            byte[] context = getContext();
            final byte[] targetInformation = getTargetInformation();
            int flags = getFlags();
            byte[] target = {};
            if ((flags & NTLMSSP_REQUEST_TARGET) != 0) {
                if (targetName != null && targetName.length() != 0) {

        }

        private void setupReferralDataWithPathConsumed(int consumed) {
            when(mockReferral.getTtl()).thenReturn(300);
            when(mockReferral.getRFlags()).thenReturn(0);
            when(mockReferral.getNode()).thenReturn("\\server\\share\\path");

            String reqPath = "\\\\server\\share\\path\\file";

    @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

                        || !msg2.getFlag(NtlmFlags.NTLMSSP_NEGOTIATE_EXTENDED_SESSIONSECURITY))) {
                    throw new SmbUnsupportedOperationException("Server does not support extended NTLMv2 key exchange");
                }

                if (!msg2.getFlag(NtlmFlags.NTLMSSP_NEGOTIATE_128)) {

            if ((getFlags() & NTLMSSP_NEGOTIATE_SIGN) == NTLMSSP_NEGOTIATE_SIGN) {
                final HMACT64 hmac = new HMACT64(responseKeyNT);
                hmac.update(ntResponse, 0, 16); // only first 16 bytes of ntResponse
                final byte[] userSessionKey = hmac.digest();

                if ((getFlags() & NTLMSSP_NEGOTIATE_KEY_EXCH) != 0) {