assertNotNull(leaseManager.getLease(key1));
        assertNotNull(leaseManager.getLease(key2));

        leaseManager.releaseAll();

        assertNull(leaseManager.getLease(key1));
        assertNull(leaseManager.getLease(key2));
        assertTrue(leaseManager.getAllLeases().isEmpty());
    }

    @Test

public boolean canCacheRead() {
    if (leaseKey != null) {
        LeaseEntry entry = leaseManager.getLease(leaseKey);
        return entry != null && entry.hasReadCache();
    }
    return false;  // Fall back to oplock logic
}

public boolean canCacheWrite() {
    if (leaseKey != null) {
        LeaseEntry entry = leaseManager.getLease(leaseKey);
        return entry != null && entry.hasWriteCache();
    }
    return false;

        try {
            Thread.sleep(10);
        } catch (InterruptedException e) {
            // Ignore
        }

        // When lease is lost, cache entry should be removed
        when(baseLeaseManager.getLease(leaseKey)).thenReturn(null);

        assertNull(directoryLeaseManager.getCacheEntry(directoryPath));
    }

    @Test
    public void testCanCacheDirectoryListing() {

        // Check if cache entry is valid
        if (entry != null && entry.needsRefresh()) {
            // Check if lease is still valid
            LeaseManager.LeaseEntry leaseEntry = baseLeaseManager.getLease(entry.getLeaseKey());
            if (leaseEntry == null || !leaseEntry.hasReadCache()) {
                // Lease lost, remove cache entry
                directoryCache.remove(directoryPath);
                return null;

        
        // Check if cache entry is valid
        if (entry != null && entry.needsRefresh()) {
            // Check if lease is still valid
            LeaseEntry leaseEntry = baseLeaseManager.getLease(entry.getLeaseKey());
            if (leaseEntry == null || !leaseEntry.hasReadCache()) {
                // Lease lost, remove cache entry
                directoryCache.remove(directoryPath);
                return null;

        }
    }

    /**
     * Get lease entry by key
     *
     * @param key lease key
     * @return lease entry or null if not found
     */
    public LeaseEntry getLease(Smb2LeaseKey key) {
        return leases.get(key);
    }

    /**
     * Get lease entry by path
     *
     * @param path file path
     * @return lease entry or null if not found
     */