}

    @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

   * 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")

import assertk.assertions.isTrue
import java.io.IOException
import java.io.InputStreamReader
import java.io.Reader
import java.nio.charset.StandardCharsets
import java.util.concurrent.atomic.AtomicBoolean
import kotlin.test.assertFailsWith
import okhttp3.MediaType.Companion.toMediaType
import okhttp3.ResponseBody.Companion.toResponseBody
import okhttp3.internal.and
import okio.Buffer
import okio.BufferedSource

    private final AtomicLong currentWindowStart = new AtomicLong(System.currentTimeMillis());

    // Cleanup scheduler
    private final ScheduledExecutorService cleanupScheduler;
    private final AtomicBoolean closed = new AtomicBoolean(false);

    // Statistics
    private final AtomicLong totalAttemptsBlocked = new AtomicLong(0);
    private final AtomicLong totalAccountsLocked = new AtomicLong(0);

    AtomicReference<AbstractFuture<String>> setFutureFuture = Atomics.newReference();
    AtomicBoolean setFutureSetSuccess = new AtomicBoolean();
    AtomicBoolean setFutureCompletionSuccess = new AtomicBoolean();
    AtomicBoolean cancellationSuccess = new AtomicBoolean();
    Runnable cancelRunnable =
        () -> {
          cancellationSuccess.set(currentFuture.get().cancel(true));

        HttpServletRequest request = createMockHttpServletRequest();
        HttpServletResponse response = createMockHttpServletResponse();
        AtomicBoolean chainCalled = new AtomicBoolean(false);
        AtomicBoolean managerCalled = new AtomicBoolean(false);

        FilterChain chain = new FilterChain() {
            @Override

    AtomicReference<AbstractFuture<String>> setFutureFuture = Atomics.newReference();
    AtomicBoolean setFutureSetSuccess = new AtomicBoolean();
    AtomicBoolean setFutureCompletionSuccess = new AtomicBoolean();
    AtomicBoolean cancellationSuccess = new AtomicBoolean();
    Runnable cancelRunnable =
        () -> {
          cancellationSuccess.set(currentFuture.get().cancel(true));

     */
    private static class TestableEntraIdAuthenticator extends EntraIdAuthenticator {
        AtomicBoolean scheduleParentGroupLookupCalled = new AtomicBoolean(false);
        AtomicBoolean processDirectMemberOfCalled = new AtomicBoolean(false);

        @Override
        protected void scheduleParentGroupLookup(EntraIdUser user, List<String> initialGroups, List<String> initialRoles,