@Test
        @DisplayName("Should add flags correctly")
        void testAddFlags() {
            assertEquals(0, testMessage.getFlags());
            testMessage.addFlags(ServerMessageBlock2.SMB2_FLAGS_SIGNED);
            assertEquals(ServerMessageBlock2.SMB2_FLAGS_SIGNED, testMessage.getFlags());
            testMessage.addFlags(ServerMessageBlock2.SMB2_FLAGS_DFS_OPERATIONS);

        if (type1 == null) {
            return getDefaultFlags(tc);
        }
        int flags = NTLMSSP_NEGOTIATE_NTLM | NTLMSSP_NEGOTIATE_VERSION;
        final int type1Flags = type1.getFlags();
        flags |= (type1Flags & NTLMSSP_NEGOTIATE_UNICODE) != 0 ? NTLMSSP_NEGOTIATE_UNICODE : NTLMSSP_NEGOTIATE_OEM;
        if ((type1Flags & NTLMSSP_REQUEST_TARGET) != 0) {

        final MsrpcSamrConnect4 request = new MsrpcSamrConnect4(server, access, this.policyHandle);

        // Then
        assertEquals(0, request.getPtype());
        assertEquals(DCERPC_FIRST_FRAG | DCERPC_LAST_FRAG, request.getFlags());
    }

        assertEquals(0, msrpcLsarClose.getPtype(), "ptype should be 0");
        assertEquals(DcerpcConstants.DCERPC_FIRST_FRAG | DcerpcConstants.DCERPC_LAST_FRAG, msrpcLsarClose.getFlags(),
                "flags should be DCERPC_FIRST_FRAG | DCERPC_LAST_FRAG");

        // Although we cannot directly assert that super(handle) was called with Mockito for a constructor,

        public InetAddress getAddress() {
            return address;
        }

        /**
         * Gets the address flags.
         *
         * @return the flags
         */
        public int getFlags() {
            return flags;
        }

        /**
         * Checks if this is an IPv4 address.
         *
         * @return true if IPv4
         */
        public boolean isIPv4() {

    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
        transformHeader.setSessionId(testSessionId);

        // Then

        when(gssContext.getConfState()).thenReturn(false);
        when(gssContext.getIntegState()).thenReturn(false);

        int flags = ctx.getFlags();
        assertEquals(0, flags);
    }

    @Test
    @DisplayName("getFlags combines all flag bits when true")
    void getFlags_allTrue() {
        when(gssContext.getCredDelegState()).thenReturn(true);

        assertEquals(0, msrpcGetMembersInAlias.getPtype(), "ptype should be initialized to 0");
        assertEquals(DCERPC_FIRST_FRAG | DCERPC_LAST_FRAG, msrpcGetMembersInAlias.getFlags(),
                "flags should be set to DCERPC_FIRST_FRAG | DCERPC_LAST_FRAG");
    }

        this.suppliedWorkstation = suppliedWorkstation;
    }

    @Override
    public byte[] toByteArray() {
        try {
            int flags = getFlags();
            int size = 8 * 4 + ((flags & NTLMSSP_NEGOTIATE_VERSION) != 0 ? 8 : 0);

            byte[] domain = {};
            final String suppliedDomainString = getSuppliedDomain();

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