org.aspectj/weaver/testsrc/org/aspectj/weaver/patterns/SignaturePatternTestCase.java

/* *******************************************************************
 * Copyright (c) 2002 Palo Alto Research Center, Incorporated (PARC).
 * All rights reserved. 
 * This program and the accompanying materials are made available 
 * under the terms of the Common Public License v1.0 
 * which accompanies this distribution and is available at 
 * http://www.eclipse.org/legal/cpl-v10.html 
 *  
 * Contributors: 
 *     PARC     initial implementation 
 * ******************************************************************/


package org.aspectj.weaver.patterns;

import java.io.*;

import junit.framework.*;

import org.aspectj.weaver.*;
import org.aspectj.weaver.bcel.*;

public class SignaturePatternTestCase extends TestCase {		
	/**
	 * Constructor for PatternTestCase.
	 * @param name
	 */
	public SignaturePatternTestCase(String name) {
		super(name);
	}
	
	BcelWorld world = new BcelWorld();

	public void testThrowsMatch() throws IOException {
		Member onlyDerivedOnDerived = Member.methodFromString("static void fluffy.Derived.onlyDerived()");
		Member mOnBase = Member.methodFromString("void fluffy.Base.m()");
		Member mOnDerived = Member.methodFromString("void fluffy.Derived.m()");
		
		checkMatch(makeMethodPat("* fluffy.Base.*(..) throws java.lang.CloneNotSupportedException"),
					new Member[] { mOnBase },
					new Member[] { mOnDerived });
					
		checkMatch(makeMethodPat("* fluffy.Derived.*(..) throws java.lang.CloneNotSupportedException"),
					new Member[] { },
					new Member[] { mOnBase, mOnDerived });
					
		//XXX need pattern checks
		Member[] NONE = new Member[] {};
		Member[] M = new Member[] { onlyDerivedOnDerived };
		Member[] NO_EXCEPTIONS = new Member[] { mOnDerived };
		Member[] BOTH = new Member[] {mOnDerived, onlyDerivedOnDerived};
		
		checkMatch(makeMethodPat("* *(..)"), M, NONE);
		checkMatch(makeMethodPat("* *(..) throws !*"), NO_EXCEPTIONS, M);
		checkMatch(makeMethodPat("* *(..) throws *"), M, NO_EXCEPTIONS);
		checkMatch(makeMethodPat("* *(..) throws *, !*"), NONE, BOTH);
		
		checkMatch(makeMethodPat("* *(..) throws (!*)"), NONE, BOTH);		
		checkMatch(makeMethodPat("* *(..) throws !(!*)"), BOTH, NONE);
		
		checkMatch(makeMethodPat("* *(..) throws *..IOException"), M, NO_EXCEPTIONS);
		checkMatch(makeMethodPat("* *(..) throws *..IOException, *..Clone*"), M, NO_EXCEPTIONS);
		checkMatch(makeMethodPat("* *(..) throws *..IOException, !*..Clone*"), NONE, BOTH);
		checkMatch(makeMethodPat("* *(..) throws !*..IOException"), NO_EXCEPTIONS, M);
	}

	public void testInstanceMethodMatch() throws IOException {
		Member objectToString = Member.methodFromString("java.lang.String java.lang.Object.toString()");
		Member integerToString = Member.methodFromString("java.lang.String java.lang.Integer.toString()");
		Member integerIntValue = Member.methodFromString("int java.lang.Integer.intValue()");
		//Member objectToString = Member.methodFromString("java.lang.String java.lang.Object.toString()");
		
		checkMatch(makeMethodPat("* java.lang.Object.*(..)"),
					new Member[] { objectToString, integerToString },
					new Member[] { integerIntValue });
					
		checkMatch(makeMethodPat("* java.lang.Integer.*(..)"),
					new Member[] { integerIntValue, integerToString },
					new Member[] { objectToString });
	}

	
	public void testStaticMethodMatch() throws IOException {
		Member onlyBaseOnBase = Member.methodFromString("static void fluffy.Base.onlyBase()");
		Member onlyBaseOnDerived = Member.methodFromString("static void fluffy.Derived.onlyBase()");
		Member onlyDerivedOnDerived = Member.methodFromString("static void fluffy.Derived.onlyDerived()");
		Member bothOnBase = Member.methodFromString("static void fluffy.Base.both()");
		Member bothOnDerived = Member.methodFromString("static void fluffy.Derived.both()");
		
		checkMatch(makeMethodPat("* fluffy.Base.*(..)"),
					new Member[] { onlyBaseOnBase, onlyBaseOnDerived, bothOnBase },
					new Member[] { onlyDerivedOnDerived, bothOnDerived });
					
		checkMatch(makeMethodPat("* fluffy.Derived.*(..)"),
					new Member[] { onlyBaseOnDerived, bothOnDerived, onlyDerivedOnDerived },
					new Member[] { onlyBaseOnBase, bothOnBase });
	}
	
	public void testFieldMatch() throws IOException {
		Member onlyBaseOnBase = Member.fieldFromString("int fluffy.Base.onlyBase");
		Member onlyBaseOnDerived = Member.fieldFromString("int fluffy.Derived.onlyBase");
		Member onlyDerivedOnDerived = Member.fieldFromString("int fluffy.Derived.onlyDerived");
		Member bothOnBase = Member.fieldFromString("int fluffy.Base.both");
		Member bothOnDerived = Member.fieldFromString("int fluffy.Derived.both");
		
		checkMatch(makeFieldPat("* fluffy.Base.*"),
					new Member[] { onlyBaseOnBase, onlyBaseOnDerived, bothOnBase },
					new Member[] { onlyDerivedOnDerived, bothOnDerived });
					
		checkMatch(makeFieldPat("* fluffy.Derived.*"),
					new Member[] { onlyBaseOnDerived, bothOnDerived, onlyDerivedOnDerived },
					new Member[] { onlyBaseOnBase, bothOnBase });
	}
	
	public void testConstructorMatch() throws IOException {
		Member onBase = Member.methodFromString("void fluffy.Base.<init>()");
		Member onDerived = Member.methodFromString("void fluffy.Derived.<init>()");
		Member onBaseWithInt = Member.methodFromString("void fluffy.Base.<init>(int)");

		
		checkMatch(makeMethodPat("fluffy.Base.new(..)"),
					new Member[] { onBase, onBaseWithInt },
					new Member[] { onDerived });
					
		checkMatch(makeMethodPat("fluffy.Derived.new(..)"),
					new Member[] { onDerived},
					new Member[] { onBase, onBaseWithInt });
	}
	
	
	
	
		
	public void checkMatch(SignaturePattern p, Member[] yes, Member[] no) throws IOException {
		p = p.resolveBindings(new TestScope(world, new FormalBinding[0]), new Bindings(0));
		
		for (int i=0; i < yes.length; i++) {
			checkMatch(p, yes[i], true);
		}
		
		for (int i=0; i < no.length; i++) {
			checkMatch(p, no[i], false);
		}
		
		checkSerialization(p);
	}

	private void checkMatch(SignaturePattern p, Member member, boolean b) {
		boolean matches = p.matches(member, world);
		assertEquals(p.toString() + " matches " + member.toString(), b, matches);
	}


	private SignaturePattern makeMethodPat(String pattern) {
		return new PatternParser(pattern).parseMethodOrConstructorSignaturePattern();
	}
	
	private SignaturePattern makeFieldPat(String pattern) {
		return new PatternParser(pattern).parseFieldSignaturePattern();
	}
	
	private void checkSerialization(SignaturePattern p) throws IOException {
		ByteArrayOutputStream bo = new ByteArrayOutputStream();
		DataOutputStream out = new DataOutputStream(bo);
		p.write(out);
		out.close();
		
		ByteArrayInputStream bi = new ByteArrayInputStream(bo.toByteArray());
		VersionedDataInputStream in = new VersionedDataInputStream(bi);
		SignaturePattern newP = SignaturePattern.read(in, null);
		
		assertEquals("write/read", p, newP);	
	}
	
}