/*
 * The contents of this file are subject to the terms of the Common Development and
 * Distribution License (the License). You may not use this file except in compliance with the
 * License.
 *
 * You can obtain a copy of the License at legal/CDDLv1.0.txt. See the License for the
 * specific language governing permission and limitations under the License.
 *
 * When distributing Covered Software, include this CDDL Header Notice in each file and include
 * the License file at legal/CDDLv1.0.txt. If applicable, add the following below the CDDL
 * Header, with the fields enclosed by brackets [] replaced by your own identifying
 * information: "Portions Copyright [year] [name of copyright owner]".
 *
 * Copyright 2016 ForgeRock AS.
 */
package org.opends.server.util;

import static java.util.concurrent.TimeUnit.*;

import static org.mockito.Mockito.*;

import java.util.concurrent.Callable;

import org.opends.server.DirectoryServerTestCase;
import org.opends.server.TestCaseUtils;
import org.opends.server.util.TestTimer.CallableVoid;
import org.testng.annotations.BeforeClass;
import org.testng.annotations.Test;

@SuppressWarnings("javadoc")
@Test(groups = "precommit", sequential = true)
public class CronExecutorServiceTest extends DirectoryServerTestCase
{
  @BeforeClass
  public void setUp() throws Exception
  {
    TestCaseUtils.startServer();
  }

  @Test
  public void execute() throws Exception
  {
    final Runnable mock = mock(Runnable.class);
    new CronExecutorService().execute(mock);

    verifyRunnableInvokedOnce(mock);
  }

  @Test
  public void submitRunnable() throws Exception
  {
    final Runnable mock = mock(Runnable.class);
    new CronExecutorService().submit(mock);

    verifyRunnableInvokedOnce(mock);
  }

  @Test
  public void submitRunnableAndReturn() throws Exception
  {
    final Runnable mock = mock(Runnable.class);
    new CronExecutorService().submit(mock, null);

    verifyRunnableInvokedOnce(mock);
  }

  @Test
  public void submitCallable() throws Exception
  {
    final Callable<Void> mock = mock(Callable.class);
    new CronExecutorService().submit(mock);

    verifyCallableInvokedOnce(mock);
  }

  @Test
  public void scheduleRunnable() throws Exception
  {
    final Runnable mock = mock(Runnable.class);
    new CronExecutorService().schedule(mock, 200, MILLISECONDS);

    verifyNoMoreInteractions(mock);

    Thread.sleep(SECONDS.toMillis(1));

    verifyRunnableInvokedOnce(mock);
  }

  @Test
  public void scheduleCallable() throws Exception
  {
    final Callable<Void> mock = mock(Callable.class);
    new CronExecutorService().schedule(mock, 200, MILLISECONDS);

    verifyNoMoreInteractions(mock);

    Thread.sleep(SECONDS.toMillis(1));

    maxOneSecond().repeatUntilSuccess(new CallableVoid()
    {
      @Override
      public void call() throws Exception
      {
        verify(mock, atLeastOnce()).call();
        verifyNoMoreInteractions(mock);
      }
    });
  }

  @Test
  public void scheduleAtFixedRate() throws Exception
  {
    final Runnable mock = mock(Runnable.class);
    new CronExecutorService().scheduleAtFixedRate(mock, 0 /* execute immediately */, 200, MILLISECONDS);

    //verifyNoMoreInteractions(mock);

    Thread.sleep(MILLISECONDS.toMillis(200));

    maxOneSecond().repeatUntilSuccess(new CallableVoid()
    {
      @Override
      public void call() throws Exception
      {
        verify(mock, atLeastOnce()).run();
        verifyNoMoreInteractions(mock);
      }
    });
  }

  private void verifyRunnableInvokedOnce(final Runnable mock) throws Exception, InterruptedException
  {
    maxOneSecond().repeatUntilSuccess(new CallableVoid()
    {
      @Override
      public void call() throws Exception
      {
        verify(mock).run();
        verifyNoMoreInteractions(mock);
      }
    });
  }

  private void verifyCallableInvokedOnce(final Callable<Void> mock) throws Exception, InterruptedException
  {
    maxOneSecond().repeatUntilSuccess(new CallableVoid()
    {
      @Override
      public void call() throws Exception
      {
        verify(mock).call();
        verifyNoMoreInteractions(mock);
      }
    });
  }

  private TestTimer maxOneSecond()
  {
    return new TestTimer.Builder()
        .maxSleep(1, SECONDS)
        .sleepTimes(10, MILLISECONDS)
        .toTimer();
  }
}