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

    /**

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

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

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

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

        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

        assertNotNull(key);
        LeaseEntry entry = leaseManager.getLease(key);
        assertNotNull(entry);
        assertEquals(requestedState, entry.getLeaseState());
        assertEquals(path, entry.getPath());
        assertEquals(1, entry.getEpoch());
        assertFalse(entry.isBreaking());
    }

    @Test
    @DisplayName("Should reuse existing lease for same path")