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

        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

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

        // 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.

    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;

     * @return the packet type
     */
    public int getPtype() {
        return ptype;
    }

    /**
     * Gets the flags of this DCERPC message
     * @return the message flags
     */
    public int getFlags() {
        return flags;
    }

    /**
     * Check if a specific flag is set.
     * @param flag the flag to check
     * @return whether flag is set
     */
    public boolean isFlagSet(final int flag) {

     */
    public final int getNumReferrals() {
        return this.numReferrals;
    }

    /**
     * Get the referral flags
     *
     * @return the tflags
     */
    public final int getTflags() {
        return this.tflags;
    }

    /**
     * Get the array of referral entries
     *
     * @return the referrals
     */
    public final Referral[] getReferrals() {

        // 1 ms = 10,000 * 100ns intervals
        return timeout100Ns / 10000L;
    }

    /**
     * Get the flags
     * @return the flags
     */
    public int getFlags() {
        return flags;
    }

    /**
     * Check if this is a persistent handle
     * @return true if persistent
     */
    public boolean isPersistent() {

        // Verify DCE/RPC message properties
        assertEquals(0, dfsRootEnum.getPtype());
        assertEquals(DcerpcConstants.DCERPC_FIRST_FRAG | DcerpcConstants.DCERPC_LAST_FRAG, dfsRootEnum.getFlags());

        // Verify DFS enumeration structure
        assertNotNull(dfsRootEnum.info);
        assertEquals(200, dfsRootEnum.info.level);
        assertNotNull(dfsRootEnum.info.e);