Browse Source
git-svn-id: https://src.springframework.org/svn/spring-framework/trunk@3753 50f2f4bb-b051-0410-bef5-90022cba6387pull/1/head
1 changed files with 178 additions and 161 deletions
@ -1,161 +1,178 @@
@@ -1,161 +1,178 @@
|
||||
package org.springframework.scheduling; |
||||
|
||||
import java.util.concurrent.Callable; |
||||
import java.util.concurrent.ExecutionException; |
||||
import java.util.concurrent.Future; |
||||
|
||||
import junit.framework.Assert; |
||||
|
||||
import static junit.framework.Assert.*; |
||||
|
||||
import org.junit.Before; |
||||
import org.junit.Test; |
||||
import org.springframework.core.task.SimpleAsyncTaskExecutor; |
||||
import org.springframework.scheduling.annotation.Async; |
||||
import org.springframework.scheduling.annotation.AsyncResult; |
||||
|
||||
/** |
||||
* Test for AnnotationDrivenAsynchronousExecutionAspect |
||||
* |
||||
* @author Ramnivas Laddad |
||||
*/ |
||||
public class AnnotationDrivenAsynchronousExecutionAspectTest { |
||||
private static final long WAIT_TIME = 1000; //milli seconds
|
||||
private CountingExecutor executor; |
||||
|
||||
@Before |
||||
public void setUp() { |
||||
executor = new CountingExecutor(); |
||||
AnnotationDrivenAsynchronousExecutionAspect.aspectOf().setExecutor(executor); |
||||
} |
||||
|
||||
@Test |
||||
public void asyncMethodGetsRoutedAsynchronously() { |
||||
ClassWithoutAsyncAnnotation obj = new ClassWithoutAsyncAnnotation(); |
||||
obj.incrementAsync(); |
||||
executor.waitForCompletion(); |
||||
assertEquals(1, obj.counter); |
||||
assertEquals(1, executor.submitStartCounter); |
||||
assertEquals(1, executor.submitCompleteCounter); |
||||
} |
||||
|
||||
@Test |
||||
public void asyncMethodReturningFutureGetsRoutedAsynchronouslyAndReturnsAFuture() throws InterruptedException, ExecutionException { |
||||
ClassWithoutAsyncAnnotation obj = new ClassWithoutAsyncAnnotation(); |
||||
Future<Integer> future = obj.incrementReturningAFuture(); |
||||
// No need to executor.waitForCompletion() as future.get() will have the same effect
|
||||
assertEquals(5, future.get().intValue()); |
||||
assertEquals(1, obj.counter); |
||||
assertEquals(1, executor.submitStartCounter); |
||||
assertEquals(1, executor.submitCompleteCounter); |
||||
} |
||||
|
||||
@Test |
||||
public void syncMethodGetsRoutedSynchronously() { |
||||
ClassWithoutAsyncAnnotation obj = new ClassWithoutAsyncAnnotation(); |
||||
obj.increment(); |
||||
assertEquals(1, obj.counter); |
||||
assertEquals(0, executor.submitStartCounter); |
||||
assertEquals(0, executor.submitCompleteCounter); |
||||
} |
||||
|
||||
@Test |
||||
public void voidMethodInAsyncClassGetsRoutedAsynchronously() { |
||||
ClassWithAsyncAnnotation obj = new ClassWithAsyncAnnotation(); |
||||
obj.increment(); |
||||
executor.waitForCompletion(); |
||||
assertEquals(1, obj.counter); |
||||
assertEquals(1, executor.submitStartCounter); |
||||
assertEquals(1, executor.submitCompleteCounter); |
||||
} |
||||
|
||||
@Test |
||||
public void methodReturningFutureInAsyncClassGetsRoutedAsynchronouslyAndReturnsAFuture() throws InterruptedException, ExecutionException { |
||||
ClassWithAsyncAnnotation obj = new ClassWithAsyncAnnotation(); |
||||
Future<Integer> future = obj.incrementReturningAFuture(); |
||||
assertEquals(5, future.get().intValue()); |
||||
assertEquals(1, obj.counter); |
||||
assertEquals(1, executor.submitStartCounter); |
||||
assertEquals(1, executor.submitCompleteCounter); |
||||
} |
||||
|
||||
@Test |
||||
public void methodReturningNonVoidNonFutureInAsyncClassGetsRoutedSynchronously() { |
||||
ClassWithAsyncAnnotation obj = new ClassWithAsyncAnnotation(); |
||||
int returnValue = obj.return5(); |
||||
assertEquals(5, returnValue); |
||||
assertEquals(0, executor.submitStartCounter); |
||||
assertEquals(0, executor.submitCompleteCounter); |
||||
} |
||||
|
||||
@SuppressWarnings("serial") |
||||
private static class CountingExecutor extends SimpleAsyncTaskExecutor { |
||||
int submitStartCounter; |
||||
int submitCompleteCounter; |
||||
|
||||
@Override |
||||
public <T> Future<T> submit(Callable<T> task) { |
||||
submitStartCounter++; |
||||
Future<T> future = super.submit(task); |
||||
submitCompleteCounter++; |
||||
synchronized (this) { |
||||
notifyAll(); |
||||
} |
||||
return future; |
||||
} |
||||
|
||||
public synchronized void waitForCompletion() { |
||||
try { |
||||
wait(WAIT_TIME); |
||||
} catch (InterruptedException e) { |
||||
Assert.fail("Didn't finish the async job in " + WAIT_TIME + " milliseconds"); |
||||
} |
||||
} |
||||
} |
||||
|
||||
static class ClassWithoutAsyncAnnotation { |
||||
int counter; |
||||
|
||||
@Async public void incrementAsync() { |
||||
counter++; |
||||
} |
||||
|
||||
public void increment() { |
||||
counter++; |
||||
} |
||||
|
||||
@Async public Future<Integer> incrementReturningAFuture() { |
||||
counter++; |
||||
return new AsyncResult<Integer>(5); |
||||
} |
||||
|
||||
// It should be an error to attach @Async to a method that returns a non-void
|
||||
// or non-Future.
|
||||
// We need to keep this commented out, otherwise there will be a compile-time error.
|
||||
// Please uncomment and re-comment this periodically to check that the compiler
|
||||
// produces an error message due to the 'declare error' statement
|
||||
// in AnnotationDrivenAsynchronousExecutionAspect
|
||||
// @Async public int getInt() {
|
||||
// return 0;
|
||||
// }
|
||||
} |
||||
|
||||
@Async |
||||
static class ClassWithAsyncAnnotation { |
||||
int counter; |
||||
|
||||
public void increment() { |
||||
counter++; |
||||
} |
||||
|
||||
// Manually check that there is a warning from the 'declare warning' statement in AnnotationDrivenAsynchronousExecutionAspect
|
||||
public int return5() { |
||||
return 5; |
||||
} |
||||
|
||||
public Future<Integer> incrementReturningAFuture() { |
||||
counter++; |
||||
return new AsyncResult<Integer>(5); |
||||
} |
||||
} |
||||
} |
||||
/* |
||||
* Copyright 2002-2010 the original author or authors. |
||||
* |
||||
* 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.springframework.scheduling.aspectj; |
||||
|
||||
import java.util.concurrent.Callable; |
||||
import java.util.concurrent.ExecutionException; |
||||
import java.util.concurrent.Future; |
||||
|
||||
import junit.framework.Assert; |
||||
|
||||
import static junit.framework.Assert.*; |
||||
|
||||
import org.junit.Before; |
||||
import org.junit.Test; |
||||
import org.springframework.core.task.SimpleAsyncTaskExecutor; |
||||
import org.springframework.scheduling.annotation.Async; |
||||
import org.springframework.scheduling.annotation.AsyncResult; |
||||
|
||||
/** |
||||
* @author Ramnivas Laddad |
||||
*/ |
||||
public class AnnotationDrivenAsynchronousExecutionAspectTests { |
||||
|
||||
private static final long WAIT_TIME = 1000; //milli seconds
|
||||
|
||||
private CountingExecutor executor; |
||||
|
||||
@Before |
||||
public void setUp() { |
||||
executor = new CountingExecutor(); |
||||
AnnotationDrivenAsynchronousExecutionAspect.aspectOf().setExecutor(executor); |
||||
} |
||||
|
||||
@Test |
||||
public void asyncMethodGetsRoutedAsynchronously() { |
||||
ClassWithoutAsyncAnnotation obj = new ClassWithoutAsyncAnnotation(); |
||||
obj.incrementAsync(); |
||||
executor.waitForCompletion(); |
||||
assertEquals(1, obj.counter); |
||||
assertEquals(1, executor.submitStartCounter); |
||||
assertEquals(1, executor.submitCompleteCounter); |
||||
} |
||||
|
||||
@Test |
||||
public void asyncMethodReturningFutureGetsRoutedAsynchronouslyAndReturnsAFuture() throws InterruptedException, ExecutionException { |
||||
ClassWithoutAsyncAnnotation obj = new ClassWithoutAsyncAnnotation(); |
||||
Future<Integer> future = obj.incrementReturningAFuture(); |
||||
// No need to executor.waitForCompletion() as future.get() will have the same effect
|
||||
assertEquals(5, future.get().intValue()); |
||||
assertEquals(1, obj.counter); |
||||
assertEquals(1, executor.submitStartCounter); |
||||
assertEquals(1, executor.submitCompleteCounter); |
||||
} |
||||
|
||||
@Test |
||||
public void syncMethodGetsRoutedSynchronously() { |
||||
ClassWithoutAsyncAnnotation obj = new ClassWithoutAsyncAnnotation(); |
||||
obj.increment(); |
||||
assertEquals(1, obj.counter); |
||||
assertEquals(0, executor.submitStartCounter); |
||||
assertEquals(0, executor.submitCompleteCounter); |
||||
} |
||||
|
||||
@Test |
||||
public void voidMethodInAsyncClassGetsRoutedAsynchronously() { |
||||
ClassWithAsyncAnnotation obj = new ClassWithAsyncAnnotation(); |
||||
obj.increment(); |
||||
executor.waitForCompletion(); |
||||
assertEquals(1, obj.counter); |
||||
assertEquals(1, executor.submitStartCounter); |
||||
assertEquals(1, executor.submitCompleteCounter); |
||||
} |
||||
|
||||
@Test |
||||
public void methodReturningFutureInAsyncClassGetsRoutedAsynchronouslyAndReturnsAFuture() throws InterruptedException, ExecutionException { |
||||
ClassWithAsyncAnnotation obj = new ClassWithAsyncAnnotation(); |
||||
Future<Integer> future = obj.incrementReturningAFuture(); |
||||
assertEquals(5, future.get().intValue()); |
||||
assertEquals(1, obj.counter); |
||||
assertEquals(1, executor.submitStartCounter); |
||||
assertEquals(1, executor.submitCompleteCounter); |
||||
} |
||||
|
||||
@Test |
||||
public void methodReturningNonVoidNonFutureInAsyncClassGetsRoutedSynchronously() { |
||||
ClassWithAsyncAnnotation obj = new ClassWithAsyncAnnotation(); |
||||
int returnValue = obj.return5(); |
||||
assertEquals(5, returnValue); |
||||
assertEquals(0, executor.submitStartCounter); |
||||
assertEquals(0, executor.submitCompleteCounter); |
||||
} |
||||
|
||||
@SuppressWarnings("serial") |
||||
private static class CountingExecutor extends SimpleAsyncTaskExecutor { |
||||
int submitStartCounter; |
||||
int submitCompleteCounter; |
||||
|
||||
@Override |
||||
public <T> Future<T> submit(Callable<T> task) { |
||||
submitStartCounter++; |
||||
Future<T> future = super.submit(task); |
||||
submitCompleteCounter++; |
||||
synchronized (this) { |
||||
notifyAll(); |
||||
} |
||||
return future; |
||||
} |
||||
|
||||
public synchronized void waitForCompletion() { |
||||
try { |
||||
wait(WAIT_TIME); |
||||
} catch (InterruptedException e) { |
||||
Assert.fail("Didn't finish the async job in " + WAIT_TIME + " milliseconds"); |
||||
} |
||||
} |
||||
} |
||||
|
||||
static class ClassWithoutAsyncAnnotation { |
||||
int counter; |
||||
|
||||
@Async public void incrementAsync() { |
||||
counter++; |
||||
} |
||||
|
||||
public void increment() { |
||||
counter++; |
||||
} |
||||
|
||||
@Async public Future<Integer> incrementReturningAFuture() { |
||||
counter++; |
||||
return new AsyncResult<Integer>(5); |
||||
} |
||||
|
||||
// It should be an error to attach @Async to a method that returns a non-void
|
||||
// or non-Future.
|
||||
// We need to keep this commented out, otherwise there will be a compile-time error.
|
||||
// Please uncomment and re-comment this periodically to check that the compiler
|
||||
// produces an error message due to the 'declare error' statement
|
||||
// in AnnotationDrivenAsynchronousExecutionAspect
|
||||
// @Async public int getInt() {
|
||||
// return 0;
|
||||
// }
|
||||
} |
||||
|
||||
@Async |
||||
static class ClassWithAsyncAnnotation { |
||||
int counter; |
||||
|
||||
public void increment() { |
||||
counter++; |
||||
} |
||||
|
||||
// Manually check that there is a warning from the 'declare warning' statement in AnnotationDrivenAsynchronousExecutionAspect
|
||||
public int return5() { |
||||
return 5; |
||||
} |
||||
|
||||
public Future<Integer> incrementReturningAFuture() { |
||||
counter++; |
||||
return new AsyncResult<Integer>(5); |
||||
} |
||||
} |
||||
|
||||
} |
||||
Loading…
Reference in new issue