assertEquals(0, msrpcSamrCloseHandle.getPtype(), "ptype should be 0");
        assertEquals(MsrpcSamrCloseHandle.DCERPC_FIRST_FRAG | MsrpcSamrCloseHandle.DCERPC_LAST_FRAG, msrpcSamrCloseHandle.getFlags(),
                "flags should be DCERPC_FIRST_FRAG | DCERPC_LAST_FRAG");
        // Verify that the super constructor was called with the correct handle.

    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

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

        }
        ret.append("]");
        return ret.toString();
    }

    /**
     * {@inheritDoc}
     *
     * @see jcifs.smb.SSPContext#getFlags()
     */
    @Override
    public int getFlags() {
        return 0;
    }

    /**
     *
     * {@inheritDoc}
     *
     * @see jcifs.smb.SSPContext#isSupported(org.bouncycastle.asn1.ASN1ObjectIdentifier)
     */

        int expectedFlags = DcerpcConstants.DCERPC_FIRST_FRAG | DcerpcConstants.DCERPC_LAST_FRAG;
        assertEquals(expectedFlags, lsarClose.getFlags(), "The flags should be set to DCERPC_FIRST_FRAG | DCERPC_LAST_FRAG.");
    }

    public void setServerAddress(String serverAddress) {
        this.serverAddress = serverAddress;
    }

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

    /**
     * Sets the registration flags.
     *
     * @param flags the flags
     */
    public void setFlags(int flags) {
        this.flags = flags;

        // Verify ptype and flags are set correctly
        assertEquals(0, policy.getPtype(), "ptype should be 0");
        assertEquals(DcerpcConstants.DCERPC_FIRST_FRAG | DcerpcConstants.DCERPC_LAST_FRAG, policy.getFlags(),
                "flags should be DCERPC_FIRST_FRAG | DCERPC_LAST_FRAG");
    }

     */
    public AvFlags(final int flags) {
        this(encode(flags));
    }

    /**
     * Gets the flags as an integer value
     * @return flags
     */
    public int getFlags() {
        return SMBUtil.readInt4(this.getRaw(), 0);
    }

    private static byte[] encode(final int flags) {
        final byte[] raw = new byte[4];
        SMBUtil.writeInt4(flags, raw, 0);
        return raw;

        // Verify that the 'flags' field is set to DCERPC_FIRST_FRAG | DCERPC_LAST_FRAG
        int expectedFlags = DcerpcConstants.DCERPC_FIRST_FRAG | DcerpcConstants.DCERPC_LAST_FRAG;
        assertEquals(expectedFlags, msrpc.getFlags(), "The 'flags' field should be initialized to DCERPC_FIRST_FRAG | DCERPC_LAST_FRAG.");

        // Although we cannot directly verify the super constructor call with Mockito for a real class,

        // Assert ptype and flags using getters
        assertEquals(0, msrpcLsarOpenPolicy2.getPtype());
        assertEquals(MsrpcLsarOpenPolicy2.DCERPC_FIRST_FRAG | MsrpcLsarOpenPolicy2.DCERPC_LAST_FRAG, msrpcLsarOpenPolicy2.getFlags());

        // Verify that the policyHandle passed to the constructor is the one used
        // This is implicitly tested by the object being constructed without errors.