/*
 * Copyright OpenSearch Contributors
 * SPDX-License-Identifier: Apache-2.0
 */

package org.opensearch.search.asynchronous.context.permits;

import org.opensearch.search.asynchronous.context.AsynchronousSearchContextId;
import org.opensearch.search.asynchronous.context.active.AsynchronousSearchContextClosedException;
import org.opensearch.action.ActionListener;
import org.opensearch.action.LatchedActionListener;
import org.opensearch.common.unit.TimeValue;
import org.opensearch.test.OpenSearchTestCase;
import org.junit.Assert;

import java.util.UUID;
import java.util.concurrent.CountDownLatch;

public class NoopAsynchronousSearchContextPermitsTests extends OpenSearchTestCase {

    public void testAcquireAllPermits() {
        NoopAsynchronousSearchContextPermits permits = new NoopAsynchronousSearchContextPermits(
                new AsynchronousSearchContextId(UUID.randomUUID().toString(), randomNonNegativeLong()));
        expectThrows(IllegalStateException.class,
                () -> permits.asyncAcquireAllPermits(ActionListener.wrap(Assert::fail), TimeValue.ZERO, "reason"));
    }

    public void testAcquireSinglePermit() throws InterruptedException {
        NoopAsynchronousSearchContextPermits permits = new NoopAsynchronousSearchContextPermits(
                new AsynchronousSearchContextId(UUID.randomUUID().toString(), randomNonNegativeLong()));
        CountDownLatch countDownLatch = new CountDownLatch(1);
        permits.asyncAcquirePermit(new LatchedActionListener<>(ActionListener.wrap(r -> {},
                e -> fail("expected permit acquisition to succeed")), countDownLatch), TimeValue.ZERO, "reason");
        countDownLatch.await();
    }

    public void testAcquireSinglePermitAfterClosure() throws InterruptedException {
        AsynchronousSearchContextId contextId = new AsynchronousSearchContextId(UUID.randomUUID().toString(), randomNonNegativeLong());
        NoopAsynchronousSearchContextPermits permits = new NoopAsynchronousSearchContextPermits(
                contextId);
        permits.close();
        CountDownLatch countDownLatch = new CountDownLatch(1);
        permits.asyncAcquirePermit(new LatchedActionListener<>(ActionListener.wrap(
                r -> fail("expected permit acquisition to fail due to permit closure"),
                e -> {
                    assertTrue("expected context closed exception, got " + e.getClass(),
                            e instanceof AsynchronousSearchContextClosedException);
                    assertTrue(((AsynchronousSearchContextClosedException) e).getAsynchronousSearchContextId().equals(contextId));
                }), countDownLatch), TimeValue.ZERO, "reason");
        countDownLatch.await();
    }

}