import jcifs.CIFSContext;
import jcifs.smb.SmbFile;

/**
 * SMB2/SMB3 Lease Manager
 *
 * Manages lease state for SMB2/SMB3 connections
 */
public class LeaseManager {

    private static final Logger log = LoggerFactory.getLogger(LeaseManager.class);

    private final ConcurrentHashMap<Smb2LeaseKey, LeaseEntry> leases;
    private final ConcurrentHashMap<String, Smb2LeaseKey> pathToLease;

        Smb2LeaseKey key1 = leaseManager.requestLease(path1, Smb2LeaseState.SMB2_LEASE_READ_CACHING);
        Smb2LeaseKey key2 = leaseManager.requestLease(path2, Smb2LeaseState.SMB2_LEASE_WRITE_CACHING);

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

        leaseManager.releaseAll();

        assertNull(leaseManager.getLease(key1));

    /**
     * Create a new directory lease manager
     *
     * @param context CIFS context
     * @param leaseManager base lease manager
     */
    public DirectoryLeaseManager(CIFSContext context, LeaseManager leaseManager) {
        this.context = context;
        this.baseLeaseManager = leaseManager;
        this.directoryCache = new ConcurrentHashMap<>();
        this.leaseToPath = new ConcurrentHashMap<>();

    
    // Change notification integration
    private final DirectoryChangeNotifier changeNotifier;
    
    public DirectoryLeaseManager(CIFSContext context, LeaseManager leaseManager) {
        this.context = context;
        this.baseLeaseManager = leaseManager;
        this.directoryCache = new ConcurrentHashMap<>();
        this.leaseToPath = new ConcurrentHashMap<>();
        this.scheduler = Executors.newScheduledThreadPool(1);

// In SmbFile.java
private Smb2LeaseKey leaseKey;
private int leaseState;
private LeaseManager leaseManager;

protected void doConnect() throws IOException {
    // ... existing connection logic ...
    
    if (context.getConfig().isUseLeases() && tree.getSession().supports(SMB3_0)) {
        // Request lease when opening file
        leaseManager = tree.getSession().getLeaseManager();
        int requestedState = isDirectory() ? 

        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);

    private static final long MAX_RETRY_DELAY = 30000;
    private static final int MAX_BACKOFF_SHIFT = 3; // Maximum 8 seconds (2^3 * 1000ms)

    private final DirectoryLeaseManager leaseManager;
    private final ConcurrentHashMap<String, ChangeNotificationHandle> activeWatchers;
    private final ConcurrentHashMap<String, Integer> failureCounts;

    /**
     * Directory change types
     */

                    }
                }

                if (this.leaseManager != null) {
                    try {
                        this.leaseManager.shutdown();
                        this.leaseManager = null;
                    } catch (Exception e) {
                        log.debug("Error shutting down lease manager during emergency cleanup", e);

    @Mock
    private CIFSContext context;

    @Mock
    private Configuration config;

    @Mock
    private SmbFile mockFile1;

    @Mock
    private SmbFile mockFile2;

    private LeaseManager baseLeaseManager;
    private DirectoryLeaseManager directoryLeaseManager;
    private AutoCloseable mocks;

    @BeforeEach
    public void setUp() throws Exception {

                    try {
                        LeaseManager leaseManager = session.getLeaseManager();
                        if (leaseManager != null) {
                            // Request appropriate lease state based on file type and access
                            int requestedState = determineLeaseState(access);

                            Smb2LeaseKey leaseKey = leaseManager.requestLease(uncPath, requestedState);