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

        }
    }

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

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