}

    /**
     * Check if directory listing can be cached
     *
     * @param directoryPath directory path
     * @return true if caching is allowed
     */
    public boolean canCacheDirectoryListing(String directoryPath) {
        DirectoryCacheEntry entry = getCacheEntry(directoryPath);
        if (entry == null)
            return false;

                directoryCache.remove(directoryPath);
                return null;
            }
        }
        
        return entry;
    }
    
    public boolean canCacheDirectoryListing(String directoryPath) {
        DirectoryCacheEntry entry = getCacheEntry(directoryPath);
        if (entry == null) return false;
        

        when(baseLeaseManager.getLease(leaseKey)).thenReturn(mockLeaseEntry);

        assertTrue(directoryLeaseManager.canCacheDirectoryListing(directoryPath));

        // When read cache is lost
        when(mockLeaseEntry.hasReadCache()).thenReturn(false);
        assertFalse(directoryLeaseManager.canCacheDirectoryListing(directoryPath));
    }

    @Test
    public void testGetCachedDirectoryListing() throws IOException {

        assertFalse(cacheEntry.isComplete());

        // Step 3: Verify can cache directory listing
        assertTrue(directoryLeaseManager.canCacheDirectoryListing(directoryPath));

        // Step 4: Initially no cached listing
        assertNull(directoryLeaseManager.getCachedDirectoryListing(directoryPath));

        String directoryPath = smbFile.getPath();

        // Check if we can use cached directory listing
        if (dirManager.canCacheDirectoryListing(directoryPath)) {
            List<SmbFile> cachedFiles = dirManager.getCachedDirectoryListing(directoryPath);
            if (cachedFiles != null) {