}

    @Test
    public void test_store_and_stop_lifecycle() {
        // Test full lifecycle: store then stop
        final AtomicBoolean storeCalled = new AtomicBoolean(false);
        final AtomicBoolean stopCalled = new AtomicBoolean(false);
        final AtomicBoolean isActive = new AtomicBoolean(false);

        dataStore = new DataStore() {
            @Override

  private ServerSocket makeMyServerExampleWithFrq(AtomicBoolean finalizeReferentRan)
      throws IOException {
    MyServerExampleWithFrq myServer = MyServerExampleWithFrq.create(finalizeReferentRan);
    assertThat(myServer.serverSocket.isClosed()).isFalse();
    return myServer.serverSocket;
  }

  @Test
  public void testMyServerExampleWithFrq() throws Exception {
    AtomicBoolean finalizeReferentRan = new AtomicBoolean(false);

import org.opensearch.index.query.QueryBuilders;

public class QueryProcessorTest extends UnitFessTestCase {

    @Test
    public void test_executeWithFilter() {
        final AtomicBoolean called = new AtomicBoolean(false);
        QueryProcessor queryProcessor = new QueryProcessor() {
            protected FilterChain createDefaultFilterChain() {
                return (context, query, boost) -> {

        assertEquals("Executed:  ", result);
    }

    @Test
    public void test_addShutdownListener() {
        // Test adding shutdown listener
        AtomicBoolean shutdownCalled = new AtomicBoolean(false);
        ShutdownListener listener = new ShutdownListener() {
            @Override
            public void onShutdown() {
                shutdownCalled.set(true);
            }
        };

        // Verify poll calls sendHeartbeat, fetchNewEvents+dispatch, cleanupExpiredDocuments
        final AtomicBoolean heartbeatCalled = new AtomicBoolean(false);
        final AtomicBoolean fetchCalled = new AtomicBoolean(false);
        final AtomicBoolean cleanupCalled = new AtomicBoolean(false);
        final CoordinatorHelper helper = new CoordinatorHelper() {
            @Override

      Monitor monitor,
      boolean expectedIsOccupied,
      boolean expectedIsOccupiedByCurrentThread,
      int expectedOccupiedDepth) {
    AtomicBoolean actualIsOccupied = new AtomicBoolean();
    AtomicBoolean actualIsOccupiedByCurrentThread = new AtomicBoolean();
    AtomicInteger actualOccupiedDepth = new AtomicInteger();
    AtomicReference<Throwable> thrown = new AtomicReference<>();
    joinUninterruptibly(

      Monitor monitor,
      boolean expectedIsOccupied,
      boolean expectedIsOccupiedByCurrentThread,
      int expectedOccupiedDepth) {
    AtomicBoolean actualIsOccupied = new AtomicBoolean();
    AtomicBoolean actualIsOccupiedByCurrentThread = new AtomicBoolean();
    AtomicInteger actualOccupiedDepth = new AtomicInteger();
    AtomicReference<Throwable> thrown = new AtomicReference<>();
    joinUninterruptibly(

   * of the {@code Map} method calls and check that {@code setUpRan} is true.
   */
  private static class CheckSetUpHashMapGenerator extends WrappedHashMapGenerator {
    private final AtomicBoolean setUpRan;

    CheckSetUpHashMapGenerator(AtomicBoolean setUpRan) {
      this.setUpRan = setUpRan;
    }

    @Override
    Map<String, String> wrap(HashMap<String, String> map) {
      @SuppressWarnings("unchecked")

    @Test
    public void test_startWithParams_passesParamsToLaunchNow() {
        final AtomicReference<LaunchNowOption> capturedOption = new AtomicReference<>();
        final AtomicBoolean launchNowCalled = new AtomicBoolean(false);

        final LaScheduledJob mockLaJob = createMockLaScheduledJob(opCall -> {
            final LaunchNowOption option = new LaunchNowOption();
            opCall.callback(option);

   * of the {@code Map} method calls and check that {@code setUpRan} is true.
   */
  private static class CheckSetUpHashMapGenerator extends WrappedHashMapGenerator {
    private final AtomicBoolean setUpRan;

    CheckSetUpHashMapGenerator(AtomicBoolean setUpRan) {
      this.setUpRan = setUpRan;
    }

    @Override
    Map<String, String> wrap(HashMap<String, String> map) {
      @SuppressWarnings("unchecked")