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

        DirectoryLeaseContext context = new DirectoryLeaseContext(key, leaseState, scope);

        assertEquals(key, context.getLeaseKey());
        assertEquals(leaseState, context.getLeaseState());
        assertEquals(scope, context.getCacheScope());
        assertEquals(30000L, context.getMaxCacheAge());
        assertTrue(context.isNotificationEnabled());
        assertEquals(0, context.getNotificationFilter());

        assertEquals(0, defaultContext.getLeaseState());
        assertEquals(0, defaultContext.getLeaseFlags());
    }

    @Test
    @DisplayName("Should initialize with provided values")
    void testParameterizedConstructor() {
        assertEquals(testKey, leaseContext.getLeaseKey());
        assertEquals(testState, leaseContext.getLeaseState());
        assertEquals(0, leaseContext.getLeaseFlags());
    }

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

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

        decodedContext.decode(buffer, 0, buffer.length);

        // Verify roundtrip integrity
        assertEquals(context.getLeaseState(), decodedContext.getLeaseState());
        assertEquals(context.getCacheScope(), decodedContext.getCacheScope());
        assertEquals(context.getMaxCacheAge(), decodedContext.getMaxCacheAge());

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