*/
    public int getLeaseState() {
        LeaseV2CreateContextResponse v2 = getLeaseV2Context();
        if (v2 != null) {
            return v2.getLeaseState();
        }
        LeaseV1CreateContextResponse v1 = getLeaseV1Context();
        if (v1 != null) {
            return v1.getLeaseState();
        }
        return 0;
    }

    /**

        if (entry != null) {
            log.info("Handling lease break for {} from state 0x{} to 0x{}", key, Integer.toHexString(entry.getLeaseState()),
                    Integer.toHexString(newState));

            entry.setBreaking(true);
            int oldState = entry.getLeaseState();
            entry.updateState(newState);

            // Flush any cached data if losing write cache

     */
    public Smb2LeaseKey getLeaseKey() {
        return leaseKey;
    }

    /**
     * Gets the lease state granted by the server
     * @return the granted lease state
     */
    public int getLeaseState() {
        return leaseState;
    }

    /**
     * Gets the lease flags from the server response
     * @return the lease flags
     */
    public int getLeaseFlags() {
        return leaseFlags;

     */
    public Smb2LeaseKey getLeaseKey() {
        return leaseKey;
    }

    /**
     * Gets the lease state granted by the server for V2
     * @return the granted lease state
     */
    public int getLeaseState() {
        return leaseState;
    }

    /**
     * Gets the lease flags from the V2 server response
     * @return the lease flags
     */
    public int getLeaseFlags() {
        return leaseFlags;

     */
    public Smb2LeaseKey getLeaseKey() {
        return leaseKey;
    }

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

    /**
     * Gets the lease flags for this acknowledgment
     * @return the lease flags
     */
    public int getLeaseFlags() {
        return flags;
    }

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

     */
    public Smb2LeaseKey getLeaseKey() {
        return leaseKey;
    }

    /**
     * Gets the granted lease state from the break response
     * @return the lease state
     */
    public int getLeaseState() {
        return leaseState;
    }

    /**
     * Gets the lease flags from the break response
     * @return the lease flags
     */
    public int getLeaseFlags() {
        return flags;
    }

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

        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