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) {
            entry.breaking = true;
            entry.updateState(newState);

        Smb2LeaseKey key = leaseManager.requestLease(path, initialState);
        leaseManager.updateLease(key, initialState);

        int initialEpoch = leaseManager.getLease(key).getEpoch();
        leaseManager.handleLeaseBreak(key, newState);

        LeaseEntry entry = leaseManager.getLease(key);
        assertEquals(newState, entry.getLeaseState());
        assertEquals(initialEpoch + 1, entry.getEpoch());

    }

    /**
     * Handle a lease break notification
     *
     * @param key lease key
     * @param newState new lease state
     */
    public void handleLeaseBreak(Smb2LeaseKey key, int newState) {
        LeaseEntry entry = leases.get(key);
        if (entry != null) {

        // Cache should be invalidated
        assertFalse(entry.isComplete());

        // Verify base lease manager was called
        verify(baseLeaseManager).handleLeaseBreak(leaseKey, newState);
    }

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

            // Lost handle cache - may need to close directory handle
            changeNotifier.stopWatching(directoryPath);
        }

        // Forward to base lease manager
        baseLeaseManager.handleLeaseBreak(leaseKey, newState);
    }

    /**
     * Release directory lease
     *
     * @param directoryPath directory path
     */
    public void releaseDirectoryLease(String directoryPath) {

            // Lost handle cache - may need to close directory handle
            changeNotifier.stopWatching(directoryPath);
        }
        
        // Forward to base lease manager
        baseLeaseManager.handleLeaseBreak(leaseKey, newState);
    }
    
    public void releaseDirectoryLease(String directoryPath) {
        DirectoryCacheEntry entry = directoryCache.remove(directoryPath);
        if (entry != null) {