}

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

        assertEquals(100, decodedSize);
        assertEquals(3, referral.getVersion());
        assertEquals(100, referral.getSize());
        assertEquals(1, referral.getServerType());
        assertEquals(0, referral.getRFlags());
        assertEquals(5, referral.getProximity());
        assertEquals(300, referral.getTtl());
        assertEquals(path, referral.getRpath());
        assertEquals(altPath, referral.getAltPath());

        final DfsReferralDataImpl dr = new DfsReferralDataImpl();
        final String[] arr = new String[4];
        dr.ttl = ref.getTtl();
        dr.rflags = ref.getRFlags();
        dr.expiration = expire;
        if ((dr.rflags & Trans2GetDfsReferralResponse.FLAGS_NAME_LIST_REFERRAL) == Trans2GetDfsReferralResponse.FLAGS_NAME_LIST_REFERRAL) {

        assertTrue((unicodeType1.getFlags() & NtlmFlags.NTLMSSP_NEGOTIATE_UNICODE) != 0);
        // Default is OEM, so both flags are set due to OR operation
        assertTrue((unicodeType1.getFlags() & NtlmFlags.NTLMSSP_NEGOTIATE_OEM) != 0);

        // OEM flag is default when isUseUnicode() is false
        assertTrue((oemType1.getFlags() & NtlmFlags.NTLMSSP_NEGOTIATE_OEM) != 0);

        message.setFlag(0b0001, true);
        assertEquals(0b0001, message.getFlags(), "Flag 0b0001 should be set.");

        message.setFlag(0b0100, true);
        assertEquals(0b0101, message.getFlags(), "Flag 0b0100 should be added to existing flags.");

        // Test clearing a flag
        message.setFlag(0b0001, false);
        assertEquals(0b0100, message.getFlags(), "Flag 0b0001 should be cleared.");

    }

    @Test
    @DisplayName("Test getFlags returns correct value")
    void testGetFlags() {
        ace.flags = 0x00;
        assertEquals(0x00, ace.getFlags());

        ace.flags = 0xFF;
        assertEquals(0xFF, ace.getFlags());

        ace.flags = ACE.FLAGS_OBJECT_INHERIT | ACE.FLAGS_CONTAINER_INHERIT | ACE.FLAGS_INHERIT_ONLY;
        assertEquals(0x0B, ace.getFlags());
    }

    @Test

        // Act
        lookupSids = new MsrpcLookupSids(mockPolicyHandle, testSids);

        // Assert
        assertTrue((lookupSids.getFlags() & DcerpcConstants.DCERPC_FIRST_FRAG) != 0);
        assertTrue((lookupSids.getFlags() & DcerpcConstants.DCERPC_LAST_FRAG) != 0);
    }

    @Test
    void constructor_shouldPassCorrectCountToSuperclass() throws Exception {
        // Arrange

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

            return this.supportedMechs.length > 0 && Objects.equals(this.supportedMechs[0], selectedMech);
        }

        @Override
        public int getFlags() {
            return this.flags;
        }

        @Override
        public ASN1ObjectIdentifier[] getSupportedMechs() {
            return this.supportedMechs.clone();
        }

            // When
            Type2Message parsedMessage = new Type2Message(rawMessage);

            // Then
            assertNotNull(parsedMessage);
            assertEquals(originalMessage.getFlags(), parsedMessage.getFlags());
            assertNull(parsedMessage.getChallenge()); // Should be null if all zeros
            assertNull(parsedMessage.getTarget());