assertEquals(0L, callback.getDocumentSize());
        assertEquals(0L, callback.getExecuteTime());

        // Test store method
        DataStoreParams params = new DataStoreParams();
        Map<String, Object> data = new HashMap<>();
        data.put("key1", "value1");
        callback.store(params, data);

        assertEquals(1L, callback.getDocumentSize());
        assertEquals(100L, callback.getExecuteTime());

            @Override
            public void store(DataStoreParams paramMap, Map<String, Object> dataMap) {
            }

            @Override
            public long getDocumentSize() {
                return 0;
            }

            @Override
            public long getExecuteTime() {
                return 0;
            }

            @Override

        // Execute store
        dataStore.store(config, callback, params);

        // Verify callback was used correctly
        assertEquals(2, callback.getDocumentSize());
        assertTrue(callback.isCommitted());
    }

    @Test
    public void test_multiple_datastore_instances() {
        // Test multiple DataStore instances working independently

        @Override
        public long getExecuteTime() {
            return 100L;
        }

        @Override
        public long getDocumentSize() {
            return 10L;
        }

        @Override
        public void commit() {
            /* No-op */ }

        @Override

        assertEquals(0, indexUpdateCallback.getDocumentSize());

        DataStoreParams paramMap = new DataStoreParams();
        Map<String, Object> dataMap = new HashMap<>();
        dataMap.put("url", "http://example.com/test");

        indexUpdateCallback.store(paramMap, dataMap);
        assertEquals(1, indexUpdateCallback.getDocumentSize());

        dataMap.put("url", "http://example.com/test2");

        this.finishCrawling = finishCrawling;
    }

    /**
     * Gets the total number of documents processed.
     *
     * @return the total document count
     */
    public long getDocumentSize() {
        return documentSize;
    }

    /**
     * Sets the uncaught exception handler for this IndexUpdater thread.
     *
     * @param eh the uncaught exception handler to set
     */

        crawlingInfoHelper.putToInfoMap(Constants.WEB_FS_INDEX_EXEC_TIME, Long.toString(indexUpdater.getExecuteTime()));
        crawlingInfoHelper.putToInfoMap(Constants.WEB_FS_INDEX_SIZE, Long.toString(indexUpdater.getDocumentSize()));

        if (systemHelper.isForceStop()) {
            return;
        }

        for (final String sid : sessionIdList) {
            // remove config