public void setLeaseKey(Smb2LeaseKey leaseKey) {
        this.leaseKey = leaseKey;
    }

    /**
     * Gets the requested lease state flags
     * @return the requested lease state
     */
    public int getLeaseState() {
        return leaseState;
    }

    /**
     * Sets the requested lease state flags
     * @param leaseState the lease state to set
     */
    public void setLeaseState(int leaseState) {

        this.leaseKey = leaseKey;
    }

    /**
     * Gets the requested lease state flags for V2
     * @return the requested lease state
     */
    public int getLeaseState() {
        return leaseState;
    }

    /**
     * Sets the requested lease state flags for V2
     * @param leaseState the lease state to set
     */
    public void setLeaseState(int leaseState) {

        LeaseEntry entry = manager.getLease(key);
        assertNotNull(entry);
        assertEquals(requestedState, entry.getLeaseState());
        
        // Test lease break
        manager.handleLeaseBreak(key, Smb2LeaseState.SMB2_LEASE_READ_CACHING);
        assertEquals(Smb2LeaseState.SMB2_LEASE_READ_CACHING, entry.getLeaseState());
    }
}
```

### 8.2 Integration Tests
```java
@Test

        this.leaseKey = leaseKey;
    }

    /**
     * Gets the current lease state flags for this directory
     * @return the lease state
     */
    public int getLeaseState() {
        return leaseState;
    }

    /**
     * Sets the lease state flags for this directory
     * @param leaseState the lease state to set
     */
    public void setLeaseState(int leaseState) {

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