SearchLogHelper mockSearchLogHelper = new SearchLogHelper() {
            private int callCount = 0;

            @Override
            public void storeSearchLog() {
                callCount++;
                throw new RuntimeException("Error " + callCount);
            }
        };
        ComponentUtil.register(mockSearchLogHelper, "searchLogHelper");

    @Test
    public void test_evaluate_concurrency() {
        queryRescorer = new QueryRescorer() {
            private int callCount = 0;

            @Override
            public synchronized RescorerBuilder<?> evaluate(Map<String, Object> params) {
                callCount++;
                if (callCount % 2 == 0) {
                    return createMockRescorerBuilder();
                }
                return null;

            ThumbnailManager mockThumbnailManager = new ThumbnailManager() {
                private int callCount = 0;

                @Override
                public int generate(ExecutorService executorService, boolean cleanup) {
                    // Return 0 to exit the loop
                    if (callCount++ == 0) {
                        return 1;
                    }
                    return 0;

        SmbWatchHandle realHandle = new SmbWatchHandle() {
            private int callCount = 0;

            @Override
            public List<FileNotifyInformation> watch() throws CIFSException {
                callCount++;
                if (callCount == 1) {
                    return Arrays.asList(fileNotifyInfo1);
                }
                return Collections.emptyList();

              public Object load(String key) throws InterruptedException {
                callCount.incrementAndGet();
                startSignal.await();
                return result;
              }
            });

    List<Object> resultArray = doConcurrentGet(cache, "bar", count, startSignal);

    assertThat(callCount.get()).isEqualTo(1);
    for (int i = 0; i < count; i++) {

    @Test
    public void test_addShutdownListener_replace() {
        // Test replacing shutdown listener
        AtomicInteger callCount = new AtomicInteger(0);
        ShutdownListener listener1 = new ShutdownListener() {
            @Override
            public void onShutdown() {
                callCount.incrementAndGet();
            }
        };

        ShutdownListener listener2 = new ShutdownListener() {
            @Override

              public Object load(String key) throws InterruptedException {
                callCount.incrementAndGet();
                startSignal.await();
                return result;
              }
            });

    List<Object> resultArray = doConcurrentGet(cache, "bar", count, startSignal);

    assertThat(callCount.get()).isEqualTo(1);
    for (int i = 0; i < count; i++) {

        // Test storing various data types
        final AtomicInteger callCount = new AtomicInteger(0);

        IndexUpdateCallback callback = new IndexUpdateCallback() {
            @Override
            public void store(DataStoreParams paramMap, Map<String, Object> dataMap) {
                assertNotNull(dataMap);
                callCount.incrementAndGet();
            }

            @Override

    public void test_getRunningJobCount() {
        // Skip this test - requires complex DB setup
        if (true)
            return;
        // Setup test with mock ScheduledJobBhv
        final AtomicInteger callCount = new AtomicInteger(0);
        final List<ScheduledJob> scheduledJobs = new ArrayList<>();

        // Create test scheduled jobs
        ScheduledJob job1 = new ScheduledJob() {
            @Override

        // Calling completeOperation twice must not throw (idempotent release)
        final AtomicInteger callCount = new AtomicInteger(0);
        final CoordinatorHelper helper = new CoordinatorHelper() {
            @Override
            public void completeOperation(final String operationName) {
                callCount.incrementAndGet();
                super.completeOperation(operationName);
            }