aspectjprecedenceawareordercomparatortests.java

struts+spring 源码 希望能给大家带来帮助

/*
 * Copyright 2002-2006 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.
 *
 * Created on 15-Feb-2006 by Adrian Colyer
 */
package org.springframework.aop.aspectj.autoproxy;

import java.lang.reflect.Method;

import org.aopalliance.aop.Advice;
import org.springframework.aop.Advisor;
import org.springframework.aop.AfterReturningAdvice;
import org.springframework.aop.BeforeAdvice;
import org.springframework.aop.aspectj.AbstractAspectJAdvice;
import org.springframework.aop.aspectj.AspectJAfterAdvice;
import org.springframework.aop.aspectj.AspectJAfterReturningAdvice;
import org.springframework.aop.aspectj.AspectJAfterThrowingAdvice;
import org.springframework.aop.aspectj.AspectJAroundAdvice;
import org.springframework.aop.aspectj.AspectJExpressionPointcut;
import org.springframework.aop.aspectj.AspectJMethodBeforeAdvice;
import org.springframework.aop.aspectj.AspectJPointcutAdvisor;
import org.springframework.aop.support.DefaultPointcutAdvisor;

import junit.framework.TestCase;

/**
 * @author Adrian Colyer
 * @since 2.0
 */
public class AspectJPrecedenceAwareOrderComparatorTests extends TestCase {

	/*
	 * Specification for the comparator (as defined in the 
	 * AspectJPrecedenceAwareOrderComparator class)
	 * 
	 * <p>
	 * Orders AspectJ advice/advisors by invocation order.
	 * </p>
	 * <p>
	 * Given two pieces of advice, <code>a</code> and <code>b</code>:
	 * </p>
	 * <ul>
	 *   <li>if <code>a</code> and <code>b</code> are defined in different
	 *   aspects, then the advice in the aspect with the lowest order
	 *   value has the highest precedence</li>
	 *   <li>if <code>a</code> and <code>b</code> are defined in the same
	 *   aspect, then if one of <code>a</code> or <code>b</code> is a form of
	 *   after advice, then the advice declared last in the aspect has the
	 *   highest precedence. If neither <code>a</code> nor <code>b</code> is a
	 *   form of after advice, then the advice declared first in the aspect has
	 *   the highest precedence.</li>
	 * </ul>
	 */
	
	private static final int HIGH_PRECEDENCE_ADVISOR_ORDER = 100;
	private static final int LOW_PRECEDENCE_ADVISOR_ORDER = 200;
	private static final int EARLY_ADVICE_DECLARATION_ORDER = 5;
	private static final int LATE_ADVICE_DECLARATION_ORDER = 10;
	
	private AspectJPrecedenceAwareOrderComparator comparator;
	private Method anyOldMethod;
	private AspectJExpressionPointcut anyOldPointcut;

	protected void setUp() throws Exception {
		super.setUp();
		this.comparator = new AspectJPrecedenceAwareOrderComparator();
		this.anyOldMethod = getClass().getMethods()[0];
		this.anyOldPointcut = new AspectJExpressionPointcut();
		this.anyOldPointcut.setExpression("get(* *");
	}

	public void testSameAspectNoAfterAdvice() {
		Advisor advisor1 = createAspectJBeforeAdvice(HIGH_PRECEDENCE_ADVISOR_ORDER,EARLY_ADVICE_DECLARATION_ORDER,"someAspect");
		Advisor advisor2 = createAspectJBeforeAdvice(HIGH_PRECEDENCE_ADVISOR_ORDER,LATE_ADVICE_DECLARATION_ORDER,"someAspect");
		
		assertEquals("advisor1 sorted before advisor2",-1,this.comparator.compare(advisor1, advisor2));

		advisor1 = createAspectJBeforeAdvice(HIGH_PRECEDENCE_ADVISOR_ORDER,LATE_ADVICE_DECLARATION_ORDER,"someAspect");
		advisor2 = createAspectJAroundAdvice(HIGH_PRECEDENCE_ADVISOR_ORDER,EARLY_ADVICE_DECLARATION_ORDER,"someAspect");
		
		assertEquals("advisor2 sorted before advisor1",1,this.comparator.compare(advisor1, advisor2));
	}
	
	public void testSameAspectAfterAdvice() {
		Advisor advisor1 = createAspectJAfterAdvice(HIGH_PRECEDENCE_ADVISOR_ORDER,EARLY_ADVICE_DECLARATION_ORDER,"someAspect");
		Advisor advisor2 = createAspectJAroundAdvice(HIGH_PRECEDENCE_ADVISOR_ORDER,LATE_ADVICE_DECLARATION_ORDER,"someAspect");
		
		assertEquals("advisor2 sorted before advisor1",1,this.comparator.compare(advisor1, advisor2));

		advisor1 = createAspectJAfterReturningAdvice(HIGH_PRECEDENCE_ADVISOR_ORDER,LATE_ADVICE_DECLARATION_ORDER,"someAspect");
		advisor2 = createAspectJAfterThrowingAdvice(HIGH_PRECEDENCE_ADVISOR_ORDER,EARLY_ADVICE_DECLARATION_ORDER,"someAspect");
		
		assertEquals("advisor1 sorted before advisor2",-1,this.comparator.compare(advisor1, advisor2));		
	}
	
	public void testSameAspectOneOfEach() {
		Advisor advisor1 = createAspectJAfterAdvice(HIGH_PRECEDENCE_ADVISOR_ORDER,EARLY_ADVICE_DECLARATION_ORDER,"someAspect");
		Advisor advisor2 = createAspectJBeforeAdvice(HIGH_PRECEDENCE_ADVISOR_ORDER,LATE_ADVICE_DECLARATION_ORDER,"someAspect");
		
		assertEquals("advisor1 and advisor2 not comparable",0,this.comparator.compare(advisor1, advisor2));
		
	}
	
	public void testSameAdvisorPrecedenceDifferentAspectNoAfterAdvice() {
		Advisor advisor1 = createAspectJBeforeAdvice(HIGH_PRECEDENCE_ADVISOR_ORDER,EARLY_ADVICE_DECLARATION_ORDER,"someAspect");
		Advisor advisor2 = createAspectJBeforeAdvice(HIGH_PRECEDENCE_ADVISOR_ORDER,LATE_ADVICE_DECLARATION_ORDER,"someOtherAspect");
		
		assertEquals("nothing to say about order here",0,this.comparator.compare(advisor1, advisor2));

		advisor1 = createAspectJBeforeAdvice(HIGH_PRECEDENCE_ADVISOR_ORDER,LATE_ADVICE_DECLARATION_ORDER,"someAspect");
		advisor2 = createAspectJAroundAdvice(HIGH_PRECEDENCE_ADVISOR_ORDER,EARLY_ADVICE_DECLARATION_ORDER,"someOtherAspect");
		
		assertEquals("nothing to say about order here",0,this.comparator.compare(advisor1, advisor2));
		
	}
	
	public void testSameAdvisorPrecedenceDifferentAspectAfterAdvice() {
		Advisor advisor1 = createAspectJAfterAdvice(HIGH_PRECEDENCE_ADVISOR_ORDER,EARLY_ADVICE_DECLARATION_ORDER,"someAspect");
		Advisor advisor2 = createAspectJAroundAdvice(HIGH_PRECEDENCE_ADVISOR_ORDER,LATE_ADVICE_DECLARATION_ORDER,"someOtherAspect");
		
		assertEquals("nothing to say about order here",0,this.comparator.compare(advisor1, advisor2));

		advisor1 = createAspectJAfterReturningAdvice(HIGH_PRECEDENCE_ADVISOR_ORDER,LATE_ADVICE_DECLARATION_ORDER,"someAspect");
		advisor2 = createAspectJAfterThrowingAdvice(HIGH_PRECEDENCE_ADVISOR_ORDER,EARLY_ADVICE_DECLARATION_ORDER,"someOtherAspect");
		
		assertEquals("nothing to say about order here",0,this.comparator.compare(advisor1, advisor2));
	}
	
	public void testHigherAdvisorPrecedenceNoAfterAdvice() {
		Advisor advisor1 = createSpringAOPBeforeAdvice(HIGH_PRECEDENCE_ADVISOR_ORDER);
		Advisor advisor2 = createAspectJBeforeAdvice(LOW_PRECEDENCE_ADVISOR_ORDER,EARLY_ADVICE_DECLARATION_ORDER,"someOtherAspect");
		
		assertEquals("advisor1 sorted before advisor2",-1,this.comparator.compare(advisor1, advisor2));

		advisor1 = createAspectJBeforeAdvice(HIGH_PRECEDENCE_ADVISOR_ORDER,LATE_ADVICE_DECLARATION_ORDER,"someAspect");
		advisor2 = createAspectJAroundAdvice(LOW_PRECEDENCE_ADVISOR_ORDER,EARLY_ADVICE_DECLARATION_ORDER,"someOtherAspect");
		
		assertEquals("advisor1 sorted before advisor2",-1,this.comparator.compare(advisor1, advisor2));		
	}
	
	public void testHigherAdvisorPrecedenceAfterAdvice() {
		Advisor advisor1 = createAspectJAfterAdvice(HIGH_PRECEDENCE_ADVISOR_ORDER,EARLY_ADVICE_DECLARATION_ORDER,"someAspect");
		Advisor advisor2 = createAspectJAroundAdvice(LOW_PRECEDENCE_ADVISOR_ORDER,LATE_ADVICE_DECLARATION_ORDER,"someOtherAspect");
		
		assertEquals("advisor1 sorted before advisor2",-1,this.comparator.compare(advisor1, advisor2));

		advisor1 = createAspectJAfterReturningAdvice(HIGH_PRECEDENCE_ADVISOR_ORDER,LATE_ADVICE_DECLARATION_ORDER,"someAspect");
		advisor2 = createAspectJAfterThrowingAdvice(LOW_PRECEDENCE_ADVISOR_ORDER,EARLY_ADVICE_DECLARATION_ORDER,"someOtherAspect");
		
		assertEquals("advisor2 sorted after advisor1",-1,this.comparator.compare(advisor1, advisor2));				
	}
	
	public void testLowerAdvisorPrecedenceNoAfterAdvice() {
		Advisor advisor1 = createAspectJBeforeAdvice(LOW_PRECEDENCE_ADVISOR_ORDER,EARLY_ADVICE_DECLARATION_ORDER,"someAspect");
		Advisor advisor2 = createAspectJBeforeAdvice(HIGH_PRECEDENCE_ADVISOR_ORDER,EARLY_ADVICE_DECLARATION_ORDER,"someOtherAspect");
		
		assertEquals("advisor1 sorted after advisor2",1,this.comparator.compare(advisor1, advisor2));

		advisor1 = createAspectJBeforeAdvice(LOW_PRECEDENCE_ADVISOR_ORDER,LATE_ADVICE_DECLARATION_ORDER,"someAspect");
		advisor2 = createAspectJAroundAdvice(HIGH_PRECEDENCE_ADVISOR_ORDER,EARLY_ADVICE_DECLARATION_ORDER,"someOtherAspect");
		
		assertEquals("advisor1 sorted after advisor2",1,this.comparator.compare(advisor1, advisor2));				
	}
	
	public void testLowerAdvisorPrecedenceAfterAdvice() {
		Advisor advisor1 = createAspectJAfterAdvice(LOW_PRECEDENCE_ADVISOR_ORDER,EARLY_ADVICE_DECLARATION_ORDER,"someAspect");
		Advisor advisor2 = createAspectJAroundAdvice(HIGH_PRECEDENCE_ADVISOR_ORDER,LATE_ADVICE_DECLARATION_ORDER,"someOtherAspect");
		
		assertEquals("advisor1 sorted after advisor2",1,this.comparator.compare(advisor1, advisor2));

		advisor1 = createSpringAOPAfterAdvice(LOW_PRECEDENCE_ADVISOR_ORDER);
		advisor2 = createAspectJAfterThrowingAdvice(HIGH_PRECEDENCE_ADVISOR_ORDER,EARLY_ADVICE_DECLARATION_ORDER,"someOtherAspect");
		
		assertEquals("advisor1 sorted after advisor2",1,this.comparator.compare(advisor1, advisor2));						
	}
	
	// helpers...
	private Advisor createAspectJBeforeAdvice(int advisorOrder, int adviceDeclarationOrder, String aspectName) {
		AspectJMethodBeforeAdvice advice = 
			new AspectJMethodBeforeAdvice(this.anyOldMethod,this.anyOldPointcut,null);
		return createAspectJAdvice(advisorOrder, adviceDeclarationOrder, aspectName, advice);
	}

	private Advisor createAspectJAroundAdvice(int advisorOrder, int adviceDeclarationOrder, String aspectName) {
		AspectJAroundAdvice advice = 
			new AspectJAroundAdvice(this.anyOldMethod,this.anyOldPointcut,null,null);
		return createAspectJAdvice(advisorOrder, adviceDeclarationOrder, aspectName, advice);
	}

	private Advisor createAspectJAfterAdvice(int advisorOrder, int adviceDeclarationOrder, String aspectName) {
		AspectJAfterAdvice advice = 
			new AspectJAfterAdvice(this.anyOldMethod,this.anyOldPointcut,null);
		return createAspectJAdvice(advisorOrder, adviceDeclarationOrder, aspectName, advice);
	}

	private Advisor createAspectJAfterReturningAdvice(int advisorOrder, int adviceDeclarationOrder, String aspectName) {
		AspectJAfterReturningAdvice advice = 
			new AspectJAfterReturningAdvice(this.anyOldMethod,this.anyOldPointcut,null);
		return createAspectJAdvice(advisorOrder, adviceDeclarationOrder, aspectName, advice);
	}
	
	private Advisor createAspectJAfterThrowingAdvice(int advisorOrder, int adviceDeclarationOrder, String aspectName) {
		AspectJAfterThrowingAdvice advice = 
			new AspectJAfterThrowingAdvice(this.anyOldMethod,this.anyOldPointcut,null);
		return createAspectJAdvice(advisorOrder, adviceDeclarationOrder, aspectName, advice);
	}

	private Advisor createAspectJAdvice(int advisorOrder, int adviceDeclarationOrder, String aspectName, AbstractAspectJAdvice advice) {
		advice.setDeclarationOrder(adviceDeclarationOrder);
		advice.setAspectName(aspectName);
		AspectJPointcutAdvisor advisor = new AspectJPointcutAdvisor();
		advisor.setPointcut(this.anyOldPointcut);
		advisor.setAdvice((Advice)advice);
		advisor.setOrder(advisorOrder);
		return advisor;
	}
	
	private Advisor createSpringAOPAfterAdvice(int order) {
		AfterReturningAdvice advice = new AfterReturningAdvice() {

			public void afterReturning(Object returnValue, Method method, Object[] args, Object target) throws Throwable {
				// TODO Auto-generated method stub
				
			} };
		DefaultPointcutAdvisor advisor = new DefaultPointcutAdvisor(this.anyOldPointcut,advice);
		advisor.setOrder(order);
		return advisor;
	}
	
	private Advisor createSpringAOPBeforeAdvice(int order) {
		BeforeAdvice advice = new BeforeAdvice() { };
		DefaultPointcutAdvisor advisor = new DefaultPointcutAdvisor(this.anyOldPointcut,advice);
		advisor.setOrder(order);
		return advisor;		
	}
}