DirectoryCacheScope scope = DirectoryCacheScope.IMMEDIATE_CHILDREN;

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

        DirectoryCacheEntry entry = directoryCache.remove(directoryPath);
        if (entry != null) {
            leaseToPath.remove(entry.getLeaseKey());
            changeNotifier.stopWatching(directoryPath);
            baseLeaseManager.releaseLease(entry.getLeaseKey());
        }
    }

    /**
     * Shutdown the directory lease manager
     */
    public void shutdown() {

        assertNotNull(defaultContext.getLeaseKey());
        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());

     */
    public jcifs.internal.smb2.lease.Smb2LeaseKey getLeaseKey() {
        LeaseV2CreateContextResponse v2 = getLeaseV2Context();
        if (v2 != null) {
            return v2.getLeaseKey();
        }
        LeaseV1CreateContextResponse v1 = getLeaseV1Context();
        if (v1 != null) {
            return v1.getLeaseKey();
        }
        return null;
    }

    /**

        // request.addCreateContext(new DurableHandleReconnect(info.getFileId()));

        // Add lease context if needed
        // if (info.getLeaseKey() != null) {
        //     request.addLeaseV1Context(info.getLeaseKey(), Smb2LeaseState.SMB2_LEASE_NONE);
        // }

        // return request;

    @Override
    public byte[] getName() {
        return CONTEXT_NAME_BYTES;
    }

    /**
     * Gets the lease key for this lease request
     * @return the lease key
     */
    public Smb2LeaseKey getLeaseKey() {
        return leaseKey;
    }

    /**
     * Sets the lease key for this lease request
     * @param leaseKey the lease key to set
     */
    public void setLeaseKey(Smb2LeaseKey leaseKey) {

    public Smb2LeaseBreakNotification(Configuration config) {
        super(config);
    }

    /**
     * Gets the lease key that is being broken
     * @return the lease key
     */
    public Smb2LeaseKey getLeaseKey() {
        return leaseKey;
    }

    /**
     * Gets the current lease state before the break
     * @return the current lease state
     */
    public int getCurrentLeaseState() {

    @Override
    public byte[] getName() {
        return CONTEXT_NAME_BYTES;
    }

    /**
     * Gets the lease key from the server response
     * @return the lease key
     */
    public Smb2LeaseKey getLeaseKey() {
        return leaseKey;
    }

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

    public byte[] getName() {
        return CONTEXT_NAME_BYTES;
    }

    /**
     * Gets the lease key from the V2 server response
     * @return the lease key
     */
    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;

    public Smb2LeaseBreakResponse(Configuration config) {
        super(config);
    }

    /**
     * Gets the lease key from the break response
     * @return the lease key
     */
    public Smb2LeaseKey getLeaseKey() {
        return leaseKey;
    }

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