package Sequenic.T2ext.Instrumenter;

import java.lang.reflect.Method;

import org.junit.Test;

import Sequenic.Graph.PathUtil;

public class MethodInstrumenterTest1 {
	
	public Method getMethod(String mname, Class C){
		Method[] methods = C.getDeclaredMethods() ;
		for (int i=0; i<methods.length; i++)
			if (methods[i].getName().equals(mname)) return methods[i] ;
		throw new Error() ;
	}
	
	@Test
	public void test1(){
		Class C = Simple.class ;
		Method m = getMethod("foo",C) ;
		MethodInstrumenter MI = MethodInstrumenter.createMethodInstrumenter(m, C) ;
		CFG CFG = MI.doInstrumenting(true);
		CFG.simplePrint() ;
	}
	
	@Test
	public void test2(){
		Class C = SimpleTargetClass1.class ;
		Method m = getMethod("methodWithSwitch",C) ;
		MethodInstrumenter MI = MethodInstrumenter.createMethodInstrumenter(m, C) ;
		CFG CFG = MI.doInstrumenting(true);
		CFG.simplePrint() ;
		assert CFG.targetPaths.size() == 4 ;
		assert PathUtil.pathSimpleShow(CFG.targetPaths.get(0)).equals("[ 0, 1, 5 ]") ;
		assert PathUtil.pathSimpleShow(CFG.targetPaths.get(1)).equals("[ 0, 2, 3, 4, 5 ]") ;
		assert PathUtil.pathSimpleShow(CFG.targetPaths.get(2)).equals("[ 0, 3, 4, 5 ]") ;
		assert PathUtil.pathSimpleShow(CFG.targetPaths.get(3)).equals("[ 0, 4, 5 ]") ;
	}
	
	@Test
	public void test3(){
		Class C = SimpleTargetClass1.class ;
		Method m = getMethod("methodWithNestedIf",C) ;
		MethodInstrumenter MI = MethodInstrumenter.createMethodInstrumenter(m, C) ;
		CFG CFG = MI.doInstrumenting(true);
		CFG.simplePrint() ;
		assert CFG.targetPaths.size() == 3 ;
		assert PathUtil.pathSimpleShow(CFG.targetPaths.get(0)).equals("[ 0, 1, 2, 5 ]") ;
		assert PathUtil.pathSimpleShow(CFG.targetPaths.get(1)).equals("[ 0, 1, 3, 5 ]") ;
		assert PathUtil.pathSimpleShow(CFG.targetPaths.get(2)).equals("[ 0, 4, 5 ]") ;
	}

	@Test
	public void test4(){
		Class C = SimpleTargetClass1.class ;
		Method m = getMethod("methodWithLoop",C) ;
		MethodInstrumenter MI = MethodInstrumenter.createMethodInstrumenter(m, C) ;
		CFG CFG = MI.doInstrumenting(true);
		CFG.simplePrint() ;
		assert CFG.targetPaths.size() == 2 ;
		assert PathUtil.pathSimpleShow(CFG.targetPaths.get(0)).equals("[ 0, 2, 3 ]") ;
		assert PathUtil.pathSimpleShow(CFG.targetPaths.get(1)).equals("[ 2, 1, 2 ]") ;
	}

	@Test
	public void test4b(){
		Class C = SimpleTargetClass1.class ;
		Method m = getMethod("methodWithLoopAndIf",C) ;
		MethodInstrumenter MI = MethodInstrumenter.createMethodInstrumenter(m, C) ;
		CFG CFG = MI.doInstrumenting(true);
		CFG.simplePrint() ;
		assert CFG.targetPaths.size() == 3 ;
		assert PathUtil.pathSimpleShow(CFG.targetPaths.get(0)).equals("[ 0, 4, 5 ]") ;
		assert PathUtil.pathSimpleShow(CFG.targetPaths.get(1)).equals("[ 4, 1, 2, 4 ]") ;
		assert PathUtil.pathSimpleShow(CFG.targetPaths.get(2)).equals("[ 4, 1, 3, 4 ]") ;
	}

	@Test
	public void test4c(){
		Class C = SimpleTargetClass1.class ;
		Method m = getMethod("methodWithNestedLoop",C) ;
		MethodInstrumenter MI = MethodInstrumenter.createMethodInstrumenter(m, C) ;
		CFG CFG = MI.doInstrumenting(true);
		CFG.simplePrint() ;
		assert CFG.targetPaths.size() == 3 ;
		assert PathUtil.pathSimpleShow(CFG.targetPaths.get(0)).equals("[ 0, 4, 5 ]") ;
		assert PathUtil.pathSimpleShow(CFG.targetPaths.get(1)).equals("[ 3, 2, 3 ]") ;
		assert PathUtil.pathSimpleShow(CFG.targetPaths.get(2)).equals("[ 4, 1, 3, 4 ]") ;
	}
	
	@Test
	public void test6(){
		Class C = SimpleTargetClass1.class ;
		Method m = getMethod("methodWithThrow",C) ;
		MethodInstrumenter MI = MethodInstrumenter.createMethodInstrumenter(m, C) ;
		CFG CFG = MI.doInstrumenting(true);
		CFG.simplePrint() ;
		assert CFG.targetPaths.size() == 3 ;
		assert PathUtil.pathSimpleShow(CFG.targetPaths.get(0)).equals("[ 0, 1 ]") ;
		assert PathUtil.pathSimpleShow(CFG.targetPaths.get(1)).equals("[ 0, 3 ]") ;
		assert PathUtil.pathSimpleShow(CFG.targetPaths.get(2)).equals("[ 2, 3 ]") ;
	}
	
	@Test
	public void test7(){
		Class C = SimpleTargetClass1.class ;
		Method m = getMethod("methodWithAssert1",C) ;
		MethodInstrumenter MI = MethodInstrumenter.createMethodInstrumenter(m, C) ;
		CFG CFG = MI.doInstrumenting(true);
		CFG.simplePrint() ;
		assert CFG.targetPaths.size() == 1 ;
		assert PathUtil.pathSimpleShow(CFG.targetPaths.get(0)).equals("[ 0, 1, 3 ]") ;
	}
	
	@Test
	public void test7b(){
		Class C = SimpleTargetClass1.class ;
		Method m = getMethod("methodWithAssert2",C) ;
		MethodInstrumenter MI = MethodInstrumenter.createMethodInstrumenter(m, C) ;
		CFG CFG = MI.doInstrumenting(true);
		CFG.simplePrint() ;
		assert CFG.targetPaths.size() == 2 ;
		assert PathUtil.pathSimpleShow(CFG.targetPaths.get(0)).equals("[ 0, 1, 3 ]") ;
		assert PathUtil.pathSimpleShow(CFG.targetPaths.get(1)).equals("[ 0, 3 ]") ;
	}
	
	@Test
	public void test7c(){
		Class C = SimpleTargetClass1.class ;
		Method m = getMethod("methodWithAssert3",C) ;
		MethodInstrumenter MI = MethodInstrumenter.createMethodInstrumenter(m, C) ;
		CFG CFG = MI.doInstrumenting(true);
		CFG.simplePrint() ;
		assert CFG.targetPaths.size() == 2 ;
		assert PathUtil.pathSimpleShow(CFG.targetPaths.get(0)).equals("[ 0, 1, 3 ]") ;
		assert PathUtil.pathSimpleShow(CFG.targetPaths.get(1)).equals("[ 0, 3 ]") ;
	}
	
	@Test
	public void test7d(){
		Class C = SimpleTargetClass1.class ;
		Method m = getMethod("methodWithAssertCatch",C) ;
		MethodInstrumenter MI = MethodInstrumenter.createMethodInstrumenter(m, C) ;
		CFG CFG = MI.doInstrumenting(true);
		CFG.simplePrint() ;
		assert CFG.targetPaths.size() == 2 ;
		assert PathUtil.pathSimpleShow(CFG.targetPaths.get(0)).equals("[ 0, 3 ]") ;
		assert PathUtil.pathSimpleShow(CFG.targetPaths.get(1)).equals("[ 2, 3 ]") ;
	}
	
	@Test
	public void test7e(){
		Class C = SimpleTargetClass1.class ;
		Method m = getMethod("methodWithAssertCatch2",C) ;
		MethodInstrumenter MI = MethodInstrumenter.createMethodInstrumenter(m, C) ;
		CFG CFG = MI.doInstrumenting(true);
		CFG.simplePrint() ;
		assert CFG.targetPaths.size() == 2 ;
		assert PathUtil.pathSimpleShow(CFG.targetPaths.get(0)).equals("[ 0, 2, 4 ]") ;
		assert PathUtil.pathSimpleShow(CFG.targetPaths.get(1)).equals("[ 3, 4 ]") ;
	}
	
	@Test
	public void test7f(){
		Class C = SimpleTargetClass1.class ;
		Method m = getMethod("methodWithAssertCatch3",C) ;
		MethodInstrumenter MI = MethodInstrumenter.createMethodInstrumenter(m, C) ;
		CFG CFG = MI.doInstrumenting(true);
		CFG.simplePrint() ;
		assert CFG.targetPaths.size() == 2 ;
		assert PathUtil.pathSimpleShow(CFG.targetPaths.get(0)).equals("[ 0, 2 ]") ;
		assert PathUtil.pathSimpleShow(CFG.targetPaths.get(1)).equals("[ 3 ]") ;
	}
	
	@Test
	public void test8(){
		Class C = SimpleTargetClass1.class ;
		Method m = getMethod("methodWithFinally",C) ;
		MethodInstrumenter MI = MethodInstrumenter.createMethodInstrumenter(m, C) ;
		CFG CFG = MI.doInstrumenting(true);
		CFG.simplePrint() ;
		assert CFG.targetPaths.size() == 3 ;
		assert PathUtil.pathSimpleShow(CFG.targetPaths.get(0)).equals("[ 0 ]") ;
		assert PathUtil.pathSimpleShow(CFG.targetPaths.get(1)).equals("[ 1 ]") ;
		assert PathUtil.pathSimpleShow(CFG.targetPaths.get(2)).equals("[ 2 ]") ;
	}
}