package org.opendaylight.yangtools.util.concurrent;
-import static org.junit.Assert.*;
-
-import java.util.concurrent.Callable;
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertNotNull;
+import static org.junit.Assert.assertTrue;
+import static org.opendaylight.yangtools.util.concurrent.AsyncNotifyingListeningExecutorServiceTest.testListenerCallback;
+import static org.opendaylight.yangtools.util.concurrent.CommonTestUtils.SUBMIT_CALLABLE;
+import static org.opendaylight.yangtools.util.concurrent.CommonTestUtils.SUBMIT_RUNNABLE;
+import static org.opendaylight.yangtools.util.concurrent.CommonTestUtils.SUBMIT_RUNNABLE_WITH_RESULT;
+
+import com.google.common.base.Supplier;
+import com.google.common.util.concurrent.FutureCallback;
+import com.google.common.util.concurrent.Futures;
+import com.google.common.util.concurrent.ListenableFuture;
+import com.google.common.util.concurrent.ListeningExecutorService;
+import com.google.common.util.concurrent.ThreadFactoryBuilder;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.ExecutionException;
+import java.util.concurrent.Executor;
+import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicReference;
-
+import org.junit.After;
import org.junit.Before;
import org.junit.Test;
-
-import com.google.common.base.Function;
-import com.google.common.util.concurrent.FutureCallback;
-import com.google.common.util.concurrent.Futures;
-import com.google.common.util.concurrent.ListenableFuture;
-import com.google.common.util.concurrent.ListeningExecutorService;
+import org.opendaylight.yangtools.util.concurrent.CommonTestUtils.Invoker;
/**
* Unit tests for DeadlockDetectingListeningExecutorService.
*/
public class DeadlockDetectingListeningExecutorServiceTest {
- interface Invoker {
- ListenableFuture<?> invokeExecutor( ListeningExecutorService executor );
- };
-
- static final Invoker SUBMIT_CALLABLE = new Invoker() {
- @Override
- public ListenableFuture<?> invokeExecutor( ListeningExecutorService executor ) {
- return executor.submit( new Callable<String>() {
- @Override
- public String call() throws Exception{
- return "foo";
- }
- } );
- }
- };
-
- static final Invoker SUBMIT_RUNNABLE = new Invoker() {
- @Override
- public ListenableFuture<?> invokeExecutor( ListeningExecutorService executor ) {
- return executor.submit( new Runnable() {
- @Override
- public void run(){
- }
- } );
- }
- };
-
- static final Invoker SUBMIT_RUNNABLE_WITH_RESULT = new Invoker() {
- @Override
- public ListenableFuture<?> invokeExecutor( ListeningExecutorService executor ) {
- return executor.submit( new Runnable() {
- @Override
- public void run(){
- }
- }, "foo" );
- }
- };
-
interface InitialInvoker {
- void invokeExecutor( ListeningExecutorService executor, Runnable task );
- };
+ void invokeExecutor(ListeningExecutorService executor, Runnable task);
+ }
- static final InitialInvoker SUBMIT = new InitialInvoker() {
- @Override
- public void invokeExecutor( ListeningExecutorService executor, Runnable task ) {
- executor.submit( task );
- }
- };
+ static final InitialInvoker SUBMIT = ListeningExecutorService::submit;
- static final InitialInvoker EXECUTE = new InitialInvoker() {
- @Override
- public void invokeExecutor( ListeningExecutorService executor, Runnable task ) {
- executor.execute( task );
- }
- };
+ static final InitialInvoker EXECUTE = Executor::execute;
- @SuppressWarnings("serial")
public static class TestDeadlockException extends Exception {
+ private static final long serialVersionUID = 1L;
+
}
- public static Function<Void, Exception> DEADLOCK_EXECUTOR_FUNCTION = new Function<Void, Exception>() {
- @Override
- public Exception apply( Void notUsed ) {
- return new TestDeadlockException();
- }
- };
+ private static final Supplier<Exception> DEADLOCK_EXECUTOR_SUPPLIER = TestDeadlockException::new;
DeadlockDetectingListeningExecutorService executor;
@Before
public void setup() {
- executor = new DeadlockDetectingListeningExecutorService( Executors.newSingleThreadExecutor(),
- DEADLOCK_EXECUTOR_FUNCTION );
+ }
+
+ @After
+ public void tearDown() {
+ if (executor != null) {
+ executor.shutdownNow();
+ }
+ }
+
+ DeadlockDetectingListeningExecutorService newExecutor() {
+ return new DeadlockDetectingListeningExecutorService(Executors.newSingleThreadExecutor(),
+ DEADLOCK_EXECUTOR_SUPPLIER);
}
@Test
public void testBlockingSubmitOffExecutor() throws Exception {
+ executor = newExecutor();
+
// Test submit with Callable.
- ListenableFuture<String> future = executor.submit( new Callable<String>() {
- @Override
- public String call() throws Exception{
- return "foo";
- }
- } );
+ ListenableFuture<String> future = executor.submit(() -> "foo");
- assertEquals( "Future result", "foo", future.get( 5, TimeUnit.SECONDS ) );
+ assertEquals( "Future result", "foo", future.get( 5, TimeUnit.SECONDS));
// Test submit with Runnable.
- executor.submit( new Runnable() {
- @Override
- public void run(){
- }
- } ).get();
+ executor.submit(() -> {
+ }).get();
// Test submit with Runnable and value.
- future = executor.submit( new Runnable() {
- @Override
- public void run(){
- }
- }, "foo" );
+ future = executor.submit(() -> { }, "foo");
- assertEquals( "Future result", "foo", future.get( 5, TimeUnit.SECONDS ) );
+ assertEquals("Future result", "foo", future.get(5, TimeUnit.SECONDS));
}
@Test
public void testNonBlockingSubmitOnExecutorThread() throws Throwable {
- testNonBlockingSubmitOnExecutorThread( SUBMIT, SUBMIT_CALLABLE );
- testNonBlockingSubmitOnExecutorThread( SUBMIT, SUBMIT_RUNNABLE );
- testNonBlockingSubmitOnExecutorThread( SUBMIT, SUBMIT_RUNNABLE_WITH_RESULT );
+ executor = newExecutor();
+
+ testNonBlockingSubmitOnExecutorThread(SUBMIT, SUBMIT_CALLABLE);
+ testNonBlockingSubmitOnExecutorThread(SUBMIT, SUBMIT_RUNNABLE);
+ testNonBlockingSubmitOnExecutorThread(SUBMIT, SUBMIT_RUNNABLE_WITH_RESULT);
- testNonBlockingSubmitOnExecutorThread( EXECUTE, SUBMIT_CALLABLE );
+ testNonBlockingSubmitOnExecutorThread(EXECUTE, SUBMIT_CALLABLE);
}
- void testNonBlockingSubmitOnExecutorThread( InitialInvoker initialInvoker,
- final Invoker invoker ) throws Throwable {
+ void testNonBlockingSubmitOnExecutorThread(final InitialInvoker initialInvoker, final Invoker invoker)
+ throws Throwable {
final AtomicReference<Throwable> caughtEx = new AtomicReference<>();
- final CountDownLatch futureCompletedLatch = new CountDownLatch( 1 );
+ final CountDownLatch futureCompletedLatch = new CountDownLatch(1);
- Runnable task = new Runnable() {
- @SuppressWarnings({ "unchecked", "rawtypes" })
+ Runnable task = () -> Futures.addCallback(invoker.invokeExecutor(executor, null), new FutureCallback<Object>() {
@Override
- public void run() {
-
- Futures.addCallback( invoker.invokeExecutor( executor ), new FutureCallback() {
- @Override
- public void onSuccess( Object result ) {
- futureCompletedLatch.countDown();
- }
-
- @Override
- public void onFailure( Throwable t ) {
- caughtEx.set( t );
- futureCompletedLatch.countDown();
- }
- } );
+ public void onSuccess(final Object result) {
+ futureCompletedLatch.countDown();
}
- };
+ @Override
+ public void onFailure(final Throwable t) {
+ caughtEx.set(t);
+ futureCompletedLatch.countDown();
+ }
+ });
- initialInvoker.invokeExecutor( executor, task );
+ initialInvoker.invokeExecutor(executor, task);
- assertTrue( "Task did not complete - executor likely deadlocked",
- futureCompletedLatch.await( 5, TimeUnit.SECONDS ) );
+ assertTrue("Task did not complete - executor likely deadlocked",
+ futureCompletedLatch.await(5, TimeUnit.SECONDS));
- if( caughtEx.get() != null ) {
+ if (caughtEx.get() != null) {
throw caughtEx.get();
}
}
@Test
public void testBlockingSubmitOnExecutorThread() throws Exception {
- testBlockingSubmitOnExecutorThread( SUBMIT, SUBMIT_CALLABLE );
- testBlockingSubmitOnExecutorThread( SUBMIT, SUBMIT_RUNNABLE );
- testBlockingSubmitOnExecutorThread( SUBMIT, SUBMIT_RUNNABLE_WITH_RESULT );
+ executor = newExecutor();
- testBlockingSubmitOnExecutorThread( EXECUTE, SUBMIT_CALLABLE );
+ testBlockingSubmitOnExecutorThread(SUBMIT, SUBMIT_CALLABLE);
+ testBlockingSubmitOnExecutorThread(SUBMIT, SUBMIT_RUNNABLE);
+ testBlockingSubmitOnExecutorThread(SUBMIT, SUBMIT_RUNNABLE_WITH_RESULT);
+
+ testBlockingSubmitOnExecutorThread(EXECUTE, SUBMIT_CALLABLE);
}
- void testBlockingSubmitOnExecutorThread( InitialInvoker initialInvoker,
- final Invoker invoker ) throws Exception {
+ void testBlockingSubmitOnExecutorThread(final InitialInvoker initialInvoker, final Invoker invoker)
+ throws Exception {
final AtomicReference<Throwable> caughtEx = new AtomicReference<>();
- final CountDownLatch latch = new CountDownLatch( 1 );
-
- Runnable task = new Runnable() {
- @Override
- public void run() {
-
- try {
- invoker.invokeExecutor( executor ).get();
- } catch( ExecutionException e ) {
- caughtEx.set( e.getCause() );
- } catch( Throwable e ) {
- caughtEx.set( e );
- } finally {
- latch.countDown();
- }
+ final CountDownLatch latch = new CountDownLatch(1);
+
+ Runnable task = () -> {
+
+ try {
+ invoker.invokeExecutor(executor, null).get();
+ } catch(ExecutionException e) {
+ caughtEx.set(e.getCause());
+ } catch(Throwable e) {
+ caughtEx.set(e);
+ } finally {
+ latch.countDown();
}
-
};
- initialInvoker.invokeExecutor( executor, task );
+ initialInvoker.invokeExecutor(executor, task);
- assertTrue( "Task did not complete - executor likely deadlocked",
- latch.await( 5, TimeUnit.SECONDS ) );
+ assertTrue("Task did not complete - executor likely deadlocked", latch.await( 5, TimeUnit.SECONDS));
+ assertNotNull("Expected exception thrown", caughtEx.get());
+ assertEquals("Caught exception type", TestDeadlockException.class, caughtEx.get().getClass());
+ }
- assertNotNull( "Expected exception thrown", caughtEx.get() );
- assertEquals( "Caught exception type", TestDeadlockException.class, caughtEx.get().getClass() );
+ @Test
+ public void testListenableFutureCallbackWithExecutor() throws InterruptedException {
+
+ String listenerThreadPrefix = "ListenerThread";
+ ExecutorService listenerExecutor = Executors.newFixedThreadPool(1,
+ new ThreadFactoryBuilder().setNameFormat(listenerThreadPrefix + "-%d").build());
+
+ executor = new DeadlockDetectingListeningExecutorService(
+ Executors.newSingleThreadExecutor(new ThreadFactoryBuilder().setNameFormat("SingleThread").build()),
+ DEADLOCK_EXECUTOR_SUPPLIER, listenerExecutor);
+
+ try {
+ testListenerCallback(executor, SUBMIT_CALLABLE, listenerThreadPrefix);
+ testListenerCallback(executor, SUBMIT_RUNNABLE, listenerThreadPrefix);
+ testListenerCallback(executor, SUBMIT_RUNNABLE_WITH_RESULT, listenerThreadPrefix);
+ } finally {
+ listenerExecutor.shutdownNow();
+ }
}
}
Code Review / yangtools.git / blobdiff