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

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

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

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

        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