package com.wsdjeg.util;

import java.lang.reflect.Constructor;

import java.lang.reflect.Method;

public class TestMethod{

public static void main(String[] args) {

if(args.length == 2){

testSpecifiedMethod(args[0],args[1]);

}else if (args.length == 1) {

testAllMethods(args[0]);

}else if (args.length > 2){

testMethods(args);

}

}

public static void testAllMethods(String className){

Class<?> clazz = null;

try {

clazz = Class.forName(className);

} catch(Exception e){

e.printStackTrace();

}

Method[] mds = clazz.getMethods();

for (int i = 0; i < mds.length; i++) {

if (mds[i].getName().startsWith("test")) {

testSpecifiedMethod(className,mds[i].getName());

}

}

}

public static void testMethods(String[] args) {

for (int i = 1; i < args.length; i++) {

testSpecifiedMethod(args[0],args[i]);

}

}

@SuppressWarnings("unchecked")

public static void testSpecifiedMethod(String className,String methodName){

try{

System.out.println("========================= JavaUnite Test =============================");

System.out.println("CLASS:" + className);

System.out.println("METHOD:" + methodName + "()");

System.out.println("============================= OUTPUT =================================");

Class<?> clazz = Class.forName(className);

Constructor<?> c = clazz.getDeclaredConstructor();

c.setAccessible(true);

long startTime = System.currentTimeMillis();

clazz.getMethod(methodName).invoke(c.newInstance());

long endTime = System.currentTimeMillis();

System.out.println("================ SUCCESS: " + (endTime - startTime) + "ms ======================");

}catch(Exception e){

System.out.println("============================= FAILED ================================");

e.printStackTrace();

}

}

}