/* Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.camunda.bpm.engine.test.util;

import java.util.concurrent.atomic.AtomicInteger;

import org.camunda.bpm.engine.impl.util.SingleConsumerCondition;
import org.junit.Assert;
import org.junit.Test;

public class SingleConsumerConditionTest {

  @Test(timeout=10000)
  public void shouldNotBlockIfSignalAvailable() {
    SingleConsumerCondition condition = new SingleConsumerCondition(Thread.currentThread());

    // given
    condition.signal();

    // then
    condition.await(100000);
  }

  @Test(timeout=10000)
  public void shouldNotBlockIfSignalAvailableDifferentThread() throws InterruptedException {

    final SingleConsumerCondition condition = new SingleConsumerCondition(Thread.currentThread());

    Thread consumer = new Thread() {
      @Override
      public void run() {
        condition.signal();
      }
    };

    consumer.start();
    consumer.join();

    // then
    condition.await(100000);
  }

  @Test
  public void cannotAwaitFromDifferentThread() {
    // given
    SingleConsumerCondition condition = new SingleConsumerCondition(new Thread());

    // when then
    try {
      condition.await(0);
      Assert.fail("expected exception");
    }
    catch (RuntimeException e) {
      // expected
    }
  }

  @Test
  public void cannotCreateWithNull() {
    try {
      new SingleConsumerCondition(null);
      Assert.fail("expected exception");
    }
    catch (IllegalArgumentException e) {
      // expected
    }
  }

  @Test(timeout = 10000)
  public void conditionStressTest() throws InterruptedException {

    final SingleConsumerCondition condition = new SingleConsumerCondition(Thread.currentThread());

    final AtomicInteger signalCounter = new AtomicInteger();

    Thread consumer = new Thread() {
      @Override
      public void run() {
        for (int i = 0; i < 500000; i++) {

          condition.signal();

          while (signalCounter.get() == i) {
            Thread.yield();
          }
        }
      }
    };

    consumer.start();

    for (int i = 0; i < 500000; i++) {
      condition.await(100000);
      signalCounter.set(i + 1);
    }

    consumer.join();
  }

}

/* Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.camunda.bpm.engine.test.util;

import java.util.concurrent.atomic.AtomicInteger;

import org.camunda.bpm.engine.impl.util.SingleConsumerCondition;
import org.junit.Assert;
import org.junit.Test;

public class SingleConsumerConditionTest {

  @Test(timeout=10000)
  public void shouldNotBlockIfSignalAvailable() {
    SingleConsumerCondition condition = new SingleConsumerCondition(Thread.currentThread());

    // given
    condition.signal();

    // then
    condition.await(100000);
  }

  @Test(timeout=10000)
  public void shouldNotBlockIfSignalAvailableDifferentThread() throws InterruptedException {

    final SingleConsumerCondition condition = new SingleConsumerCondition(Thread.currentThread());

    Thread consumer = new Thread() {
      @Override
      public void run() {
        condition.signal();
      }
    };

    consumer.start();
    consumer.join();

    // then
    condition.await(100000);
  }

  @Test
  public void cannotAwaitFromDifferentThread() {
    // given
    SingleConsumerCondition condition = new SingleConsumerCondition(new Thread());

    // when then
    try {
      condition.await(0);
      Assert.fail("expected exception");
    }
    catch (RuntimeException e) {
      // expected
    }
  }

  @Test
  public void cannotCreateWithNull() {
    try {
      new SingleConsumerCondition(null);
      Assert.fail("expected exception");
    }
    catch (IllegalArgumentException e) {
      // expected
    }
  }

  @Test(timeout = 100000)
  public void conditionStressTest() throws InterruptedException {

    final SingleConsumerCondition condition = new SingleConsumerCondition(Thread.currentThread());

    final AtomicInteger signalCounter = new AtomicInteger();

    Thread consumer = new Thread() {
      @Override
      public void run() {
        for (int i = 0; i < 500000; i++) {

          condition.signal();

          while (signalCounter.get() == i) {
            Thread.yield();
          }
        }
      }
    };

    consumer.start();

    for (int i = 0; i < 500000; i++) {
      condition.await(100000);
      signalCounter.set(i + 1);
    }

    consumer.join();
  }

}

increase Timeout value for SingleConsumerConditionTest