* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 */
package jcifs.internal.smb2.lease;

/**
 * SMB2 Lease State constants
 *
 * MS-SMB2 2.2.13.2.8
 */
public class Smb2LeaseState {

    /**
     * No lease caching
     */
    public static final int SMB2_LEASE_NONE = 0x00;

    /**
     * Read caching lease (R)
     */
    public static final int SMB2_LEASE_READ_CACHING = 0x01;

        assertTrue(Smb2LeaseState.hasReadCaching(Smb2LeaseState.SMB2_LEASE_READ_WRITE));
        assertTrue(Smb2LeaseState.hasReadCaching(Smb2LeaseState.SMB2_LEASE_FULL));

        assertFalse(Smb2LeaseState.hasReadCaching(Smb2LeaseState.SMB2_LEASE_NONE));
        assertFalse(Smb2LeaseState.hasReadCaching(Smb2LeaseState.SMB2_LEASE_HANDLE_CACHING));
        assertFalse(Smb2LeaseState.hasReadCaching(Smb2LeaseState.SMB2_LEASE_WRITE_CACHING));
    }

    @Test
    @DisplayName("Should update lease state")
    void testUpdateLease() {
        String path = "/share/file.txt";
        int requestedState = Smb2LeaseState.SMB2_LEASE_FULL;
        int grantedState = Smb2LeaseState.SMB2_LEASE_READ_WRITE;

        Smb2LeaseKey key = leaseManager.requestLease(path, requestedState);
        leaseManager.updateLease(key, grantedState);

        LeaseEntry entry = leaseManager.getLease(key);

    public void testDirectoryLeaseConstants() {
        // Test that directory lease states use correct base values
        assertEquals(Smb2LeaseState.SMB2_LEASE_READ_CACHING | Smb2LeaseState.SMB2_LEASE_HANDLE_CACHING,
                DirectoryLeaseState.DIRECTORY_READ_HANDLE);

        assertEquals(Smb2LeaseState.SMB2_LEASE_FULL, DirectoryLeaseState.DIRECTORY_FULL);
    }

    @Test
    public void testCanCacheEnumeration() {

    }
    
    @Test
    public void testLeaseStateFlags() {
        int state = Smb2LeaseState.SMB2_LEASE_READ_WRITE;
        
        assertTrue((state & Smb2LeaseState.SMB2_LEASE_READ_CACHING) != 0);
        assertTrue((state & Smb2LeaseState.SMB2_LEASE_WRITE_CACHING) != 0);
        assertFalse((state & Smb2LeaseState.SMB2_LEASE_HANDLE_CACHING) != 0);
    }
    
    @Test
    public void testLeaseManager() {

     */
    public static final int DIRECTORY_READ_HANDLE = Smb2LeaseState.SMB2_LEASE_READ_CACHING | Smb2LeaseState.SMB2_LEASE_HANDLE_CACHING;

    /**
     * Directory Full caching (RWH) - all three lease types
     */
    public static final int DIRECTORY_FULL = Smb2LeaseState.SMB2_LEASE_FULL;

    /**
     * Check if directory can cache enumeration results
     * @param state lease state

            // Flush any cached data if losing write cache
            if (Smb2LeaseState.hasWriteCaching(oldState) && !Smb2LeaseState.hasWriteCaching(newState)) {
                flushCachedWrites(entry.getPath());
            }

            // Invalidate cached data if losing read cache
            if (Smb2LeaseState.hasReadCaching(oldState) && !Smb2LeaseState.hasReadCaching(newState)) {
                invalidateReadCache(entry.getPath());

import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.DisplayName;
import org.junit.jupiter.api.Test;

import jcifs.internal.smb2.lease.Smb2LeaseKey;
import jcifs.internal.smb2.lease.Smb2LeaseState;
import jcifs.internal.util.SMBUtil;

@DisplayName("LeaseV1CreateContextRequest Tests")
class LeaseV1CreateContextRequestTest {

    private LeaseV1CreateContextRequest leaseContext;

        Smb2LeaseKey leaseKey = dirManager.requestDirectoryLease(directoryPath,
                Smb2LeaseState.SMB2_LEASE_READ_CACHING | Smb2LeaseState.SMB2_LEASE_HANDLE_CACHING, scope);

        // Perform actual directory enumeration
        SmbFile[] files = smbFile.listFiles();

        // Update cache if we have a directory lease

        if (newState != 0) {
            if ((newState & Smb2LeaseState.SMB2_LEASE_READ_CACHING) == 0) {
                // Lost read cache - invalidate directory cache
                entry.invalidate();
            }
        } else {
            // Lease completely broken - invalidate cache
            entry.invalidate();
        }

        if ((newState & Smb2LeaseState.SMB2_LEASE_HANDLE_CACHING) == 0) {