Smb2LeaseKey key2 = new Smb2LeaseKey();
        DirectoryLeaseContext context = new DirectoryLeaseContext(key1, 0, DirectoryCacheScope.IMMEDIATE_CHILDREN);

        context.setLeaseKey(key2);
        assertEquals(key2, context.getLeaseKey());

        context.setLeaseState(DirectoryLeaseState.DIRECTORY_FULL);
        assertEquals(DirectoryLeaseState.DIRECTORY_FULL, context.getLeaseState());

```java
// In Smb2CreateRequest.java
public void addLeaseContext(Smb2LeaseKey key, int requestedState, boolean isV2) {
    Smb2LeaseContext leaseContext = new Smb2LeaseContext();
    leaseContext.setLeaseKey(key);
    leaseContext.setLeaseState(requestedState);
    if (isV2) {
        leaseContext.setEpoch(1);
        // Set parent lease key if available
    }
    addCreateContext(leaseContext);
}
```

    public static final int DIRECTORY_LEASE_FLAG_NOTIFICATIONS = 0x00000002;
    
    public DirectoryLeaseContext(Smb2LeaseKey key, int leaseState, DirectoryCacheScope scope) {
        super();
        setLeaseKey(key);
        setLeaseState(leaseState);
        this.cacheScope = scope;
        this.maxCacheAge = 30000;  // 30 seconds default
        this.notificationEnabled = true;
    }
    
    @Override