}

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

    @Nested
    @DisplayName("Constants Validation Tests")
    class ConstantsValidationTests {

        @Test
        @DisplayName("All fragment flags should have unique values")
        void testFragmentFlagsUniqueness() {
            // Verify that fragment flags are unique and follow bit pattern
            assertEquals(1, Integer.bitCount(DcerpcConstants.DCERPC_FIRST_FRAG), "DCERPC_FIRST_FRAG should be a single bit");

    }

    /**
     * Gets the lease flags for this request
     * @return the lease flags
     */
    public int getLeaseFlags() {
        return leaseFlags;
    }

    /**
     * Sets the lease flags for this request
     * @param leaseFlags the lease flags to set
     */
    public void setLeaseFlags(int leaseFlags) {
        this.leaseFlags = leaseFlags;
    }

    @Override

     * @return the break reason
     */
    public int getBreakReason() {
        return breakReason;
    }

    /**
     * Gets the lease flags from the notification
     * @return the lease flags
     */
    public int getLeaseFlags() {
        return flags;
    }

    /**
     * Gets the access mask hint for optimizing lease handling
     * @return the access mask hint
     */

    }

    boolean allow;
    int flags;
    int access;
    SID sid;

    @Override
    public boolean isAllow() {
        return this.allow;
    }

    @Override
    public boolean isInherited() {
        return (this.flags & FLAGS_INHERITED) != 0;
    }

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

    @Override

    @DisplayName("Test setQueryFlags method")
    void testSetQueryFlags() {
        request = new Smb2QueryDirectoryRequest(mockConfig);
        byte flags = Smb2QueryDirectoryRequest.SMB2_RESTART_SCANS | Smb2QueryDirectoryRequest.SMB2_RETURN_SINGLE_ENTRY;

        request.setQueryFlags(flags);

        // Verify by writing to buffer and checking the queryFlags position
        byte[] buffer = new byte[1024];

            // Test with both flags
            setPrivateField(response, "shareFlags",
                    Smb2TreeConnectResponse.SMB2_SHAREFLAG_DFS | Smb2TreeConnectResponse.SMB2_SHAREFLAG_DFS_ROOT);
            assertTrue(response.isShareDfs(), "Should detect DFS from combined flags");

            // Test without DFS flags
            setPrivateField(response, "shareFlags", 0);

  }

  data class Request(
    val network: Network?,
    val domain: String,
    val nsType: Int,
    val flags: Int,
    val callback: DnsResolver.Callback<List<InetAddress>>,
  )

  @Implementation
  fun query(
    network: Network?,
    domain: String,
    nsType: Int,
    flags: Int,
    executor: Executor,
    cancellationSignal: CancellationSignal?,
    callback: DnsResolver.Callback<List<InetAddress>>,

			return o, err
		}
		x.Type = VersionType(zb0002)
	}
	{
		var zb0003 uint8
		zb0003, bts, err = msgp.ReadUint8Bytes(bts)
		if err != nil {
			err = msgp.WrapError(err, "Flags")
			return o, err
		}
		x.Flags = xlFlags(zb0003)
	}
	o = bts
	return o, err
}

// unmarshalV unmarshals with a specific metadata version.
func (j *xlMetaV2Version) unmarshalV(v uint8, bts []byte) (o []byte, err error) {

        Field flagsField = DcerpcMessage.class.getDeclaredField("flags");
        flagsField.setAccessible(true);
        int flagsValue = (int) flagsField.get(request);
        int expectedFlags = DcerpcConstants.DCERPC_FIRST_FRAG | DcerpcConstants.DCERPC_LAST_FRAG;
        assertEquals(expectedFlags, flagsValue, "The 'flags' field should be initialized to DCERPC_FIRST_FRAG | DCERPC_LAST_FRAG.");
    }