* @return the lease key
     */
    public Smb2LeaseKey getLeaseKey() {
        return leaseKey;
    }

    /**
     * Sets the lease key for this directory lease context
     * @param leaseKey the lease key to set
     */
    public void setLeaseKey(Smb2LeaseKey leaseKey) {
        this.leaseKey = leaseKey;
    }

    /**
     * Gets the current lease state flags for this directory

     * @return the lease key
     */
    public Smb2LeaseKey getLeaseKey() {
        return leaseKey;
    }

    /**
     * Sets the lease key for this lease request
     * @param leaseKey the lease key to set
     */
    public void setLeaseKey(Smb2LeaseKey leaseKey) {
        this.leaseKey = leaseKey;
    }

    /**
     * Gets the requested lease state flags
     * @return the requested lease state

        this.leaseState = leaseState;
        this.flags = 0;
    }

    /**
     * Create a lease break acknowledgment
     *
     * @param context CIFS context
     * @param leaseKey lease key
     * @param leaseState acknowledged lease state
     */
    public Smb2LeaseBreakAcknowledgment(CIFSContext context, Smb2LeaseKey leaseKey, int leaseState) {
        this(context.getConfig(), leaseKey, leaseState);

        // Flags (4 bytes)
        this.flags = SMBUtil.readInt4(buffer, bufferIndex);
        bufferIndex += 4;

        // LeaseKey (16 bytes)
        byte[] keyBytes = new byte[16];
        System.arraycopy(buffer, bufferIndex, keyBytes, 0, 16);
        this.leaseKey = new Smb2LeaseKey(keyBytes);
        bufferIndex += 16;

        // CurrentLeaseState (4 bytes)

     * @return the lease key
     */
    public Smb2LeaseKey getLeaseKey() {
        return leaseKey;
    }

    /**
     * Sets the lease key for this V2 lease request
     * @param leaseKey the lease key to set
     */
    public void setLeaseKey(Smb2LeaseKey leaseKey) {
        this.leaseKey = leaseKey;
    }

    /**
     * Gets the requested lease state flags for V2

 */
public class DirectoryCacheEntryTest {

    private DirectoryCacheEntry entry;
    private Smb2LeaseKey leaseKey;
    private String directoryPath = "/test/dir";

    @BeforeEach
    public void setUp() {
        leaseKey = new Smb2LeaseKey();
        entry = new DirectoryCacheEntry(directoryPath, leaseKey, DirectoryCacheScope.IMMEDIATE_CHILDREN);
    }

    @Test
    public void testConstructor() {

        // Flags (4 bytes)
        this.flags = SMBUtil.readInt4(buffer, bufferIndex);
        bufferIndex += 4;

        // LeaseKey (16 bytes)
        byte[] keyBytes = new byte[16];
        System.arraycopy(buffer, bufferIndex, keyBytes, 0, 16);
        this.leaseKey = new Smb2LeaseKey(keyBytes);
        bufferIndex += 16;

        // LeaseState (4 bytes)

     * @param fileId the 16-byte file ID
     * @param type the handle type
     * @param timeout the timeout in milliseconds
     * @param leaseKey the associated lease key (can be null)
     */
    public HandleInfo(String path, HandleGuid guid, byte[] fileId, HandleType type, long timeout, Smb2LeaseKey leaseKey) {
        this.path = path;
        this.createGuid = guid;
        this.fileId = Arrays.copyOf(fileId, 16);
        this.type = type;

    /**
     * Context name for lease response
     */
    public static final String CONTEXT_NAME = "RqLs";

    private static final byte[] CONTEXT_NAME_BYTES = CONTEXT_NAME.getBytes();

    private Smb2LeaseKey leaseKey;
    private int leaseState;
    private int leaseFlags;

    /**
     * Create a new lease V1 context response
     */
    public LeaseV1CreateContextResponse() {
    }

    @Override

    /**
     * Context name for lease V2 response
     */
    public static final String CONTEXT_NAME = "RqL2";

    private static final byte[] CONTEXT_NAME_BYTES = CONTEXT_NAME.getBytes();

    private Smb2LeaseKey leaseKey;
    private int leaseState;
    private int leaseFlags;
    private Smb2LeaseKey parentLeaseKey;
    private int epoch;

    /**
     * Create a new lease V2 context response
     */