int requestedState = Smb2LeaseState.SMB2_LEASE_FULL;
        int grantedState = Smb2LeaseState.SMB2_LEASE_READ_WRITE;

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

        LeaseEntry entry = leaseManager.getLease(key);
        assertEquals(grantedState, entry.getLeaseState());
    }

    @Test
    @DisplayName("Should handle lease break")

            pathToLease.put(path, newKey);
            
            return newKey;
        } finally {
            lock.writeLock().unlock();
        }
    }
    
    public void updateLease(Smb2LeaseKey key, int grantedState) {
        LeaseEntry entry = leases.get(key);
        if (entry != null) {
            entry.updateState(grantedState);
        }
    }
    

        }
    }

    /**
     * Update lease state after receiving response
     *
     * @param key lease key
     * @param grantedState granted lease state
     */
    public void updateLease(Smb2LeaseKey key, int grantedState) {
        LeaseEntry entry = leases.get(key);
        if (entry != null) {
            entry.updateState(grantedState);

                if (config.isUseLeases() && resp.isLeaseGranted()) {
                    LeaseManager leaseManager = session.getLeaseManager();
                    if (leaseManager != null) {
                        leaseManager.updateLease(resp.getLeaseKey(), resp.getLeaseState());
                    }
                }

                fh = new SmbFileHandleImpl(config, resp.getFileId(), h, uncPath, flags, access, 0, 0, resp.getEndOfFile());