// Force to no caching state
                entry.updateState(Smb2LeaseState.SMB2_LEASE_NONE);
            }
            releaseLease(key);
        } catch (Exception e) {
            log.error("Error handling lease break for key: " + key, e);
            // On error, ensure we're in safe state
            LeaseEntry entry = leases.get(key);

            if (!entry.hasReadCache()) {
                invalidateReadCache(entry.path);
            }
            entry.breaking = false;
        }
    }
    
    public void releaseLease(Smb2LeaseKey key) {
        lock.writeLock().lock();
        try {
            LeaseEntry entry = leases.remove(key);
            if (entry != null) {
                pathToLease.remove(entry.path);
            }

        // Entry should be removed
        assertNull(directoryLeaseManager.getCacheEntry(directoryPath));

        // Verify base lease manager was called
        verify(baseLeaseManager).releaseLease(leaseKey);
    }

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

        // Create cache entry

        int requestedState = Smb2LeaseState.SMB2_LEASE_READ_HANDLE;

        Smb2LeaseKey key = leaseManager.requestLease(path, requestedState);
        assertNotNull(leaseManager.getLease(key));

        leaseManager.releaseLease(key);

        assertNull(leaseManager.getLease(key));
        assertNull(leaseManager.getLeaseByPath(path));
    }

    @Test
    @DisplayName("Should get lease by path")

        if (entry != null) {
            leaseToPath.remove(entry.getLeaseKey());
            changeNotifier.stopWatching(directoryPath);
            baseLeaseManager.releaseLease(entry.getLeaseKey());
        }
    }

    /**
     * Shutdown the directory lease manager
     */
    public void shutdown() {
        scheduler.shutdown();
        try {

        if (entry != null) {
            leaseToPath.remove(entry.getLeaseKey());
            changeNotifier.stopWatching(directoryPath);
            baseLeaseManager.releaseLease(entry.getLeaseKey());
        }
    }
    
    private void cleanupExpiredEntries() {
        List<String> expiredPaths = new ArrayList<>();