assertTrue("Then callback should complete", latch.await(10, TimeUnit.SECONDS));
        Thread.sleep(100); // Wait a bit to ensure error callback isn't called
        assertEquals("Then callback should be invoked", 1, thenCount.get());
        assertEquals("Error callback should not be invoked", 0, errorCount.get());
    }

    @Test

        IndexUpdateCallback callback1 = createMockCallback(10, 500);
        IndexUpdateCallback callback2 = createMockCallback(20, 1000);

        assertEquals(10L, callback1.getDocumentSize());
        assertEquals(500L, callback1.getExecuteTime());

        assertEquals(20L, callback2.getDocumentSize());
        assertEquals(1000L, callback2.getExecuteTime());

        // Test store operations

        DataStoreParams params = new DataStoreParams();

        // 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 that the functional interface works correctly
        XContentBuilderCallback callback = (builder, params) -> {
            builder.startObject();
            builder.field("callback", "test");
            builder.endObject();
            return builder;
        };

        assertNotNull(callback);

        // Verify the interface can be used in context

    dispatcher.maxRequests = 2
    client.newCall(newRequest("http://a/1")).enqueue(callback)
    client.newCall(newRequest("http://b/1")).enqueue(callback)
    client.newCall(newRequest("http://c/1")).enqueue(callback)
    client.newCall(newRequest("http://a/2")).enqueue(callback)
    client.newCall(newRequest("http://b/2")).enqueue(callback)

    val dispatcherQueueStartC1 = eventRecorder.removeUpToEvent<DispatcherQueueStart>()

			fchown(fd, uid, gid, callback) { callback(enosys()); },
			fstat(fd, callback) { callback(enosys()); },
			fsync(fd, callback) { callback(null); },
			ftruncate(fd, length, callback) { callback(enosys()); },
			lchown(path, uid, gid, callback) { callback(enosys()); },
			link(path, link, callback) { callback(enosys()); },
			lstat(path, callback) { callback(enosys()); },
			mkdir(path, perm, callback) { callback(enosys()); },

import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;

import java.util.HashMap;

import javax.security.auth.Subject;
import javax.security.auth.callback.Callback;
import javax.security.auth.callback.NameCallback;
import javax.security.auth.callback.PasswordCallback;

import org.junit.jupiter.api.DisplayName;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;

     *
     * @param request the chat request
     * @param callback the callback to receive streaming chunks
     * @throws LlmException if LLM is not available or an error occurs
     */
    public void streamChat(final LlmChatRequest request, final LlmStreamCallback callback) {
        final long startTime = System.currentTimeMillis();
        final String llmType = getLlmType();

        String input = "callback1\ncallback2";
        InputStream is = new ByteArrayInputStream(input.getBytes(StandardCharsets.UTF_8));

        List<String> callbackResults = new ArrayList<>();
        Consumer<String> callback = line -> callbackResults.add(line);

        InputStreamThread thread = new InputStreamThread(is, StandardCharsets.UTF_8, 10, callback);
        thread.start();
        thread.join(1000);

     * Uses OpenSearch scroll API for efficient iteration over large result sets.
     *
     * @param sessionId The session ID.
     * @param callback The callback to execute for each access result.
     */
    @Override
    public void iterate(final String sessionId, final AccessResultCallback<OpenSearchAccessResult> callback) {