// Request base lease
        Smb2LeaseKey leaseKey = baseLeaseManager.requestLease(directoryPath, requestedState);

        // Create directory cache entry
        DirectoryCacheEntry cacheEntry = new DirectoryCacheEntry(directoryPath, leaseKey, scope);
        directoryCache.put(directoryPath, cacheEntry);
        leaseToPath.put(leaseKey, directoryPath);

        // Start change notification if enabled

     * @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

        verify(baseLeaseManager).releaseLease(leaseKey);
    }

    @Test
    public void testShutdown() {
        String directoryPath = "/test/dir";
        Smb2LeaseKey leaseKey = new Smb2LeaseKey();

        // Create cache entry
        when(baseLeaseManager.requestLease(anyString(), anyInt())).thenReturn(leaseKey);

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

     * @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;

        // Request base lease
        Smb2LeaseKey leaseKey = baseLeaseManager.requestLease(directoryPath, requestedState);
        
        // Create directory cache entry
        DirectoryCacheEntry cacheEntry = new DirectoryCacheEntry(directoryPath, leaseKey, scope);
        directoryCache.put(directoryPath, cacheEntry);
        leaseToPath.put(leaseKey, directoryPath);