// Clean up oldest leases
                evictOldestLeases(1);
            }

            // Create new lease
            Smb2LeaseKey newKey = new Smb2LeaseKey();
            LeaseEntry newEntry = new LeaseEntry(newKey, path, requestedState);
            leases.put(newKey, newEntry);
            pathToLease.put(path, newKey);

            log.debug("Created new lease for path: {} with key: {}", path, newKey);

        LeaseEntry entry = leases.get(key);
        if (entry != null) {
            entry.updateState(grantedState);
        }
    }
    
    public LeaseEntry getLease(Smb2LeaseKey key) {
        return leases.get(key);
    }
    
    public void handleLeaseBreak(Smb2LeaseKey key, int newState) {
        LeaseEntry entry = leases.get(key);
        if (entry != null) {

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

        int requestedState = Smb2LeaseState.SMB2_LEASE_READ_CACHING;

        Smb2LeaseKey key = leaseManager.requestLease(path, requestedState);
        LeaseEntry entryByKey = leaseManager.getLease(key);
        LeaseEntry entryByPath = leaseManager.getLeaseByPath(path);

        assertEquals(entryByKey, entryByPath);
        assertEquals(path, entryByPath.getPath());
    }

    @Test

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

        directoryLeaseManager.requestDirectoryLease(directoryPath, DirectoryLeaseState.DIRECTORY_READ_HANDLE,
                DirectoryCacheScope.IMMEDIATE_CHILDREN);

        LeaseManager.LeaseEntry mockLeaseEntry = mock(LeaseManager.LeaseEntry.class);
        when(mockLeaseEntry.hasReadCache()).thenReturn(true);
        when(baseLeaseManager.getLease(leaseKey)).thenReturn(mockLeaseEntry);