// Then
        assertArrayEquals(testFileId, customRequest.getFileId(), "Custom implementation should store file ID correctly");
        assertEquals(1, customRequest.getSetCount(), "Should track set count correctly");
    }

    protected String charset = Constants.UTF_8;

    /** Helper for managing content length limits */
    @Resource
    protected ContentLengthHelper contentLengthHelper;

    /** Flag to track initialization status */
    protected AtomicBoolean isInit = new AtomicBoolean(false);

    /**
     * Constructs a new FileSystemClient.
     */
    public FileSystemClient() {
        // Default constructor

import org.opensearch.search.aggregations.bucket.terms.TermsAggregationBuilder;

/**
 * Helper class for managing crawling information and statistics.
 * Provides functionality to track crawling sessions, manage document expiration,
 * and handle crawling information storage and retrieval.
 */
public class CrawlingInfoHelper {

    /**
     * Creates a new instance of CrawlingInfoHelper.
     */

        assertFalse(exception.isAuthenticationError());
        assertFalse(exception.isFileSystemError());
        assertTrue(exception.isTransientError());
    }

    @Test
    @DisplayName("Should track elapsed time")
    void testElapsedTime() throws InterruptedException {
        // Given
        exception = new SmbOperationException(SmbOperationException.ErrorCode.BUSY, "Server busy");

        // When

        SmbTransportImpl pooled = pool.getSmbTransport(ctx, address, 445, false);
        SmbTransportImpl nonPooled = pool.getSmbTransport(ctx, address, 445, true);

        // Then: Pool should track pooled but not non-pooled
        assertTrue(pool.contains(pooled), "Pooled connection should be tracked");
        assertFalse(pool.contains(nonPooled), "Non-pooled connection should not be tracked");
    }

    @Test

        if (log.isDebugEnabled()) {
            log.debug("New transport connection " + conn + " (pool size: " + connections.size() + "/" + maxPoolSize + ")");
        }

        // Track connection metrics
        String key = getConnectionKey(address, port);
        connectionMetrics.computeIfAbsent(key, k -> new ConnectionMetrics()).recordConnection();
        activeConnections.incrementAndGet();

     * Defaults to normal thread priority.
     */
    protected int crawlerPriority = Thread.NORM_PRIORITY;

    /**
     * Thread-safe list of active data crawling threads.
     * Used to track and manage all currently running crawler threads.
     */
    protected final List<DataCrawlingThread> dataCrawlingThreadList = Collections.synchronizedList(new ArrayList<>());

    /**

        when(fe2.getFileIndex()).thenReturn(2);

        FileEntry fe3 = mock(FileEntry.class);
        lenient().when(fe3.getName()).thenReturn("file3");
        when(fe3.getFileIndex()).thenReturn(3);

        // Track query count to handle multiple queries properly
        final int[] queryCount = { 0 };

        // Mock send(create) to wire a query response with fe1, fe2
        doAnswer(inv -> {

    }

    @Test
    @DisplayName("send retries on transport errors and restores request/response state")
    void send_retries_on_transportError() throws Exception {
        // Create a spy to track method calls
        SmbTreeConnection c = spy(newConn());

        // Prepare a minimal locator
        SmbResourceLocatorImpl loc = new SmbResourceLocatorImpl(ctx, smbUrl("smb://srv/share/path"));

            transformHeader.encode(result, 0);
            System.arraycopy(encResult.ciphertext, 0, result, Smb2TransformHeader.TRANSFORM_HEADER_SIZE, encResult.ciphertext.length);

            // Track encrypted bytes for key rotation - lock-free atomic operation
            bytesEncrypted.addAndGet(message.length);

            return result;
        } catch (final Exception e) {